Adhidarma Hadiwinoto / ym-store-n-forward

Download as
Browse Code ยป

Commit 776e8a8525ab386a2ed06412ea22da318e94a833

Authored by Adhidarma Hadiwinoto
1 parent e6fc9c4a99
Exists in master

clean/raw relay, close #3

Showing 1 changed file with 12 additions and 0 deletions Inline Diff

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 loggedPM = []; 10 var loggedPM = [];
11 11
12 var logger = require('winston'); 12 var logger = require('winston');
13 logger.remove(logger.transports.Console); 13 logger.remove(logger.transports.Console);
14 logger.add(logger.transports.Console, { timestamp: function() { return (strftime('%F %T')); }}); 14 logger.add(logger.transports.Console, { timestamp: function() { return (strftime('%F %T')); }});
15 15
16 var db = require('./db.js'); 16 var db = require('./db.js');
17 db_connection = db.start(config.db.host, config.db.username, config.db.password, config.db.name, logger); 17 db_connection = db.start(config.db.host, config.db.username, config.db.password, config.db.name, logger);
18 18
19 function isAdmin(user) { 19 function isAdmin(user) {
20 var admins = config.yahoomessenger.admin.split(','); 20 var admins = config.yahoomessenger.admin.split(',');
21 21
22 return (admins.indexOf(user) >= 0); 22 return (admins.indexOf(user) >= 0);
23 } 23 }
24 24
25 function broadcast(destinations, message, exclude) { 25 function broadcast(destinations, message, exclude) {
26 var destinations = destinations.split(','); 26 var destinations = destinations.split(',');
27 var destinationCount = destinations.length; 27 var destinationCount = destinations.length;
28 28
29 for (var i=0; i < destinationCount; i++) { 29 for (var i=0; i < destinationCount; i++) {
30 destination = destinations[i]; 30 destination = destinations[i];
31 if (destination == exclude) { 31 if (destination == exclude) {
32 continue; 32 continue;
33 } 33 }
34 34
35 logger.info('Sending message to ' + destination); 35 logger.info('Sending message to ' + destination);
36 YahooMessenger.sendPM(destination, message); 36 YahooMessenger.sendPM(destination, message);
37 } 37 }
38 } 38 }
39 39
40 var _keepAliveLoop = function() { 40 var _keepAliveLoop = function() {
41 logger.info('Keep alive loop'); 41 logger.info('Keep alive loop');
42 42
43 try { 43 try {
44 YahooMessenger.keepAlive(); 44 YahooMessenger.keepAlive();
45 } 45 }
46 catch(e) { 46 catch(e) {
47 logger.error("Something wrong"); 47 logger.error("Something wrong");
48 logger.error(e); 48 logger.error(e);
49 logger.info("Try to reconnecting to yahoo messenger"); 49 logger.info("Try to reconnecting to yahoo messenger");
50 YahooMessenger.login(config.yahoomessenger.username, config.yahoomessenger.password); 50 YahooMessenger.login(config.yahoomessenger.username, config.yahoomessenger.password);
51 return; 51 return;
52 } 52 }
53 53
54 keepAliveLoop(); 54 keepAliveLoop();
55 } 55 }
56 56
57 function keepAliveLoop() { 57 function keepAliveLoop() {
58 setTimeout(_keepAliveLoop, config.yahoomessenger.keepalive_interval * 1000); 58 setTimeout(_keepAliveLoop, config.yahoomessenger.keepalive_interval * 1000);
59 } 59 }
60 60
61 function insertMessages(db_connection, logger_id, direction, partner, message, sentdate) { 61 function insertMessages(db_connection, logger_id, direction, partner, message, sentdate) {
62 62
63 db_connection.query( 63 db_connection.query(
64 'INSERT INTO messages SET ?', 64 'INSERT INTO messages SET ?',
65 { 65 {
66 transport: 'YAHOO', 66 transport: 'YAHOO',
67 logger: logger_id, 67 logger: logger_id,
68 direction: direction, 68 direction: direction,
69 partner: partner, 69 partner: partner,
70 message: message, 70 message: message,
71 sentdate: sentdate, 71 sentdate: sentdate,
72 }, 72 },
73 73
74 function(err) { 74 function(err) {
75 if (err != null) { 75 if (err != null) {
76 logger.error("Error on inserting messages to DB, error: " + err); 76 logger.error("Error on inserting messages to DB, error: " + err);
77 } 77 }
78 } 78 }
79 ); 79 );
80 } 80 }
81 81
82 var striptags = require('striptags'); 82 var striptags = require('striptags');
83 83
84 var YahooMessenger = require("yahoomessenger"); 84 var YahooMessenger = require("yahoomessenger");
85 YahooMessenger.newInstance(); 85 YahooMessenger.newInstance();
86 86
87 function onYmLoginSuccesful(data){ 87 function onYmLoginSuccesful(data){
88 logger.info('Online on YM as ' + data.user_id + ' (' + data.firstname + ' ' + data.lastname + ')'); 88 logger.info('Online on YM as ' + data.user_id + ' (' + data.firstname + ' ' + data.lastname + ')');
89 89
90 if (config.yahoomessenger.status) { 90 if (config.yahoomessenger.status) {
91 logger.info('Set status to: ' + config.yahoomessenger.status); 91 logger.info('Set status to: ' + config.yahoomessenger.status);
92 YahooMessenger.setCustomAwayStatus(config.yahoomessenger.status); 92 YahooMessenger.setCustomAwayStatus(config.yahoomessenger.status);
93 }; 93 };
94 94
95 broadcast(config.yahoomessenger.report_to, config.yahoomessenger.report_message); 95 broadcast(config.yahoomessenger.report_to, config.yahoomessenger.report_message);
96 96
97 keepAliveLoop(); 97 keepAliveLoop();
98 } 98 }
99 99
100 function onYmPM(data) { 100 function onYmPM(data) {
101 var direction = 'IN'; 101 var direction = 'IN';
102 102
103 if (config.globals.debug == 1) { 103 if (config.globals.debug == 1) {
104 console.log(data); 104 console.log(data);
105 } 105 }
106 106
107 // check duplicate message 107 // check duplicate message
108 if (loggedPM.indexOf(data.message_id) >= 0) { 108 if (loggedPM.indexOf(data.message_id) >= 0) {
109 return; 109 return;
110 } 110 }
111 111
112 loggedPM.unshift(data.message_id); 112 loggedPM.unshift(data.message_id);
113 if (loggedPM.length > 30) { 113 if (loggedPM.length > 30) {
114 loggedPM = loggedPM.slice(0, 10); 114 loggedPM = loggedPM.slice(0, 10);
115 } 115 }
116 116
117 partner = data.sender; 117 partner = data.sender;
118 if (partner == data.user_id) { 118 if (partner == data.user_id) {
119 partner = data.target_user; 119 partner = data.target_user;
120 direction = 'OUT'; 120 direction = 'OUT';
121 } 121 }
122 122
123 var message = striptags(data.message).trim(); 123 var message = striptags(data.message).trim();
124 124
125 logger.info('New ' + direction + ' message from ' + data.sender + ' to ' + data.target_user + ': ' + message); 125 logger.info('New ' + direction + ' message from ' + data.sender + ' to ' + data.target_user + ': ' + message);
126 126
127 if (isAdmin(data.sender)) { 127 if (isAdmin(data.sender)) {
128 logger.info('Sender (' + data.sender + ') is an admin'); 128 logger.info('Sender (' + data.sender + ') is an admin');
129 } 129 }
130 130
131 var sentdate = Date.parse(data.time); 131 var sentdate = Date.parse(data.time);
132 var strftime = require('strftime'); 132 var strftime = require('strftime');
133 sentdate = strftime('%Y-%m-%d %H:%M:%S', sentdate); 133 sentdate = strftime('%Y-%m-%d %H:%M:%S', sentdate);
134 134
135 insertMessages( 135 insertMessages(
136 db_connection, 136 db_connection,
137 data.user_id, 137 data.user_id,
138 direction, 138 direction,
139 partner, 139 partner,
140 message, 140 message,
141 sentdate 141 sentdate
142 ); 142 );
143 143
144 broadcast(config.yahoomessenger.report_to, '@' + data.sender + ': ' + message, partner); 144 broadcast(config.yahoomessenger.report_to, '@' + data.sender + ': ' + message, partner);
145 145
146 if ((direction == 'IN') && (isAdmin(partner))) { 146 if ((direction == 'IN') && (isAdmin(partner))) {
147 if (message.charAt(0) == '@') { 147 if (message.charAt(0) == '@') {
148 target = message.split(' ', 1).join(); 148 target = message.split(' ', 1).join();
149 target = target.slice(1); 149 target = target.slice(1);
150 150
151 var messageToForward = message.replace('@' + target, '').trim(); 151 var messageToForward = message.replace('@' + target, '').trim();
152 if (messageToForward == '') { 152 if (messageToForward == '') {
153 return; 153 return;
154 } 154 }
155 155
156 logger.info('Forward to ' + target + ': ' + messageToForward); 156 logger.info('Forward to ' + target + ': ' + messageToForward);
157 YahooMessenger.sendPM(target, '@' + partner + ': ' + messageToForward); 157 YahooMessenger.sendPM(target, '@' + partner + ': ' + messageToForward);
158
159 } else if (message.charAt(0) == '+') {
160 target = message.split(' ', 1).join();
161 target = target.slice(1);
162
163 var messageToForward = message.replace('+' + target, '').trim();
164 if (messageToForward == '') {
165 return;
166 }
167
168 logger.info('Clean forward to ' + target + ': ' + messageToForward);
169 YahooMessenger.sendPM(target, messageToForward);
158 } 170 }
159 171
160 } 172 }
161 } 173 }
162 174
163 function onYmBuddyAddRequest(data) { 175 function onYmBuddyAddRequest(data) {
164 logger.info("Got request to add buddy from " + data.username); 176 logger.info("Got request to add buddy from " + data.username);
165 YahooMessenger.acceptAddBuddy(data.username); 177 YahooMessenger.acceptAddBuddy(data.username);
166 } 178 }
167 179
168 180
169 function onYmReady(){ 181 function onYmReady(){
170 logger.info("Going to online on YM") 182 logger.info("Going to online on YM")
171 YahooMessenger.login(config.yahoomessenger.username, config.yahoomessenger.password); 183 YahooMessenger.login(config.yahoomessenger.username, config.yahoomessenger.password);
172 } 184 }
173 185
174 YahooMessenger.on('ready', onYmReady); 186 YahooMessenger.on('ready', onYmReady);
175 YahooMessenger.on('loginSuccessful', onYmLoginSuccesful); 187 YahooMessenger.on('loginSuccessful', onYmLoginSuccesful);
176 YahooMessenger.on('pm', onYmPM); 188 YahooMessenger.on('pm', onYmPM);
177 YahooMessenger.on('buddyAddRequest', onYmBuddyAddRequest); 189 YahooMessenger.on('buddyAddRequest', onYmBuddyAddRequest);
178 190
179 var http = require('http'); 191 var http = require('http');
180 var url = require('url'); 192 var url = require('url');
181 193
182 function onHttpRequest(request, response) { 194 function onHttpRequest(request, response) {
183 var pathname = url.parse(request.url).pathname; 195 var pathname = url.parse(request.url).pathname;
184 logger.info('Got HTTP request on ' + pathname); 196 logger.info('Got HTTP request on ' + pathname);
185 197
186 var queries = url.parse(request.url, true).query; 198 var queries = url.parse(request.url, true).query;
187 console.log(queries); 199 console.log(queries);
188 200
189 response.writeHead(200, {"Content-Type": "text/html"}); 201 response.writeHead(200, {"Content-Type": "text/html"});
190 response.write("Hello"); 202 response.write("Hello");
191 response.end(); 203 response.end();
192 } 204 }
193 205
194 http.createServer(onHttpRequest).listen(config.httpserver.listen_port); 206 http.createServer(onHttpRequest).listen(config.httpserver.listen_port);
195 logger.info("HTTP server listening on " + config.httpserver.listen_port); 207 logger.info("HTTP server listening on " + config.httpserver.listen_port);
196 208