synapsecns/sanguine

import express from 'express'
import swaggerUi from 'swagger-ui-express'

import { specs } from './swagger'
import routes from './routes'
import { logger } from './middleware/logger'
import {
  isRFQAPIRequest,
  isRFQIndexerRequest,
  rfqApiProxy,
  rfqIndexerProxy,
} from './utils/isGatewayRoute'

const app = express()
const port = process.env.PORT || 3000

app.use((req, res, next) => {
  res.setHeader('Access-Control-Allow-Origin', '*')
  res.setHeader('Access-Control-Allow-Methods', 'GET, OPTIONS')
  res.setHeader('Access-Control-Allow-Headers', '*')

  if (req.method === 'OPTIONS') {
    res.sendStatus(200)
    return
  }

  next()
})

app.use(express.json())

app.use((req, res, next) => {
  logger.info({
    msg: `Incoming request ${req.path}`,
    method: req.method,
    path: req.path,
    query: req.query,
    body: req.method === 'POST' || req.method === 'PUT' ? req.body : undefined,
  })

  const originalPath = req.path
  const originalJson = res.json
  res.json = function (body) {
    logger.info({
      msg: `Outgoing response ${originalPath}`,
      method: req.method,
      path: originalPath,
      statusCode: res.statusCode,
      body:
        originalPath === '/' || originalPath.toLowerCase() === '/tokenlist'
          ? '[truncated for size]'
          : body,
    })
    return originalJson.call(this, body)
  }

  if (isRFQAPIRequest(originalPath)) {
    return rfqApiProxy(req, res, next)
  }

  if (isRFQIndexerRequest(originalPath)) {
    return rfqIndexerProxy(req, res, next)
  }

  return next()
})

app.listen(port, () => {
  logger.info(`Server is listening on port ${port}`)
})

app.get('/openapi.json', (_req, res) => {
  res.set(
    'Cache-Control',
    'no-store, no-cache, must-revalidate, proxy-revalidate'
  )
  res.set('Pragma', 'no-cache')
  res.set('Expires', '0')
  res.set('Surrogate-Control', 'no-store')
  res.json(specs)
})

app.use(
  '/api-docs',
  (_req, res, next) => {
    res.set(
      'Cache-Control',
      'no-store, no-cache, must-revalidate, proxy-revalidate'
    )
    res.set('Pragma', 'no-cache')
    res.set('Expires', '0')
    res.set('Surrogate-Control', 'no-store')
    next()
  },
  swaggerUi.serve,
  swaggerUi.setup(specs)
)

app.use('/', routes)

app.use((err, _req, res, _next) => {
  logger.error(`Express error: ${err.message}`, { stack: err.stack })
  res.status(500).json({ error: 'Something went wrong', details: err.message })
})

process.on('uncaughtException', (err) => {
  logger.error(`Uncaught Exception: ${err.message}`, { stack: err.stack })
  process.exit(1)
})

process.on('unhandledRejection', (reason, promise) => {
  logger.error('Unhandled Rejection at:', promise, 'reason:', reason)
})