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

import express from 'express'
import { Server } from 'http'
import { getPort, randomListen, terminateServer } from './utils'

export class MockInstancesIndex {
  private server: Server

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

  async initialize () {
    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
      })
    })

    this.server = await randomListen(app)

    return getPort(this.server)
  }

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

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