Created commands/create-database.js

scripts/core/db.js

@@ -0,0 +1,61 @@
+const Database = require('nedb-promises')
+const file = require('./file')
+
+const DB_FILEPATH = process.env.DB_FILEPATH || './scripts/channels.db'
+
+const nedb = Database.create({
+  filename: file.resolve(DB_FILEPATH),
+  autoload: true,
+  onload(err) {
+    if (err) console.error(err)
+  },
+  compareStrings: (a, b) => {
+    a = a.replace(/\s/g, '_')
+    b = b.replace(/\s/g, '_')
+
+    return a.localeCompare(b, undefined, {
+      sensitivity: 'accent',
+      numeric: true
+    })
+  }
+})
+
+const db = {}
+
+db.removeIndex = function (field) {
+  return nedb.removeIndex(field)
+}
+
+db.addIndex = function (options) {
+  return nedb.ensureIndex(options)
+}
+
+db.compact = function () {
+  return nedb.persistence.compactDatafile()
+}
+
+db.reset = function () {
+  return file.clear(DB_FILEPATH)
+}
+
+db.count = function (query) {
+  return nedb.count(query)
+}
+
+db.insert = function (doc) {
+  return nedb.insert(doc)
+}
+
+db.update = function (query, update) {
+  return nedb.update(query, update)
+}
+
+db.find = function (query) {
+  return nedb.find(query)
+}
+
+db.remove = function (query, options) {
+  return nedb.remove(query, options)
+}
+
+module.exports = db

scripts/core/file.js

@@ -0,0 +1,67 @@
+const path = require('path')
+const glob = require('glob')
+const fs = require('mz/fs')
+
+const file = {}
+
+file.list = function (pattern) {
+  return new Promise(resolve => {
+    glob(pattern, function (err, files) {
+      resolve(files)
+    })
+  })
+}
+
+file.getFilename = function (filepath) {
+  return path.parse(filepath).name
+}
+
+file.createDir = async function (dir) {
+  if (await file.exists(dir)) return
+
+  return fs.mkdir(dir, { recursive: true }).catch(console.error)
+}
+
+file.exists = function (filepath) {
+  return fs.exists(path.resolve(filepath))
+}
+
+file.read = function (filepath) {
+  return fs.readFile(path.resolve(filepath), { encoding: 'utf8' }).catch(console.error)
+}
+
+file.append = function (filepath, data) {
+  return fs.appendFile(path.resolve(filepath), data).catch(console.error)
+}
+
+file.create = function (filepath, data = '') {
+  filepath = path.resolve(filepath)
+  const dir = path.dirname(filepath)
+
+  return file
+    .createDir(dir)
+    .then(() => fs.writeFile(filepath, data, { encoding: 'utf8', flag: 'w' }))
+    .catch(console.error)
+}
+
+file.write = function (filepath, data = '') {
+  return fs.writeFile(path.resolve(filepath), data).catch(console.error)
+}
+
+file.clear = function (filepath) {
+  return file.write(filepath, '')
+}
+
+file.resolve = function (filepath) {
+  return path.resolve(filepath)
+}
+
+file.dirname = function (filepath) {
+  return path.dirname(filepath)
+}
+
+file.basename = function (filepath) {
+  return path.basename(filepath)
+}
+
+module.exports = file

scripts/core/index.js

@@ -0,0 +1,4 @@
+exports.db = require('./db')
+exports.logger = require('./logger')
+exports.file = require('./file')
+exports.parser = require('./parser')

scripts/core/logger.js

@@ -0,0 +1,42 @@
+const { createLogger, format, transports, addColors } = require('winston')
+const { combine, timestamp, printf } = format
+
+const consoleFormat = ({ level, message, timestamp }) => {
+  if (typeof message === 'object') return JSON.stringify(message)
+  return message
+}
+
+const config = {
+  levels: {
+    error: 0,
+    warn: 1,
+    info: 2,
+    failed: 3,
+    success: 4,
+    http: 5,
+    verbose: 6,
+    debug: 7,
+    silly: 8
+  },
+  colors: {
+    info: 'white',
+    success: 'green',
+    failed: 'red'
+  }
+}
+
+const t = [
+  new transports.Console({
+    format: format.combine(format.printf(consoleFormat))
+  })
+]
+
+const logger = createLogger({
+  transports: t,
+  levels: config.levels,
+  level: 'verbose'
+})
+
+addColors(config.colors)
+
+module.exports = logger

scripts/core/parser.js

@@ -0,0 +1,31 @@
+const logger = require('./logger')
+const file = require('./file')
+const grabber = require('epg-grabber')
+
+const parser = {}
+
+parser.parseChannels = async function(filepath) {
+  const content = await file.read(filepath)
+  const channels = grabber.parseChannels(content)
+
+  return channels
+}
+
+parser.parseLogs = async function (filepath) {
+  const content = await file.read(filepath)
+  if (!content) return []
+  const lines = content.split('\n')
+
+  return lines.map(line => (line ? JSON.parse(line) : null)).filter(l => l)
+}
+
+parser.parseNumber = function (string) {
+  const parsed = parseInt(string)
+  if (isNaN(parsed)) {
+    throw new Error('scripts/core/parser.js:parseNumber() Input value is not a number')
+  }
+
+  return parsed
+}
+
+module.exports = parser