mirror of
https://github.com/iptv-org/iptv.git
synced 2024-09-06 00:02:24 +02:00
83 lines
1.4 KiB
JavaScript
83 lines
1.4 KiB
JavaScript
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
|