embark-area-51/lib/modules/vyper/index.js

let async = require('../../utils/async_extend.js');
const shelljs = require('shelljs');
const path = require('path');

class Vyper {

  constructor(embark, options) {
    this.logger = embark.logger;
    this.events = embark.events;
    this.contractDirectories = options.contractDirectories;

    embark.registerCompiler(".py", this.compile_vyper.bind(this));
  }

  compile_vyper(contractFiles, cb) {
    let self = this;
    async.waterfall([
      function compileContracts(callback) {
        self.logger.info("compiling vyper contracts...");
        const compiled_object = {};
        async.each(contractFiles,
          function (file, fileCb) {
            const className = path.basename(file.filename).split('.')[0];
            compiled_object[className] = {};
            async.parallel([
              function getByteCode(paraCb) {
                shelljs.exec(`vyper ${file.filename}`, {silent: true}, (code, stdout, stderr) => {
                  if (stderr) {
                    return paraCb(stderr);
                  }
                  if (code !== 0) {
                    return paraCb(`Vyper exited with error code ${code}`);
                  }
                  if (!stdout) {
                    return paraCb('Execution returned no bytecode');
                  }
                  const byteCode = stdout.replace(/\n/g, '');
                  compiled_object[className].runtimeBytecode = byteCode;
                  compiled_object[className].realRuntimeBytecode = byteCode;
                  compiled_object[className].code = byteCode;
                  paraCb();
                });
              },
              function getABI(paraCb) {
                shelljs.exec(`vyper -f json ${file.filename}`, {silent: true}, (code, stdout, stderr) => {
                  if (stderr) {
                    return paraCb(stderr);
                  }
                  if (code !== 0) {
                    return paraCb(`Vyper exited with error code ${code}`);
                  }
                  if (!stdout) {
                    return paraCb('Execution returned no ABI');
                  }
                  let ABI = [];
                  try {
                    ABI = JSON.parse(stdout.replace(/\n/g, ''));
                  } catch (e) {
                    return paraCb('ABI is not valid JSON');
                  }
                  compiled_object[className].abiDefinition = ABI;
                  paraCb();
                });
              }
            ], fileCb);
          },
          function (err) {
            callback(err, compiled_object);
          });
      }
    ], function (err, result) {
      cb(err, result);
    });
  }

}

module.exports = Vyper;