chomens-bot-js/plugins/console.js

/* eslint-disable require-jsdoc */
/* eslint-disable max-len */
const moment = require('moment-timezone');
const util = require('util');

function inject(bot, _dcclient, config, rl) {
  // readline > fix on log
  function log(...args) {
    rl.output.write('\x1b[2K\r');
    console.log(args.toString());
    rl._refreshLine();
  };

  const chatMessage = require('prismarine-chat')(bot.version);

  function prefix(prefix, _message) {
    const message = `[${moment().format('HH:mm:ss')} ${prefix}§r] [${bot.options.host}] `;
    const component = chatMessage.MessageBuilder.fromString(message).toJSON();
    return chatMessage.fromNotch(component).toAnsi() + _message;
  }

  bot.console = {};
  bot.console.host = 'all';
  bot.console.log = function(message) {
    log(prefix('&6LOG', message));
  };
  bot.console.info = function(message) {
    log(prefix('&aINFO', message));
  };
  bot.console.error = function(error) {
    log(prefix('&cERROR', typeof error === 'string' ? error : error.stack));
  };

  // bot.consoleQueue = [];
  // const consoleQueueInterval = setInterval(function() {
  //   if (!bot.messageLogging) return;
  //   if (bot.consoleQueue.length > 50) bot.consoleQueue = [];
  //   if (bot.consoleQueue[0] || bot.consoleQueue[0] === '') {
  //     bot.console.log(bot.consoleQueue[0].substring(0, 10000));
  //     bot.consoleQueue.shift();
  //   }
  // }, 100);

  bot.on('parsed_chat', (message) => {
    if (!bot.messageLogging) return;
    bot.console.log(message.toAnsi());
  });

  // bot.once('end', () => {
  //   clearInterval(consoleQueueInterval);
  //   rl.removeAllListeners();
  // });
}

function oneTimeInject(bot, _dcclient, config, rl) {
  if (!config.console) return;

  bot.console.setRl = setRl;
  function setRl(rl) {
    rl.on('line', function(line) {
      try {
        if (line.toLowerCase() === '' ||
          line.toLowerCase().startsWith(' ')) return;

        if (line.startsWith('.csvr ')) {
          const host = line.substring(6);
          bot.getBots().forEach((eachBot) => eachBot.console.host = host);
          bot.console.info(`Host set to: ${host}`);
          return;
        }

        if (bot.options.host !== bot.console.host && bot.console.host !== 'all') return;
        if (line.toLowerCase() === '.exit' || line.toLowerCase() === '.end') {
          bot.end('end command');
          return;
        }
        if (line.toLowerCase().startsWith('.servereval ')) {
          try {
            bot.tellraw('@a', {
              text: `${util.inspect(eval(`${line.substring(12)}`))}`,
              color: 'green',
            });
            return;
          } catch (err) {
            bot.tellraw('@a', {text: `${util.inspect(err)}`, color: 'red'});
            return;
          }
        }
        // if (line === '.clearconsolequeue') {
        //   consoleQueue = [];
        //   return;
        // }
        if (line === '.messagelogging on') {
          bot.consoleQueue = [];
          bot.messageLogging = true;
          return;
        }
        if (line === '.messagelogging off') {
          bot.consoleQueue = [];
          bot.messageLogging = false;
          return;
        }
        if (line === '.kill') process.exit();

        if (line.startsWith('.')) {
          return bot.command_handler.run(
              bot.username,
              bot.username,
              '*' + line.substring(1),
              bot.uuid,
              null,
              'h',
              'o',
          );
        }
        bot.tellraw('@a', [
          {
            text: '[',
            color: 'dark_gray',
          },
          {
            text: `${bot.username} Console`,
            color: 'gray',
          },
          {
            text: '] ',
            color: 'dark_gray',
          },
          {
            text: 'chayapak ',
            color: 'green',
          },
          {
            text: '\u203a ',
            color: 'dark_gray',
          },
          chatMessage.MessageBuilder.fromString('&7' + line),
        ]);
      } catch (e) {
        bot.console.error(e);
      }
    });
  }
};

module.exports = {inject, oneTimeInject};