project.js

'use strict';

const YAML = require('yamljs');
const filesize = require('filesize');
const getStream = require('get-stream');
const Controller = require('egg').Controller;

const swaggerConvert = require('../../util').swaggerConvert;

class ProjectController extends Controller {

  async showAll() {
    const ctx = this.ctx;
    const res = [];
    const _res = await ctx.service.project.queryAllProject();
    for (const _item of _res) {
      const item = _item.get({ plain: true });
      res.push(item);
    }
    ctx.success(res);
  }

  async statistics() {
    const ctx = this.ctx;
    const res = [];
    const _res = await ctx.service.project.queryAllProject();
    for (const _item of _res) {
      const item = _item.get({ plain: true });
      const interfaceList = await ctx.service.interface.queryInterfaceByProjectUniqId({
        projectUniqId: item.uniqId,
      });
      const allSceneList = await Promise.all(interfaceList.map(({ uniqId: interfaceUniqId }) => {
        return ctx.service.scene.querySceneByInterfaceUniqId({ interfaceUniqId });
      }));
      let bufSize = 0;
      for (const sceneList of allSceneList) {
        for (const scene of sceneList) {
          const buf = new Buffer.from(JSON.stringify(scene.data));
          bufSize += buf.length;
        }
      }

      item.capacity = {
        count: interfaceList.length,
        size: filesize(bufSize),
      };
      res.push(item);
    }
    ctx.success(res);
  }

  async show() {
    const ctx = this.ctx;
    const { uniqId } = ctx.params;
    const res = await ctx.service.project.queryProjectByUniqId({ uniqId });
    ctx.success(res);
  }

  async create() {
    const ctx = this.ctx;
    const {
      projectName,
      description,
      globalProxy,
    } = ctx.request.body;
    ctx.assertParam({ projectName, description });
    const res = await ctx.service.project.createProject({
      projectName,
      description,
      globalProxy,
    });
    ctx.success(res);
  }

  async update() {
    const ctx = this.ctx;
    const { uniqId } = ctx.params;
    const payload = {};
    [ 'description', 'projectName', 'globalProxy' ].forEach(i => {
      if (ctx.request.body[i]) payload[i] = ctx.request.body[i];
    });
    const res = await ctx.service.project.updateProject({
      uniqId,
      payload,
    });
    ctx.success(res);
  }

  async delete() {
    const ctx = this.ctx;
    const { uniqId } = ctx.params;
    const res = await ctx.service.project.deleteProjectByUniqId({ uniqId });
    ctx.success(res);
  }

  async download() {
    const ctx = this.ctx;
    const { uniqId } = ctx.params;
    const res = await ctx.service.transfer.downloadProject({ uniqId });

    ctx.body = JSON.stringify(res.dataGroupList, null, 2);
    ctx.attachment(res.fileName);
  }

  async upload() {
    const ctx = this.ctx;
    const stream = await this.ctx.getFileStream();
    const projectUniqId = stream.fieldname;
    let projectData = null;
    const projectString = await getStream(stream);

    if (/.yaml$/.test(stream.filename)) {
      projectData = YAML.parse(projectString);
    } else {
      projectData = JSON.parse(projectString);
    }

    if (projectData.swagger) {
      projectData = swaggerConvert(projectData);
    }

    const res = await ctx.service.transfer.uploadProject({
      projectData,
      projectUniqId,
    });
    ctx.body = res;
  }
}

module.exports = ProjectController;