electron-egg/app/controller/v1/example.js

'use strict';

const BaseController = require('../base');
const os = require('os');
const fs = require('fs');
const path = require('path');

class ExampleController extends BaseController {

  async openLocalDir() {
    const self = this;
    const { ctx, service } = this;
    const body = ctx.request.body;
    const id = body.id;
    const data = {};
    let dir = '';
    switch (id) {
      case 'download' :
        dir = os.userInfo().homedir + '/Downloads';
        break;
      case 'picture' :
        dir = os.userInfo().homedir + '/Pictures';
        break;
      case 'doc' :
        dir = os.userInfo().homedir + '/Documents';
        break;
      case 'music' :
        dir = os.userInfo().homedir + '/Music';
        break;
    }

    await service.example.openLocalDir(dir);

    self.sendSuccess(data);
  }

  async executeJS() {
    const self = this;
    const { ctx, service } = this;
    const body = ctx.request.body;
    const str = body.str;
    let data = await service.example.executeJS(str);
    self.sendSuccess(data);
  }

  async uploadFile() {
    const self = this;
    const { ctx, service } = this;
    let tmpDir = service.storage.getStorageDir();
    // for (const file of ctx.request.files) {
    //   this.app.logger.info('file:', file);

    //   try {
    //     let tmpFile = fs.readFileSync(file.filepath)
    //     fs.writeFileSync(path.join(tmpDir, file.filename), tmpFile)
    //   } finally {
    //     await fs.unlink(file.filepath, function(){});
    //   }
    //   const fileStream = fs.createReadStream(path.join(tmpDir, file.filename))
    //   const uploadRes = await service.example.uploadFileToSMMS(fileStream);
    // }
    const file = ctx.request.files[0];
    //this.app.logger.info('file:', file);

    try {
      let tmpFile = fs.readFileSync(file.filepath)
      fs.writeFileSync(path.join(tmpDir, file.filename), tmpFile)
    } finally {
      await fs.unlink(file.filepath, function(){});
    }
    const fileStream = fs.createReadStream(path.join(tmpDir, file.filename))
    const uploadRes = await service.example.uploadFileToSMMS(fileStream);

    self.sendData(uploadRes);
  }

  async getWsUrl() {
    const self = this;
    const { service } = this;
    const data = {};

    const addr = await service.socket.getWsUrl();
    data.url = addr;

    self.sendSuccess(data);
  }

  async addTestData() {
    const self = this;
    const { service } = this;
    const data = {};

    const userInfo = {
      name: 'jame',
      age: 18,
      gender: 'man'
    }
    await service.storage.addTestData(userInfo);

    self.sendSuccess(data);
  }

  async delTestData() {
    const self = this;
    const { service } = this;
    const data = {};
    const name = 'jame';
    await service.storage.delTestData(name);

    self.sendSuccess(data);
  }

  async updateTestData() {
    const self = this;
    const { service } = this;
    const data = {};
    const name = 'jame';
    const age = 20;
    await service.storage.updateTestData(name, age);

    self.sendSuccess(data);
  }

  async getTestData() {
    const self = this;
    const { service } = this;
    const data = {};
    const name = 'jame';
    const user = await service.storage.getTestData(name);
    data.user = user;

    self.sendSuccess(data);
  }
}

module.exports = ExampleController;