Discord Bot Integration

require('dotenv').config();
const { Client, GatewayIntentBits, AttachmentBuilder } = require('discord.js');

const client = new Client({
  intents: [
    GatewayIntentBits.Guilds,
    GatewayIntentBits.GuildMessages,
    GatewayIntentBits.MessageContent,
  ],
});

const ANIMEAPI_KEY = process.env.ANIMEAPI_KEY;

client.on('ready', () => {
  console.log(`Logged in as ${client.user.tag}`);
});

client.on('messageCreate', async (message) => {
  // Ignore bot messages
  if (message.author.bot) return;

  // Check for command
  if (!message.content.startsWith('!anime ')) return;

  const prompt = message.content.slice(7).trim();
  if (!prompt) {
    return message.reply('Please provide a prompt! Example: `!anime cute anime girl`');
  }

  // Send "generating" message
  const statusMsg = await message.reply('Generating your anime image...');

  try {
    // Call AnimeAPI
    const response = await fetch('https://api.animeapi.com/api/generate', {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${ANIMEAPI_KEY}`,
        'Content-Type': 'application/json',
      },
      body: JSON.stringify({
        prompt,
        orientation: 'portrait', // portrait, square, or landscape
        allowNSFW: true,
      }),
    });

    const data = await response.json();

    if (!response.ok) {
      throw new Error(data.error || 'Generation failed');
    }

    // Download the image and send it
    const imageResponse = await fetch(data.image_url);
    const imageBuffer = Buffer.from(await imageResponse.arrayBuffer());
    const attachment = new AttachmentBuilder(imageBuffer, { name: 'anime.webp' });

    await statusMsg.edit({
      content: `Here's your anime image! (Balance: $${data.balance_usd.toFixed(2)})`,
      files: [attachment],
    });
  } catch (error) {
    console.error('Generation error:', error);
    await statusMsg.edit(`Failed to generate image: ${error.message}`);
  }
});

client.login(process.env.DISCORD_TOKEN);