commands.cpp

Go to the documentation of this file.

#include <boost/algorithm/string.hpp>
#include <boost/date_time/posix_time/conversion.hpp>
#include <boost/date_time/posix_time/posix_time.hpp>
#include <cstdint>
#include <iostream>
#include <leptonica/allheaders.h>
#include <string>
#include <tgbot/tgbot.h>
#include <vector>
 
#include "config.hpp"
#include "metar.hpp"
#include "ocr.hpp"
#include "tgbot/types/Message.h"
#include "utils.hpp"
 
const std::string PARSE_MODE("Markdown");
 
const std::string HELP_MESSAGE =
    "Hello, I'm " BOT_NAME "!\n"
    "I'm a bot created by " BOT_AUTHOR "!\n"
    "My commands:\n"
    "`/start` - _start the bot_\n"
    "`/help` - _show this message_\n"
    "`/info` - _show compilation machine information_\n"
    "`/ocr` - _optical Character Recognition with tesseract_\n"
    "`/metar <ICAO Code>` - _get METAR information_\n"
    "\n"
    "My source code is available at " BOT_GITHUB_REPO "\n"
    "Version: " BOT_VERSION "\n";
 
const std::string INFO_MESSAGE = "System Informations\n\n"
                                 "*OS*: _"
#if defined(__linux__)
                                 "Linux"
#elif defined(__APPLE__)
                                 "MacOS"
#elif defined(_WIN32)
#if defined(_WIN64)
                                 "Windows x86-64"
#else
                                 "Windows x86"
#endif
#endif
                                 " with "
#if defined(__amd64__)
                                 "CPU x86-64"
#elif defined(__i386__)
                                 "CPU x86"
#elif defined(__arm__)
                                 "SoC ARMv7"
#elif defined(__arm64__)
                                 "SoC ARMv8"
#endif
                                 "_\n"
                                 "*Compiler*: _"
#if defined(__clang__)
                                 "CLANG " __clang_version__
#elif defined(__GNUC__)
                                 "GCC (GNU C Compiler) " __GNUC__
                                 "." __GNUC_MINOR__
#else
                                 "Unknown"
#endif
                                 "_\n";
 
namespace Bot {
void register_handlers(TgBot::Bot *bot) {
 
  bot->getEvents().onCommand("start", [bot](TgBot::Message::Ptr message) {
    tools::log_cmd("start", message->from->username, message->date, LOG::Info);
    bot->getApi().sendMessage(message->chat->id, "Hi!");
  });
 
  bot->getEvents().onCommand("help", [bot](TgBot::Message::Ptr message) {
    tools::log_cmd("help", message->from->username, message->date, LOG::Info);
    bot->getApi().sendMessage(message->chat->id, HELP_MESSAGE, false, 0,
                              nullptr, PARSE_MODE);
  });
 
  bot->getEvents().onCommand("info", [bot](TgBot::Message::Ptr message) {
    tools::log_cmd("info", message->from->username, message->date, LOG::Info);
    bot->getApi().sendMessage(message->chat->id, INFO_MESSAGE, false, 0,
                              nullptr, PARSE_MODE);
  });
 
  bot->getEvents().onCommand("ocr", [bot](TgBot::Message::Ptr message) {
    tools::log_cmd("ocr", message->from->username, message->date, LOG::Info);
 
    TgBot::Message::Ptr reply = message->replyToMessage;
 
    if (reply == nullptr) {
      tools::log_cmd("No Message Reply", message->from->username,
              boost::posix_time::to_time_t(
                  boost::posix_time::second_clock::universal_time()),
              LOG::Warn);
      bot->getApi().sendMessage(message->chat->id,
                                "You should reply to a message");
      return;
    }
 
    TgBot::File::Ptr file =
        bot->getApi().getFile(reply->photo[reply->photo.size() - 1]->fileId);
#if defined(DEBUG)
    tools::log_cmd("DOWNLOADED FILE", reply->from->username, reply->date, LOG::Debug);
#endif
    std::string fileContent = bot->getApi().downloadFile(file->filePath);
 
    Pix *pixImage =
        pixReadMem(reinterpret_cast<const l_uint8 *>(fileContent.c_str()),
                   fileContent.size());
 
    if (pixImage == nullptr) {
      tools::log_cmd("Failed to read the image from memory", "PROGRAM",
              boost::posix_time::to_time_t(
                  boost::posix_time::second_clock::universal_time()),
              LOG::Error);
      return;
    }
 
    bot->getApi().sendMessage(message->chat->id, tools::getTextFromImage(pixImage));
  });
 
  bot->getEvents().onCommand("metar", [bot](TgBot::Message::Ptr message) {
    tools::log_cmd("metar", message->from->username, message->date, LOG::Info);
 
    std::vector<std::string> text;
    boost::split(text, message->text, [](char c) { return c == ' '; });
 
    bot->getApi().sendMessage(message->chat->id,
                              "✈️ METAR for " + text[1] + "\n🕒 " +
                                  tools::getMetar(text[1]),
                              false, message->messageId);
  });
}
 
} // namespace Bot