Ssh-chat, част 2

Здравей, Хабр. Това е втората статия от цикъла ssh-chat.

Какво ще направим:

• Ще добавим възможност за създаване на собствени функции за форматиране.
• Ще добавим поддръжка на markdown.
• Ще добавим поддръжка на ботове.
• Ще увеличим сигурността на паролите (хеширане и сол).
  За съжаление, функцията за изпращане на файлове няма да бъде налична.

Потребителски функции за форматиране.

В момента е реализирана поддръжка на следните функции за форматиране:

• @color
• @bold
• @underline
• @hex
• @box
  Но трябва да добавим възможност за създаване на собствени функции:
  Всички функции се съхраняват в обект, наречен methods.
  Така че ще бъде достатъчно да създадем функция registerMethod.:

// parserExec.js at end
module.exports.registerMethod  =  function(name, func) {
  methods[name] =  func
}

Също така, трябва да върнем този метод след създаването на сървъра.

// index.js at require part
const { registerMethod } = require('./parserExec')

// index.js at end
module.exports.registerMethod  =  registerMethod

Сега при създаването на сървъра можем да регистрираме методи за форматиране. Пример:

const chat = require('.')
const { formatNick } = require('./format')

chat({})

chat.registerMethod('hello', function(p, name){
  return 'Здравей, ' + formatNick(name) + '!'
})

Поддръжка на markdown.

Markdown е много удобен, така че ще го добавим с помощта на marked terminal.

// format.js near require
const marked = require('marked');
const TerminalRenderer = require('marked-terminal');

marked.setOptions({
  renderer: new TerminalRenderer()
});

// format.js line 23
message = marked(message)

Ботове.

Как ще работи това?

let writeBotBob = chat.registerBot({
  name: 'botBob',

  onConnect(nick, write){
    write('@hello{' + nick + '}')
  },

  onDisconnect(nick, write){},

  onMessage(nick, message, write) {
    if(message == 'botBob!') write('Аз съм тук')
  },

  onCommand(command, write) {
    write('Изпълнявам ' + command)
  }
})

onCommand може да бъде извикана с помощта на @bot(botBob){Command}

Всичко, свързано с работата с ботове, е описано в файла:

let bots = []; // Все боты

let onWrite = () => {}; 

function getWrite(bot) { // Генерирует метод отправки сообщения для бота
  return msg => {
    onWrite(bot.name, msg);
  };
}

module.exports.message = function message(nick, message) { // index.js выполнит эту функцию после отправки сообщения
  bots.forEach(bot => {
    try {
      bot.onMessage(nick, message, getWrite(bot));
    } catch (e) {
      console.error(e);
    }
  });
};

module.exports.connect = function message(nick) { // При соединении
  bots.forEach(bot => {
    try {
      bot.onConnect(nick, getWrite(bot));
    } catch (e) {
      console.error(e);
    }
  });
};

module.exports.disConnect = function message(nick) { // При отсоединении
  bots.forEach(bot => {
    try {
      bot.onDisconnect(nick, message, getWrite(bot));
    } catch (e) {
      console.error(e);
    }
  });
};

module.exports.command = function message(name, message) { // При выполнении команды
  bots.forEach(bot => {
    if (bot.name == name) {
      try {
        bot.onCommand(message, getWrite(bot));
      } catch (e) {
        console.error(e);
      }
    }
  });
};

module.exports.registerBot = function(bot) {
  bots.push(bot);
  return  getWrite(bot)
};

module.exports.onMessage = func => {
  onWrite = func;
};

Какво може да се направи с ботовете:

• Монитор на натоварването.
• Разгърни
• Дъска със задачи.

Хеш и сол.

Защо не ssh ключове? Защото ssh ключовете ще са различни на различни устройства.
Ще създадем файл, който ще отговаря за проверката и създаването на пароли.

// crypto.js
const crypto = require('crypto');

function genRandomString(length) {
  return crypto
    .randomBytes(Math.ceil(length / 2))
    .toString('hex')
    .slice(0, length);
}

function sha512(password, salt){
  const hash = crypto.createHmac('sha512', salt); /** Hashing algorithm sha512 */
  hash.update(password);
  const value = hash.digest('hex');
  return value
};

function checkPass(pass, obj){
  return obj.password == sha512(pass, obj.salt)
}

function encodePass(pass){
  const salt = genRandomString(16)
  return JSON.stringify({
    salt,
    password: sha512(pass, salt)
  })
}

module.exports.encodePass = encodePass
module.exports.checkPass = checkPass

Също така, скрипт за солене и хеширане на паролата.

// To generate password run node ./encryptPassword password
const { encodePass } =require('./crypto')
console.log(encodePass(process.argv[2]))

Обновяваме в users.json и вместо сравнение в lobby.js използваме checkPassword.

Резюме

В резултат на това имаме чат по ssh с възможности за форматиране и ботове.
Финален репозиторий.

Източник: habr.com