czip/protoc/protoc.js

Nodejs console app for file encryption

const path = require('path');
const fs = require('fs-extra');
const { exec } = require('child_process');

class Protoc {
  constructor() {
    this.projectPath = path.resolve(__dirname, '../');

    const projectName = path.parse(this.projectPath).name;
    if (!fs.existsSync(this.projectPath)) {
      this.error(`Project "${projectName}" not found`);
    }

    this.projectProtoConfigPath = path.join(this.projectPath, 'protoc', 'proto');
    if (!fs.existsSync(this.projectProtoConfigPath + '.json')) {
      this.error('Project does not contain proto.json');
    }

    this.start();
  }

  start() {
    // List with names of .proto files. It's used in protoc.
    this.protocFilenameList = [];
    // List with absolute paths to each copied .proto file. To delete the files after use.
    this.protoFileList = [];

    // Load proto config of the project
    const projectProtoConfig = require(this.projectProtoConfigPath);
    if (!projectProtoConfig.exportPath) {
      this.error('Invalid proto.json. exportPath is not set');
    }
    this.projectExportPath = path.join(this.projectPath, projectProtoConfig.exportPath);

    // Clean proto folder
    fs.removeSync(this.projectExportPath);
    fs.mkdirSync(this.projectExportPath);

    // Copy project's .proto files to proto folder
    if (!projectProtoConfig.protoList || projectProtoConfig.protoList.length === 0) {
      this.error('proto.json does not contain any file');
    }
    for (const protoFile of projectProtoConfig.protoList) {
      this.copyPackage(path.join(this.projectPath, protoFile));
    }

    // Create proto functions from the .proto files in the proto folder
    const protocImports = this.protocFilenameList.join(' ');
    exec(
      'protoc ' +
      `--proto_path=${this.projectExportPath} ` +
      `--js_out=import_style=commonjs,binary:${this.projectExportPath} ` +
      protocImports, (error) => {
        this.removeProtoFiles(error);
      }
    );
  }

  // Copies a package to proto folder
  copyPackage(originProtoPath) {
    if (!fs.existsSync(originProtoPath)) {
      this.error(originProtoPath + ' not found');
    }
    const filename = path.parse(originProtoPath).base;
    const newPackagePath = path.join(this.projectExportPath, filename);

    if (this.protoFileList.indexOf(newPackagePath) === -1) {
      // If the file wasn't copied before, then copy it now
      this.protocFilenameList.push(filename);
      this.protoFileList.push(newPackagePath);
      fs.copySync(originProtoPath, newPackagePath);
    }
  }

  // Removes all *.proto files from the proto folder
  removeProtoFiles(error) {
    if (error) {
      throw error;
    }
    for (const protoFile of this.protoFileList) {
      fs.removeSync(protoFile);
    }
    this.onLoad();
  }

  error(message) {
    console.log('\x1b[31mERROR:\x1b[0m ' + message);
    process.exit();
  }

  onLoad() {
    console.log('\x1b[32m%s\x1b[0m', 'Proto functions are successfully created');
  }
}
const protoc = new Protoc();