2 files changed, 13 insertions, 4 deletions
diff --git a/src/irc/irc_client.cpp b/src/irc/irc_client.cpp
index e134bea..0d5c4e3 100644
--- a/src/irc/irc_client.cpp
+++ b/src/irc/irc_client.cpp
@@ -134,9 +134,9 @@ void IrcClient::send_topic_command(const std::string& chan_name, const std::stri
   this->send_message(IrcMessage("TOPIC", {chan_name, topic}));
 }
 
-void IrcClient::send_quit_command()
+void IrcClient::send_quit_command(const std::string& reason)
 {
-  this->send_message(IrcMessage("QUIT", {"gateway shutdown"}));
+  this->send_message(IrcMessage("QUIT", {reason}));
 }
 
 void IrcClient::send_join_command(const std::string& chan_name)
@@ -403,7 +403,6 @@ void IrcClient::on_part(const IrcMessage& message)
       iid.chan = chan_name;
       iid.server = this->hostname;
       bool self = channel->get_self()->nick == nick;
-      this->bridge->send_muc_leave(std::move(iid), std::move(nick), std::move(txt), self);
       if (self)
       {
         channel->joined = false;
@@ -411,6 +410,7 @@ void IrcClient::on_part(const IrcMessage& message)
         // channel pointer is now invalid
         channel = nullptr;
       }
+      this->bridge->send_muc_leave(std::move(iid), std::move(nick), std::move(txt), self);
     }
 }
 
@@ -608,3 +608,8 @@ void IrcClient::on_user_mode(const IrcMessage& message)
                                   std::string("User mode for ") + message.arguments[0] +
                                   " is [" + message.arguments[1] + "]");
 }
+
+size_t IrcClient::number_of_joined_channels() const
+{
+  return this->channels.size();
+}
diff --git a/src/irc/irc_client.hpp b/src/irc/irc_client.hpp
index 7380a8d..5cd1403 100644
--- a/src/irc/irc_client.hpp
+++ b/src/irc/irc_client.hpp
@@ -101,7 +101,7 @@ public:
   /**
    * Send the QUIT irc command
    */
-  void send_quit_command();
+  void send_quit_command(const std::string& reason);
   /**
    * Send a message to the gateway user, not generated by the IRC server,
    * but that might be useful because we want to be verbose (for example we
@@ -186,6 +186,10 @@ public:
    */
   void on_channel_mode(const IrcMessage& message);
   void on_quit(const IrcMessage& message);
+  /**
+   * Return the number of joined channels
+   */
+  size_t number_of_joined_channels() const;
 
 private:
   /**