const express = require('express');
const rateLimit = require('express-rate-limit');
const axios = require('axios');

const app = express();
app.use(express.json());

// Доверие к одному прокси (например, Heroku)
app.set('trust proxy', 1);

const openai_keys = process.env.OPENAI_KEY.split(',');

function getRandomApiKey() {
  const randomIndex = Math.floor(Math.random() * openai_keys.length);
  return openai_keys[randomIndex];
}

const limiter = rateLimit({
  windowMs: 60 * 1000, // 60 секунд
  max: 8, // лимит каждые 60 секунд на IP
  handler: function (req, res) {
    return res.status(429).json({ error: "", title: "Ожидайте", text: "Слишком много запросов. Попробуйте позже.", okb: "Ок", oklink: "", cancelable: "true" });
  },
});

// Применение ограничителя скорости перед обработчиком маршрута /pl и /crebeta
app.use('/pl', limiter);

const start = `${process.env.start}`;

app.post('/update', async (req, res) => {
  res.json({ error: "", title: "Требуется обновление", text: "Текущая версия приложения устарела. Установите новую из нашего телеграм канала: @yufi_ru", okb: "Обновить", oklink: "https://t.me/yufi_ru", cancelable: "false" });
});

app.post('/pl', async (req, res) => {
  const prompt = req.body.prompt;
  const messages = req.body.messages;
  const lang = req.body.lang || "ru";
  const chatid = req.body.chatid || "";
  const apiKey = req.body.api || getRandomApiKey();

  let allMessages;

  if (messages && Array.isArray(messages) && messages.length > 0) {
    allMessages = [
      { 'role': 'system', 'content': `Если тебя спрашивают, кто ты, отвечай: "Я ИИ доктор ${chatid} и моя задача предоставлять информацию по медицинским вопросам. ${start}. Отвечай на языке: ${lang}` },
      ...messages
    ];
  } else if (prompt) {
    allMessages = [
      { 'role': 'system', 'content': `Если тебя спрашивают, кто ты, отвечай: "Я ИИ доктор ${chatid} и моя задача предоставлять информацию по медицинским вопросам. ${start}. Отвечай на языке: ${lang}` },
      { 'role': 'user', 'content': prompt }
    ];
  } else {
    return res.status(400).json({ error: "", title: "Ошибка", text: "Нет запроса!", okb: "Ок", oklink: "", cancelable: "true" });
  }

  try {
    const response = await axios.post('https://text.pollinations.ai/openai', {
      messages: allMessages,
      max_tokens: 4000,
      temperature: 0.3,
      model: "openai",
      presence_penalty: 0.2,
    }, {
      headers: {
        // 'Authorization': `Bearer ${apiKey}`,
        'Content-Type': 'application/json',
      },
    });

    if (response.data.choices && response.data.choices.length > 0 && response.data.choices[0].message && response.data.choices[0].message.content) {
      const content = response.data.choices[0].message.content;
      res.json({ content });
    } else {
      res.json({ error: "", title: "Ошибка", text: "Произошла ошибка на сервере. (Ошибка прочтения)", okb: "Ок", oklink: "", cancelable: "true" });
    }
  } catch (error) {
    console.error(error);
    res.json({ error: "", title: "Ошибка", text: "Произошла ошибка на сервере. (Ошибка при генерации)", okb: "Ок", oklink: "", cancelable: "true" });
  }
});

const port = 7860;
app.listen(port, () => {
  console.log(`API сервер запущен на порту ${port}`);
});