iptv/scripts/core/db.js

const nedb = require('nedb-promises')
const fs = require('fs-extra')
const file = require('./file')

const DB_DIR = process.env.DB_DIR || './scripts/tmp/database'

fs.ensureDirSync(DB_DIR)

class Database {
  constructor(filepath) {
    this.filepath = filepath
  }

  load() {
    this.db = nedb.create({
      filename: file.resolve(this.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
        })
      }
    })
  }

  removeIndex(field) {
    return this.db.removeIndex(field)
  }

  addIndex(options) {
    return this.db.ensureIndex(options)
  }

  compact() {
    return this.db.persistence.compactDatafile()
  }

  stopAutocompact() {
    return this.db.persistence.stopAutocompaction()
  }

  reset() {
    return file.clear(this.filepath)
  }

  count(query) {
    return this.db.count(query)
  }

  insert(doc) {
    return this.db.insert(doc)
  }

  update(query, update) {
    return this.db.update(query, update)
  }

  find(query) {
    return this.db.find(query)
  }

  all() {
    return this.find({})
  }

  remove(query, options) {
    return this.db.remove(query, options)
  }
}

const db = {}

db.streams = new Database(`${DB_DIR}/streams.db`)

module.exports = db