From 5ad89cf008979390f63033721608675980259782 Mon Sep 17 00:00:00 2001
From: freearhey <free.arhey@gmail.com>
Date: Sat, 11 Apr 2020 03:54:32 +0300
Subject: [PATCH] Add custom http User-Agent and make code prettier

---
 scripts/test.js | 53 +++++++++++++++++++++----------------------------
 1 file changed, 23 insertions(+), 30 deletions(-)

diff --git a/scripts/test.js b/scripts/test.js
index 0b008adf2..2dbb5a664 100644
--- a/scripts/test.js
+++ b/scripts/test.js
@@ -17,67 +17,60 @@ let stats = {
 }
 
 async function test() {
-
   const playlist = helper.parsePlaylist('index.m3u')
-  
-  const countries = helper.filterPlaylists(playlist.items, config.country, config.exclude)
 
-  for(let country of countries) {
+  const countries = helper.filterPlaylists(playlist.items, config.country, config.exclude)
 
+  for (let country of countries) {
     stats.playlists++
 
     console.log(`Processing '${country.url}'...`)
 
     const playlist = helper.parsePlaylist(country.url)
 
-    for(let item of playlist.items) {
-
+    for (let item of playlist.items) {
       stats.channels++
 
-      if(config.debug) { console.log(`Checking '${item.url}'...`) }
+      if (config.debug) {
+        console.log(`Checking '${item.url}'...`)
+      }
 
       await new Promise(resolve => {
-        
         const timeout = setTimeout(() => {
-
           resolve()
-        
         }, config.timeout * 1000)
 
-        ffmpeg(item.url, { timeout: 60 }).ffprobe((err) => {
-      
-          if(err) {
-            const message = helper.parseMessage(err, item.url)
-
-            stats.failures++
+        ffmpeg(item.url, { timeout: 60 }).ffprobe(
+          ['-user_agent', `"Mozilla/5.0 (Windows NT 10.0; Win64; x64)"`],
+          err => {
+            if (err) {
+              const message = helper.parseMessage(err, item.url)
 
-            helper.writeToLog(country.url, message, item.url)
+              stats.failures++
 
-            console.log(`${message} '${item.url}'`)
-          }
+              helper.writeToLog(country.url, message, item.url)
 
-          clearTimeout(timeout)
+              console.log(`${message} '${item.url}'`)
+            }
 
-          resolve()
+            clearTimeout(timeout)
 
-        })
+            resolve()
+          }
+        )
       })
-
     }
   }
 
-  if(stats.failures === 0) {
-
+  if (stats.failures === 0) {
     console.log(`OK (${stats.playlists} playlists, ${stats.channels} channels)`)
-    
   } else {
-
-    console.log(`FAILURES! (${stats.playlists} playlists, ${stats.channels} channels, ${stats.failures} failures)`)
+    console.log(
+      `FAILURES! (${stats.playlists} playlists, ${stats.channels} channels, ${stats.failures} failures)`
+    )
 
     process.exit(1)
-
   }
-
 }
 
 console.log('Test is running...')