Adhidarma Hadiwinoto / ym-store-n-forward

Browse Code »

Commit 53b5c5b0e3f733c4f5cf31a2ea8b8bab0cc9bb25

Authored by Adhidarma Hadiwinoto 10 years ago

1 parent 901cf9158a

Exists in master

relogin by http req

Showing 2 changed files with 60 additions and 46 deletions Inline Diff

index.js
ym.js

index.js

Diff comments View file @ 53b5c5b

ym.js

Diff comments View file @ 53b5c5b

 var YahooMessenger = require("yahoomessenger");
 var striptags = require('striptags');
 var db = require('./db.js');
+var oldMessages = [];
+var config;
+var logger;
 function insertMessages(db_connection, logger_id, direction, partner, message, sentdate) {
     db_connection.query(
         'INSERT INTO messages SET ?',
         {
             transport: 'YAHOO',
             logger: logger_id,
             direction: direction,
             partner: partner,
             message: message,
             sentdate: sentdate,
         },
         function(err) {
             if (err != null) {
                 logger.error("Error on inserting messages to DB, error: " + err);
             }
         }
     );
 }
-function start(config, logger) {
+function pm(destination, message) {
-    var oldMessages = [];
+    logger.info("Sending to " + destination + ": " + message);
-    var keepAliveTimer;
+    YahooMessenger.sendPM(destination, message);
+}
-    var _keepAliveLoop = function() {
-        logger.info('Keep alive loop');
-        YahooMessenger.keepAlive();
+function broadcast(destinations, message, exclude) {
-        keepAliveLoop();
+    var destinations = destinations.split(',');
-    }
+    var destinationCount = destinations.length;
-    function isAdmin(user) {
+    for (var i=0; i < destinationCount; i++) {
-      var admins = config.yahoomessenger.admin.split(',');
+        destination = destinations[i];
+        if (destination == exclude) {
+            continue;
+        }
-      return (admins.indexOf(user) >= 0);
+        pm(destination, message);
     }
+}
-    function keepAliveLoop() {
+var keepAliveTimer;
-        keepAliveTimer = setTimeout(_keepAliveLoop, config.yahoomessenger.keepalive_interval * 1000);
-    }
-    function login(){
+var _keepAliveLoop = function() {
-        logger.info("Login to YM as " + config.yahoomessenger.username)
+    logger.info('Keep alive loop');
-        YahooMessenger.login(config.yahoomessenger.username, config.yahoomessenger.password);
-    }
-    function pm(destination, message) {
+    YahooMessenger.keepAlive();
-        logger.info("Sending to " + destination + ": " + message);
+    keepAliveLoop();
+}
-        YahooMessenger.sendPM(destination, message);
+function isAdmin(user) {
-    }
+  var admins = config.yahoomessenger.admin.split(',');
-    function sendPM(destination, message) {
+  return (admins.indexOf(user) >= 0);
-        pm(destination, message);
+}
-    }
-    function broadcast(destinations, message, exclude) {
+function keepAliveLoop() {
-        var destinations = destinations.split(',');
+    keepAliveTimer = setTimeout(_keepAliveLoop, config.yahoomessenger.keepalive_interval * 1000);
-        var destinationCount = destinations.length;
+}
-        for (var i=0; i < destinationCount; i++) {
+function login(){
-            destination = destinations[i];
+    logger.info("Login to YM as " + config.yahoomessenger.username)
-            if (destination == exclude) {
+    YahooMessenger.login(config.yahoomessenger.username, config.yahoomessenger.password);
-                continue;
+}
-            }
-            sendPM(destination, message);
+function logout() {
-        }
+    logger.info("Logout from YM");
-    }
+    clearTimeout(keepAliveTimer);
+    YahooMessenger.logout();
+}
+function relogin() {
+    logout();
+    YahooMessenger.newInstance();
+}
+function start(_config, _logger) {
+    config = _config;
+    logger = _logger;
     function onReady() {
         login();
     }
     function onLoginSuccesful(data) {
         logger.info('Online on YM as ' + data.user_id + ' (' + data.firstname + ' ' + data.lastname + ')');
         if (config.yahoomessenger.status) {
             logger.info('Set status to: ' + config.yahoomessenger.status);
             YahooMessenger.setCustomAwayStatus(config.yahoomessenger.status);
         };
         broadcast(config.yahoomessenger.report_to, config.yahoomessenger.report_message);
         keepAliveLoop();
     }
     function onBuddyAddRequest(data) {
         logger.info("Got request to add buddy from " + data.username);
         YahooMessenger.acceptAddBuddy(data.username);
     }
     function onPM(data) {
         var direction = 'IN';
         if (config.globals.debug == 1) {
           console.log(data);
         }
         // check duplicate message
         if (oldMessages.indexOf(data.message_id) >= 0) {
             return;
         }
         oldMessages.unshift(data.message_id);
         if (oldMessages.length > 30) {
             oldMessages = oldMessages.slice(0, 10);
         }
         partner = data.sender;
         if (partner == data.user_id) {
             partner = data.target_user;
             direction = 'OUT';
         }
         var message = striptags(data.message).trim();
         logger.info('New ' + direction + ' message from ' + data.sender + ' to ' +  data.target_user + ': ' + message);
         if (isAdmin(data.sender)) {
             logger.info('Sender (' + data.sender + ') is an admin');
         }
         var sentdate = Date.parse(data.time);
         var strftime = require('strftime');
         sentdate = strftime('%Y-%m-%d %H:%M:%S', sentdate);
         insertMessages(
             db_connection,
             data.user_id,
             direction,
             partner,
             message,
             sentdate
         );
         broadcast(config.yahoomessenger.report_to, '@' + data.sender + ': ' + message, partner);
         if ((direction == 'IN') && (isAdmin(partner))) {
             if (message.charAt(0) == '@') {
                 target = message.split(' ', 1).join();
                 target = target.slice(1);
                 var messageToForward = message.replace('@' + target, '').trim();
                 if (messageToForward == '') {
                     return;
                 }
                 logger.info('Forward to ' + target + ': ' + messageToForward);
-                sendPM(target, '@' + partner + ': ' + messageToForward);
+                pm(target, '@' + partner + ': ' + messageToForward);
             } else if (message.charAt(0) == '+') {
                 target = message.split(' ', 1).join();
                 target = target.slice(1);
                 var messageToForward = message.replace('+' + target, '').trim();
                 if (messageToForward == '') {
                     return;
                 }
                 logger.info('Clean forward to ' + target + ': ' + messageToForward);
-                sendPM(target, messageToForward);
+                pm(target, messageToForward);
             }
         }
     }
     db_connection = db.start(config.db.host, config.db.username, config.db.password, config.db.name, logger);
     YahooMessenger.newInstance();
     YahooMessenger.on('ready', onReady);
     YahooMessenger.on('loginSuccessful', onLoginSuccesful);
     YahooMessenger.on('buddyAddRequest', onBuddyAddRequest);
     YahooMessenger.on('pm', onPM);
     return YahooMessenger;

1	var iniparser = require('iniparser');	1	var iniparser = require('iniparser');
2	var config = iniparser.parseSync('./config.ini');	2	var config = iniparser.parseSync('./config.ini');
3		3
4	if (config.yahoomessenger.keepalive_interval == null) {	4	if (config.yahoomessenger.keepalive_interval == null) {
5	config.yahoomessenger.keepalive_interval = 60;	5	config.yahoomessenger.keepalive_interval = 60;
6	}	6	}
7		7
8	var strftime = require('strftime');	8	var strftime = require('strftime');
9		9
10	var logger = require('winston');	10	var logger = require('winston');
11	logger.remove(logger.transports.Console);	11	logger.remove(logger.transports.Console);
12	logger.add(logger.transports.Console, { timestamp: function() { return (strftime('%F %T')); }});	12	logger.add(logger.transports.Console, { timestamp: function() { return (strftime('%F %T')); }});
13		13
14		14
15	var ym = require('./ym');	15	var ym = require('./ym');
16	var YahooMessenger = ym.start(config, logger);	16	var YahooMessenger = ym.start(config, logger);
17		17
18	var http = require('http');	18	var http = require('http');
19	var nsr = require('node-simple-router');	19	var nsr = require('node-simple-router');
20	var router = nsr();	20	var router = nsr();
21		21
22	router.get('/hello', function(request, response) {	22	router.get('/hello', function(request, response) {
23	response.end('Hello bro');	23	response.end('Hello bro');
24	});	24	});
25		25
26	router.get('/broadcastPing', function(request, response) {	26	router.get('/broadcastPing', function(request, response) {
27	response.end('Ok');	27	response.end('Ok');
28	broadcast(config.yahoomessenger.report_to, 'PING');	28	ym.broadcast(config.yahoomessenger.report_to, 'PING');
29	});	29	});
30		30
31	router.get('/status', function(request, response) {	31	router.get('/status', function(request, response) {
32	response.writeHead(200, {"Content-Type": "text/plain"});	32	response.writeHead(200, {"Content-Type": "text/plain"});
33	response.write(config.yahoomessenger.status);	33	response.write(config.yahoomessenger.status);
34	response.end();	34	response.end();
35	});	35	});
36		36
37	router.post('/status', function(request, response) {	37	router.post('/status', function(request, response) {
38	var apikey = request.body.apikey;	38	var apikey = request.body.apikey;
39	if (apikey != config.httpserver.apikey) {	39	if (apikey != config.httpserver.apikey) {
40	response.end('Invalid APIKEY: ' + apikey);	40	response.end('Invalid APIKEY: ' + apikey);
41	return;	41	return;
42	}	42	}
43	response.end('Ok');	43	response.end('Ok');
44		44
45	config.yahoomessenger.status = request.body.status;	45	config.yahoomessenger.status = request.body.status;
46	YahooMessenger.setCustomAwayStatus(config.yahoomessenger.status);	46	YahooMessenger.setCustomAwayStatus(config.yahoomessenger.status);
47	});	47	});
48		48
49	router.get('/relogin/:apikey', function(request, response) {	49	router.get('/relogin/:apikey', function(request, response) {
50	var apikey = request.params.apikey;	50	var apikey = request.params.apikey;
51		51
52	if (apikey != config.httpserver.apikey){	52	if (apikey != config.httpserver.apikey){
53	response.end('Invalid APIKEY: ' + apikey);	53	response.end('Invalid APIKEY: ' + apikey);
54	return;	54	return;
55	}	55	}
56		56
57	logger.info("Logging out");	57	logger.info("Got relogin request by http");
58	YahooMessenger.logout();	58	ym.relogin();
59		59	response.end('Ok');
60	logger.info("Logout");
61	clearTimeout(keepAliveTimer);
62
63	ymLogin();
64	});	60	});
65		61
66	logger.info("Starting http server on port " + config.httpserver.listen_port);	62	logger.info("Starting http server on port " + config.httpserver.listen_port);
67	var httpServer = http.createServer(router).listen(config.httpserver.listen_port);	63	var httpServer = http.createServer(router).listen(config.httpserver.listen_port);
68		64
69		65