Create channels/parse.test.js

scripts/commands/channels/parse.js

@@ -1,8 +1,7 @@
-const { db, logger } = require('../../core')
+const { logger, file, xml } = require('../../core')
 const { Command } = require('commander')
 const path = require('path')
 const _ = require('lodash')
-const fs = require('fs')
 
 const program = new Command()
 program
@@ -14,37 +13,25 @@ program
 const options = program.opts()
 
 async function main() {
-  await db.channels.load()
   const config = require(path.resolve(options.config))
   const args = {}
   options.set.forEach(arg => {
     const [key, value] = arg.split(':')
     args[key] = value
   })
+
   let channels = config.channels(args)
   if (isPromise(channels)) {
     channels = await channels
   }
-  channels = _.uniqBy(channels, 'site_id')
-
-  const siteChannels = await db.channels.find({ site: config.site })
-  for (const channel of channels) {
-    if (channel.xmltv_id) continue
-    const data = siteChannels.find(c => c.site_id === channel.site_id.toString())
-    if (data) {
-      channel.xmltv_id = data.xmltv_id
-      channel.name = data.name
-    }
-  }
-
   channels = _.sortBy(channels, 'xmltv_id')
 
-  const xml = json2xml(channels, config.site)
+  const dir = file.dirname(options.config)
+  const outputFilepath = options.output || `${dir}/${config.site}.channels.xml`
 
-  const dir = path.parse(options.config).dir
-  const output = options.output || `${dir}/${config.site}.channels.xml`
+  const output = xml.create(channels, config.site)
 
-  fs.writeFileSync(path.resolve(output), xml)
+  await file.write(outputFilepath, output)
 
   logger.info(`File '${output}' successfully saved`)
 }
@@ -54,51 +41,3 @@ main()
 function isPromise(promise) {
   return !!promise && typeof promise.then === 'function'
 }
-
-function json2xml(items, site) {
-  let output = `<?xml version="1.0" encoding="UTF-8"?>\r\n<site site="${site}">\r\n  <channels>\r\n`
-
-  items.forEach(channel => {
-    const logo = channel.logo ? ` logo="${channel.logo}"` : ''
-    const xmltv_id = channel.xmltv_id || ''
-    const lang = channel.lang || ''
-    const site_id = channel.site_id || ''
-    output += `    <channel lang="${lang}" xmltv_id="${escapeString(
-      xmltv_id
-    )}" site_id="${site_id}"${logo}>${escapeString(channel.name)}</channel>\r\n`
-  })
-
-  output += `  </channels>\r\n</site>\r\n`
-
-  return output
-}
-
-function escapeString(string, defaultValue = '') {
-  if (!string) return defaultValue
-
-  const regex = new RegExp(
-    '((?:[\0-\x08\x0B\f\x0E-\x1F\uFFFD\uFFFE\uFFFF]|[\uD800-\uDBFF](?![\uDC00-\uDFFF])|(?:[^\uD800-\uDBFF]|^)[\uDC00-\uDFFF]))|([\\x7F-\\x84]|[\\x86-\\x9F]|[\\uFDD0-\\uFDEF]|(?:\\uD83F[\\uDFFE\\uDFFF])|(?:\\uD87F[\\uDF' +
-      'FE\\uDFFF])|(?:\\uD8BF[\\uDFFE\\uDFFF])|(?:\\uD8FF[\\uDFFE\\uDFFF])|(?:\\uD93F[\\uDFFE\\uD' +
-      'FFF])|(?:\\uD97F[\\uDFFE\\uDFFF])|(?:\\uD9BF[\\uDFFE\\uDFFF])|(?:\\uD9FF[\\uDFFE\\uDFFF])' +
-      '|(?:\\uDA3F[\\uDFFE\\uDFFF])|(?:\\uDA7F[\\uDFFE\\uDFFF])|(?:\\uDABF[\\uDFFE\\uDFFF])|(?:\\' +
-      'uDAFF[\\uDFFE\\uDFFF])|(?:\\uDB3F[\\uDFFE\\uDFFF])|(?:\\uDB7F[\\uDFFE\\uDFFF])|(?:\\uDBBF' +
-      '[\\uDFFE\\uDFFF])|(?:\\uDBFF[\\uDFFE\\uDFFF])(?:[\\0-\\t\\x0B\\f\\x0E-\\u2027\\u202A-\\uD7FF\\' +
-      'uE000-\\uFFFF]|[\\uD800-\\uDBFF][\\uDC00-\\uDFFF]|[\\uD800-\\uDBFF](?![\\uDC00-\\uDFFF])|' +
-      '(?:[^\\uD800-\\uDBFF]|^)[\\uDC00-\\uDFFF]))',
-    'g'
-  )
-
-  string = String(string || '').replace(regex, '')
-
-  return string
-    .replace(/&/g, '&amp;')
-    .replace(/</g, '&lt;')
-    .replace(/>/g, '&gt;')
-    .replace(/"/g, '&quot;')
-    .replace(/'/g, '&apos;')
-    .replace(/\n|\r/g, ' ')
-    .replace(/  +/g, ' ')
-    .trim()
-}
-
-module.exports = { json2xml }

tests/__data__/expected/sites/parse-channels.channels.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<site site="parse-channels.com">
+  <channels>
+    <channel lang="en" xmltv_id="CNNInternational.us" site_id="140">CNN International</channel>
+  </channels>
+</site>

tests/__data__/input/sites/parse-channels.config.js

@@ -0,0 +1,19 @@
+module.exports = {
+  site: 'parse-channels.com',
+  url() {
+    return `https://parse-channels.com`
+  },
+  parser() {
+    return []
+  },
+  channels() {
+    return [
+      {
+        lang: 'en',
+        xmltv_id: 'CNNInternational.us',
+        site_id: 140,
+        name: 'CNN International'
+      }
+    ]
+  }
+}

tests/commands/channels/parse.test.js

@@ -0,0 +1,24 @@
+const { execSync } = require('child_process')
+const fs = require('fs-extra')
+const path = require('path')
+
+beforeEach(() => {
+  fs.emptyDirSync('tests/__data__/output')
+
+  const stdout = execSync(
+    'npm run channels:parse -- --config=tests/__data__/input/sites/parse-channels.config.js --output=tests/__data__/output/channels.xml',
+    { encoding: 'utf8' }
+  )
+})
+
+it('can parse channels', () => {
+  expect(content('tests/__data__/output/channels.xml')).toEqual(
+    content('tests/__data__/expected/sites/parse-channels.channels.xml')
+  )
+})
+
+function content(filepath) {
+  return fs.readFileSync(path.resolve(filepath), {
+    encoding: 'utf8'
+  })
+}