summaryrefslogtreecommitdiff
path: root/src/irc/message.hpp
blob: 94541a6868013e9e2c56418ea0c39c0e5d283669 (plain)
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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#pragma once

#include <functional>
#include <optional>
#include <sstream>
#include <string>
#include <vector>

#include "../utils/string.hpp"

namespace bot {
  namespace irc {
    enum MessageType { Privmsg, Notice };
    std::optional<MessageType> define_message_type(const std::string &msg);

    struct MessageSender {
        std::string login;
        std::string display_name;
        int id;

        // More fields will be here
    };

    struct MessageSource {
        std::string login;
        int id;
    };

    template <MessageType T>
    struct Message;

    template <>
    struct Message<MessageType::Privmsg> {
        MessageSender sender;
        MessageSource source;
        std::string message;
    };

    template <MessageType T>
    std::optional<Message<T>> parse_message(const std::string &msg) {
      std::vector<std::string> parts = utils::string::split_text(msg, ' ');

      if (T == MessageType::Privmsg) {
        MessageSender sender;
        MessageSource source;

        Message<MessageType::Privmsg> message;

        std::string tags = parts[0];
        tags = tags.substr(1, tags.length());
        parts.erase(parts.begin());

        std::string user = parts[0];
        user = user.substr(1, user.length());

        std::vector<std::string> user_parts =
            utils::string::split_text(user, '!');

        sender.login = user_parts[0];

        parts.erase(parts.begin(), parts.begin() + 2);

        std::string channel_login = parts[0];
        source.login = channel_login.substr(1, channel_login.length());

        parts.erase(parts.begin());

        std::string chat_message = utils::string::join_vector(parts, ' ');
        message.message = chat_message.substr(1, chat_message.length());

        std::vector<std::string> tags_parts =
            utils::string::split_text(tags, ';');

        for (const std::string &tag : tags_parts) {
          std::istringstream iss(tag);
          std::string key;
          std::string value;

          std::getline(iss, key, '=');
          std::getline(iss, value);

          if (key == "display-name") {
            sender.display_name = value;
          } else if (key == "room-id") {
            source.id = std::stoi(value);
          } else if (key == "user-id") {
            sender.id = std::stoi(value);
          }
        }

        message.sender = sender;
        message.source = source;

        return message;
      }

      return std::nullopt;
    }

    template <MessageType T>
    struct MessageHandler;

    template <>
    struct MessageHandler<MessageType::Privmsg> {
        using fn = std::function<void(Message<Privmsg> message)>;
    };

  }
}