2017-06-04 19:15:50 +02:00

317 lines
10 KiB
JavaScript
Executable File

#!/usr/bin/env node
const cli = require("commander");
const watchman = require('fb-watchman');
const fs = require('fs');
const chalk = require('chalk');
const mdns = require('mdns');
const pkgJson = require(__dirname + '/package.json');
const client = new watchman.Client();
const defaultIp = process.env.STATUS_DEVICE_IP;
const statusDebugServerPort = 5561;
const StatusDev = require('./index.js');
function createStatusDev() {
const ip = cli.ip || defaultIp;
if (ip == null) {
console.error(chalk.red("You have provide your device IP using --ip."));
process.exit(1);
}
return new StatusDev({ip: ip});
}
function fromAscii(str) {
var hex = "";
for(var i = 0; i < str.length; i++) {
var code = str.charCodeAt(i);
var n = code.toString(16);
hex += n.length < 2 ? '0' + n : n;
}
return "0x" + hex;
};
function makeSubscription(client, watch, relativePath, contactData) {
sub = {
expression: ["allof", ["match", "*.*"]],
fields: ["name"]
};
if (relativePath) {
sub.relative_root = relativePath;
}
client.command(['subscribe', watch, 'contact-subscription', sub],
function (error, resp) {
if (error) {
console.error('Failed to subscribe: ', error);
return;
}
console.log('Subscription established');
}
);
client.on('subscription', function (resp) {
if (resp.subscription !== 'contact-subscription') return;
resp.files.forEach(function (file) {
console.log('File changed: ' + file);
});
var statusDev = new StatusDev({ip: cli.ip || defaultIp});
statusDev.refreshContact(contactData, function(err, body) {
// nothing
});
});
}
function getCurrentPackageData() {
var obj = {};
if (fs.existsSync(process.cwd() + '/package.json')) {
var json = JSON.parse(fs.readFileSync(process.cwd() + '/package.json', 'utf8'));
obj["name"] = json.name;
obj["whisper-identity"] = "dapp-" + fromAscii(json.name);
obj["dapp-url"] = json["dapp-url"] || cli.dappUrl;
obj["bot-url"] = json["bot-url"] || cli.botUrl;
}
return obj;
}
function getPackageData(contact) {
var contactData = null;
if (!contact) {
var data = getCurrentPackageData();
if (data != null) {
contactData = JSON.stringify(data);
}
} else {
contactData = contact;
}
return contactData;
}
function printMan() {
console.error(chalk.red("Cannot connect to Status."));
console.log("1. Please, ensure that your device is accessible from your computer;");
console.log("2. If it is, ensure that you're logged in and the debug mode is enabled;");
console.log();
console.log("Check our docs for more information:");
console.log("http://docs.status.im/");
}
function printServerProblem() {
console.log("Server doesn't respond properly.");
console.log("Please, re-run it or re-login to your account.");
console.log();
console.log("Check our docs for more information:");
console.log("http://docs.status.im/");
}
cli.command("add [contact]")
.description("Adds a contact")
.action(function (contact) {
var statusDev = createStatusDev();
var contactData = getPackageData(contact);
if (contactData) {
statusDev.addContact(contactData, function(err, body) {
if (err) {
printMan();
} else if (body.type == "error") {
console.log(chalk.red(body.text));
} else {
console.log(chalk.green(body.text));
}
});
}
});
cli.command("remove [contactIdentity]")
.description("Removes a contact")
.action(function (contactIdentity) {
var statusDev = createStatusDev();
var contact = null;
if (contactIdentity) {
contact = JSON.stringify({"whisper-identity": contactIdentity});
}
var contactData = getPackageData(contact);
if (contactData) {
statusDev.removeContact(contactData, function(err, body) {
if (err) {
printMan();
} else if (body.type == "error") {
console.log(chalk.red(body.text));
} else {
console.log(chalk.green(body.text));
}
});
}
});
cli.command("refresh [contactIdentity]")
.description("Refreshes a debuggable contact")
.action(function (contactIdentity) {
var statusDev = createStatusDev();
var contact = null;
if (contactIdentity) {
contact = JSON.stringify({"whisper-identity": contactIdentity});
}
var contactData = getPackageData(contact);
if (contactData) {
statusDev.refreshContact(contactData, function(err, body) {
if (err) {
printMan();
} else if (body.type == "error") {
console.log(chalk.red(body.text));
} else {
console.log(chalk.green(body.text));
}
});
}
});
cli.command("switch-node <url>")
.description("Switches the current RPC node")
.action(function (url) {
var statusDev = createStatusDev();
statusDev.switchNode(url, function(err, body) {
if (err) {
printMan();
} else if (body.type == "error") {
console.log(chalk.red(body.text));
} else {
console.log(chalk.green(body.text));
}
});
});
cli.command("list")
.description("Displays all debuggable DApps and bots")
.action(function () {
var statusDev = createStatusDev();
statusDev.listDApps(function (err, body) {
if (err) {
printMan();
} else if (body === undefined) {
printServerProblem();
} else if (body.data === undefined) {
console.log(chalk.red("No DApps or bots."));
} else {
body.data.forEach(function(item, i, arr) {
if (item["dapp-url"]) {
console.log(chalk.green(chalk.bold(item["whisper-identity"]) +
" (Name: \"" + item.name + "\", DApp URL: \"" + item["dapp-url"] + "\")"));
} else if (item["bot-url"]) {
console.log(chalk.cyan(chalk.bold(item["whisper-identity"]) +
" (Name: \"" + item.name + "\", Bot URL: \"" + item["bot-url"] + "\")"));
}
});
}
});
});
cli.command("log <contactIdentity>")
.description("Returns log for a specified DApp or bot")
.action(function (contactIdentity) {
var statusDev = createStatusDev();
statusDev.getLog(contactIdentity, function (err, body) {
if (err) {
printMan();
} else if (body === undefined) {
printServerProblem();
} else if (body.type == "error") {
console.log(chalk.red(body.text));
} else {
body.data.forEach(function(item, i, arr) {
var time = new Date(item.timestamp).toLocaleString();
if (item.content.startsWith("error:")) {
console.log(chalk.red(time + " " + item.content));
} else if (item.content.startsWith("warn:")) {
console.log(chalk.cyan(time + " " + item.content));
} else {
console.log(time + " " + item.content);
}
});
}
});
});
cli.command("scan")
.description("Scans your network and searches for Status debug servers")
.action(function () {
console.log("Searching for connected devices...");
var sequence = [
mdns.rst.DNSServiceResolve(),
'DNSServiceGetAddrInfo' in mdns.dns_sd ? mdns.rst.DNSServiceGetAddrInfo({families:[0]}) : mdns.rst.getaddrinfo({families:[0]}),
mdns.rst.makeAddressesUnique()
];
var browser = mdns.createBrowser(mdns.tcp('http'), {resolverSequence: sequence});
browser.on('serviceUp', function(service) {
if (service.port == statusDebugServerPort) {
console.log(chalk.green(chalk.bold(service.name) + " (" + service.addresses.join(", ") + ")"));
}
});
browser.start();
});
cli.command("watch [dir] [contactIdentity]")
.description("Starts watching for contact changes")
.action(function (dir, contactIdentity) {
var contact = null;
if (contactIdentity) {
contact = JSON.stringify({"whisper-identity": contactIdentity});
}
var contactData = getPackageData(contact);
if (!contactData) {
return;
}
contactDir = dir || process.cwd();
console.log("Watching for changes in " + contactDir);
client.capabilityCheck(
{optional:[], required:['relative_root']},
function (error, resp) {
if (error) {
console.log(error);
client.end();
return;
}
client.command(
['watch-project', contactDir],
function (error, resp) {
if (error) {
console.error('Error initiating watch:', error);
return;
}
if ('warning' in resp) {
console.log('Warning: ', resp.warning);
}
makeSubscription(
client,
resp.watch,
resp.relative_path,
contactData
);
}
);
}
);
});
cli.on("*", function(command) {
console.error("Unknown command " + command[0] + ". See --help for valid commands.")
});
cli.version(pkgJson.version)
.option("--ip [ip]", "IP address of your device")
.option("--dappUrl [url]", "Custom DApp URL (overrides the one from the package.json)")
.option("--botUrl [url]", "Custom bot URL (overrides the one from the package.json)")
.parse(process.argv);