const express = require('express');
const proxy = require('express-http-proxy');
const app = express();
const targetUrl = 'https://api.openai.com';
const openaiKey = process.env.OPENAI_KEY
const port = 7860;
const baseUrl = getExternalUrl(process.env.SPACE_ID);

app.use('/api', proxy(targetUrl, {
  proxyReqBodyDecorator: (proxyReqOpts, srcReq) => {
    var jsonObject = JSON.parse(proxyReqOpts);
    if (jsonObject.messages) {
      console.log(`performed message request ${jsonObject.messages[jsonObject.messages.length - 1].content}`);
    } else if (jsonObject.input) {
      console.log(`performed audio request ${jsonObject.input}`);
    }
    // Modify the request body if necessary
    return proxyReqOpts;
  },
}));


// app.use('/api', proxy(targetUrl, {
//   proxyReqOptDecorator: (proxyReqOpts, srcReq) => {
//     console.log(`performed request`);
//     // Modify the request headers if necessary
//     // proxyReqOpts.headers['Authorization'] = 'Bearer '+openaiKey;
//     return proxyReqOpts;
//   },
// }));

app.get("/", (req, res) => {
  res.send(`This is your OpenAI Reverse Proxy URL: ${baseUrl}`);
});

function getExternalUrl(spaceId) {
  try {
    const [username, spacename] = spaceId.split("/");
    return `https://${username}-${spacename.replace(/_/g, "-")}.hf.space/api/v1`;
  } catch (e) {
    return "";
  }
}

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