PeerTube/server/tests/shared/mock-servers/mock-object-storage.ts

import express from 'express'
import got, { RequestError } from 'got'
import { Server } from 'http'
import { pipeline } from 'stream'
import { ObjectStorageCommand } from '@shared/server-commands'
import { getPort, randomListen, terminateServer } from './shared'

export class MockObjectStorage {
  private server: Server

  async initialize () {
    const app = express()

    app.get('/:bucketName/:path(*)', (req: express.Request, res: express.Response, next: express.NextFunction) => {
      const url = `http://${req.params.bucketName}.${ObjectStorageCommand.getEndpointHost()}/${req.params.path}`

      if (process.env.DEBUG) {
        console.log('Receiving request on mocked server %s.', req.url)
        console.log('Proxifying request to %s', url)
      }

      return pipeline(
        got.stream(url, { throwHttpErrors: false }),
        res,
        (err: RequestError) => {
          if (!err) return

          console.error('Pipeline failed.', err)
        }
      )
    })

    this.server = await randomListen(app)

    return getPort(this.server)
  }

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