ee-core 1.5.1 → 1.5.2-beta.1

package/lib/storage/lowdb/adapters/Base.js

@@ -0,0 +1,15 @@
+const stringify = require('./_stringify')
+
+class Base {
+  constructor(
+    source,
+    { defaultValue = {}, serialize = stringify, deserialize = JSON.parse } = {}
+  ) {
+    this.source = source
+    this.defaultValue = defaultValue
+    this.serialize = serialize
+    this.deserialize = deserialize
+  }
+}
+
+module.exports = Base

package/lib/storage/lowdb/adapters/FileAsync.js

@@ -0,0 +1,41 @@
+// Not using async/await on purpose to avoid adding regenerator-runtime
+// to lowdb dependencies
+const fs = require('graceful-fs')
+const pify = require('pify')
+const steno = require('steno')
+const Base = require('./Base')
+
+const readFile = pify(fs.readFile)
+const writeFile = pify(steno.writeFile)
+
+class FileAsync extends Base {
+  read() {
+    // fs.exists is deprecated but not fs.existsSync
+    if (fs.existsSync(this.source)) {
+      // Read database
+      return readFile(this.source, 'utf-8')
+        .then(data => {
+          // Handle blank file
+          const trimmed = data.trim()
+          return trimmed ? this.deserialize(trimmed) : this.defaultValue
+        })
+        .catch(e => {
+          if (e instanceof SyntaxError) {
+            e.message = `Malformed JSON in file: ${this.source}\n${e.message}`
+          }
+          throw e
+        })
+    } else {
+      // Initialize
+      return writeFile(this.source, this.serialize(this.defaultValue)).then(
+        () => this.defaultValue
+      )
+    }
+  }
+
+  write(data) {
+    return writeFile(this.source, this.serialize(data))
+  }
+}
+
+module.exports = FileAsync

package/lib/storage/lowdb/adapters/FileSync.js

@@ -0,0 +1,39 @@
+const fs = require('graceful-fs')
+const Base = require('./Base')
+const fileSystem = require('fs')
+
+//const readFile = fs.readFileSync
+const writeFile = fs.writeFileSync
+
+// Same code as in FileAsync, minus `await`
+class FileSync extends Base {
+  read() {
+    // fs.exists is deprecated but not fs.existsSync
+    if (fs.existsSync(this.source)) {
+      // Read database
+      try {
+        // 使用 fileSystem的readFileSync
+        //const data = readFile(this.source, 'utf-8').trim()
+        const data = fileSystem.readFileSync(this.source, {encoding: 'utf-8'})
+
+        // Handle blank file
+        return data ? this.deserialize(data) : this.defaultValue
+      } catch (e) {
+        if (e instanceof SyntaxError) {
+          e.message = `Malformed JSON in file: ${this.source}\n${e.message}`
+        }
+        throw e
+      }
+    } else {
+      // Initialize
+      writeFile(this.source, this.serialize(this.defaultValue))
+      return this.defaultValue
+    }
+  }
+
+  write(data) {
+    return writeFile(this.source, this.serialize(data))
+  }
+}
+
+module.exports = FileSync

package/lib/storage/lowdb/adapters/LocalStorage.js

@@ -0,0 +1,20 @@
+/* global localStorage */
+const Base = require('./Base')
+
+class LocalStorage extends Base {
+  read() {
+    const data = localStorage.getItem(this.source)
+    if (data) {
+      return this.deserialize(data)
+    } else {
+      localStorage.setItem(this.source, this.serialize(this.defaultValue))
+      return this.defaultValue
+    }
+  }
+
+  write(data) {
+    localStorage.setItem(this.source, this.serialize(data))
+  }
+}
+
+module.exports = LocalStorage

package/lib/storage/lowdb/adapters/Memory.js

@@ -0,0 +1,8 @@
+const Base = require('./Base')
+
+module.exports = class Memory extends Base {
+  read() {
+    return this.defaultValue
+  }
+  write() {}
+}

package/lib/storage/lowdb/adapters/_stringify.js

@@ -0,0 +1,4 @@
+// Pretty stringify
+module.exports = function stringify(obj) {
+  return JSON.stringify(obj, null, 2)
+}

package/lib/storage/lowdb/common.js

@@ -0,0 +1,33 @@
+const isPromise = require('./is-promise')
+
+const init = (db, key, adapter) => {
+  db.read = () => {
+    const r = adapter.read()
+
+    return isPromise(r) ? r.then(db.plant) : db.plant(r)
+  }
+
+  db.write = (value = db.getState()) => {
+    const w = adapter.write(db.getState())
+
+    return isPromise(w) ? w.then(() => value) : value
+  }
+
+  db.plant = state => {
+    db[key] = state
+    return db
+  }
+
+  db.getState = () => db[key]
+
+  db.setState = state => {
+    db.plant(state)
+    return db
+  }
+
+  return db.read()
+}
+
+module.exports = {
+  init
+}

package/lib/storage/lowdb/fp.js

@@ -0,0 +1,23 @@
+const flow = require('lodash/flow')
+const get = require('lodash/get')
+const set = require('lodash/set')
+const common = require('./common')
+
+module.exports = function(adapter) {
+  function db(path, defaultValue) {
+    function getValue(funcs) {
+      const result = get(db.getState(), path, defaultValue)
+      return flow(funcs)(result)
+    }
+
+    getValue.write = (...funcs) => {
+      const result = getValue(...funcs)
+      set(db.getState(), path, result)
+      return db.write()
+    }
+
+    return getValue
+  }
+
+  return common.init(db, '__state__', adapter)
+}

package/lib/storage/lowdb/is-promise.js

@@ -0,0 +1,6 @@
+module.exports = isPromise;
+module.exports.default = isPromise;
+
+function isPromise(obj) {
+  return !!obj && (typeof obj === 'object' || typeof obj === 'function') && typeof obj.then === 'function';
+}

package/lib/storage/lowdb/main.js

@@ -0,0 +1,46 @@
+const lodash = require('lodash')
+const isPromise = require('is-promise')
+
+module.exports = function(adapter) {
+  if (typeof adapter !== 'object') {
+    throw new Error(
+      'An adapter must be provided, see https://github.com/typicode/lowdb/#usage'
+    )
+  }
+
+  // Create a fresh copy of lodash
+  const _ = lodash.runInContext()
+  const db = _.chain({})
+
+  // Add write function to lodash
+  // Calls save before returning result
+  _.prototype.write = _.wrap(_.prototype.value, function(func) {
+    const funcRes = func.apply(this)
+    return db.write(funcRes)
+  })
+
+  function plant(state) {
+    db.__wrapped__ = state
+    return db
+  }
+
+  // Lowdb API
+  // Expose _ for mixins
+  db._ = _
+
+  db.read = () => {
+    const r = adapter.read()
+    return isPromise(r) ? r.then(plant) : plant(r)
+  }
+
+  db.write = returnValue => {
+    const w = adapter.write(db.getState())
+    return isPromise(w) ? w.then(() => returnValue) : returnValue
+  }
+
+  db.getState = () => db.__wrapped__
+
+  db.setState = state => plant(state)
+
+  return db.read()
+}

package/lib/storage/lowdb/nano.js

@@ -0,0 +1,5 @@
+const common = require('./common')
+
+module.exports = function(adapter) {
+  return common.init({}, '__state__', adapter)
+}

package/lib/storage/lowdbStorage.js

@@ -3,8 +3,8 @@
 const assert = require('assert');
 const fs = require('fs');
 const path = require('path');
-const lowdb = require('lowdb');
-const FileSync = require('lowdb/adapters/FileSync');
+const lowdb = require('./lowdb/main');
+const FileSync = require('./lowdb/adapters/FileSync');
 const _ = require('lodash');
 const constant = require('../constant');
 const utilsCommon = require('../../utils/common');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ee-core",
-  "version": "1.5.1",
+  "version": "1.5.2-beta.1",
   "description": "ee core",
   "main": "index.js",
   "scripts": {
@@ -25,8 +25,10 @@
     "fs-extra": "^10.0.0",
     "get-port": "^5.1.1",
     "globby": "^10.0.0",
+    "graceful-fs": "^4.1.3",
     "humanize-ms": "^1.2.1",
     "is-type-of": "^1.2.1",
+    "javascript-obfuscator": "^4.0.0",
     "koa": "^2.13.4",
     "koa-body": "^5.0.0",
     "koa-convert": "^2.0.0",
@@ -35,9 +37,10 @@
     "lodash": "^4.17.21",
     "lowdb": "^1.0.0",
     "path-to-regexp": "^6.2.0",
+    "pify": "^6.1.0",
     "socket.io": "^4.4.1",
     "socket.io-client": "^4.4.1",
-    "javascript-obfuscator": "^4.0.0",
+    "steno": "^0.4.1",
     "urllib": "^2.38.0",
     "utility": "^1.17.0"
   },