Reading nick on welcome message fix
[KiwiIRC.git] / node / app.js
1 var tls = null;
2 var net = null;
3 var http = null;
4 var https = null;
5 var fs = null;
6 var url = null;
7 var dns = null;
8 var crypto = null;
9 var ws = null;
10 var jsp = null;
11 var pro = null;
12 var _ = null;
13 var starttls = null;
14 var kiwi = null;
15
16 this.init = function (objs) {
17 tls = objs.tls;
18 net = objs.net;
19 http = objs.http;
20 https = objs.https;
21 fs = objs.fs;
22 url = objs.url;
23 dns = objs.dns;
24 crypto = objs.crypto;
25 ws = objs.ws;
26 jsp = objs.jsp;
27 pro = objs.pro;
28 _ = objs._;
29 starttls = objs.starttls;
30 kiwi = require('./kiwi.js');
31 }
32
33
34
35
36
37
38 /*
39 * Some process changes
40 */
41 this.setTitle = function () {
42 process.title = 'kiwiirc';
43 }
44
45 this.changeUser = function () {
46 if (typeof kiwi.config.group !== 'undefined' && kiwi.config.group !== '') {
47 try {
48 process.setgid(kiwi.config.group);
49 } catch (err) {
50 console.log('Failed to set gid: ' + err);
51 process.exit();
52 }
53 }
54
55 if (typeof kiwi.config.user !== 'undefined' && kiwi.config.user !== '') {
56 try {
57 process.setuid(kiwi.config.user);
58 } catch (e) {
59 console.log('Failed to set uid: ' + e);
60 process.exit();
61 }
62 }
63 };
64
65
66
67
68
69
70
71
72
73 var ircNumerics = {
74 RPL_WELCOME: '001',
75 RPL_MYINFO: '004',
76 RPL_ISUPPORT: '005',
77 RPL_WHOISUSER: '311',
78 RPL_WHOISSERVER: '312',
79 RPL_WHOISOPERATOR: '313',
80 RPL_WHOISIDLE: '317',
81 RPL_ENDOFWHOIS: '318',
82 RPL_WHOISCHANNELS: '319',
83 RPL_NOTOPIC: '331',
84 RPL_TOPIC: '332',
85 RPL_NAMEREPLY: '353',
86 RPL_ENDOFNAMES: '366',
87 RPL_MOTD: '372',
88 RPL_WHOISMODES: '379',
89 ERR_NOSUCHNICK: '401',
90 ERR_CANNOTSENDTOCHAN: '404',
91 ERR_TOOMANYCHANNELS: '405',
92 ERR_NICKNAMEINUSE: '433',
93 ERR_USERNOTINCHANNEL: '441',
94 ERR_NOTONCHANNEL: '442',
95 ERR_NOTREGISTERED: '451',
96 ERR_LINKCHANNEL: '470',
97 ERR_CHANNELISFULL: '471',
98 ERR_INVITEONLYCHAN: '473',
99 ERR_BANNEDFROMCHAN: '474',
100 ERR_BADCHANNELKEY: '475',
101 ERR_CHANOPRIVSNEEDED: '482',
102 RPL_STARTTLS: '670'
103 };
104
105
106
107 this.parseIRCMessage = function (websocket, ircSocket, data) {
108 /*global ircSocketDataHandler */
109 var msg, regex, opts, options, opt, i, j, matches, nick, users, chan, channel, params, prefix, prefixes, nicklist, caps, rtn, obj;
110 //regex = /^(?::(?:([a-z0-9\x5B-\x60\x7B-\x7D\.\-]+)|([a-z0-9\x5B-\x60\x7B-\x7D\.\-]+)!([a-z0-9~\.\-_|]+)@?([a-z0-9\.\-:\/]+)?) )?([a-z0-9]+)(?:(?: ([^:]+))?(?: :(.+))?)$/i;
111 //regex = /^(?::(\S+) )?(\S+)(?: (?!:)(.+?))?(?: :(.+))?$/i;
112 regex = /^(?::(?:([a-z0-9\x5B-\x60\x7B-\x7D\.\-]+)|([a-z0-9\x5B-\x60\x7B-\x7D\.\-]+)!([a-z0-9~\.\-_|]+)@?([a-z0-9\.\-:\/]+)?) )?(\S+)(?: (?!:)(.+?))?(?: :(.+))?$/i;
113
114 msg = regex.exec(data);
115 if (msg) {
116 msg = {
117 prefix: msg[1],
118 nick: msg[2],
119 ident: msg[3],
120 hostname: msg[4] || '',
121 command: msg[5],
122 params: msg[6] || '',
123 trailing: (msg[7]) ? msg[7].trim() : ''
124 };
125
126 switch (msg.command.toUpperCase()) {
127 case 'PING':
128 websocket.sendServerLine('PONG ' + msg.trailing);
129 break;
130 case ircNumerics.RPL_WELCOME:
131 if (ircSocket.IRC.CAP.negotiating) {
132 ircSocket.IRC.CAP.negotiating = false;
133 ircSocket.IRC.CAP.enabled = [];
134 ircSocket.IRC.CAP.requested = [];
135 ircSocket.IRC.registered = true;
136 }
137 //regex = /([a-z0-9\x5B-\x60\x7B-\x7D\.\-]+)!([a-z0-9~\.\-_|]+)@?([a-z0-9\.\-:\/]+)/i;
138 //matches = regex.exec(msg.trailing);
139 nick = msg.params.split(' ')[0];
140 websocket.sendClientEvent('connect', {connected: true, host: null, nick: nick});
141 break;
142 case ircNumerics.RPL_ISUPPORT:
143 opts = msg.params.split(" ");
144 options = [];
145 for (i = 0; i < opts.length; i++) {
146 opt = opts[i].split("=", 2);
147 opt[0] = opt[0].toUpperCase();
148 ircSocket.IRC.options[opt[0]] = (typeof opt[1] !== 'undefined') ? opt[1] : true;
149 if (_.include(['NETWORK', 'PREFIX', 'CHANTYPES'], opt[0])) {
150 if (opt[0] === 'PREFIX') {
151 regex = /\(([^)]*)\)(.*)/;
152 matches = regex.exec(opt[1]);
153 if ((matches) && (matches.length === 3)) {
154 ircSocket.IRC.options[opt[0]] = [];
155 for (j = 0; j < matches[2].length; j++) {
156 //ircSocket.IRC.options[opt[0]][matches[2].charAt(j)] = matches[1].charAt(j);
157 ircSocket.IRC.options[opt[0]].push({symbol: matches[2].charAt(j), mode: matches[1].charAt(j)});
158 }
159
160 }
161 }
162 }
163 }
164
165 websocket.sendClientEvent('options', {server: '', "options": ircSocket.IRC.options});
166 break;
167 case ircNumerics.RPL_WHOISUSER:
168 case ircNumerics.RPL_WHOISSERVER:
169 case ircNumerics.RPL_WHOISOPERATOR:
170 case ircNumerics.RPL_ENDOFWHOIS:
171 case ircNumerics.RPL_WHOISCHANNELS:
172 case ircNumerics.RPL_WHOISMODES:
173 websocket.sendClientEvent('whois', {server: '', nick: msg.params.split(" ", 3)[1], "msg": msg.trailing});
174 break;
175 case ircNumerics.RPL_WHOISIDLE:
176 params = msg.params.split(" ", 4);
177 rtn = {server: '', nick: params[1], idle: params[2]};
178 if (params[3]) {
179 rtn.logon = params[3];
180 }
181 websocket.sendClientEvent('whois', rtn);
182 break;
183 case ircNumerics.RPL_MOTD:
184 websocket.sendClientEvent('motd', {server: '', "msg": msg.trailing});
185 break;
186 case ircNumerics.RPL_NAMEREPLY:
187 params = msg.params.split(" ");
188 nick = params[0];
189 chan = params[2];
190 users = msg.trailing.split(" ");
191 prefixes = _.values(ircSocket.IRC.options.PREFIX);
192 nicklist = {};
193 i = 0;
194 _.each(users, function (user) {
195 if (_.include(prefix, user.charAt(0))) {
196 prefix = user.charAt(0);
197 user = user.substring(1);
198 nicklist[user] = prefix;
199 } else {
200 nicklist[user] = '';
201 }
202 if (i++ >= 50) {
203 websocket.sendClientEvent('userlist', {server: '', 'users': nicklist, channel: chan});
204 nicklist = {};
205 i = 0;
206 }
207 });
208 if (i > 0) {
209 websocket.sendClientEvent('userlist', {server: '', "users": nicklist, channel: chan});
210 } else {
211 console.log("oops");
212 }
213 break;
214 case ircNumerics.RPL_ENDOFNAMES:
215 websocket.sendClientEvent('userlist_end', {server: '', channel: msg.params.split(" ")[1]});
216 break;
217 case ircNumerics.ERR_LINKCHANNEL:
218 params = msg.params.split(" ");
219 websocket.sendClientEvent('channel_redirect', {from: params[1], to: params[2]});
220 break;
221 case ircNumerics.ERR_NOSUCHNICK:
222 websocket.sendClientEvent('irc_error', {error: 'no_such_nick', nick: msg.params.split(" ")[1], reason: msg.trailing});
223 break;
224 case 'JOIN':
225 // Some BNC's send malformed JOIN causing the channel to be as a
226 // parameter instead of trailing.
227 if (typeof msg.trailing === 'string' && msg.trailing !== '') {
228 channel = msg.trailing;
229 } else if (typeof msg.params === 'string' && msg.params !== '') {
230 channel = msg.params;
231 }
232
233 websocket.sendClientEvent('join', {nick: msg.nick, ident: msg.ident, hostname: msg.hostname, channel: channel});
234 if (msg.nick === ircSocket.IRC.nick) {
235 websocket.sendServerLine('NAMES ' + msg.trailing);
236 }
237 break;
238 case 'PART':
239 websocket.sendClientEvent('part', {nick: msg.nick, ident: msg.ident, hostname: msg.hostname, channel: msg.params.trim(), message: msg.trailing});
240 break;
241 case 'KICK':
242 params = msg.params.split(" ");
243 websocket.sendClientEvent('kick', {kicked: params[1], nick: msg.nick, ident: msg.ident, hostname: msg.hostname, channel: params[0].trim(), message: msg.trailing});
244 break;
245 case 'QUIT':
246 websocket.sendClientEvent('quit', {nick: msg.nick, ident: msg.ident, hostname: msg.hostname, message: msg.trailing});
247 break;
248 case 'NOTICE':
249 if ((msg.trailing.charAt(0) === String.fromCharCode(1)) && (msg.trailing.charAt(msg.trailing.length - 1) === String.fromCharCode(1))) {
250 // It's a CTCP response
251 websocket.sendClientEvent('ctcp_response', {nick: msg.nick, ident: msg.ident, hostname: msg.hostname, channel: msg.params.trim(), msg: msg.trailing.substr(1, msg.trailing.length - 2)});
252 } else {
253 websocket.sendClientEvent('notice', {nick: msg.nick, ident: msg.ident, hostname: msg.hostname, target: msg.params.trim(), msg: msg.trailing});
254 }
255 break;
256 case 'NICK':
257 websocket.sendClientEvent('nick', {nick: msg.nick, ident: msg.ident, hostname: msg.hostname, newnick: msg.trailing});
258 break;
259 case 'TOPIC':
260 obj = {nick: msg.nick, channel: msg.params, topic: msg.trailing};
261 websocket.sendClientEvent('topic', obj);
262 break;
263 case ircNumerics.RPL_TOPIC:
264 obj = {nick: '', channel: msg.params.split(" ")[1], topic: msg.trailing};
265 websocket.sendClientEvent('topic', obj);
266 break;
267 case ircNumerics.RPL_NOTOPIC:
268 obj = {nick: '', channel: msg.params.split(" ")[1], topic: ''};
269 websocket.sendClientEvent('topic', obj);
270 break;
271 case 'MODE':
272 opts = msg.params.split(" ");
273 params = {nick: msg.nick};
274 switch (opts.length) {
275 case 1:
276 params.effected_nick = opts[0];
277 params.mode = msg.trailing;
278 break;
279 case 2:
280 params.channel = opts[0];
281 params.mode = opts[1];
282 break;
283 default:
284 params.channel = opts[0];
285 params.mode = opts[1];
286 params.effected_nick = opts[2];
287 break;
288 }
289 websocket.sendClientEvent('mode', params);
290 break;
291 case 'PRIVMSG':
292 if ((msg.trailing.charAt(0) === String.fromCharCode(1)) && (msg.trailing.charAt(msg.trailing.length - 1) === String.fromCharCode(1))) {
293 // It's a CTCP request
294 if (msg.trailing.substr(1, 6) === 'ACTION') {
295 websocket.sendClientEvent('action', {nick: msg.nick, ident: msg.ident, hostname: msg.hostname, channel: msg.params.trim(), msg: msg.trailing.substr(7, msg.trailing.length - 2)});
296 } else if (msg.trailing.substr(1, 7) === 'VERSION') {
297 ircSocket.write('NOTICE ' + msg.nick + ' :' + String.fromCharCode(1) + 'VERSION KiwiIRC' + String.fromCharCode(1) + '\r\n');
298 } else {
299 websocket.sendClientEvent('ctcp_request', {nick: msg.nick, ident: msg.ident, hostname: msg.hostname, channel: msg.params.trim(), msg: msg.trailing.substr(1, msg.trailing.length - 2)});
300 }
301 } else {
302 obj = {nick: msg.nick, ident: msg.ident, hostname: msg.hostname, channel: msg.params.trim(), msg: msg.trailing};
303 websocket.sendClientEvent('msg', obj);
304 }
305 break;
306 case 'CAP':
307 caps = kiwi.config.cap_options;
308 options = msg.trailing.split(" ");
309 switch (_.last(msg.params.split(" "))) {
310 case 'LS':
311 opts = '';
312 _.each(_.intersect(caps, options), function (cap) {
313 if (opts !== '') {
314 opts += " ";
315 }
316 opts += cap;
317 ircSocket.IRC.CAP.requested.push(cap);
318 });
319 if (opts.length > 0) {
320 websocket.sendServerLine('CAP REQ :' + opts);
321 } else {
322 websocket.sendServerLine('CAP END');
323 }
324 // TLS is special
325 /*if (_.include(options, 'tls')) {
326 websocket.sendServerLine('STARTTLS');
327 ircSocket.IRC.CAP.requested.push('tls');
328 }*/
329 break;
330 case 'ACK':
331 _.each(options, function (cap) {
332 ircSocket.IRC.CAP.enabled.push(cap);
333 });
334 if (_.last(msg.params.split(" ")) !== '*') {
335 ircSocket.IRC.CAP.requested = [];
336 ircSocket.IRC.CAP.negotiating = false;
337 websocket.sendServerLine('CAP END');
338 }
339 break;
340 case 'NAK':
341 ircSocket.IRC.CAP.requested = [];
342 ircSocket.IRC.CAP.negotiating = false;
343 websocket.sendServerLine('CAP END');
344 break;
345 }
346 break;
347 /*case ircNumerics.RPL_STARTTLS:
348 try {
349 IRC = ircSocket.IRC;
350 listeners = ircSocket.listeners('data');
351 ircSocket.removeAllListeners('data');
352 ssl_socket = starttls(ircSocket, {}, function () {
353 ssl_socket.on("data", function (data) {
354 ircSocketDataHandler(data, websocket, ssl_socket);
355 });
356 ircSocket = ssl_socket;
357 ircSocket.IRC = IRC;
358 _.each(listeners, function (listener) {
359 ircSocket.addListener('data', listener);
360 });
361 });
362 //console.log(ircSocket);
363 } catch (e) {
364 console.log(e);
365 }
366 break;*/
367 case ircNumerics.ERR_CANNOTSENDTOCHAN:
368 websocket.sendClientEvent('irc_error', {error: 'cannot_send_to_chan', channel: msg.params.split(" ")[1], reason: msg.trailing});
369 break;
370 case ircNumerics.ERR_TOOMANYCHANNELS:
371 websocket.sendClientEvent('irc_error', {error: 'too_many_channels', channel: msg.params.split(" ")[1], reason: msg.trailing});
372 break;
373 case ircNumerics.ERR_USERNOTINCHANNEL:
374 params = msg.params.split(" ");
375 websocket.sendClientEvent('irc_error', {error: 'user_not_in_channel', nick: params[0], channel: params[1], reason: msg.trainling});
376 break;
377 case ircNumerics.ERR_NOTONCHANNEL:
378 websocket.sendClientEvent('irc_error', {error: 'not_on_channel', channel: msg.params.split(" ")[1], reason: msg.trailing});
379 break;
380 case ircNumerics.ERR_CHANNELISFULL:
381 websocket.sendClientEvent('irc_error', {error: 'channel_is_full', channel: msg.params.split(" ")[1], reason: msg.trailing});
382 break;
383 case ircNumerics.ERR_INVITEONLYCHAN:
384 websocket.sendClientEvent('irc_error', {error: 'invite_only_channel', channel: msg.params.split(" ")[1], reason: msg.trailing});
385 break;
386 case ircNumerics.ERR_BANNEDFROMCHAN:
387 websocket.sendClientEvent('irc_error', {error: 'banned_from_channel', channel: msg.params.split(" ")[1], reason: msg.trailing});
388 break;
389 case ircNumerics.ERR_BADCHANNELKEY:
390 websocket.sendClientEvent('irc_error', {error: 'bad_channel_key', channel: msg.params.split(" ")[1], reason: msg.trailing});
391 break;
392 case ircNumerics.ERR_CHANOPRIVSNEEDED:
393 websocket.sendClientEvent('irc_error', {error: 'chanop_privs_needed', channel: msg.params.split(" ")[1], reason: msg.trailing});
394 break;
395 case ircNumerics.ERR_NICKNAMEINUSE:
396 websocket.sendClientEvent('irc_error', {error: 'nickname_in_use', nick: _.last(msg.params.split(" ")), reason: msg.trailing});
397 break;
398 case 'ERROR':
399 ircSocket.end();
400 websocket.sendClientEvent('irc_error', {error: 'error', reason: msg.trailing});
401 websocket.disconnect();
402 break;
403 case ircNumerics.ERR_NOTREGISTERED:
404 if (ircSocket.IRC.registered) {
405 console.log('Kiwi thinks user is registered, but the IRC server thinks differently');
406 }
407 break;
408 default:
409 console.log("Unknown command (" + String(msg.command).toUpperCase() + ")");
410 }
411 } else {
412 console.log("Malformed IRC line: " + data);
413 }
414 };
415
416
417
418
419
420
421 /*
422 * NOTE: Some IRC servers or BNC's out there incorrectly use
423 * only \n as a line splitter.
424 */
425 this.ircSocketDataHandler = function (data, websocket, ircSocket) {
426 var i;
427 if ((ircSocket.holdLast) && (ircSocket.held !== '')) {
428 data = ircSocket.held + data;
429 ircSocket.holdLast = false;
430 ircSocket.held = '';
431 }
432 if (data.substr(-1) !== '\n') {
433 ircSocket.holdLast = true;
434 }
435 data = data.split("\n");
436 for (i = 0; i < data.length; i++) {
437 if (data[i]) {
438 if ((ircSocket.holdLast) && (i === data.length - 1)) {
439 ircSocket.held = data[i];
440 break;
441 }
442
443 // We have a complete line of data, parse it!
444 kiwi.parseIRCMessage(websocket, ircSocket, data[i].replace(/^\r+|\r+$/, ''));
445 }
446 }
447 };
448
449
450
451
452
453 this.httpHandler = function (request, response) {
454 var uri, subs, useragent, agent, server_set, server, nick, debug, touchscreen, hash,
455 min = {}, public_http_path;
456 if (kiwi.config.handle_http) {
457 uri = url.parse(request.url, true);
458 subs = uri.pathname.substr(0, 4);
459 if (uri.pathname === '/js/all.js') {
460 if (kiwi.cache.alljs === '') {
461 public_http_path = kiwi.kiwi_root + '/' + kiwi.config.public_http;
462
463 min.util = fs.readFileSync(public_http_path + 'js/util.js');
464 min.gateway = fs.readFileSync(public_http_path + 'js/gateway.js');
465 min.front = fs.readFileSync(public_http_path + 'js/front.js');
466 min.iscroll = fs.readFileSync(public_http_path + 'js/iscroll.js');
467 min.ast = jsp.parse(min.util + min.gateway + min.front + min.iscroll);
468 min.ast = pro.ast_mangle(min.ast);
469 min.ast = pro.ast_squeeze(min.ast);
470 min.final_code = pro.gen_code(min.ast);
471 kiwi.cache.alljs = min.final_code;
472 hash = crypto.createHash('md5').update(kiwi.cache.alljs);
473 kiwi.cache.alljs_hash = hash.digest('base64');
474 }
475 if (request.headers['if-none-match'] === kiwi.cache.alljs_hash) {
476 response.statusCode = 304;
477 } else {
478 response.setHeader('ETag', kiwi.cache.alljs_hash);
479 response.write(kiwi.cache.alljs);
480 }
481 response.end();
482 } else if ((subs === '/js/') || (subs === '/css') || (subs === '/img')) {
483 request.addListener('end', function () {
484 kiwi.fileServer.serve(request, response);
485 });
486 } else if (uri.pathname === '/') {
487 useragent = (response.headers) ? response.headers['user-agent'] : '';
488 if (useragent.indexOf('android') !== -1) {
489 agent = 'android';
490 touchscreen = true;
491 } else if (useragent.indexOf('iphone') !== -1) {
492 agent = 'iphone';
493 touchscreen = true;
494 } else if (useragent.indexOf('ipad') !== -1) {
495 agent = 'ipad';
496 touchscreen = true;
497 } else if (useragent.indexOf('ipod') !== -1) {
498 agent = 'ipod';
499 touchscreen = true;
500 } else {
501 agent = 'normal';
502 touchscreen = false;
503 }
504
505 if (uri.query) {
506 server_set = ((typeof uri.query.server !== 'undefined') && (uri.query.server !== ''));
507 server = uri.query.server || 'irc.anonnet.org';
508 nick = uri.query.nick || '';
509 debug = (uri.query.debug !== '');
510 } else {
511 server_set = false;
512 server = 'irc.anonnet.org';
513 nick = '';
514 }
515 response.setHeader('X-Generated-By', 'KiwiIRC');
516 hash = crypto.createHash('md5').update(touchscreen ? 't' : 'f').update(debug ? 't' : 'f').update(server_set ? 't' : 'f').update(server).update(nick).update(agent).update(JSON.stringify(kiwi.config)).digest('base64');
517 if (kiwi.cache.html[hash]) {
518 if (request.headers['if-none-match'] === kiwi.cache.html[hash].hash) {
519 response.statusCode = 304;
520 } else {
521 response.setHeader('Etag', kiwi.cache.html[hash].hash);
522 response.write(kiwi.cache.html[hash].html);
523 }
524 response.end();
525 } else {
526 kiwi.jade.renderFile(__dirname + '/client/index.html.jade', { locals: { "touchscreen": touchscreen, "debug": debug, "server_set": server_set, "server": server, "nick": nick, "agent": agent, "config": kiwi.config }}, function (err, html) {
527 if (!err) {
528 var hash2 = crypto.createHash('md5').update(html).digest('base64');
529 kiwi.cache.html[hash] = {"html": html, "hash": hash2};
530 if (request.headers['if-none-match'] === hash2) {
531 response.statusCode = 304;
532 } else {
533 response.setHeader('Etag', hash2);
534 response.write(html);
535 }
536 } else {
537 response.statusCode = 500;
538 }
539 response.end();
540 });
541 }
542 } else if (uri.pathname.substr(0, 10) === '/socket.io') {
543 return;
544 } else {
545 response.statusCode = 404;
546 response.end();
547 }
548 }
549 };
550
551
552
553
554 this.websocketListen = function (port, host, handler, secure, key, cert) {
555 if (kiwi.httpServer) {
556 kiwi.httpServer.close();
557 }
558 if (secure) {
559 kiwi.httpServer = https.createServer({key: fs.readFileSync(__dirname + '/' + key), cert: fs.readFileSync(__dirname + '/' + cert)}, handler);
560 kiwi.io = ws.listen(kiwi.httpServer, {secure: true});
561 kiwi.httpServer.listen(port, host);
562 } else {
563 kiwi.httpServer = http.createServer(handler);
564 kiwi.io = ws.listen(kiwi.httpServer, {secure: false});
565 kiwi.httpServer.listen(port, host);
566 }
567
568 kiwi.io.set('log level', 1);
569 kiwi.io.enable('browser client minification');
570 kiwi.io.enable('browser client etag');
571 kiwi.io.set('transports', kiwi.config.transports);
572
573 kiwi.io.of('/kiwi').authorization(function (handshakeData, callback) {
574 var address = handshakeData.address.address;
575
576 if (typeof kiwi.connections[address] === 'undefined') {
577 kiwi.connections[address] = {count: 0, sockets: []};
578 }
579 callback(null, true);
580 }).on('connection', kiwi.websocketConnection);
581 };
582
583
584
585
586
587
588 this.websocketConnection = function (websocket) {
589 var con;
590 websocket.kiwi = {address: websocket.handshake.address.address};
591 con = kiwi.connections[websocket.kiwi.address];
592
593 if (con.count >= kiwi.config.max_client_conns) {
594 websocket.emit('too_many_connections');
595 websocket.disconnect();
596 } else {
597 con.count += 1;
598 con.sockets.push(websocket);
599
600 websocket.sendClientEvent = function (event_name, data) {
601 var ev = kiwi.kiwi_mod.run(event_name, data, {websocket: this});
602 if(ev === null) return;
603
604 data.event = event_name;
605 websocket.emit('message', data);
606 };
607
608 websocket.sendServerLine = function (data, eol) {
609 eol = (typeof eol === 'undefined') ? '\r\n' : eol;
610 websocket.ircSocket.write(data + eol);
611 };
612
613 websocket.on('irc connect', kiwi.websocketIRCConnect);
614 websocket.on('message', kiwi.websocketMessage);
615 websocket.on('disconnect', kiwi.websocketDisconnect);
616 }
617 };
618
619
620
621
622
623 this.websocketIRCConnect = function (websocket, nick, host, port, ssl, callback) {
624 var ircSocket;
625 //setup IRC connection
626 if (!ssl) {
627 ircSocket = net.createConnection(port, host);
628 } else {
629 ircSocket = tls.connect(port, host);
630 }
631 ircSocket.setEncoding('ascii');
632 ircSocket.IRC = {options: {}, CAP: {negotiating: true, requested: [], enabled: []}, registered: false};
633 ircSocket.on('error', function (e) {
634 if (ircSocket.IRC.registered) {
635 websocket.emit('disconnect');
636 } else {
637 websocket.emit('error', e.message);
638 }
639 });
640 websocket.ircSocket = ircSocket;
641 ircSocket.holdLast = false;
642 ircSocket.held = '';
643
644 ircSocket.on('data', function (data) {
645 kiwi.ircSocketDataHandler(data, websocket, ircSocket);
646 });
647
648 ircSocket.IRC.nick = nick;
649 // Send the login data
650 dns.reverse(websocket.kiwi.address, function (err, domains) {
651 //console.log(domains);
652 websocket.kiwi.hostname = (err) ? websocket.kiwi.address : _.first(domains);
653 if ((kiwi.config.webirc) && (kiwi.config.webirc_pass[host])) {
654 websocket.sendServerLine('WEBIRC ' + kiwi.config.webirc_pass[host] + ' KiwiIRC ' + websocket.kiwi.hostname + ' ' + websocket.kiwi.address);
655 }
656 websocket.sendServerLine('CAP LS');
657 websocket.sendServerLine('NICK ' + nick);
658 websocket.sendServerLine('USER ' + nick.replace(/[^0-9a-zA-Z\-_.]/, '') + '_kiwi 0 0 :' + nick);
659
660 if ((callback) && (typeof (callback) === 'function')) {
661 callback();
662 }
663 });
664 };
665
666
667
668 this.websocketMessage = function (websocket, msg, callback) {
669 var args, obj;
670 try {
671 msg.data = JSON.parse(msg.data);
672 args = msg.data.args;
673 switch (msg.data.method) {
674 case 'msg':
675 if ((args.target) && (args.msg)) {
676 obj = kiwi.kiwi_mod.run('msgsend', args, {websocket: websocket});
677 if (obj !== null) {
678 websocket.sendServerLine('PRIVMSG ' + args.target + ' :' + args.msg);
679 }
680 }
681 break;
682 case 'action':
683 if ((args.target) && (args.msg)) {
684 websocket.sendServerLine('PRIVMSG ' + args.target + ' :\ 1' + String.fromCharCode(1) + 'ACTION ' + args.msg + String.fromCharCode(1));
685 }
686 break;
687 case 'raw':
688 websocket.sendServerLine(args.data);
689 break;
690 case 'join':
691 if (args.channel) {
692 _.each(args.channel.split(","), function (chan) {
693 websocket.sendServerLine('JOIN ' + chan);
694 });
695 }
696 break;
697 case 'topic':
698 if (args.channel) {
699 websocket.sendServerLine('TOPIC ' + args.channel + ' :' + args.topic);
700 }
701 break;
702 case 'quit':
703 websocket.ircSocket.end('QUIT :' + args.message + '\r\n');
704 websocket.sentQUIT = true;
705 websocket.ircSocket.destroySoon();
706 websocket.disconnect();
707 break;
708 case 'notice':
709 if ((args.target) && (args.msg)) {
710 websocket.sendServerLine('NOTICE ' + args.target + ' :' + args.msg);
711 }
712 break;
713 default:
714 }
715 if ((callback) && (typeof (callback) === 'function')) {
716 callback();
717 }
718 } catch (e) {
719 console.log("Caught error: " + e);
720 }
721 };
722
723
724
725 this.websocketDisconnect = function (websocket) {
726 var con;
727
728 if ((!websocket.sentQUIT) && (websocket.ircSocket)) {
729 try {
730 websocket.ircSocket.end('QUIT :' + kiwi.config.quit_message + '\r\n');
731 websocket.sentQUIT = true;
732 websocket.ircSocket.destroySoon();
733 } catch (e) {
734 }
735 }
736 con = kiwi.connections[websocket.kiwi.address];
737 con.count -= 1;
738 con.sockets = _.reject(con.sockets, function (sock) {
739 return sock === websocket;
740 });
741 };
742
743
744
745
746
747
748 this.rehash = function () {
749 var changes, i,
750 reload_config = kiwi.loadConfig();
751
752 // If loading the new config errored out, dont attempt any changes
753 if (reload_config === false) {
754 return false;
755 }
756
757 // We just want the settings that have been changed
758 changes = reload_config[1];
759
760 if (Object.keys(changes).length !== 0) {
761 console.log('%s config changes: \n', Object.keys(changes).length, changes);
762 for (i in changes) {
763 switch (i) {
764 case 'port':
765 case 'bind_address':
766 case 'listen_ssl':
767 case 'ssl_key':
768 case 'ssl_cert':
769 kiwi.websocketListen(kiwi.config.port, kiwi.config.bind_address, kiwi.httpHandler, kiwi.config.listen_ssl, kiwi.config.ssl_key, kiwi.config.ssl_cert);
770 delete changes.port;
771 delete changes.bind_address;
772 delete changes.listen_ssl;
773 delete changes.ssl_key;
774 delete changes.ssl_cert;
775 break;
776 case 'user':
777 case 'group':
778 kiwi.changeUser();
779 delete changes.user;
780 delete changes.group;
781 break;
782 case 'module_dir':
783 case 'modules':
784 kiwi.kiwi_mod.loadModules(kiwi_root, kiwi.config);
785 kiwi.kiwi_mod.printMods();
786 delete changes.module_dir;
787 delete changes.modules;
788 break;
789 }
790 }
791 }
792
793 // Also clear the kiwi.cached javascript and HTML
794 if (kiwi.config.handle_http) {
795 kiwi.cache = {alljs: '', html: []};
796 }
797
798 return true;
799 };
800
801
802
803
804
805 /*
806 * KiwiIRC controlling via STDIN
807 */
808 this.manageControll = function (data) {
809 var parts = data.toString().trim().split(' ');
810 switch (parts[0]) {
811 case 'rehash':
812 console.log('Rehashing...');
813 console.log(kiwi.rehash() ? 'Rehash complete' : 'Rehash failed');
814 break;
815
816 case 'recode':
817 console.log('Recoding...');
818 console.log(kiwi.recode() ? 'Recode complete' : 'Recode failed');
819 break;
820
821 case 'mod':
822 if (parts[1] === 'reload') {
823 console.log('Reloading module (' + parts[2] + ')..');
824 kiwi.kiwi_mod.reloadModule(parts[2]);
825 }
826 break;
827
828 case 'cache':
829 if (parts[1] === 'clear') {
830 kiwi.cache.html = {};
831 console.log('HTML cache cleared');
832 }
833 break;
834
835 case 'status':
836 var connections_cnt = 0;
837 for (var i in kiwi.connections) {
838 connections_cnt = connections_cnt + parseInt(kiwi.connections[i].count, 10);
839 }
840 console.log(connections_cnt.toString() + ' connected clients');
841 break;
842
843 default:
844 console.log('Unknown command \'' + parts[0] + '\'');
845 }
846 };