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 pkg = require('../../package.json');
const storageDir = path.normalize(os.userInfo().homedir + '/' + pkg.name + '/');
const storageDb = 'db.json';

class StorageService extends BaseService {
  /*
   * instance
   */
  instance(file = null) {
    if (!file) {
        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() {
    return storageDir;
  }

  /*
   * add 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;
  }

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

    return data;
  }

  /*
   * update Test data
   */
  async updateTestData(name= '', age = 0) {
    const key = storageKey.TEST_DATA;
    const data = this.instance()
    .get(key)
    .find({name: name})
    .assign({ age: age})
    .write();

    return data;
  }

  /*
   * get Test data
   */
  async getTestData(name = '') {
    const key = storageKey.TEST_DATA;
    const data = this.instance()
    .get(key)
    .find({name: name})
    .value();

    return data;
  }
}

module.exports = StorageService;