server.ts

import $RefParser from '@apidevtools/json-schema-ref-parser'
import express, { Request, Response } from 'express'
import { copyFile } from 'fs/promises'
import { Server } from 'http'
import { bootstrapSops } from 'src/cmd/bootstrap'
import { decrypt, encrypt } from 'src/common/crypt'
import { terminal } from 'src/common/debug'
import { hfValues } from './common/hf'
import { setValuesFile, unsetValuesFile } from './common/repo'
import { rootDir } from './common/utils'
import { objectToYaml } from './common/values'

const d = terminal('server')
const app = express()
let server: Server

export const stopServer = (): void => {
  server?.close()
}
// Return statement is not needed anymore in express 5.x and above
app.get('/', async (req: Request, res: Response): Promise<void> => {
  res.send({ status: 'ok' })
})

type QueryParams = {
  envDir: string
  files?: string[]
}

app.get('/init', async (req: Request, res: Response): Promise<void> => {
  const { envDir } = req.query as QueryParams
  try {
    d.log('Request to initialize values repo on', envDir)
    await decrypt(envDir)
    res.status(200).send('ok')
  } catch (error) {
    d.error(error)
    res.status(500).send(`${error}`)
  }
})

app.get('/prepare', async (req: Request, res: Response): Promise<void> => {
  const { envDir, files } = req.query as QueryParams
  try {
    d.log('Request to prepare values repo on', envDir)
    const file = '.editorconfig'
    await copyFile(`${rootDir}/.values/${file}`, `${envDir}/${file}`)
    await bootstrapSops(envDir)
    await setValuesFile(envDir)
    // Encrypt ensures that a brand new secret file is encrypted in place
    await encrypt(envDir, ...(files ?? []))
    // Decrypt ensures that a brand new encrypted secret file is decrypted to the .dec file
    await decrypt(envDir, ...(files ?? []))
    res.status(200).send('ok')
  } catch (error) {
    const err = `${error}`
    let status = 500
    d.error(`Request to prepare values went wrong: ${err}`)
    if (err.includes('Values validation FAILED')) {
      status = 422
    }
    res.status(status).send(err)
  } finally {
    await unsetValuesFile(envDir)
  }
})

function parseBoolean(string: any, defaultValue = false): boolean {
  return string === 'true' ? true : string === 'false' ? false : defaultValue
}
app.get('/otomi/values', async (req: Request, res: Response): Promise<void> => {
  const { envDir } = req.query as QueryParams

  const filesOnly = parseBoolean(req.query.filesOnly, true)
  const excludeSecrets = parseBoolean(req.query.excludeSecrets, true)
  const withWorkloadValues = parseBoolean(req.query.withWorkloadValues, true)
  d.log('Get otomi values', req.query)
  try {
    const data = await hfValues({ filesOnly, excludeSecrets, withWorkloadValues }, envDir)
    res.setHeader('Content-type', 'text/plain')
    const yamlData = objectToYaml(data!)
    res.status(200).send(yamlData)
  } catch (error) {
    const status = 500
    d.error(error)
    res.status(status).send(error)
  }
})

app.get('/apl/schema', async (req: Request, res: Response): Promise<void> => {
  const derefSchema = await $RefParser.dereference(`${rootDir}/values-schema.yaml`)
  res.setHeader('Content-type', 'application/json')
  res.status(200).send(derefSchema)
})

export const startServer = (): void => {
  server = app
    .listen(17771, '127.0.0.1', () => {
      d.log(`Server listening on http://127.0.0.1:17771`)
    })
    .on('error', (e) => {
      console.error(e)
    })
}

// Add this at the bottom of the file or in a lifecycle manager
process.on('SIGINT', () => {
  d.info('Shutting down server')
  stopServer()
  process.exit(0)
})

process.on('SIGTERM', () => {
  d.info('Shutting down server')
  stopServer()
  process.exit(0)
})