From 2aa55af99e75485a785896e47c1eca723e41edc7 Mon Sep 17 00:00:00 2001
From: Aleksandr Statciuk <free.arhey@gmail.com>
Date: Wed, 8 Sep 2021 20:03:08 +0300
Subject: [PATCH] Update format.js

---
 scripts/format.js | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/scripts/format.js b/scripts/format.js
index f1060295d..e5599a2e8 100644
--- a/scripts/format.js
+++ b/scripts/format.js
@@ -1,3 +1,4 @@
+const glob = require('glob')
 const IPTVChecker = require('iptv-checker')
 const normalize = require('normalize-url')
 const { program } = require('commander')
@@ -24,14 +25,17 @@ const checker = new IPTVChecker({
 })
 
 let buffer, origins
-async function main() {
+function main() {
   log.start()
 
-  let playlists = parser.parseIndex().filter(i => i.url !== 'channels/unsorted.m3u')
-  playlists = utils.filterPlaylists(playlists, config.country, config.exclude)
-  if (!playlists.length) log.print(`No playlist is selected\n`)
-  for (const playlist of playlists) {
-    await parser.parsePlaylist(playlist.url).then(updatePlaylist).then(savePlaylist)
+  glob('channels/*.m3u', { ignore: ['unsorted.m3u'] }, handleFiles)
+}
+
+async function handleFiles(er, files) {
+  const filtered = utils.filterFiles(files, config.country, config.exclude)
+  if (!filtered.length) log.print(`No files is selected\n`)
+  for (const file of filtered) {
+    await parser.parsePlaylist(file).then(updatePlaylist).then(savePlaylist)
   }
 
   log.finish()