server/irc/state.js

   1 var util            = require('util'),
   2     events          = require('events'),
   3     _               = require('lodash'),
   4     IrcConnection   = require('./connection.js').IrcConnection;
   5
   6 var State = function (client, save_state) {
   7     var that = this;
   8
   9     events.EventEmitter.call(this);
  10     this.client = client;
  11     this.save_state = save_state || false;
  12
  13     this.irc_connections = [];
  14     this.next_connection = 0;
  15
  16     this.client.on('dispose', function () {
  17         if (!that.save_state) {
  18             _.each(that.irc_connections, function (irc_connection, i, cons) {
  19                 if (irc_connection) {
  20                     irc_connection.end('QUIT :' + (global.config.quit_message || ''));
  21                     irc_connection.dispose();
  22                     global.servers.removeConnection(irc_connection);
  23                     cons[i] = null;
  24                 }
  25             });
  26
  27             that.dispose();
  28         }
  29     });
  30 };
  31
  32 util.inherits(State, events.EventEmitter);
  33
  34 module.exports = State;
  35
  36 State.prototype.connect = function (hostname, port, ssl, nick, user, pass, callback) {
  37     var that = this;
  38     var con, con_num;
  39
  40     // Check the per-server limit on the number of connections
  41     if ((global.config.max_server_conns > 0) &&
  42         (!global.config.restrict_server) &&
  43         (!(global.config.webirc_pass && global.config.webirc_pass[hostname])) &&
  44         (!(global.config.ip_as_username && _.contains(global.config.ip_as_username, hostname))) &&
  45         (global.servers.numOnHost(hostname) >= global.config.max_server_conns))
  46     {
  47         return callback('Too many connections to host', {host: hostname, limit: global.config.max_server_conns});
  48     }
  49
  50     con = new IrcConnection(
  51         hostname,
  52         port,
  53         ssl,
  54         nick,
  55         user,
  56         pass,
  57         this);
  58
  59     con_num = this.next_connection++;
  60     this.irc_connections[con_num] = con;
  61     con.con_num = con_num;
  62
  63     new IrcCommands(con, con_num, this).bindEvents();
  64
  65     con.on('connected', function () {
  66         global.servers.addConnection(this);
  67         return callback(null, con_num);
  68     });
  69
  70     con.on('error', function (err) {
  71         console.log('irc_connection error (' + hostname + '):', err);
  72         return callback(err.code, {server: con_num, error: err});
  73     });
  74
  75     con.on('close', function () {
  76         that.irc_connections[con_num] = null;
  77         global.servers.removeConnection(this);
  78     });
  79 };
  80
  81 State.prototype.sendIrcCommand = function () {
  82     this.client.sendIrcCommand.apply(this.client, arguments);
  83 };
  84
  85 State.prototype.sendKiwiCommand = function () {
  86     this.client.sendKiwicommand.apply(this.client, arguments);
  87 };
  88
  89 State.prototype.dispose = function () {
  90     this.emit('dispose');
  91     this.removeAllListeners();
  92 };