// Cloudflare Worker pour l'API Mistral - Data Stories AI
// Ce worker agit comme un proxy sécurisé entre votre site statique et l'API Mistral

addEventListener('fetch', event => {
  event.respondWith(handleRequest(event.request))
})

async function handleRequest(request) {
  // 1. Vérification de la méthode
  if (request.method !== 'POST') {
    return new Response(JSON.stringify({ error: 'Method not allowed' }), {
      status: 405,
      headers: { 'Content-Type': 'application/json', 'Allow': 'POST' }
    })
  }

  // 2. Vérification de l'origine (sécurité CORS)
  const origin = request.headers.get('Origin')
  const allowedOrigins = [
    'https://hericlibong.github.io',
    'https://votre-domaine.com'
  ]

  if (!allowedOrigins.includes(origin)) {
    return new Response(JSON.stringify({ error: 'Origin not allowed' }), {
      status: 403,
      headers: { 'Content-Type': 'application/json' }
    })
  }

  try {
    // 3. Lire le corps de la requête
    const { message, corpusUrl } = await request.json()

    if (!message) {
      return new Response(JSON.stringify({ error: 'Message is required' }), {
        status: 400,
        headers: { 'Content-Type': 'application/json' }
      })
    }

    // 4. Récupérer le corpus de données (RAG)
    let ragContext = ''
    try {
      const corpusResponse = await fetch(corpusUrl || 'https://hericlibong.github.io/assets/data/ai_corpus.json')
      const corpus = await corpusResponse.json()

      // Construire le contexte RAG
      const articles = corpus.filter(item => item.type === 'article')
      const projects = corpus.filter(item => item.type === 'project')
      const pages = corpus.filter(item => item.type === 'page')

      ragContext = `
Contexte Data Stories (${corpus.length} éléments) :
- Articles (${articles.length}) : ${articles.map(a => a.title).join(', ')}
- Projets (${projects.length}) : ${projects.map(p => p.title).join(', ')}
- Pages (${pages.length}) : ${pages.map(p => p.title).join(', ')}

Contenu pertinent : ${corpus.slice(0, 5).map(item => `${item.title}: ${item.content.substring(0, 100)}...`).join(' | ')}
`
    } catch (error) {
      console.error('Error fetching corpus:', error)
      ragContext = 'Contexte: Data Stories - Contenu sur l\'analyse de données et la visualisation.'
    }

    // 5. Préparer la requête pour Mistral
    const mistralRequest = {
      model: "mistral-tiny",
      messages: [
        {
          role: "system",
          content: `Vous êtes Data Stories AI, un assistant spécialisé en analyse de données et visualisation.
          Répondez aux questions en vous basant sur le contexte suivant :
          ${ragContext}
          
          Règles :
          - Soyez concis et professionnel
          - Si la réponse n'est pas dans le contexte, dites "Je ne sais pas"
          - Répondez en français
          - Mentionnez la source quand c'est pertinent`
        },
        {
          role: "user",
          content: message
        }
      ],
      temperature: 0.7,
      max_tokens: 500
    }

    // 6. Appeler l'API Mistral
    const MISTRAL_API_KEY = MISTRAL_API_KEY // Variable d'environnement
    const MISTRAL_API_URL = 'https://api.mistral.ai/v1/chat/completions'

    const response = await fetch(MISTRAL_API_URL, {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
        'Authorization': `Bearer ${MISTRAL_API_KEY}`,
        'Accept': 'application/json'
      },
      body: JSON.stringify(mistralRequest)
    })

    const mistralResponse = await response.json()

    // 7. Retourner la réponse
    return new Response(JSON.stringify({
      success: true,
      response: mistralResponse.choices[0].message.content,
      usage: mistralResponse.usage
    }), {
      status: 200,
      headers: {
        'Content-Type': 'application/json',
        'Access-Control-Allow-Origin': origin,
        'Access-Control-Allow-Methods': 'POST',
        'Access-Control-Allow-Headers': 'Content-Type'
      }
    })

  } catch (error) {
    console.error('Error:', error)
    return new Response(JSON.stringify({
      success: false,
      error: 'Une erreur est survenue. Veuillez réessayer.',
      details: error.message
    }), {
      status: 500,
      headers: {
        'Content-Type': 'application/json',
        'Access-Control-Allow-Origin': origin
      }
    })
  }
}

function sendChatMessage() {
  const input = document.getElementById('ai-chat-input');
  const message = input.value.trim();
  
  if (message) {
    addChatMessage(message, 'ai-user');
    input.value = '';
    
    // Show typing indicator
    const typingIndicator = document.createElement('div');
    typingIndicator.className = 'ai-chat-message ai-assistant';
    typingIndicator.id = 'typing-indicator';
    typingIndicator.innerHTML = '<div class="typing-dots"><span>●</span><span>●</span><span>●</span></div>';
    document.getElementById('ai-chat-messages').appendChild(typingIndicator);
    
    // Scroll to bottom
    scrollChatToBottom();
    
    // Call the Cloudflare Worker
    callAIWorker(message);
  }
}

async function callAIWorker(message) {
  try {
    // Remplacez par l'URL de votre Worker
    const WORKER_URL = 'https://data-stories-ai.votre-sous-domaine.workers.dev';
    const CORPUS_URL = 'https://hericlibong.github.io/assets/data/ai_corpus.json';
    
    const response = await fetch(WORKER_URL, {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
        'Origin': window.location.origin
      },
      body: JSON.stringify({
        message: message,
        corpusUrl: CORPUS_URL
      })
    });
    
    const data = await response.json();
    
    removeTypingIndicator();
    
    if (data.success && data.response) {
      addChatMessage(data.response, 'ai-assistant');
    } else {
      addChatMessage(data.error || "Désolé, je n'ai pas pu obtenir de réponse.", 'ai-assistant');
    }
    
    scrollChatToBottom();
    
  } catch (error) {
    console.error('Error calling AI Worker:', error);
    removeTypingIndicator();
    addChatMessage("Une erreur est survenue. Veuillez vérifier votre connexion.", 'ai-assistant');
    scrollChatToBottom();
  }
}