Merge branch 'server-time' of https://github.com/M2Ys4U/KiwiIRC into M2ys4U-server...
[KiwiIRC.git] / server / client.js
1 var util = require('util'),
2 events = require('events'),
3 crypto = require('crypto'),
4 _ = require('lodash'),
5 State = require('./irc/state.js'),
6 IrcConnection = require('./irc/connection.js').IrcConnection,
7 ClientCommands = require('./clientcommands.js'),
8 WebsocketRpc = require('./websocketrpc.js');
9
10
11 var Client = function (websocket) {
12 var that = this;
13
14 events.EventEmitter.call(this);
15 this.websocket = websocket;
16 this.rpc = new WebsocketRpc(this.websocket);
17
18 // Clients address
19 this.real_address = this.websocket.meta.real_address;
20
21 // A hash to identify this client instance
22 this.hash = crypto.createHash('sha256')
23 .update(this.real_address)
24 .update('' + Date.now())
25 .update(Math.floor(Math.random() * 100000).toString())
26 .digest('hex');
27
28 this.state = new State(this);
29
30 this.buffer = {
31 list: [],
32 motd: ''
33 };
34
35 // Handler for any commands sent from the client
36 this.client_commands = new ClientCommands(this);
37
38 this.rpc.on('irc', function (response, data) {
39 handleClientMessage.call(that, data, response);
40 });
41 this.rpc.on('kiwi', function (response, data) {
42 kiwiCommand.call(that, data, response);
43 });
44 websocket.on('close', function () {
45 websocketDisconnect.apply(that, arguments);
46 });
47 websocket.on('error', function () {
48 websocketError.apply(that, arguments);
49 });
50
51 this.disposed = false;
52 };
53 util.inherits(Client, events.EventEmitter);
54
55 module.exports.Client = Client;
56
57 // Callback API:
58 // Callbacks SHALL accept 2 arguments, error and response, in that order.
59 // error MUST be null where the command is successul.
60 // error MUST otherwise be a truthy value and SHOULD be a string where the cause of the error is known.
61 // response MAY be given even if error is truthy
62
63 Client.prototype.sendIrcCommand = function (command, data, callback) {
64 var c = {command: command, data: data};
65 this.rpc.call('irc', c, callback);
66 };
67
68 Client.prototype.sendKiwiCommand = function (command, data, callback) {
69 var c = {command: command, data: data};
70 this.rpc.call('kiwi', c, callback);
71 };
72
73 Client.prototype.dispose = function () {
74 this.disposed = true;
75 this.rpc.dispose();
76 this.emit('dispose');
77 this.removeAllListeners();
78 };
79
80 function handleClientMessage(msg, callback) {
81 var server;
82
83 // Make sure we have a server number specified
84 if ((msg.server === null) || (typeof msg.server !== 'number')) {
85 return (typeof callback === 'function') ? callback('server not specified') : undefined;
86 } else if (!this.state.irc_connections[msg.server]) {
87 return (typeof callback === 'function') ? callback('not connected to server') : undefined;
88 }
89
90 // The server this command is directed to
91 server = this.state.irc_connections[msg.server];
92
93 if (typeof callback !== 'function') {
94 callback = null;
95 }
96
97 try {
98 msg.data = JSON.parse(msg.data);
99 } catch (e) {
100 kiwi.log('[handleClientMessage] JSON parsing error ' + msg.data);
101 return;
102 }
103
104 // Run the client command
105 this.client_commands.run(msg.data.method, msg.data.args, server, callback);
106 }
107
108
109
110
111 function kiwiCommand(command, callback) {
112 if (typeof callback !== 'function') {
113 callback = function () {};
114 }
115
116 switch (command.command) {
117 case 'connect':
118 if (command.hostname && command.port && command.nick) {
119 var options = {};
120
121 // Get any optional parameters that may have been passed
122 if (command.encoding)
123 options.encoding = command.encoding;
124
125 options.password = global.config.restrict_server_password || command.password;
126
127 this.state.connect(
128 (global.config.restrict_server || command.hostname),
129 (global.config.restrict_server_port || command.port),
130 (typeof global.config.restrict_server_ssl !== 'undefined' ?
131 global.config.restrict_server_ssl :
132 command.ssl),
133 command.nick,
134 {hostname: this.websocket.meta.revdns, address: this.websocket.meta.real_address},
135 options,
136 callback);
137 } else {
138 return callback('Hostname, port and nickname must be specified');
139 }
140 break;
141 default:
142 callback();
143 }
144 }
145
146
147 // Websocket has disconnected, so quit all the IRC connections
148 function websocketDisconnect() {
149 this.emit('disconnect');
150
151 this.dispose();
152 }
153
154
155 // TODO: Should this close all the websocket connections too?
156 function websocketError() {
157 this.dispose();
158 }