bot/src/stream.hpp


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

#pragma once

#include <set>
#include <vector>

#include "api/kick.hpp"
#include "api/twitch/helix_client.hpp"
#include "api/twitch/schemas/stream.hpp"
#include "config.hpp"
#include "irc/client.hpp"
#include "schemas/stream.hpp"

namespace bot::stream {
  enum StreamerType { TWITCH, KICK };

  struct StreamerData {
      int id;
      StreamerType type;
      bool is_live;
      std::string title;
      std::string game;
  };

  class StreamListenerClient {
    public:
      StreamListenerClient(const api::twitch::HelixClient &helix_client,
                           const api::KickAPIClient &kick_api_client,
                           irc::Client &irc_client,
                           const Configuration &configuration)
          : helix_client(helix_client),
            kick_api_client(kick_api_client),
            irc_client(irc_client),
            configuration(configuration) {};
      ~StreamListenerClient() = default;

      void run();
      void listen_channel(const int &id, const StreamerType &type);
      void unlisten_channel(const int &id, const StreamerType &type);

    private:
      void check();
      void handler(const schemas::EventType &type,
                   const api::twitch::schemas::Stream &stream,
                   const StreamerData &data);
      void update_channel_ids();

      const api::twitch::HelixClient &helix_client;
      const api::KickAPIClient &kick_api_client;
      irc::Client &irc_client;
      const Configuration &configuration;

      std::vector<StreamerData> streamers;
  };
}