BybitBot/Bot.hpp at main · andarut/BybitBot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#ifndef BOT_H
#define BOT_H

#include "Utils/Utils.hpp"
#include "Utils/Logger.hpp"
#include "json.hpp"

#include "Bybit.hpp"

#include <tgbot/tgbot.h>

/* TOneverDO: Make parent TgBot class */
class BybitBot {
public:
    BybitBot(const std::string &apiKey);
    void run();

    void setAdmin(const s64 devId) { m_devId = devId; }

    void sendGreetings(const s64& chatId);
    void sendHelp(const s64& chatId);
    void sendBalance(const s64& chatId);
    void sendP2POffers(const s64& chatId, const u64& amount);

    void sendAmountSetup(const s64& chatId);
    void sendApiKeySetup(const s64& chatId);
    void sendApiSecretSetup(const s64& chatId);
    void sendPaymentsSetup(const s64& chatId);

    void sendUnknownCommand(const s64& chatId);
private:
    void m_sendToUser(const s64& chatId, const std::string& message, const TgBot::InlineKeyboardMarkup::Ptr& keyboard=nullptr);
    void m_sendToUser(const s64& chatId, const char *message, const TgBot::InlineKeyboardMarkup::Ptr& keyboard=nullptr);

    void m_sendToAdmin(const std::string& message);
    void m_sendToAdmin(const char *message);

    void m_deleteMessage(const s64& chatId, const s64& messageId);

    TgBot::InlineKeyboardMarkup::Ptr m_paymentsKeyboard(const s64& chatId);
    TgBot::InlineKeyboardMarkup::Ptr m_offersKeyboard(std::array<BybitP2POffer, 10> offers);

    TgBot::InlineKeyboardMarkup::Ptr m_menuKeyboard(const s64& chatId);


    // TODO
    // void m_sendToAll(const std::string& message);

    s64 m_devId = -1;

    /* DATA */
    struct BybitBotUserData {
        /* default read-only API keys from test net for demo*/
        std::string apiKey="JwmWRC3AeLxMQ5HHN1";
        std::string apiSecret="JcKWltDJjjjf5c4SSLg8ZguAHJyG3FJ7niH4";
        std::list<u64> paymentMethods;
    };
    std::unordered_map<s64, BybitBotUserData> m_data;

    void m_saveData();
    void m_loadData();

    /* STATE */
    enum BybitBotUserState {
        IDLE,
        API_KEY_INPUT,
        API_SECRET_INPUT,
        PAYMENTS_INPUT,
        AMOUNT_INPUT,
        TOKENS_INPUT,
    };
    std::unordered_map<s64, BybitBotUserState> m_state;

    void m_saveState();
    void m_loadState();

    const std::map<std::string, std::function<void(u64)>> m_menu = {
        {"Balance",               [this](u64 chatId) { sendBalance(chatId); }},
        {"P2P Offers",            [this](u64 chatId) { sendAmountSetup(chatId); }},
        {"Payment methods setup", [this](u64 chatId) { sendPaymentsSetup(chatId); }},
        {"API_KEY setup",         [this](u64 chatId) { sendApiKeySetup(chatId); }},
        {"API_SECRET setup",      [this](u64 chatId) { sendApiSecretSetup(chatId); }},
        {"Help",                  [this](u64 chatId) { sendHelp(chatId); }}
    };

    /* tgbot objects */
    TgBot::Bot m_bot;
};

#endif // BOT_H