biboumi_component.hpp 3.95 KB
Newer Older
1 2
#pragma once

3 4 5 6 7 8 9 10 11

#include <xmpp/xmpp_component.hpp>

#include <bridge/bridge.hpp>

#include <memory>
#include <string>
#include <map>

12 13 14 15
namespace db
{
class MucLogLine;
}
louiz’'s avatar
louiz’ committed
16
struct ListElement;
17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32

/**
 * A callback called when the waited iq result is received (it is matched
 * against the iq id)
 */
using iq_responder_callback_t = std::function<void(Bridge* bridge, const Stanza& stanza)>;

/**
 * Interact with the Biboumi Bridge
 */
class BiboumiComponent: public XmppComponent
{
public:
  explicit BiboumiComponent(std::shared_ptr<Poller> poller, const std::string& hostname, const std::string& secret);
  ~BiboumiComponent() = default;

louiz’'s avatar
louiz’ committed
33 34 35 36 37
  BiboumiComponent(const BiboumiComponent&) = delete;
  BiboumiComponent(BiboumiComponent&&) = delete;
  BiboumiComponent& operator=(const BiboumiComponent&) = delete;
  BiboumiComponent& operator=(BiboumiComponent&&) = delete;

38 39 40 41
  /**
   * Returns the bridge for the given user. If it does not exist, return
   * nullptr.
   */
42
  Bridge* find_user_bridge(const std::string& full_jid);
43 44 45
  /**
   * Return a list of all the managed bridges.
   */
louiz’'s avatar
louiz’ committed
46
  std::vector<Bridge*> get_bridges() const;
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63

  /**
   * Send a "close" message to all our connected peers.  That message
   * depends on the protocol used (this may be a QUIT irc message, or a
   * <stream/>, etc).  We may also directly close the connection, or we may
   * wait for the remote peer to acknowledge it before closing.
   */
  void shutdown();
  /**
   * Run a check on all bridges, to remove all disconnected (socket is
   * closed, or no channel is joined) IrcClients. Some kind of garbage collector.
   */
  void clean();
  /**
   * Send a result IQ with the gateway disco informations.
   */
  void send_self_disco_info(const std::string& id, const std::string& jid_to);
64 65 66 67
  /**
   * Send a result IQ with the disco informations regarding IRC server JIDs.
   */
  void send_irc_server_disco_info(const std::string& id, const std::string& jid_to, const std::string& jid_from);
68 69 70 71 72 73 74 75 76 77 78 79 80 81
  /**
   * Send an iq version request
   */
  void send_iq_version_request(const std::string& from,
                               const std::string& jid_to);
  /**
   * Send a ping request
   */
  void send_ping_request(const std::string& from,
                         const std::string& jid_to,
                         const std::string& id);
  /**
   * Send the channels list in one big stanza
   */
82 83 84
  void send_iq_room_list_result(const std::string& id, const std::string& to_jid, const std::string& from,
                                const ChannelList& channel_list, std::vector<ListElement>::const_iterator begin,
                                std::vector<ListElement>::const_iterator end, const ResultSetInfo& rs_info);
louiz’'s avatar
louiz’ committed
85
  void send_invitation(const std::string& room_target, const std::string& jid_to, const std::string& author_nick);
86 87 88 89 90 91 92
  /**
   * Handle the various stanza types
   */
  void handle_presence(const Stanza& stanza);
  void handle_message(const Stanza& stanza);
  void handle_iq(const Stanza& stanza);

93 94 95 96 97 98
#ifdef USE_DATABASE
  bool handle_mam_request(const Stanza& stanza);
  void send_archived_message(const db::MucLogLine& log_line, const std::string& from, const std::string& to,
                             const std::string& queryid);
#endif

99 100
private:
  /**
101
   * Return the bridge associated with the bare JID. Create a new one
102 103 104 105 106 107 108 109 110 111 112 113 114
   * if none already exist.
   */
  Bridge* get_user_bridge(const std::string& user_jid);

  /**
   * A map of id -> callback.  When we want to wait for an iq result, we add
   * the callback to this map, with the iq id as the key. When an iq result
   * is received, we look for a corresponding callback in this map. If
   * found, we call it and remove it.
   */
  std::map<std::string, iq_responder_callback_t> waiting_iq;

  /**
115
   * One bridge for each user of the component. Indexed by the user's bare
116 117 118 119
   * jid
   */
  std::unordered_map<std::string, std::unique_ptr<Bridge>> bridges;

120
  AdhocCommandsHandler irc_server_adhoc_commands_handler;
121
  AdhocCommandsHandler irc_channel_adhoc_commands_handler;
122 123
};

124