Komodo / komodo-center-xmpp-messaging

Browse Code »

Commit b2a835600625377b4bdd59200c50bf64eaf77ab8

Authored by Adhidarma Hadiwinoto 4 years ago

1 parent 39e412b9c1

Exists in master

Register bot for custom-ping on connected

Showing 2 changed files with 6 additions and 0 deletions Inline Diff

lib/custom-ping.js
lib/transport.js

lib/custom-ping.js

Diff comments View file @ b2a8356

 const MODULE_NAME = 'CUSTOM-PING';
 const uniqid = require('uniqid');
 const config = require('komodo-sdk/config');
 const logger = require('tektrans-logger');
 let bot;
 const allowedFromPartnerList = (
     (config.custom_ping && config.custom_ping.allowed_from_partners) || []
 )
     .filter((item) => typeof item === 'string')
     .map((item) => (item || '').trim().toUpperCase())
     .filter((item) => item);
 const setBot = (botFromCaller) => {
+    logger.verbose(`${MODULE_NAME} A063F39F: Bot registered for custom ping`);
     bot = botFromCaller;
 };
 exports.setBot = setBot;
 const isPingMessage = (msg) => {
     if (!msg) return false;
     if (typeof msg !== 'string') return false;
     return ((msg || '').trim().toUpperCase().search(/^PING($| )/) === 0);
 };
 exports.isPingMessage = isPingMessage;
 const isPongMessage = (msg) => {
     if (!msg) return false;
     if (typeof msg !== 'string') return false;
     return ((msg || '').trim().toUpperCase().search(/^PONG($| )/) === 0);
 };
 exports.isPongMessage = isPongMessage;
 const isAllowedPartner = (partner) => {
     if (!config.custom_ping) return false;
     if (config.custom_ping.disable_pong) return false;
     if (config.custom_ping.allowed_from_all) return true;
     return allowedFromPartnerList.indexOf((partner || '').trim().toUpperCase()) >= 0;
 };
 exports.isAllowedPartner = isAllowedPartner;
 const sendPong = (xid, partner, pingMessage) => {
     if (!bot) return;
     if (!partner || typeof partner !== 'string' || !partner.trim()) return;
     const tokens = pingMessage.trim().split(/ +/);
     const pongMessage = [
         'PONG',
         (tokens && tokens[1]) || null,
     ].filter((item) => item)
         .join(' ');
     if (config.custom_ping && config.custom_ping.verbose) {
         logger.verbose(`${MODULE_NAME} F0D18EF4: Responding PING message`, {
             xid, partner, pongMessage,
         });
     }
     bot.send(partner.trim(), pongMessage);
 };
 exports.sendPong = sendPong;
 const pingSender = () => {
     const xid = uniqid();
     if (
         !bot
         || !config.custom_ping
         || !config.custom_ping.send_to
         || !Array.isArray(config.custom_ping.send_to)
         || !config.custom_ping.send_to.length
     ) return;
     config.custom_ping.send_to.forEach((partner) => {
         if (config.custom_ping.verbose) {
             logger.verbose(`${MODULE_NAME} 21358F45: Sending custom ping`, {
                 xid,
                 partner,
             });
         }
         bot.send(partner, `PING ${xid}`);
     });
 };
 if (config.custom_ping && config.custom_ping.send_to) {
     const intervalMs = config.custom_ping.interval_ms || (25 * 1000);
     logger.verbose(`${MODULE_NAME} 324034EE: Registering ping sender`, {
         intervalMs,
         partners: config.custom_ping.send_to,
     });
     setInterval(() => {
         pingSender();
     }, intervalMs);
 }

lib/transport.js

Diff comments View file @ b2a8356

 const MODULE_NAME = 'TRANSPORT';
 const SLEEP_BEFORE_TERMINATE_ON_ERROR_MS = 5 * 1000;
 const bot = require('simple-xmpp');
 const config = require('komodo-sdk/config');
 const logger = require('tektrans-logger');
 const messagingService = require('komodo-center-messaging-client-lib');
 const uniqid = require('uniqid');
 const customPing = require('./custom-ping');
 let isReady;
 bot.on('online', (data) => {
     logger.info(`XMPP transport connected, JID: ${data.jid.user}`);
+    if (config.custom_ping) {
+        customPing.setBot(bot);
+    }
     bot.getRoster();
     setTimeout(
         () => {
             isReady = true;
             logger.verbose('Transport is ready');
             if (messagingService.onOnline) {
                 messagingService.onOnline({
                     me: config.username,
                 });
             }
         },
         config.warming_up_ms || (30 * 1000),
     );
 });
 bot.on('chat', (partner, msg) => {
     if (!msg || !msg.trim()) {
         return;
     }
     if (partner === config.username.replace(/\/.*$/, '')) {
         return;
     }
     const xid = uniqid();
     if (customPing.isPongMessage(msg)) {
         if (config.custom_ping && config.custom_ping.verbose) {
             logger.verbose(`${MODULE_NAME} 70CDD087: Got PONG message`, {
                 xid,
                 partner,
                 msg,
             });
         }
         return;
     }
     if (customPing.isPingMessage(msg) && customPing.isAllowedPartner(partner)) {
         if (config.custom_ping && config.custom_ping.verbose) {
             logger.verbose(`${MODULE_NAME} ED8C8786: Processing PING message`, {
                 xid,
                 partner,
                 msg,
             });
         }
         customPing.sendPong(xid, partner, msg);
         return;
     }
     if (!isReady) {
         logger.warn('Warming up is not finished yet, ignoring message', {
             xid, me: config.username, partner, msg,
         });
         return;
     }
     logger.info('Incoming message via XMPP transport', {
         xid, me: config.username, partner, msg,
     });
     if (messagingService && messagingService.onIncomingMessage) {
         messagingService.onIncomingMessage(
             {
                 me: config.username,
                 partner,
                 msg: msg.trim(),
                 xid,
             },
         );
     }
 });
 bot.on('error', (err) => {
     const xid = uniqid();
     logger.warn(`${MODULE_NAME} F2E53C12: Error detected.`, {
         xid,
         eCode: err.code,
         eMessage: err.message,
     });
     if (!config.do_not_terminate_on_error) {
         logger.warn(`${MODULE_NAME} BA6C0C55: Terminating on error`, {
             xid,
             millisecondSleepBeforeTerminate: SLEEP_BEFORE_TERMINATE_ON_ERROR_MS,
         });
         setTimeout(() => {
             process.exit(1);
         }, SLEEP_BEFORE_TERMINATE_ON_ERROR_MS);
     }
 });
 function send(partner, msg) {
     logger.verbose('Sending message via XMPP transport', {
         transport: 'xmpp', me: config.username, partner, msg,
     });
     bot.send(partner, msg);
 }
 bot.on('subscribe', (from) => {
     logger.verbose(`Incoming subscribe request from ${from}`);
     bot.acceptSubscription(from);
     bot.subscribe(from);
 });
 function ping() {
     if (isReady) bot.send(config.username, 'PING!');
 }
 function init() {
     messagingService.setTransport(exports);
     bot.connect({
         jid: config.username,
         password: config.password,
         host: config.xmpp_host,
     });
     setInterval(
         ping,
         config.ping_interval_ms || 60000,
     );
 }
 init();
 exports.init = init;
 exports.send = send;