diff --git a/scripts/commands/playlist/test.ts b/scripts/commands/playlist/test.ts index ca53d56ad1..f4d59cac1a 100644 --- a/scripts/commands/playlist/test.ts +++ b/scripts/commands/playlist/test.ts @@ -30,6 +30,12 @@ program os.cpus().length ) .option('-x, --proxy ', 'Use the specified proxy') + .option( + '-t, --timeout ', + 'The number of milliseconds before the request will be aborted', + (value: string) => parseInt(value), + 30000 + ) .parse(process.argv) const options: OptionValues = program.opts() @@ -96,7 +102,7 @@ async function runTest(stream: Stream) { const result: TestResult = await tester.test(stream) let status = '' - const errorStatusCodes = ['HTTP_404_NOT_FOUND'] + const errorStatusCodes = ['ENOTFOUND', 'HTTP_404_NOT_FOUND'] if (result.status.ok) status = chalk.green('OK') else if (errorStatusCodes.includes(result.status.code)) { status = chalk.red(result.status.code) diff --git a/scripts/core/streamTester.ts b/scripts/core/streamTester.ts index b6c6540617..d5a4f5f591 100644 --- a/scripts/core/streamTester.ts +++ b/scripts/core/streamTester.ts @@ -19,6 +19,7 @@ export type StreamTesterProps = { export class StreamTester { client: AxiosInstance + options: OptionValues constructor({ options }: StreamTesterProps) { const proxyParser = new ProxyParser() @@ -42,6 +43,7 @@ export class StreamTester { } this.client = axios.create(request) + this.options = options } async test(stream: Stream): Promise { @@ -51,10 +53,8 @@ export class StreamTester { return results[stream.url as keyof typeof results] } else { try { - const timeout = 10000 - const res = await this.client(stream.url, { - signal: AbortSignal.timeout(timeout), + signal: AbortSignal.timeout(this.options.timeout), headers: { 'User-Agent': stream.getUserAgent() || 'Mozilla/5.0', Referer: stream.getReferrer()