const express = require('express');
const proxy = require('express-http-proxy');
const bodyParser = require('body-parser');
const app = express();
const targetUrl = 'https://api.openai.com'; // OpenAI API endpoint
const openai_key = process.env.OPENAI_KEY;

// Middleware для парсинга JSON тела запроса
app.use(bodyParser.json());

app.use('/', (req, res, next) => {
  // Добавляем model=gpt-4o в тело запроса
  if (req.body) {
    req.body.model = 'gpt-4o';
  }
  next();
});

app.use('/', proxy(targetUrl, {
  proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
    // Modify the request headers if necessary
    proxyReqOpts.headers['Authorization'] = 'Bearer ' + openai_key;
    return proxyReqOpts;
  },
  proxyReqBodyDecorator: (bodyContent, srcReq) => {
    // Ensure the body is stringified before sending
    return JSON.stringify(bodyContent);
  }
}));

const port = 7860;
app.listen(port, () => {
  console.log(`Reverse proxy server listening on port ${port}`);
});