electron-egg/app/service/storage.js

'use strict';

const BaseService = require('./base');
const path = require('path');
const _ = require('lodash');
const lowdb = require('lowdb');
const FileSync = require('lowdb/adapters/FileSync');
const storageKey = require('../const/storageKey');
const fs = require('fs');
const os = require('os');
const utils = require('../utils/utils');
const pkg = require('../../package.json');
const storageDb = 'db.json';

class StorageService extends BaseService {
  /*
   * instance
   */
  instance(file = null) {
    if (!file) {
      const storageDir = this.getStorageDir();
      if (!fs.existsSync(storageDir)) {
        utils.mkdir(storageDir);
        utils.chmodPath(storageDir, '777');
      }
      file = path.normalize(storageDir + storageDb);
    }
    const isExist = fs.existsSync(file);
    if (!isExist) {
        return null;
    }
  
    const adapter = new FileSync(file);
    const db = lowdb(adapter);
  
    return db;
  }

  /*
   * getElectronIPCPort
   */
  getElectronIPCPort() {
    const key = storageKey.ELECTRON_IPC + '.port';
    const port = this.instance()
    .get(key)
    .value();
  
    return port;
  }

  /*
   * getStorageDir
   */
  getStorageDir() {
    const userHomeDir = os.userInfo().homedir;
    const storageDir = path.normalize(userHomeDir + '/' + pkg.name + '/');

    return storageDir;
  }

  /*
   * 增 Test data
   */
  async addTestData(user) {
    const key = storageKey.TEST_DATA;
    if (!this.instance().has(key).value()) {
      this.instance().set(key, []).write();
    }
    
    const data = this.instance()
    .get(key)
    .push(user)
    .write();

    return data;
  }

  /*
   * 删 Test data
   */
  async delTestData(name = '') {
    const key = storageKey.TEST_DATA;
    const data = this.instance()
    .get(key)
    .remove({name: name})
    .write();

    return data;
  }

  /*
   * 改 Test data
   */
  async updateTestData(name= '', age = 0) {
    const key = storageKey.TEST_DATA;
    const data = this.instance()
    .get(key)
    .find({name: name}) // 修改找到的第一个数据，貌似无法批量修改 todo
    .assign({age: age})
    .write();

    return data;
  }

  /*
   * 查 Test data
   */
  async getTestData(age = 0) {
    const key = storageKey.TEST_DATA;
    let data = this.instance()
    .get(key)
    //.find({age: age}) 查找单个
    .filter(function(o) {
      let isHas = true;
      isHas = age === o.age ? true : false;
      return isHas;
    })
    //.orderBy(['age'], ['name']) 排序
    //.slice(0, 10) 分页
    .value();

    if (_.isEmpty(data)) {
      data = []
    }

    return data;
  }

  /*
   * all Test data
   */
    async getAllTestData() {
      const key = storageKey.TEST_DATA;
      if (!this.instance().has(key).value()) {
        this.instance().set(key, []).write();
      }
      let data = this.instance()
      .get(key)
      .value();
  
      if (_.isEmpty(data)) {
        data = []
      }
  
      return data;
    }
}

module.exports = StorageService;