Add --timeout option

scripts/commands/playlist/test.ts

@@ -30,6 +30,12 @@ program
     os.cpus().length
   )
   .option('-x, --proxy <url>', 'Use the specified proxy')
+  .option(
+    '-t, --timeout <number>',
+    '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)

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<TestResult> {
@@ -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()