|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322 |
- /* Copyright 2014-present Facebook, Inc.
- * Licensed under the Apache License, Version 2.0 */
-
- 'use strict';
-
- var net = require('net');
- var EE = require('events').EventEmitter;
- var util = require('util');
- var childProcess = require('child_process');
- var bser = require('bser');
-
- // We'll emit the responses to these when they get sent down to us
- var unilateralTags = ['subscription', 'log'];
-
- /**
- * @param options An object with the following optional keys:
- * * 'watchmanBinaryPath' (string) Absolute path to the watchman binary.
- * If not provided, the Client locates the binary using the PATH specified
- * by the node child_process's default env.
- */
- function Client(options) {
- var self = this;
- EE.call(this);
-
- this.watchmanBinaryPath = 'watchman';
- if (options && options.watchmanBinaryPath) {
- this.watchmanBinaryPath = options.watchmanBinaryPath.trim();
- };
- this.commands = [];
- }
- util.inherits(Client, EE);
-
- module.exports.Client = Client;
-
- // Try to send the next queued command, if any
- Client.prototype.sendNextCommand = function() {
- if (this.currentCommand) {
- // There's a command pending response, don't send this new one yet
- return;
- }
-
- this.currentCommand = this.commands.shift();
- if (!this.currentCommand) {
- // No further commands are queued
- return;
- }
-
- this.socket.write(bser.dumpToBuffer(this.currentCommand.cmd));
- }
-
- Client.prototype.cancelCommands = function(why) {
- var error = new Error(why);
-
- // Steal all pending commands before we start cancellation, in
- // case something decides to schedule more commands
- var cmds = this.commands;
- this.commands = [];
-
- if (this.currentCommand) {
- cmds.unshift(this.currentCommand);
- this.currentCommand = null;
- }
-
- // Synthesize an error condition for any commands that were queued
- cmds.forEach(function(cmd) {
- cmd.cb(error);
- });
- }
-
- Client.prototype.connect = function() {
- var self = this;
-
- function makeSock(sockname) {
- // bunser will decode the watchman BSER protocol for us
- self.bunser = new bser.BunserBuf();
- // For each decoded line:
- self.bunser.on('value', function(obj) {
- // Figure out if this is a unliteral response or if it is the
- // response portion of a request-response sequence. At the time
- // of writing, there are only two possible unilateral responses.
- var unilateral = false;
- for (var i = 0; i < unilateralTags.length; i++) {
- var tag = unilateralTags[i];
- if (tag in obj) {
- unilateral = tag;
- }
- }
-
- if (unilateral) {
- self.emit(unilateral, obj);
- } else if (self.currentCommand) {
- var cmd = self.currentCommand;
- self.currentCommand = null;
- if ('error' in obj) {
- var error = new Error(obj.error);
- error.watchmanResponse = obj;
- cmd.cb(error);
- } else {
- cmd.cb(null, obj);
- }
- }
-
- // See if we can dispatch the next queued command, if any
- self.sendNextCommand();
- });
- self.bunser.on('error', function(err) {
- self.emit('error', err);
- });
-
- self.socket = net.createConnection(sockname);
- self.socket.on('connect', function() {
- self.connecting = false;
- self.emit('connect');
- self.sendNextCommand();
- });
- self.socket.on('error', function(err) {
- self.connecting = false;
- self.emit('error', err);
- });
- self.socket.on('data', function(buf) {
- if (self.bunser) {
- self.bunser.append(buf);
- }
- });
- self.socket.on('end', function() {
- self.socket = null;
- self.bunser = null;
- self.cancelCommands('The watchman connection was closed');
- self.emit('end');
- });
- }
-
- // triggers will export the sock path to the environment.
- // If we're invoked in such a way, we can simply pick up the
- // definition from the environment and avoid having to fork off
- // a process to figure it out
- if (process.env.WATCHMAN_SOCK) {
- makeSock(process.env.WATCHMAN_SOCK);
- return;
- }
-
- // We need to ask the client binary where to find it.
- // This will cause the service to start for us if it isn't
- // already running.
- var args = ['--no-pretty', 'get-sockname'];
-
- // We use the more elaborate spawn rather than exec because there
- // are some error cases on Windows where process spawning can hang.
- // It is desirable to pipe stderr directly to stderr live so that
- // we can discover the problem.
- var proc = null;
- var spawnFailed = false;
-
- function spawnError(error) {
- if (spawnFailed) {
- // For ENOENT, proc 'close' will also trigger with a negative code,
- // let's suppress that second error.
- return;
- }
- spawnFailed = true;
- if (error.errno === 'EACCES') {
- error.message = 'The Watchman CLI is installed but cannot ' +
- 'be spawned because of a permission problem';
- } else if (error.errno === 'ENOENT') {
- error.message = 'Watchman was not found in PATH. See ' +
- 'https://facebook.github.io/watchman/docs/install.html ' +
- 'for installation instructions';
- }
- console.error('Watchman: ', error.message);
- self.emit('error', error);
- }
-
- try {
- proc = childProcess.spawn(this.watchmanBinaryPath, args, {
- stdio: ['ignore', 'pipe', 'pipe']
- });
- } catch (error) {
- spawnError(error);
- return;
- }
-
- var stdout = [];
- var stderr = [];
- proc.stdout.on('data', function(data) {
- stdout.push(data);
- });
- proc.stderr.on('data', function(data) {
- data = data.toString('utf8');
- stderr.push(data);
- console.error(data);
- });
- proc.on('error', function(error) {
- spawnError(error);
- });
-
- proc.on('close', function (code, signal) {
- if (code !== 0) {
- spawnError(new Error(
- self.watchmanBinaryPath + ' ' + args.join(' ') +
- ' returned with exit code=' + code + ', signal=' +
- signal + ', stderr= ' + stderr.join('')));
- return;
- }
- try {
- var obj = JSON.parse(stdout.join(''));
- if ('error' in obj) {
- var error = new Error(obj.error);
- error.watchmanResponse = obj;
- self.emit('error', error);
- return;
- }
- makeSock(obj.sockname);
- } catch (e) {
- self.emit('error', e);
- }
- });
- }
-
- Client.prototype.command = function(args, done) {
- done = done || function() {};
-
- // Queue up the command
- this.commands.push({cmd: args, cb: done});
-
- // Establish a connection if we don't already have one
- if (!this.socket) {
- if (!this.connecting) {
- this.connecting = true;
- this.connect();
- return;
- }
- return;
- }
-
- // If we're already connected and idle, try sending the command immediately
- this.sendNextCommand();
- }
-
- var cap_versions = {
- "cmd-watch-del-all": "3.1.1",
- "cmd-watch-project": "3.1",
- "relative_root": "3.3",
- "term-dirname": "3.1",
- "term-idirname": "3.1",
- "wildmatch": "3.7",
- }
-
- // Compares a vs b, returns < 0 if a < b, > 0 if b > b, 0 if a == b
- function vers_compare(a, b) {
- a = a.split('.');
- b = b.split('.');
- for (var i = 0; i < 3; i++) {
- var d = parseInt(a[i] || '0') - parseInt(b[i] || '0');
- if (d != 0) {
- return d;
- }
- }
- return 0; // Equal
- }
-
- function have_cap(vers, name) {
- if (name in cap_versions) {
- return vers_compare(vers, cap_versions[name]) >= 0;
- }
- return false;
- }
-
- // This is a helper that we expose for testing purposes
- Client.prototype._synthesizeCapabilityCheck = function(
- resp, optional, required) {
- resp.capabilities = {}
- var version = resp.version;
- optional.forEach(function (name) {
- resp.capabilities[name] = have_cap(version, name);
- });
- required.forEach(function (name) {
- var have = have_cap(version, name);
- resp.capabilities[name] = have;
- if (!have) {
- resp.error = 'client required capability `' + name +
- '` is not supported by this server';
- }
- });
- return resp;
- }
-
- Client.prototype.capabilityCheck = function(caps, done) {
- var optional = caps.optional || [];
- var required = caps.required || [];
- var self = this;
- this.command(['version', {
- optional: optional,
- required: required
- }], function (error, resp) {
- if (error) {
- done(error);
- return;
- }
- if (!('capabilities' in resp)) {
- // Server doesn't support capabilities, so we need to
- // synthesize the results based on the version
- resp = self._synthesizeCapabilityCheck(resp, optional, required);
- if (resp.error) {
- error = new Error(resp.error);
- error.watchmanResponse = resp;
- done(error);
- return;
- }
- }
- done(null, resp);
- });
- }
-
- // Close the connection to the service
- Client.prototype.end = function() {
- this.cancelCommands('The client was ended');
- if (this.socket) {
- this.socket.end();
- this.socket = null;
- }
- this.bunser = null;
- }
|