From 9b317ce5a6738cec18eb9ab96f5e24ca5647dea0 Mon Sep 17 00:00:00 2001
From: freearhey <free.arhey@gmail.com>
Date: Sat, 30 Jan 2021 06:16:44 +0300
Subject: [PATCH] Update utils.js
---
scripts/utils.js | 242 +++++++++++++++++++++++++++++++++++++----------
1 file changed, 192 insertions(+), 50 deletions(-)
diff --git a/scripts/utils.js b/scripts/utils.js
index 27e9b25fc..063f3e3ae 100644
--- a/scripts/utils.js
+++ b/scripts/utils.js
@@ -7,9 +7,157 @@ const escapeStringRegexp = require('escape-string-regexp')
const markdownInclude = require('markdown-include')
const iso6393 = require('iso-639-3')
-let helper = {}
+const utils = {}
+
+utils.supportedCategories = [
+ {
+ name: 'Auto',
+ id: 'auto',
+ nsfw: false
+ },
+ {
+ name: 'Business',
+ id: 'business',
+ nsfw: false
+ },
+ {
+ name: 'Classic',
+ id: 'classic',
+ nsfw: false
+ },
+ {
+ name: 'Comedy',
+ id: 'comedy',
+ nsfw: false
+ },
+ {
+ name: 'Documentary',
+ id: 'documentary',
+ nsfw: false
+ },
+ {
+ name: 'Education',
+ id: 'education',
+ nsfw: false
+ },
+ {
+ name: 'Entertainment',
+ id: 'entertainment',
+ nsfw: false
+ },
+ {
+ name: 'Family',
+ id: 'family',
+ nsfw: false
+ },
+ {
+ name: 'Fashion',
+ id: 'fashion',
+ nsfw: false
+ },
+ {
+ name: 'Food',
+ id: 'food',
+ nsfw: false
+ },
+ {
+ name: 'General',
+ id: 'general',
+ nsfw: false
+ },
+ {
+ name: 'Health',
+ id: 'health',
+ nsfw: false
+ },
+ {
+ name: 'History',
+ id: 'history',
+ nsfw: false
+ },
+ {
+ name: 'Hobby',
+ id: 'hobby',
+ nsfw: false
+ },
+ {
+ name: 'Kids',
+ id: 'kids',
+ nsfw: false
+ },
+ {
+ name: 'Legislative',
+ id: 'legislative',
+ nsfw: false
+ },
+ {
+ name: 'Lifestyle',
+ id: 'lifestyle',
+ nsfw: false
+ },
+ {
+ name: 'Local',
+ id: 'local',
+ nsfw: false
+ },
+ {
+ name: 'Movies',
+ id: 'movies',
+ nsfw: false
+ },
+ {
+ name: 'Music',
+ id: 'music',
+ nsfw: false
+ },
+ {
+ name: 'News',
+ id: 'news',
+ nsfw: false
+ },
+ {
+ name: 'Quiz',
+ id: 'quiz',
+ nsfw: false
+ },
+ {
+ name: 'Religious',
+ id: 'religious',
+ nsfw: false
+ },
+ {
+ name: 'Sci-Fi',
+ id: 'sci-fi',
+ nsfw: false
+ },
+ {
+ name: 'Shop',
+ id: 'shop',
+ nsfw: false
+ },
+ {
+ name: 'Sport',
+ id: 'sport',
+ nsfw: false
+ },
+ {
+ name: 'Travel',
+ id: 'travel',
+ nsfw: false
+ },
+ {
+ name: 'Weather',
+ id: 'weather',
+ nsfw: false
+ },
+ {
+ name: 'XXX',
+ id: 'xxx',
+ nsfw: true
+ }
+]
-helper.code2flag = function (code) {
+utils.code2flag = function (code) {
switch (code) {
case 'uk':
return '🇬🇧'
@@ -24,9 +172,16 @@ helper.code2flag = function (code) {
}
}
-helper.code2name = function (code) {
+utils.code2name = function (code) {
+ switch (code.toLowerCase()) {
+ case 'int':
+ return 'International'
+ case 'us':
+ return 'United States'
+ }
+
const intlDisplayNames = new Intl.DisplayNames(['en'], {
- style: 'long',
+ style: 'narrow',
type: 'region'
})
@@ -37,18 +192,34 @@ helper.code2name = function (code) {
}
}
-helper.language2code = function (name) {
+utils.language2code = function (name) {
const lang = iso6393.find(l => l.name === name)
return lang && lang.iso6393 ? lang.iso6393 : null
}
-helper.sortBy = function (arr, fields) {
+utils.sortBy = function (arr, fields) {
return arr.sort((a, b) => {
for (let field of fields) {
let propA = a[field] ? a[field].toLowerCase() : ''
let propB = b[field] ? b[field].toLowerCase() : ''
+ if (propA === 'undefined') {
+ return 1
+ }
+
+ if (propB === 'undefined') {
+ return -1
+ }
+
+ if (propA === 'other') {
+ return 1
+ }
+
+ if (propB === 'other') {
+ return -1
+ }
+
if (propA < propB) {
return -1
}
@@ -60,7 +231,7 @@ helper.sortBy = function (arr, fields) {
})
}
-helper.loadEPG = function (url) {
+utils.loadEPG = function (url) {
return new Promise((resolve, reject) => {
var buffer = []
axios({
@@ -96,11 +267,11 @@ helper.loadEPG = function (url) {
})
}
-helper.getBasename = function (filename) {
+utils.getBasename = function (filename) {
return path.basename(filename, path.extname(filename))
}
-helper.filterPlaylists = function (arr, include = '', exclude = '') {
+utils.filterPlaylists = function (arr, include = '', exclude = '') {
if (include) {
const included = include.split(',').map(filename => `channels/${filename}.m3u`)
@@ -116,7 +287,7 @@ helper.filterPlaylists = function (arr, include = '', exclude = '') {
return arr
}
-helper.generateTable = function (data, options) {
+utils.generateTable = function (data, options) {
let output = '<table>\n'
output += '\t<thead>\n\t\t<tr>'
@@ -145,77 +316,48 @@ helper.generateTable = function (data, options) {
return output
}
-helper.createDir = function (dir) {
+utils.createDir = function (dir) {
if (!fs.existsSync(dir)) {
fs.mkdirSync(dir)
}
}
-helper.readFile = function (filename) {
+utils.readFile = function (filename) {
return fs.readFileSync(path.resolve(__dirname) + `/../${filename}`, { encoding: 'utf8' })
}
-helper.appendToFile = function (filename, data) {
+utils.appendToFile = function (filename, data) {
fs.appendFileSync(path.resolve(__dirname) + '/../' + filename, data)
}
-helper.compileMarkdown = function (filepath) {
+utils.compileMarkdown = function (filepath) {
return markdownInclude.compileFiles(path.resolve(__dirname, filepath))
}
-helper.escapeStringRegexp = function (scring) {
+utils.escapeStringRegexp = function (scring) {
return escapeStringRegexp(string)
}
-helper.createFile = function (filename, data = '') {
+utils.createFile = function (filename, data = '') {
fs.writeFileSync(path.resolve(__dirname) + '/../' + filename, data)
}
-helper.writeToLog = function (country, msg, url) {
+utils.writeToLog = function (country, msg, url) {
var now = new Date()
var line = `${country}: ${msg} '${url}'`
this.appendToFile('error.log', now.toISOString() + ' ' + line + '\n')
}
-helper.filterNSFW = function (arr) {
- const sfwCategories = [
- 'Auto',
- 'Business',
- 'Classic',
- 'Comedy',
- 'Documentary',
- 'Education',
- 'Entertainment',
- 'Family',
- 'Fashion',
- 'Food',
- 'General',
- 'Health',
- 'History',
- 'Hobby',
- 'Kids',
- 'Legislative',
- 'Lifestyle',
- 'Local',
- 'Movies',
- 'Music',
- 'News',
- 'Quiz',
- 'Religious',
- 'Sci-Fi',
- 'Shop',
- 'Sport',
- 'Travel',
- 'Weather'
- ]
+utils.filterNSFW = function (arr) {
+ const sfwCategories = utils.supportedCategories.filter(c => !c.nsfw).map(c => c.name)
return arr.filter(i => sfwCategories.includes(i.category))
}
-helper.sleep = function (ms) {
+utils.sleep = function (ms) {
return function (x) {
return new Promise(resolve => setTimeout(() => resolve(x), ms))
}
}
-module.exports = helper
+module.exports = utils