]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/blob - shared/server-commands/mock-servers/mock-object-storage.ts
shared/ typescript types dir server-commands
[github/Chocobozzz/PeerTube.git] / shared / server-commands / mock-servers / mock-object-storage.ts
1 import express from 'express'
2 import got, { RequestError } from 'got'
3 import { Server } from 'http'
4 import { pipeline } from 'stream'
5 import { ObjectStorageCommand } from '../server'
6 import { getPort, randomListen, terminateServer } from './utils'
7
8 export class MockObjectStorage {
9 private server: Server
10
11 async initialize () {
12 const app = express()
13
14 app.get('/:bucketName/:path(*)', (req: express.Request, res: express.Response, next: express.NextFunction) => {
15 const url = `http://${req.params.bucketName}.${ObjectStorageCommand.getEndpointHost()}/${req.params.path}`
16
17 if (process.env.DEBUG) {
18 console.log('Receiving request on mocked server %s.', req.url)
19 console.log('Proxifying request to %s', url)
20 }
21
22 return pipeline(
23 got.stream(url, { throwHttpErrors: false }),
24 res,
25 (err: RequestError) => {
26 if (!err) return
27
28 console.error('Pipeline failed.', err)
29 }
30 )
31 })
32
33 this.server = await randomListen(app)
34
35 return getPort(this.server)
36 }
37
38 terminate () {
39 return terminateServer(this.server)
40 }
41 }