Added formatter script

This commit is contained in:
freearhey 2019-04-30 16:52:38 +03:00
parent ee754bc44e
commit de37a7efad
2 changed files with 119 additions and 1 deletions

117
helpers/formatter.js Normal file
View File

@ -0,0 +1,117 @@
const parsers = require('playlist-parser')
const M3U = parsers.M3U
const fs = require("fs")
const path = require('path')
const urlParser = require('url')
let total = 0
function init() {
let countries = loadPlaylist('index.m3u')
// countries = countries.slice(0, 10)
let channels = []
for(let country of countries) {
const playlist = loadPlaylist(country.file)
for(let item of playlist) {
let channel = {
title: parseTitle(item),
file: item.file,
id: parseTvgId(item),
name: parseTvgName(item),
logo: parseTvgLogo(item),
group: parseGroupTitle(item)
}
channels.push(channel)
}
channels = channels.sort(byTitle)
let outputPath = path.resolve(__dirname) + '/../' + country.file
fs.writeFileSync(outputPath, '#EXTM3U\n')
channels.forEach(channel => {
writeToFile(outputPath, channel)
})
// console.log(country.title, channels)
total += channels.length
channels = []
}
}
init()
console.log(`Total: ${total}.`)
function loadPlaylist(filename) {
return M3U.parse(fs.readFileSync(path.resolve(__dirname) + `/../${filename}`, { encoding: "utf8" }))
}
function writeToFile(outputPath, channel) {
let info = `#EXTINF:-1 tvg-id="${channel.id}" tvg-name="${channel.name}" tvg-logo="${channel.logo}" group-title="${channel.group}",${channel.title}`
fs.appendFileSync(outputPath, info + '\n' + channel.file + '\n')
}
function getInfo(item) {
return (item.artist) ? item.artist + '-' + item.title : item.title
}
function parseTitle(item) {
let info = getInfo(item)
let parts = info.split(',')
return parts[parts.length - 1].trim()
}
function parseTvgId(item) {
let info = getInfo(item)
let matches = info.match(/tvg\-id\=\"(.*?)\"/i)
return (matches && matches.length > 1) ? matches[1] : ''
}
function parseTvgName(item) {
let info = getInfo(item)
let matches = info.match(/tvg\-name\=\"(.*?)\"/i)
return (matches && matches.length > 1) ? matches[1] : ''
}
function parseTvgLogo(item) {
let info = getInfo(item)
let matches = info.match(/tvg\-logo\=\"(.*?)\"/i)
return (matches && matches.length > 1) ? matches[1] : ''
}
function parseGroupTitle(item) {
let info = getInfo(item)
let matches = info.match(/group\-title\=\"(.*?)\"/i)
return (matches && matches.length > 1) ? matches[1] : ''
}
function byTitle(a, b) {
var nameA = a.title.toLowerCase()
var nameB = b.title.toLowerCase()
if (nameA < nameB) {
return -1
}
if (nameA > nameB) {
return 1
}
return 0
}

View File

@ -11,7 +11,8 @@
"main": "index.m3u",
"scripts": {
"test": "node test/index.js",
"generate": "node helpers/generate.js"
"generate": "node helpers/generate.js",
"format": "node helpers/formatter.js"
},
"repository": {
"type": "git",