Adding profile command

This commit is contained in:
Anthony Laibe 2018-07-25 10:15:43 +01:00 committed by Iuri Matias
parent 7a34b9d675
commit 2f6e96db86
2 changed files with 15 additions and 1 deletions

View File

@ -3,6 +3,7 @@ const repl = require("repl");
class REPL { class REPL {
constructor(options) { constructor(options) {
this.env = options.env; this.env = options.env;
this.plugins = options.plugins;
} }
start(done) { start(done) {
@ -15,6 +16,19 @@ class REPL {
process.exit(); process.exit();
}); });
let self = this;
this.replServer.defineCommand('profile', {
help: 'Profile a contract',
action(contract) {
this.clearBufferedCommand();
let pluginCmds = self.plugins.getPluginsProperty('console', 'console');
for (let pluginCmd of pluginCmds) {
pluginCmd.call(self, `profile ${contract}`, {});
}
this.displayPrompt();
}
});
done(); done();
} }

View File

@ -281,7 +281,7 @@ class Embark {
}); });
}, },
function startREPL(callback) { function startREPL(callback) {
let repl = new REPL({env: engine.env}); let repl = new REPL({env: engine.env, plugins: engine.plugins});
repl.start(callback); repl.start(callback);
} }
], function (err, _result) { ], function (err, _result) {