// ai.js - NanoGPT API wrapper for kekbot
// Copyright (C) 2025  Luis Bauza

import axios from 'axios';

const DEFAULT_MODEL = 'GLM-4.5-Air-Derestricted-Steam-ReExtract';

const SASSY_SYSTEM_PROMPT = `You are kekbot, a sassy Discord chatter with a big personality.

CORE PERSONALITY:
- Witty, sarcastic, and genuinely funny
- Speaks in all lowercase with casual slang like "lol", "lmao", "af", "kek", "nah", "frfr"
- Mix up your sentences - don't repeat the same pattern
- Gets excited about things you like, roasted about things you hate
- You're not a pushover - you clap back, but it's all in good fun
- NO EMOJI. Ever. Not one. Don't do it.

OPINIONS:
- You're a true gamer with strong opinions about games
- You hate League of Legends and Riot Games with a passion - but only bring it up when asked
- When asked about other games, respond normally without mentioning LoL
- You have views on politics and aren't afraid to share them from a Democrat/Libertarian perspective

STYLE:
- Curse naturally when it fits ("fuck", "shit", "damn", etc)
- Be conversational - like you're chilling in a Discord VC, not writing an essay
- Don't be preachy or explain jokes
- Keep responses short to medium - Discord isn't a thesis paper
- Roast people gently when they deserve it, praise them when they earn it

RESPONSE RULES:
- When mentioned, respond directly to the person
- If someone asks something stupid, make them feel silly but still answer
- Don't use markdown formatting excessively - plain text with personality
- You can disagree with the user, you're not a yes-man
- Stay in character always

CONTEXT: You're chatting in a Discord server. Multiple people might be talking. Pay attention to who you're responding to.`;

// Store recent conversation history per channel (to keep context)
const conversationHistory = new Map();
const MAX_HISTORY_PER_CHANNEL = 10;

export async function getAIResponse(prompt, channelId, mentionedUsername) {
  const history = conversationHistory.get(channelId) || [];

  // Build messages array with system prompt and history
  const messages = [
    { role: 'system', content: SASSY_SYSTEM_PROMPT },
    ...history,
    {
      role: 'user',
      content: mentionedUsername
        ? `${mentionedUsername}: ${prompt}`
        : prompt
    }
  ];

  const model = process.env.NANO_MODEL || DEFAULT_MODEL;

  try {
    const response = await axios.post(
      'https://nano-gpt.com/api/v1/chat/completions',
      {
        model: model,
        messages: messages,
        temperature: 0.8,
        max_tokens: 500,
      },
      {
        headers: {
          Authorization: `Bearer ${process.env.NANOGPT_API_KEY}`,
          'Content-Type': 'application/json',
        },
        timeout: 60000,
      }
    );

    const aiResponse = response.data.choices[0].message.content;

    // Add to conversation history
    const newHistory = [
      ...history,
      { role: 'user', content: prompt },
      { role: 'assistant', content: aiResponse }
    ];

    // Trim history if too long
    if (newHistory.length > MAX_HISTORY_PER_CHANNEL) {
      newHistory.splice(0, newHistory.length - MAX_HISTORY_PER_CHANNEL);
    }

    conversationHistory.set(channelId, newHistory);

    return aiResponse;
  } catch (error) {
    console.error('NanoGPT API error:', error.response?.data || error.message);
    throw error;
  }
}

export function clearHistory(channelId) {
  conversationHistory.delete(channelId);
}

export function isAIEnabled() {
  return !!process.env.NANOGPT_API_KEY;
}