first commit
This commit is contained in:
185
node_modules/streamws/bin/wscat
generated
vendored
Normal file
185
node_modules/streamws/bin/wscat
generated
vendored
Normal file
@ -0,0 +1,185 @@
|
||||
#!/usr/bin/env node
|
||||
|
||||
/*!
|
||||
* ws: a node.js websocket client
|
||||
* Copyright(c) 2011 Einar Otto Stangvik <einaros@gmail.com>
|
||||
* MIT Licensed
|
||||
*/
|
||||
|
||||
/**
|
||||
* Module dependencies.
|
||||
*/
|
||||
|
||||
var WebSocket = require('../')
|
||||
, fs = require('fs')
|
||||
, program = require('commander')
|
||||
, util = require('util')
|
||||
, events = require('events')
|
||||
, readline = require('readline');
|
||||
|
||||
/**
|
||||
* InputReader - processes console input
|
||||
*/
|
||||
|
||||
function Console() {
|
||||
this.stdin = process.stdin;
|
||||
this.stdout = process.stdout;
|
||||
|
||||
this.readlineInterface = readline.createInterface(this.stdin, this.stdout);
|
||||
|
||||
var self = this;
|
||||
this.readlineInterface.on('line', function(data) {
|
||||
self.emit('line', data);
|
||||
});
|
||||
this.readlineInterface.on('close', function() {
|
||||
self.emit('close');
|
||||
});
|
||||
|
||||
this._resetInput = function() {
|
||||
self.clear();
|
||||
}
|
||||
}
|
||||
util.inherits(Console, events.EventEmitter);
|
||||
|
||||
Console.Colors = {
|
||||
Red: '\033[31m',
|
||||
Green: '\033[32m',
|
||||
Yellow: '\033[33m',
|
||||
Blue: '\033[34m',
|
||||
Default: '\033[39m'
|
||||
};
|
||||
|
||||
Console.prototype.prompt = function() {
|
||||
this.readlineInterface.prompt();
|
||||
}
|
||||
|
||||
Console.prototype.print = function(msg, color) {
|
||||
this.clear();
|
||||
color = color || Console.Colors.Default;
|
||||
this.stdout.write(color + msg + Console.Colors.Default + '\n');
|
||||
this.prompt();
|
||||
}
|
||||
|
||||
Console.prototype.clear = function() {
|
||||
this.stdout.write('\033[2K\033[E');
|
||||
}
|
||||
|
||||
Console.prototype.pause = function() {
|
||||
this.stdin.on('keypress', this._resetInput);
|
||||
}
|
||||
|
||||
Console.prototype.resume = function() {
|
||||
this.stdin.removeListener('keypress', this._resetInput);
|
||||
}
|
||||
|
||||
/**
|
||||
* The actual application
|
||||
*/
|
||||
|
||||
var version = JSON.parse(fs.readFileSync(__dirname + '/../package.json', 'utf8')).version;
|
||||
program
|
||||
.version(version)
|
||||
.usage('[options] <url>')
|
||||
.option('-l, --listen <port>', 'listen on port')
|
||||
.option('-c, --connect <url>', 'connect to a websocket server')
|
||||
.option('-p, --protocol <version>', 'optional protocol version')
|
||||
.option('-o, --origin <origin>', 'optional origin')
|
||||
.parse(process.argv);
|
||||
|
||||
if (program.listen && program.connect) {
|
||||
console.error('\033[33merror: use either --listen or --connect\033[39m');
|
||||
process.exit(-1);
|
||||
}
|
||||
else if (program.listen) {
|
||||
var wsConsole = new Console();
|
||||
wsConsole.pause();
|
||||
var options = {};
|
||||
if (program.protocol) options.protocolVersion = program.protocol;
|
||||
if (program.origin) options.origin = program.origin;
|
||||
var ws = null;
|
||||
var wss = new WebSocket.Server({port: program.listen}, function() {
|
||||
wsConsole.print('listening on port ' + program.listen + ' (press CTRL+C to quit)', Console.Colors.Green);
|
||||
wsConsole.clear();
|
||||
});
|
||||
wsConsole.on('close', function() {
|
||||
if (ws) {
|
||||
try {
|
||||
ws.close();
|
||||
}
|
||||
catch (e) {}
|
||||
}
|
||||
process.exit(0);
|
||||
});
|
||||
wsConsole.on('line', function(data) {
|
||||
if (ws) {
|
||||
ws.send(data, {mask: true});
|
||||
wsConsole.prompt();
|
||||
}
|
||||
});
|
||||
wss.on('connection', function(newClient) {
|
||||
if (ws) {
|
||||
// limit to one client
|
||||
newClient.terminate();
|
||||
return;
|
||||
};
|
||||
ws = newClient;
|
||||
wsConsole.resume();
|
||||
wsConsole.prompt();
|
||||
wsConsole.print('client connected', Console.Colors.Green);
|
||||
ws.on('close', function() {
|
||||
wsConsole.print('disconnected', Console.Colors.Green);
|
||||
wsConsole.clear();
|
||||
wsConsole.pause();
|
||||
ws = null;
|
||||
});
|
||||
ws.on('error', function(code, description) {
|
||||
wsConsole.print('error: ' + code + (description ? ' ' + description : ''), Console.Colors.Yellow);
|
||||
});
|
||||
ws.on('message', function(data, flags) {
|
||||
wsConsole.print('< ' + data, Console.Colors.Blue);
|
||||
});
|
||||
});
|
||||
wss.on('error', function(error) {
|
||||
wsConsole.print('error: ' + error.toString(), Console.Colors.Yellow);
|
||||
process.exit(-1);
|
||||
});
|
||||
}
|
||||
else if (program.connect) {
|
||||
var wsConsole = new Console();
|
||||
var options = {};
|
||||
if (program.protocol) options.protocolVersion = program.protocol;
|
||||
if (program.origin) options.origin = program.origin;
|
||||
var ws = new WebSocket(program.connect, options);
|
||||
ws.on('open', function() {
|
||||
wsConsole.print('connected (press CTRL+C to quit)', Console.Colors.Green);
|
||||
wsConsole.on('line', function(data) {
|
||||
ws.send(data, {mask: true});
|
||||
wsConsole.prompt();
|
||||
});
|
||||
});
|
||||
ws.on('close', function() {
|
||||
wsConsole.print('disconnected', Console.Colors.Green);
|
||||
wsConsole.clear();
|
||||
process.exit();
|
||||
});
|
||||
ws.on('error', function(code, description) {
|
||||
wsConsole.print('error: ' + code + (description ? ' ' + description : ''), Console.Colors.Yellow);
|
||||
process.exit(-1);
|
||||
});
|
||||
ws.on('message', function(data, flags) {
|
||||
wsConsole.print('< ' + data, Console.Colors.Blue);
|
||||
});
|
||||
wsConsole.on('close', function() {
|
||||
if (ws) {
|
||||
try {
|
||||
ws.close();
|
||||
}
|
||||
catch(e) {}
|
||||
process.exit();
|
||||
}
|
||||
});
|
||||
}
|
||||
else {
|
||||
console.error('\033[33merror: use either --listen or --connect\033[39m');
|
||||
process.exit(-1);
|
||||
}
|
Reference in New Issue
Block a user