LibTgBotPP/src/TelegramBot.cc

#include "Telegram/TelegramBot.h"

#include <iostream>

#include <curlpp/cURLpp.hpp>
#include <curlpp/Easy.hpp>
#include <curlpp/Infos.hpp>
#include <curlpp/Options.hpp>

#include "http_build_query.h"
#include "json_encode.h"
#include "std.h"

#include "Telegram/Message.h"
#include "Telegram/MessageEntity.h"

#define API_URL "https://api.telegram.org/bot"

/**
 * Constructor of Bot
 *
 * Calls init
 */
Telegram::TelegramBot::TelegramBot() {

  this->api_url = "";
  this->init();
}

/**
 * Constructor of bot
 *
 * Sets the API URL and calls init
 *
 * @param token a std::string storing the bot token
 *
 */
Telegram::TelegramBot::TelegramBot(std::string token) {

  this->api_url = API_URL + token + "/";
  this->init();
}

/**
 * Adds a callback to the command list of the bot
 *
 * @param cmd a structure of type TCommand
 */
void Telegram::TelegramBot::addCommand(TCommand cmd) {

  if (cmd.command[0] != '/') {
    cmd.command = "/" + cmd.command;
  }
  this->command_map[cmd.command] = cmd;
}

/**
 * Set the webhook
 * The web hook URL is like https://<your domain>/cgi-bin/mybot.cgi
 * A web hook must be set before you can get any messages from the Telegram
 * network!
 *
 * @param url a std::string with the URL of the hook
 */
void Telegram::TelegramBot::setWebhook(std::string url) {

  std::map<std::string, std::string> params;

  params["url"] = url;
  this->apiRequest("setWebhook", params);
}

/**
 * Process incoming message
 *
 * @param message a std::string storing the message sent by the Telegram system
 *
 */
void Telegram::TelegramBot::processMessage(std::string message) {

  this->msg = new Telegram::Message(message);

  Telegram::TMessageEntities entities = this->msg->getEntities();

  if (entities.size() && (entities[0]->getType() == "bot_command")) {
    this->sendMessage(this->processCommand(this->msg->getText()), this->msg->getChat()->getId());
  } else {
    this->sendMessage("Hello " + this->msg->getChat()->getUsername() + ", you told me: '" + msg->getText() + "'", this->msg->getChat()->getId());
  }
}

/**
 * A simple command to the API to get some information about the bot.
 *
 * @return Telegram::User* a pointer of an object of type Telegram::User
 */
Telegram::User *Telegram::TelegramBot::getMe(void) {

	Json::Value obj;
	std::map<std::string, std::string> params;

	obj = this->apiRequestJson("getMe", params);

	return(new Telegram::User(obj["result"]));
}

/**
 * Send a text message to a user or group
 *
 * @param message The text of the message
 * @param chat_id the chat id
 *
 * @return A pointer to a Telegram message object
 */
Telegram::Message *Telegram::TelegramBot::sendMessage(std::string message, Json::Int64 chat_id) {

  return(this->sendMessage(message, SSTR(chat_id)));
}

/**
 * sendMessage sends a simple text message to a given chat (might be a user or a group)
 *
 * @param message The text of the message
 * @param chat_id the chat id
 *
 * @return A pointer to a Telegram message object
 */
Telegram::Message* Telegram::TelegramBot::sendMessage(std::string message, std::string chat_id) {

	std::map<std::string, std::string> params;
	Json::Value obj;

	params["chat_id"] = chat_id;
	params["text"] = message;

	obj = this->apiRequestJson("sendMessage", params);

	return(new Telegram::Message(obj["result"]));
}

/**
 * Sends a picture from the internet to a chat
 *
 * @param URL The URL of the picture
 * @param chat_id the chat id
 *
 * @return A pointer to a Telegram message object
 */
Telegram::Message* Telegram::TelegramBot::sendPhoto(std::string URL, Json::Int64 chat_id) {

	return(this->sendPhoto(URL, SSTR(chat_id)));
}

/**
 * Sends a picture from the internet to a chat
 *
 * @param URL The URL of the picture
 * @param chat_id the chat id
 *
 * @return A pointer to a Telegram message object
 */
Telegram::Message* Telegram::TelegramBot::sendPhoto(std::string URL, std::string chat_id) {

	Json::Value obj;

	std::map<std::string, std::string> params;

	params["chat_id"] = chat_id;
	params["photo"] = URL;

	if ((URL.substr(0, 8) == "https://") || (URL.substr(0, 7) == "http://")) {
		obj = this->apiRequestJson("sendPhoto", params);
	}

	if (URL.substr(0, 7) == "file://") {
		obj = this->apiRequestFile(URL.substr(7, std::string::npos), "photo", chat_id);
	}

	return(new Telegram::Message(obj["result"]));
}

Telegram::Message *Telegram::TelegramBot::getMessage() {

  return(this->msg);
}

/**
 * getCommandMap return the complete command list of the bot
 *
 * @return Map of the commands
 */
Telegram::TCommandMap Telegram::TelegramBot::getCommandMap() {

  return(this->command_map);
}

// Private methods

/**
 * inits the bot. That is only sending a content-type to stdout in case
 * one uses the bot by an CGI script
 */
void Telegram::TelegramBot::init() {

  std::cout << "Content-type: text/html\r\n\r\n" << std::endl;
}

/**
 * A generic API request
 *
 * Might be removed in future!
 *
 * @param method might be sendPhoto, sendMessage or any command defined by Telegram's API
 * @param parameters a map of further parameters, needed by the API call
 */
void Telegram::TelegramBot::apiRequest(std::string method, std::map<std::string, std::string> parameters) {

  parameters["method"] = method;

  Json::Value jValues;
  Json::StyledWriter w;

  for(std::map<std::string, std::string>::iterator it = parameters.begin(); it != parameters.end(); ++it) {
    jValues[(*it).first] = (*it).second;
  }

  try {
    cURLpp::Easy handle;
    std::string url = this->api_url + method + "?" + http_build_query(parameters);

    handle.setOpt(cURLpp::Options::Url(url));
    handle.setOpt(cURLpp::Options::ConnectTimeout(5));
    handle.setOpt(cURLpp::Options::Timeout(60));
    handle.perform(); // Do the curl request

    int code;
    code = cURLpp::Infos::ResponseCode::get(handle);
  }
  catch(cURLpp::LogicError &e) {
    Log(e.what());
  }
  catch(cURLpp::RuntimeError &e) {
    Log(e.what());
  }
  catch(std::exception &e) {
    Log(e.what());
  }
}

/**
 * An API request, posting JSON data
 *
 * @param method might be "sendPhoto", "sendMessage" or anything else, defined by the Telegram API
 * @param parameters further parameters to the API call (e.g. the text of a message)
 *
 * @return the JSON structure, sent back by Telegram's API
 */
Json::Value Telegram::TelegramBot::apiRequestJson(std::string method, std::map<std::string, std::string> parameters) {

	std::stringstream result; // Stores the result of the api call
	parameters["method"] = method;

	try {
		cURLpp::Easy handle;
		cURLpp::Cleanup cleaner;

		std::list<std::string> header;
		header.push_back("Content-Type: application/json");

		handle.setOpt(cURLpp::Options::Url(this->api_url));
		handle.setOpt(cURLpp::Options::ConnectTimeout(5));
		handle.setOpt(cURLpp::Options::Timeout(60));
		handle.setOpt(cURLpp::Options::HttpHeader(header));
		handle.setOpt(cURLpp::Options::PostFields(json_encode(parameters)));
		handle.setOpt(cURLpp::Options::WriteStream(&result));

		handle.perform(); // Do the curl request
	}
	catch(cURLpp::LogicError &e) {
		Log(e.what());
	}
	catch(cURLpp::RuntimeError &e) {
		Log(e.what());
	}
	catch(std::exception &e) {
		Log(e.what());
	}

	Json::Reader jreader;
	Json::Value obj;
	jreader.parse(result.str(), obj);

	return (obj);
}

/**
 * Sending a file by InputFile object
 *
 * @param filename a valid file name
 * @param type may be photo, document, video
 * @param chat_id a valid chat id (might be a user or a group chat)
 *
 * @return result Json::Value
 */
Json::Value Telegram::TelegramBot::apiRequestFile(std::string filename, std::string type, std::string chat_id) {

	std::stringstream result; // Stores the result of the api call
	cURLpp::Forms formParts; // Parts to be sent

	try {
		cURLpp::Easy handle;
		cURLpp::Cleanup cleaner;

		handle.setOpt(cURLpp::Options::Url(this->api_url + "send" + type));
		{
			cURLpp::Forms parts;
			parts.push_back(new cURLpp::FormParts::Content("chat_id", chat_id));
			parts.push_back(new cURLpp::FormParts::File(type, filename));
			handle.setOpt(cURLpp::Options::HttpPost(parts));
		}

		handle.perform();

	}
	catch(cURLpp::LogicError &e) {
		Log(e.what());
	}
	catch(cURLpp::RuntimeError &e) {
		Log(e.what());
	}
	catch(std::exception &e) {
		Log(e.what());
	}

	Json::Reader jreader;
	Json::Value obj;
	jreader.parse(result.str(), obj);

	return (obj);
}

/**
 * processCommand gets a command by the user (e.g. /start) and processes
 * the command.
 * The command should be added to the command list by addCommand before,
 * otherwise "Unknwon command" is returned
 *
 * @param cmd the command, defined in the command list (e.g. /start)
 *
 * @return the output of the command
 */
std::string Telegram::TelegramBot::processCommand(std::string cmd) {

  // Process command
  std::string sCommand;
  std::string sResult;
  std::vector<std::string> vCmdLine = split(cmd, ' ');

  if((vCmdLine.size() > 0) && (this->command_map.find(vCmdLine[0]) != this->command_map.end())) {
    sCommand = vCmdLine[0];
    vCmdLine.erase(vCmdLine.begin()); // First value is the command itself
    TCommand cs = this->command_map[sCommand];
    sResult = (*cs.callback)(this, vCmdLine);
  } else {
    sResult = "Unknown command";
  }

  return(sResult);
}