path: root/src/xmpp/biboumi_adhoc_commands.cpp

#include <xmpp/biboumi_adhoc_commands.hpp>
#include <xmpp/biboumi_component.hpp>
#include <utils/scopeguard.hpp>
#include <bridge/bridge.hpp>
#include <config/config.hpp>
#include <utils/string.hpp>
#include <utils/split.hpp>
#include <xmpp/jid.hpp>
#include <algorithm>
#include <sstream>
#include <iomanip>

#include <biboumi.h>

#ifdef USE_DATABASE
#include <database/database.hpp>
#include <database/save.hpp>

static void set_desc(XmlSubNode& field, const char* text)
{
  XmlSubNode desc(field, "desc");
  desc.set_inner(text);
}

#endif

#ifndef HAS_PUT_TIME
# include <time.h>
#endif

using namespace std::string_literals;

void DisconnectUserStep1(XmppComponent& xmpp_component, AdhocSession&, XmlNode& command_node)
{
  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);

  XmlSubNode x(command_node, "jabber:x:data:x");
  x["type"] = "form";
  XmlSubNode title(x, "title");
  title.set_inner("Disconnect a user from the gateway");
  XmlSubNode instructions(x, "instructions");
  instructions.set_inner("Choose a user JID and a quit message");
  XmlSubNode jids_field(x, "field");
  jids_field["var"] = "jids";
  jids_field["type"] = "list-multi";
  jids_field["label"] = "The JIDs to disconnect";
  XmlSubNode required(jids_field, "required");
  for (Bridge* bridge: biboumi_component.get_bridges())
    {
      XmlSubNode option(jids_field, "option");
      option["label"] = bridge->get_jid();
      XmlSubNode value(option, "value");
      value.set_inner(bridge->get_jid());
    }

  XmlSubNode message_field(x, "field");
  message_field["var"] = "quit-message";
  message_field["type"] = "text-single";
  message_field["label"] = "Quit message";
  XmlSubNode message_value(message_field, "value");
  message_value.set_inner("Disconnected by admin");
}

void DisconnectUserStep2(XmppComponent& xmpp_component, AdhocSession& session, XmlNode& command_node)
{
  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);

  // Find out if the jids, and the quit message are provided in the form.
  std::string quit_message;
  const XmlNode* x = command_node.get_child("x", "jabber:x:data");
  if (x)
    {
      const XmlNode* message_field = nullptr;
      const XmlNode* jids_field = nullptr;
      for (const XmlNode* field: x->get_children("field", "jabber:x:data"))
        if (field->get_tag("var") == "jids")
          jids_field = field;
        else if (field->get_tag("var") == "quit-message")
          message_field = field;
      if (message_field)
        {
          const XmlNode* value = message_field->get_child("value", "jabber:x:data");
          if (value)
            quit_message = value->get_inner();
        }
      if (jids_field)
        {
          std::size_t num = 0;
          for (const XmlNode* value: jids_field->get_children("value", "jabber:x:data"))
            {
              Bridge* bridge = biboumi_component.find_user_bridge(value->get_inner());
              if (bridge)
                {
                  bridge->shutdown(quit_message);
                  num++;
                }
            }
          command_node.delete_all_children();

          XmlSubNode note(command_node, "note");
          note["type"] = "info";
          if (num == 0)
            note.set_inner("No user were disconnected.");
          else if (num == 1)
            note.set_inner("1 user has been disconnected.");
          else
            note.set_inner(std::to_string(num) + " users have been disconnected.");
          return;
        }
    }
  XmlSubNode error(command_node, ADHOC_NS":error");
  error["type"] = "modify";
  XmlSubNode condition(error, STANZA_NS":bad-request");
  session.terminate();
}

#ifdef USE_DATABASE

void ConfigureGlobalStep1(XmppComponent&, AdhocSession& session, XmlNode& command_node)
{
  const Jid owner(session.get_owner_jid());
  const Jid target(session.get_target_jid());

  auto options = Database::get_global_options(owner.bare());

  command_node.delete_all_children();
  XmlSubNode x(command_node, "jabber:x:data:x");
  x["type"] = "form";
  XmlSubNode title(x, "title");
  title.set_inner("Configure some global default settings.");
  XmlSubNode instructions(x, "instructions");
  instructions.set_inner("Edit the form, to configure your global settings for the component.");

  {
    XmlSubNode max_histo_length(x, "field");
    max_histo_length["var"] = "max_history_length";
    max_histo_length["type"] = "text-single";
    max_histo_length["label"] = "Max history length";
    set_desc(max_histo_length, "The maximum number of lines in the history that the server sends when joining a channel");
    {
      XmlSubNode value(max_histo_length, "value");
      value.set_inner(std::to_string(options.col<Database::MaxHistoryLength>()));
    }
  }

  {
    XmlSubNode record_history(x, "field");
    record_history["var"] = "record_history";
    record_history["type"] = "boolean";
    record_history["label"] = "Record history";
    set_desc(record_history, "Whether to save the messages into the database, or not");
    {
      XmlSubNode value(record_history, "value");
      value.set_name("value");
      if (options.col<Database::RecordHistory>())
        value.set_inner("true");
      else
        value.set_inner("false");
    }
  }

  {
    XmlSubNode persistent(x, "field");
    persistent["var"] = "persistent";
    persistent["type"] = "boolean";
    persistent["label"] = "Make all channels persistent";
    set_desc(persistent, "If true, all channels will be persistent");
    {
      XmlSubNode value(persistent, "value");
      value.set_name("value");
      if (options.col<Database::GlobalPersistent>())
        value.set_inner("true");
      else
        value.set_inner("false");
    }
  }
}

void ConfigureGlobalStep2(XmppComponent& xmpp_component, AdhocSession& session, XmlNode& command_node)
{
  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);

  const XmlNode* x = command_node.get_child("x", "jabber:x:data");
  if (x)
    {
      const Jid owner(session.get_owner_jid());
      auto options = Database::get_global_options(owner.bare());
      options.clear();
      for (const XmlNode* field: x->get_children("field", "jabber:x:data"))
        {
          const XmlNode* value = field->get_child("value", "jabber:x:data");

          if (field->get_tag("var") == "max_history_length" &&
              value && !value->get_inner().empty())
            {
              try {
                  options.col<Database::MaxHistoryLength>() = std::stol(value->get_inner().data());
              } catch (const std::logic_error&) {
                  options.col<Database::MaxHistoryLength>() = 20;
              }
            }
          else if (field->get_tag("var") == "record_history" &&
                   value && !value->get_inner().empty())
            {
              options.col<Database::RecordHistory>() = to_bool(value->get_inner());
              Bridge* bridge = biboumi_component.find_user_bridge(owner.bare());
              if (bridge)
                bridge->set_record_history(options.col<Database::RecordHistory>());
            }
          else if (field->get_tag("var") == "persistent" &&
                   value)
            options.col<Database::GlobalPersistent>() = to_bool(value->get_inner());
        }

      save(options, *Database::db);

      command_node.delete_all_children();
      XmlSubNode note(command_node, "note");
      note["type"] = "info";
      note.set_inner("Configuration successfully applied.");
      return;
    }
  XmlSubNode error(command_node, ADHOC_NS":error");
  error["type"] = "modify";
  XmlSubNode condition(error, STANZA_NS":bad-request");
  session.terminate();
}

void ConfigureIrcServerStep1(XmppComponent&, AdhocSession& session, XmlNode& command_node)
{
  const Jid owner(session.get_owner_jid());
  const Jid target(session.get_target_jid());
  std::string server_domain;
  if ((server_domain = Config::get("fixed_irc_server", "")).empty())
    server_domain = target.local;
  auto options = Database::get_irc_server_options(owner.local + "@" + owner.domain,
                                                  server_domain);
  auto commands = Database::get_after_connection_commands(options);

  command_node.delete_all_children();
  XmlSubNode x(command_node, "jabber:x:data:x");
  x["type"] = "form";
  XmlSubNode title(x, "title");
  title.set_inner("Configure the IRC server " + server_domain);
  XmlSubNode instructions(x, "instructions");
  instructions.set_inner("Edit the form, to configure the settings of the IRC server " + server_domain);

  if (Config::get("fixed_irc_server", "").empty())
  {
    XmlSubNode field(x, "field");
    field["var"] = "address";
    field["type"] = "text-single";
    field["label"] = "Address";
      set_desc(field, "The address (hostname or IP) to connect to.");
    XmlSubNode value(field, "value");
    if (options.col<Database::Address>().empty())
      value.set_inner(server_domain);
    else
      value.set_inner(options.col<Database::Address>());
  }

  {
    XmlSubNode ports(x, "field");
    ports["var"] = "ports";
    ports["type"] = "text-multi";
    ports["label"] = "Ports";
    set_desc(ports, "List of ports to try, without TLS. Defaults: 6667.");
    for (const auto& val: utils::split(options.col<Database::Ports>(), ';', false))
      {
        XmlSubNode ports_value(ports, "value");
        ports_value.set_inner(val);
      }
  }

#ifdef BOTAN_FOUND
  {
    XmlSubNode tls_ports(x, "field");
    tls_ports["var"] = "tls_ports";
    tls_ports["type"] = "text-multi";
    tls_ports["label"] = "TLS ports";
    set_desc(tls_ports, "List of ports to try, with TLS. Defaults: 6697, 6670.");
    for (const auto& val: utils::split(options.col<Database::TlsPorts>(), ';', false))
      {
        XmlSubNode tls_ports_value(tls_ports, "value");
        tls_ports_value.set_inner(val);
      }
  }

  {
    XmlSubNode verify_cert(x, "field");
    verify_cert["var"] = "verify_cert";
    verify_cert["type"] = "boolean";
    verify_cert["label"] = "Verify certificate";
    set_desc(verify_cert, "Whether or not to abort the connection if the server’s TLS certificate is invalid");
    XmlSubNode verify_cert_value(verify_cert, "value");
    if (options.col<Database::VerifyCert>())
      verify_cert_value.set_inner("true");
    else
      verify_cert_value.set_inner("false");
  }

  {
    XmlSubNode fingerprint(x, "field");
    fingerprint["var"] = "fingerprint";
    fingerprint["type"] = "text-single";
    fingerprint["label"] = "SHA-1 fingerprint of the TLS certificate to trust.";
    if (!options.col<Database::TrustedFingerprint>().empty())
      {
        XmlSubNode fingerprint_value(fingerprint, "value");
        fingerprint_value.set_inner(options.col<Database::TrustedFingerprint>());
      }
  }

  {
    XmlSubNode field(x, "field");
    field["var"] = "sasl_password";
    field["type"] = "text-private";
    field["label"] = "SASL Password";
    set_desc(field, "Use it to authenticate with your nick.");
    if (!options.col<Database::SaslPassword>().empty())
      {
        XmlSubNode value(field, "value");
        value.set_inner(options.col<Database::SaslPassword>());
      }
  }

#endif

  {
    XmlSubNode field(x, "field");
    field["var"] = "nick";
    field["type"] = "text-single";
    field["label"] = "Nickname";
    set_desc(field, "If set, will override the nickname provided in the initial presence sent to join the first server channel");
    if (!options.col<Database::Nick>().empty())
      {
        XmlSubNode value(field, "value");
        value.set_inner(options.col<Database::Nick>());
      }
  }

  {
    XmlSubNode pass(x, "field");
    pass["var"] = "pass";
    pass["type"] = "text-private";
    pass["label"] = "Server password";
    set_desc(pass, "Will be used in a PASS command when connecting");
    if (!options.col<Database::Pass>().empty())
      {
        XmlSubNode pass_value(pass, "value");
        pass_value.set_inner(options.col<Database::Pass>());
      }
  }

  {
    XmlSubNode after_cnt_cmd(x, "field");
    after_cnt_cmd["var"] = "after_connect_commands";
    after_cnt_cmd["type"] = "text-multi";
    set_desc(after_cnt_cmd, "Custom IRC commands sent after the connection is established with the server.");
    after_cnt_cmd["label"] = "After-connection IRC commands";
    for (const auto& command: commands)
      {
        XmlSubNode after_cnt_cmd_value(after_cnt_cmd, "value");
        after_cnt_cmd_value.set_inner(command.col<Database::AfterConnectionCommand>());
      }
  }

  if (Config::get("realname_customization", "true") == "true")
    {
      {
        XmlSubNode username(x, "field");
        username["var"] = "username";
        username["type"] = "text-single";
        username["label"] = "Username";
        if (!options.col<Database::Username>().empty())
          {
            XmlSubNode username_value(username, "value");
            username_value.set_inner(options.col<Database::Username>());
          }
      }

      {
        XmlSubNode realname(x, "field");
        realname["var"] = "realname";
        realname["type"] = "text-single";
        realname["label"] = "Realname";
        if (!options.col<Database::Realname>().empty())
          {
            XmlSubNode realname_value(realname, "value");
            realname_value.set_inner(options.col<Database::Realname>());
          }
      }
    }

  {
    XmlSubNode throttle_limit(x, "field");
    throttle_limit["var"] = "throttle_limit";
    throttle_limit["type"] = "text-single";
    throttle_limit["label"] = "Throttle limit";
    XmlSubNode value(throttle_limit, "value");
    value.set_inner(std::to_string(options.col<Database::ThrottleLimit>()));
  }

  {
    XmlSubNode max_history_length(x, "field");
    max_history_length["var"] = "max_history_length";
    max_history_length["type"] = "text-single";
    max_history_length["label"] = "Throttle limit";
    XmlSubNode value(max_history_length, "value");
    value.set_inner(std::to_string(options.col<Database::MaxHistoryLength>()));
  }

  {
  XmlSubNode encoding_out(x, "field");
  encoding_out["var"] = "encoding_out";
  encoding_out["type"] = "text-single";
    set_desc(encoding_out, "The encoding used when sending messages to the IRC server.");
  encoding_out["label"] = "Out encoding";
  if (!options.col<Database::EncodingOut>().empty())
    {
      XmlSubNode encoding_out_value(encoding_out, "value");
      encoding_out_value.set_inner(options.col<Database::EncodingOut>());
    }
  }

  {
    XmlSubNode encoding_in(x, "field");
    encoding_in["var"] = "encoding_in";
    encoding_in["type"] = "text-single";
    set_desc(encoding_in, "The encoding used to decode message received from the IRC server.");
    encoding_in["label"] = "In encoding";
    if (!options.col<Database::EncodingIn>().empty())
      {
        XmlSubNode encoding_in_value(encoding_in, "value");
        encoding_in_value.set_inner(options.col<Database::EncodingIn>());
      }
  }
}

void ConfigureIrcServerStep2(XmppComponent& xmpp_component, AdhocSession& session, XmlNode& command_node)
{
  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);

  const XmlNode* x = command_node.get_child("x", "jabber:x:data");
  if (x)
    {
      const Jid owner(session.get_owner_jid());
      const Jid target(session.get_target_jid());
      std::string server_domain;
      if ((server_domain = Config::get("fixed_irc_server", "")).empty())
        server_domain = target.local;
      auto options = Database::get_irc_server_options(owner.local + "@" + owner.domain,
                                                      server_domain);
      options.clear();
      Database::AfterConnectionCommands commands{};

      for (const XmlNode* field: x->get_children("field", "jabber:x:data"))
        {
          const XmlNode* value = field->get_child("value", "jabber:x:data");
          const std::vector<const XmlNode*> values = field->get_children("value", "jabber:x:data");

          if (field->get_tag("var") == "address" && value && Config::get("fixed_irc_server", "").empty())
            options.col<Database::Address>() = value->get_inner();

          if (field->get_tag("var") == "ports")
            {
              std::string ports;
              for (const auto& val: values)
                ports += val->get_inner() + ";";
              options.col<Database::Ports>() = ports;
            }

#ifdef BOTAN_FOUND
          else if (field->get_tag("var") == "tls_ports")
            {
              std::string ports;
              for (const auto& val: values)
                ports += val->get_inner() + ";";
              options.col<Database::TlsPorts>() = ports;
            }

          else if (field->get_tag("var") == "verify_cert" && value
                   && !value->get_inner().empty())
            {
              auto val = to_bool(value->get_inner());
              options.col<Database::VerifyCert>() = val;
            }

          else if (field->get_tag("var") == "fingerprint" && value)
            options.col<Database::TrustedFingerprint>() = value->get_inner();

          else if (field->get_tag("var") == "sasl_password" && value)
            options.col<Database::SaslPassword>() = value->get_inner();


#endif // BOTAN_FOUND

          else if (field->get_tag("var") == "nick" && value)
            options.col<Database::Nick>() = value->get_inner();

          else if (field->get_tag("var") == "pass" && value)
            options.col<Database::Pass>() = value->get_inner();

          else if (field->get_tag("var") == "after_connect_commands")
            {
              commands.clear();
              for (const auto& val: values)
                {
                  auto command = Database::after_connection_commands.row();
                  command.col<Database::AfterConnectionCommand>() = val->get_inner();
                  commands.push_back(std::move(command));
                }
            }

          else if (field->get_tag("var") == "username" && value)
            {
              auto username = value->get_inner();
              // The username must not contain spaces
              std::replace(username.begin(), username.end(), ' ', '_');
              options.col<Database::Username>() = username;
            }

          else if (field->get_tag("var") == "realname" && value)
            options.col<Database::Realname>() = value->get_inner();

          else if (field->get_tag("var") == "throttle_limit" && value)
            {
              try {
                options.col<Database::ThrottleLimit>() = std::stol(value->get_inner());
              } catch (const std::logic_error&) {
                options.col<Database::ThrottleLimit>() = 10;
              }
              Bridge* bridge = biboumi_component.find_user_bridge(session.get_owner_jid());
              if (bridge)
                {
                  IrcClient* client = bridge->find_irc_client(server_domain);
                  if (client)
                    client->set_throttle_limit(options.col<Database::ThrottleLimit>());
                }
            }

          else if (field->get_tag("var") == "max_history_length" && value)
            {
              try {
                options.col<Database::MaxHistoryLength>() = std::stol(value->get_inner());
              } catch (const std::logic_error&) {
                options.col<Database::MaxHistoryLength>() = 20;
              }
            }

          else if (field->get_tag("var") == "encoding_out" && value)
            options.col<Database::EncodingOut>() = value->get_inner();

          else if (field->get_tag("var") == "encoding_in" && value)
            options.col<Database::EncodingIn>() = value->get_inner();

        }
      Database::invalidate_encoding_in_cache();
      save(options, *Database::db);
      Database::set_after_connection_commands(options, commands);

      command_node.delete_all_children();
      XmlSubNode note(command_node, "note");
      note["type"] = "info";
      note.set_inner("Configuration successfully applied.");
      return;
    }
  XmlSubNode error(command_node, ADHOC_NS":error");
  error["type"] = "modify";
  XmlSubNode condition(error, STANZA_NS":bad-request");
  session.terminate();
}

void ConfigureIrcChannelStep1(XmppComponent&, AdhocSession& session, XmlNode& command_node)
{
  const Jid owner(session.get_owner_jid());
  const Jid target(session.get_target_jid());

  insert_irc_channel_configuration_form(command_node, owner, target);
}

void insert_irc_channel_configuration_form(XmlNode& node, const Jid& requester, const Jid& target)
{
  const Iid iid(target.local, {});

  auto options = Database::get_irc_channel_options_with_server_default(requester.local + "@" + requester.domain,
                                                                       iid.get_server(), iid.get_local());
  node.delete_all_children();
  XmlSubNode x(node, "jabber:x:data:x");
  x["type"] = "form";
  XmlSubNode title(x, "title");
  title.set_inner("Configure the IRC channel " + iid.get_local() + " on server " + iid.get_server());
  XmlSubNode instructions(x, "instructions");
  instructions.set_inner("Edit the form, to configure the settings of the IRC channel " + iid.get_local());

  {
    XmlSubNode record_history(x, "field");
    record_history["var"] = "record_history";
    record_history["type"] = "list-single";
    record_history["label"] = "Record history for this channel";
    set_desc(record_history, "If unset, the value is the one configured globally");
    {
      // Value selected by default
      XmlSubNode value(record_history, "value");
      value.set_inner(options.col<Database::RecordHistoryOptional>().to_string());
    }
    // All three possible values
    for (const auto& val: {"unset", "true", "false"})
      {
        XmlSubNode option(record_history, "option");
        option["label"] = val;
        XmlSubNode value(option, "value");
        value.set_inner(val);
      }
  }

  {
    XmlSubNode encoding_out(x, "field");
    encoding_out["var"] = "encoding_out";
    encoding_out["type"] = "text-single";
    set_desc(encoding_out, "The encoding used when sending messages to the IRC server. Defaults to the server's “out encoding” if unset for the channel");
    encoding_out["label"] = "Out encoding";
    if (!options.col<Database::EncodingOut>().empty())
      {
        XmlSubNode encoding_out_value(encoding_out, "value");
        encoding_out_value.set_inner(options.col<Database::EncodingOut>());
      }
  }

  {
    XmlSubNode encoding_in(x, "field");
    encoding_in["var"] = "encoding_in";
    encoding_in["type"] = "text-single";
    set_desc(encoding_in, "The encoding used to decode message received from the IRC server. Defaults to the server's “in encoding” if unset for the channel");
    encoding_in["label"] = "In encoding";
    if (!options.col<Database::EncodingIn>().empty())
      {
        XmlSubNode encoding_in_value(encoding_in, "value");
        encoding_in_value.set_inner(options.col<Database::EncodingIn>());
      }
  }

  {
    XmlSubNode persistent(x, "field");
    persistent["var"] = "persistent";
    persistent["type"] = "boolean";
    set_desc(persistent, "If set to true, when all XMPP clients have left this channel, biboumi will stay idle in it, without sending a PART command.");
    persistent["label"] = "Persistent";
    {
      XmlSubNode value(persistent, "value");
      value.set_name("value");
      if (options.col<Database::Persistent>())
        value.set_inner("true");
      else
        value.set_inner("false");
    }
  }
}

void ConfigureIrcChannelStep2(XmppComponent& xmpp_component, AdhocSession& session, XmlNode& command_node)
{
  const Jid owner(session.get_owner_jid());
  const Jid target(session.get_target_jid());

  if (handle_irc_channel_configuration_form(xmpp_component, command_node, owner, target))
    {
      command_node.delete_all_children();
      XmlSubNode note(command_node, "note");
      note["type"] = "info";
      note.set_inner("Configuration successfully applied.");
    }
  else
    {
      XmlSubNode error(command_node, ADHOC_NS":error");
      error["type"] = "modify";
      XmlSubNode condition(error, STANZA_NS":bad-request");
      session.terminate();
    }
}

bool handle_irc_channel_configuration_form(XmppComponent& xmpp_component, const XmlNode& node, const Jid& requester, const Jid& target)
{
  const XmlNode* x = node.get_child("x", "jabber:x:data");
  if (x)
    {
      if (x->get_tag("type") == "submit")
        {
          const Iid iid(target.local, {});
          auto options = Database::get_irc_channel_options(requester.bare(),
                                                           iid.get_server(), iid.get_local());
          options.clear();
          for (const XmlNode *field: x->get_children("field", "jabber:x:data"))
            {
              const XmlNode *value = field->get_child("value", "jabber:x:data");

              if (field->get_tag("var") == "encoding_out" && value)
                options.col<Database::EncodingOut>() = value->get_inner();

              else if (field->get_tag("var") == "encoding_in" && value)
                options.col<Database::EncodingIn>() = value->get_inner();

              else if (field->get_tag("var") == "persistent" && value)
                options.col<Database::Persistent>() = to_bool(value->get_inner());
              else if (field->get_tag("var") == "record_history" &&
                       value && !value->get_inner().empty())
                {
                  OptionalBool& database_value = options.col<Database::RecordHistoryOptional>();
                  if (value->get_inner() == "true")
                    database_value.set_value(true);
                  else if (value->get_inner() == "false")
                    database_value.set_value(false);
                  else
                    database_value.unset();
                  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);
                  Bridge* bridge = biboumi_component.find_user_bridge(requester.bare());
                  if (bridge)
                    {
                      if (database_value.is_set)
                        bridge->set_record_history(database_value.value);
                      else
                        { // It is unset, we need to fetch the Global option, to
                          // know if it’s enabled or not
                          auto g_options = Database::get_global_options(requester.bare());
                          bridge->set_record_history(g_options.col<Database::RecordHistory>());
                        }
                    }
                }

            }
          Database::invalidate_encoding_in_cache(requester.bare(), iid.get_server(), iid.get_local());
          save(options, *Database::db);
        }
      return true;
    }
  return false;
}
#endif  // USE_DATABASE

void DisconnectUserFromServerStep1(XmppComponent& xmpp_component, AdhocSession& session, XmlNode& command_node)
{
  const Jid owner(session.get_owner_jid());
  if (!Config::is_in_list("admin", owner.bare()))
    { // A non-admin is not allowed to disconnect other users, only
      // him/herself, so we just skip this step
      auto next_step = session.get_next_step();
      next_step(xmpp_component, session, command_node);
    }
  else
    { // Send a form to select the user to disconnect
      auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);

      XmlSubNode x(command_node, "jabber:x:data:x");
      x["type"] = "form";
      XmlSubNode title(x, "title");
      title.set_inner("Disconnect a user from selected IRC servers");
      XmlSubNode instructions(x, "instructions");
      instructions.set_inner("Choose a user JID");
      XmlSubNode jids_field(x, "field");
      jids_field["var"] = "jid";
      jids_field["type"] = "list-single";
      jids_field["label"] = "The JID to disconnect";
      XmlSubNode required(jids_field, "required");
      for (Bridge* bridge: biboumi_component.get_bridges())
        {
          XmlSubNode option(jids_field, "option");
          option["label"] = bridge->get_jid();
          XmlSubNode value(option, "value");
          value.set_inner(bridge->get_jid());
        }
    }
}

void DisconnectUserFromServerStep2(XmppComponent& xmpp_component, AdhocSession& session, XmlNode& command_node)
{
  // If no JID is contained in the command node, it means we skipped the
  // previous stage, and the jid to disconnect is the executor's jid
  std::string jid_to_disconnect = session.get_owner_jid();

  if (const XmlNode* x = command_node.get_child("x", "jabber:x:data"))
    {
      for (const XmlNode* field: x->get_children("field", "jabber:x:data"))
        if (field->get_tag("var") == "jid")
          {
            if (const XmlNode* value = field->get_child("value", "jabber:x:data"))
              jid_to_disconnect = value->get_inner();
          }
    }

  // Save that JID for the last step
  session.vars["jid"] = jid_to_disconnect;

  // Send a data form to let the user choose which server to disconnect the
  // user from
  command_node.delete_all_children();
  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);

  XmlSubNode x(command_node, "jabber:x:data:x");
  x["type"] = "form";
  XmlSubNode title(x, "title");
  title.set_inner("Disconnect a user from selected IRC servers");
  XmlSubNode instructions(x, "instructions");
  instructions.set_inner("Choose one or more servers to disconnect this JID from");
  XmlSubNode jids_field(x, "field");
  jids_field["var"] = "irc-servers";
  jids_field["type"] = "list-multi";
  jids_field["label"] = "The servers to disconnect from";
  XmlSubNode required(jids_field, "required");
  Bridge* bridge = biboumi_component.find_user_bridge(jid_to_disconnect);

  if (!bridge || bridge->get_irc_clients().empty())
    {
      XmlSubNode note(command_node, "note");
      note["type"] = "info";
      note.set_inner("User " + jid_to_disconnect + " is not connected to any IRC server.");
      session.terminate();
      return ;
    }

  for (const auto& pair: bridge->get_irc_clients())
    {
      XmlSubNode option(jids_field, "option");
      option["label"] = pair.first;
      XmlSubNode value(option, "value");
      value.set_inner(pair.first);
    }

  XmlSubNode message_field(x, "field");
  message_field["var"] = "quit-message";
  message_field["type"] = "text-single";
  message_field["label"] = "Quit message";
  XmlSubNode message_value(message_field, "value");
  message_value.set_inner("Killed by admin");
}

void DisconnectUserFromServerStep3(XmppComponent& xmpp_component, AdhocSession& session, XmlNode& command_node)
{
  const auto it = session.vars.find("jid");
  if (it == session.vars.end())
    return ;
  const auto jid_to_disconnect = it->second;

  std::vector<std::string> servers;
  std::string quit_message;

  if (const XmlNode* x = command_node.get_child("x", "jabber:x:data"))
    {
      for (const XmlNode* field: x->get_children("field", "jabber:x:data"))
        {
          if (field->get_tag("var") == "irc-servers")
            {
              for (const XmlNode* value: field->get_children("value", "jabber:x:data"))
                servers.push_back(value->get_inner());
            }
          else if (field->get_tag("var") == "quit-message")
            if (const XmlNode* value = field->get_child("value", "jabber:x:data"))
              quit_message = value->get_inner();
        }
    }

  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);
  Bridge* bridge = biboumi_component.find_user_bridge(jid_to_disconnect);
  auto& clients = bridge->get_irc_clients();

  std::size_t number = 0;

  for (const auto& hostname: servers)
    {
      auto it = clients.find(hostname);
      if (it != clients.end())
        {
          it->second->on_error({"ERROR", {quit_message}});
          clients.erase(it);
          number++;
        }
    }
  command_node.delete_all_children();
  XmlSubNode note(command_node, "note");
  note["type"] = "info";
  std::string msg = jid_to_disconnect + " was disconnected from " + std::to_string(number) + " IRC server";
  if (number > 1)
    msg += "s";
  msg += ".";
  note.set_inner(msg);
}

void GetIrcConnectionInfoStep1(XmppComponent& component, AdhocSession& session, XmlNode& command_node)
{
  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(component);

  const Jid owner(session.get_owner_jid());
  const Jid target(session.get_target_jid());

  std::string message{};

  // As the function is exited, set the message in the response.
  utils::ScopeGuard sg([&message, &command_node]()
                       {
                         command_node.delete_all_children();
                         XmlSubNode note(command_node, "note");
                         note["type"] = "info";
                         note.set_inner(message);
                       });

  Bridge* bridge = biboumi_component.get_user_bridge(owner.bare());
  if (!bridge)
    {
      message = "You are not connected to anything.";
      return;
    }

  std::string hostname;
  if ((hostname = Config::get("fixed_irc_server", "")).empty())
    hostname = target.local;

  IrcClient* irc = bridge->find_irc_client(hostname);
  if (!irc || !irc->is_connected())
    {
      message = "You are not connected to the IRC server " + hostname;
      return;
    }

  std::ostringstream ss;
  ss << "Connected to IRC server " << irc->get_hostname() << " on port " << irc->get_port();
  if (irc->is_using_tls())
    ss << " (using TLS)";
  const std::time_t now_c = std::chrono::system_clock::to_time_t(irc->connection_date);
  struct tm tm;
#ifdef HAS_PUT_TIME
  ss << " since " << std::put_time(localtime_r(&now_c, &tm), "%F %T");
#else
  constexpr std::size_t timestamp_size{10 + 1 + 8 + 1};
  char buf[timestamp_size] = {};
  const auto res = std::strftime(buf, timestamp_size, "%F %T", localtime(&now_c, &tm));
  if (res > 0)
    ss << " since " << buf;
#endif
  ss << " (" << std::chrono::duration_cast<std::chrono::seconds>(std::chrono::system_clock::now() - irc->connection_date).count() << " seconds ago).";

  for (const auto& it: bridge->resources_in_chan)
    {
      const auto& channel_key = it.first;
      const auto& irc_hostname = std::get<1>(channel_key);
      const auto& resources = it.second;

      if (irc_hostname == irc->get_hostname() && !resources.empty())
        {
          const auto& channel_name = std::get<0>(channel_key);
          ss << "\n" << channel_name << " from " << resources.size() << " resource" << (resources.size() > 1 ? "s": "") << ": ";
          for (const auto& resource: resources)
            ss << resource << " ";
        }
    }

  message = ss.str();
}