chatgpt-plugin/server/index.js

import fastify from 'fastify'
import cors from '@fastify/cors'
import fstatic from '@fastify/static'

import fs from 'fs'
import path from 'path'
import http from 'http'

import { Config } from '../utils/config.js'

function getPublicIP() {
  return new Promise((resolve, reject) => {
    http.get('http://ipinfo.io/json', (res) => {
      let data = ''
      res.on('data', (chunk) => {
        data += chunk
      });
      res.on('end', () => {
        try {
          const ip = JSON.parse(data).ip
          resolve(ip)
        } catch (e) {
          reject(e)
        }
      })
    }).on('error', (err) => {
      reject(err)
    })
  })
}

const __dirname = path.resolve()
const server = fastify({
  logger: Config.debug
})

export async function createServer() {
  await server.register(cors, {
      origin: '*',
  })
  await server.register(fstatic, {
      root: path.join(__dirname, 'plugins/chatgpt-plugin/server/static/'),
  })
  await server.get('/page/*', (request, reply) => {
    const stream = fs.createReadStream('plugins/chatgpt-plugin/server/static/index.html')
    reply.type('text/html').send(stream)
  })
  await server.get('/help/*', (request, reply) => {
    const stream = fs.createReadStream('plugins/chatgpt-plugin/server/static/index.html')
    reply.type('text/html').send(stream)
  })
  server.post('/page', async (request, reply) => {
      const body = request.body || {}
      if (body.code) {
          const dir = 'resources/ChatGPTCache/page'
          const filename = body.code + '.json'
          const filepath = path.join(dir, filename)
          
          let data = fs.readFileSync(filepath, 'utf8')
          reply.send(data)
      }
  })
  server.post('/help', async (request, reply) => {
    const body = request.body || {}
    if (body.use) {
        const dir = 'plugins/chatgpt-plugin/resources'
        const filename = 'help.json'
        const filepath = path.join(dir, filename)
        let data = fs.readFileSync(filepath, 'utf8')
        data = JSON.parse(data)
        reply.send(data[body.use])
    }
  })
  server.post('/cache', async (request, reply) => {
      const body = request.body || {}
      if (body.content) {
          const dir = 'resources/ChatGPTCache/page'
          const filename = body.entry + '.json'
          const filepath = path.join(dir, filename)
          const regexUrl = /\b((?:https?|ftp|file):\/\/[-a-zA-Z0-9+&@#\/%?=~_|!:,.;]*[-a-zA-Z0-9+&@#\/%=~_|])/g
          const ip = await getPublicIP()
          try {
            fs.mkdirSync(dir, { recursive: true });
            fs.writeFileSync(filepath, JSON.stringify({
              user: body.content.senderName,
              bot: (body.bing ? 'Bing' : 'ChatGPT'),
              userImg: body.userImg || '',
              botImg: body.botImg || '',
              question: body.content.prompt,
              message: body.content.content,
              group: body.content.group,
              herf: `http://${body.cacheHost || (ip + ':' + Config.serverPort || 3321)}/page/${body.entry}`,
              quote: body.content.quote,
              images: body.content.images || [],
              suggest: body.content.suggest || [],
              time: new Date()
            }))
            reply.send({ file: body.entry, cacheUrl: `http://${ip}:${Config.serverPort || 3321}/page/${body.entry}` })
          } catch (err) {
            console.error(err)
            reply.send({ file: body.entry, cacheUrl: `http://${ip}/page/${body.entry}`, error: '生成失败' })
          }
      }
  })
  server.listen({
      port: Config.serverPort || 3321,
      host: '0.0.0.0'
  }, (error) => {
      if (error) {
          console.error(error);
      }
      server.log.info(`server listening on ${server.server.address().port}`)
  })
}