PeerTube/shared/extra-utils/server/servers-command.ts

93 lines
2.6 KiB
TypeScript
Raw Normal View History

2021-07-13 09:43:59 +02:00
import { exec } from 'child_process'
import { copy, ensureDir, readFile, remove } from 'fs-extra'
2021-07-23 11:20:00 +02:00
import { basename, join } from 'path'
2021-07-13 09:43:59 +02:00
import { root } from '@server/helpers/core-utils'
2021-07-16 10:42:24 +02:00
import { HttpStatusCode } from '@shared/models'
2021-07-22 11:58:44 +02:00
import { getFileSize, isGithubCI, wait } from '../miscs'
2021-07-13 09:43:59 +02:00
import { AbstractCommand, OverrideCommandOptions } from '../shared'
export class ServersCommand extends AbstractCommand {
static flushTests (internalServerNumber: number) {
return new Promise<void>((res, rej) => {
const suffix = ` -- ${internalServerNumber}`
return exec('npm run clean:server:test' + suffix, (err, _stdout, stderr) => {
if (err || stderr) return rej(err || new Error(stderr))
return res()
})
})
}
ping (options: OverrideCommandOptions = {}) {
return this.getRequestBody({
...options,
path: '/api/v1/ping',
implicitToken: false,
defaultExpectedStatus: HttpStatusCode.OK_200
})
}
async cleanupTests () {
const p: Promise<any>[] = []
if (isGithubCI()) {
await ensureDir('artifacts')
2021-07-16 09:47:51 +02:00
const origin = this.buildDirectory('logs/peertube.log')
2021-07-13 09:43:59 +02:00
const destname = `peertube-${this.server.internalServerNumber}.log`
console.log('Saving logs %s.', destname)
await copy(origin, join('artifacts', destname))
}
if (this.server.parallel) {
p.push(ServersCommand.flushTests(this.server.internalServerNumber))
}
if (this.server.customConfigFile) {
p.push(remove(this.server.customConfigFile))
}
return p
}
async waitUntilLog (str: string, count = 1, strictCount = true) {
2021-08-18 09:14:51 +02:00
const logfile = this.buildDirectory('logs/peertube.log')
2021-07-13 09:43:59 +02:00
while (true) {
const buf = await readFile(logfile)
const matches = buf.toString().match(new RegExp(str, 'g'))
if (matches && matches.length === count) return
if (matches && strictCount === false && matches.length >= count) return
await wait(1000)
}
}
buildDirectory (directory: string) {
return join(root(), 'test' + this.server.internalServerNumber, directory)
}
2021-07-22 14:28:03 +02:00
buildWebTorrentFilePath (fileUrl: string) {
return this.buildDirectory(join('videos', basename(fileUrl)))
}
buildFragmentedFilePath (videoUUID: string, fileUrl: string) {
return this.buildDirectory(join('streaming-playlists', 'hls', videoUUID, basename(fileUrl)))
}
2021-08-18 09:14:51 +02:00
getLogContent () {
return readFile(this.buildDirectory('logs/peertube.log'))
}
2021-07-13 09:43:59 +02:00
async getServerFileSize (subPath: string) {
2021-07-16 09:04:35 +02:00
const path = this.server.servers.buildDirectory(subPath)
2021-07-13 09:43:59 +02:00
return getFileSize(path)
}
}