message_command.hpp

#ifndef CPPGRAM_MESSAGE_COMMAND_HPP
#define CPPGRAM_MESSAGE_COMMAND_HPP

#include "command.hpp"
#include "cppgram/types/update.hpp"

namespace cppgram
{
namespace commands
{
template <class T> class MessageCommand : public Command<T>
{
    public:
    typedef std::function<void( T &, const types::Message & )> MessageClosure;

    MessageCommand( std::string &command, MessageClosure script )
        : command( "/" + command )
        , script( script )
    {
    }

    virtual void callClosure( T &bot, const types::Update &update )
    {
        bot.setChatId( update.message->chat.id );
        script( bot, update.message.value() );
    }

    virtual bool isValid( const types::Update &update )
    {
        // If there isn't at least a command in the message
        if ( update.message->entities.size() == 0 )
        {
            return false;
        }

        types::MessageEntity entity = update.message->entities[0];

        // If the first entity is not a bot_command
        if ( entity.type != EMessageEntity::bot_command )
        {
            return false;
        }

        // If the command is not the first text in the message
        if ( entity.offset != 0 )
        {
            return false;
        }

        // Get the string that corresponds to the bot_command
        std::string entity_text = update.message->text->substr( 0, entity.length );

        // and the command is found in the message
        if ( entity_text.find( command ) != std::string::npos )
        {
            return true;
        }

        return false;
    }

    private:
    const EUpdate type = EUpdate::eMessage;

    std::string command;

    MessageClosure script;
};
}
}

#endif