PeerTube/shared/extra-utils/mock-servers/mock-instances-index.ts

import express from 'express'
import { Server } from 'http'
import { randomInt } from '@shared/core-utils'
import { terminateServer } from './utils'

export class MockInstancesIndex {
  private server: Server

  private readonly indexInstances: { host: string, createdAt: string }[] = []

  initialize () {
    return new Promise<number>(res => {
      const app = express()

      app.use('/', (req: express.Request, res: express.Response, next: express.NextFunction) => {
        if (process.env.DEBUG) console.log('Receiving request on mocked server %s.', req.url)

        return next()
      })

      app.get('/api/v1/instances/hosts', (req: express.Request, res: express.Response) => {
        const since = req.query.since

        const filtered = this.indexInstances.filter(i => {
          if (!since) return true

          return i.createdAt > since
        })

        return res.json({
          total: filtered.length,
          data: filtered
        })
      })

      const port = 42000 + randomInt(1, 1000)
      this.server = app.listen(port, () => res(port))
    })
  }

  addInstance (host: string) {
    this.indexInstances.push({ host, createdAt: new Date().toISOString() })
  }

  terminate () {
    return terminateServer(this.server)
  }
}