Add methods to know which resource is on which server or channel

2 files changed, 44 insertions, 3 deletions

diff --git a/src/bridge/bridge.cpp b/src/bridge/bridge.cpp
index 6008dfc..484c860 100644
--- a/src/bridge/bridge.cpp
+++ b/src/bridge/bridge.cpp
@@ -46,11 +46,12 @@ Bridge::Bridge(const std::string& user_jid, BiboumiComponent& xmpp, std::shared_
 }
 
 /**
- * Return the role and affiliation, corresponding to the given irc mode */
+ * Return the role and affiliation, corresponding to the given irc mode
+ */
 static std::tuple<std::string, std::string> get_role_affiliation_from_irc_mode(const char mode)
 {
-  if (mode == 'a' || mode == 'q')
-    return std::make_tuple("moderator", "owner");
+  if (mode == 'a' || mode == 'q'){
+    return std::make_tuple("moderator", "owner");}
   else if (mode == 'o')
     return std::make_tuple("moderator", "admin");
   else if (mode == 'h')
@@ -720,3 +721,32 @@ std::unordered_map<std::string, std::shared_ptr<IrcClient>>& Bridge::get_irc_cli
 {
   return this->irc_clients;
 }
+
+void Bridge::add_resource_to_chan(const std::string& channel, const std::string& resource)
+{
+  auto it = this->resources_in_chan.find(channel);
+  if (it == this->resources_in_chan.end())
+    this->resources_in_chan[channel] = {resource};
+  else
+    it->second.insert(resource);
+}
+
+void Bridge::remove_resource_from_chan(const std::string& channel, const std::string& resource)
+{
+  auto it = this->resources_in_chan.find(channel);
+  if (it != this->resources_in_chan.end())
+    {
+      it->second.erase(resource);
+      if (it->second.empty())
+        this->resources_in_chan.erase(it);
+    }
+}
+
+bool Bridge::is_resource_in_chan(const std::string& channel, const std::string& resource) const
+{
+  auto it = this->resources_in_chan.find(channel);
+  if (it != this->resources_in_chan.end())
+    if (it->second.count(resource) == 1)
+      return true;
+  return false;
+}
diff --git a/src/bridge/bridge.hpp b/src/bridge/bridge.hpp
index 2676d1d..b852a30 100644
--- a/src/bridge/bridge.hpp
+++ b/src/bridge/bridge.hpp
@@ -194,6 +194,13 @@ public:
   void trigger_on_irc_message(const std::string& irc_hostname, const IrcMessage& message);
   std::unordered_map<std::string, std::shared_ptr<IrcClient>>& get_irc_clients();
 
+  /**
+   * Manage which resource is in which channel
+   */
+  void add_resource_to_chan(const std::string& channel, const std::string& resource);
+  void remove_resource_from_chan(const std::string& channel, const std::string& resource);
+  bool is_resource_in_chan(const std::string& channel, const std::string& resource) const;
+
 private:
   /**
    * Returns the client for the given hostname, create one (and use the
@@ -244,6 +251,10 @@ private:
    * response iq.
    */
   std::list<irc_responder_callback_t> waiting_irc;
+  /**
+   * Keep track of which resource is in which channel.
+   */
+  std::map<std::string, std::set<std::string>> resources_in_chan;
 };
 
 struct IRCNotConnected: public std::exception