biboumi_adhoc_commands.cpp 26.9 KB
Newer Older
1
#include <xmpp/biboumi_adhoc_commands.hpp>
2
#include <xmpp/biboumi_component.hpp>
3 4
#include <utils/scopeguard.hpp>
#include <bridge/bridge.hpp>
5
#include <config/config.hpp>
6
#include <utils/string.hpp>
7
#include <utils/split.hpp>
8
#include <xmpp/jid.hpp>
louiz’'s avatar
louiz’ committed
9
#include <algorithm>
louiz’'s avatar
louiz’ committed
10
#include <sstream>
11
#include <iomanip>
12 13 14 15 16 17 18

#include <biboumi.h>

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

19 20 21 22
#ifndef HAS_PUT_TIME
#include <ctime>
#endif

23
using namespace std::string_literals;
24

25
void DisconnectUserStep1(XmppComponent& xmpp_component, AdhocSession&, XmlNode& command_node)
26
{
louiz’'s avatar
louiz’ committed
27
  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);
28

29
  XmlSubNode x(command_node, "jabber:x:data:x");
30
  x["type"] = "form";
31
  XmlSubNode title(x, "title");
32
  title.set_inner("Disconnect a user from the gateway");
33
  XmlSubNode instructions(x, "instructions");
34
  instructions.set_inner("Choose a user JID and a quit message");
35
  XmlSubNode jids_field(x, "field");
36 37 38
  jids_field["var"] = "jids";
  jids_field["type"] = "list-multi";
  jids_field["label"] = "The JIDs to disconnect";
39
  XmlSubNode required(jids_field, "required");
40
  for (Bridge* bridge: biboumi_component.get_bridges())
41
    {
42
      XmlSubNode option(jids_field, "option");
43
      option["label"] = bridge->get_jid();
44
      XmlSubNode value(option, "value");
45 46 47
      value.set_inner(bridge->get_jid());
    }

48
  XmlSubNode message_field(x, "field");
49 50 51
  message_field["var"] = "quit-message";
  message_field["type"] = "text-single";
  message_field["label"] = "Quit message";
52
  XmlSubNode message_value(message_field, "value");
53 54 55
  message_value.set_inner("Disconnected by admin");
}

56
void DisconnectUserStep2(XmppComponent& xmpp_component, AdhocSession& session, XmlNode& command_node)
57
{
louiz’'s avatar
louiz’ committed
58
  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);
59

60 61
  // Find out if the jids, and the quit message are provided in the form.
  std::string quit_message;
62
  const XmlNode* x = command_node.get_child("x", "jabber:x:data");
63 64
  if (x)
    {
65 66 67
      const XmlNode* message_field = nullptr;
      const XmlNode* jids_field = nullptr;
      for (const XmlNode* field: x->get_children("field", "jabber:x:data"))
68 69 70 71 72 73
        if (field->get_tag("var") == "jids")
          jids_field = field;
        else if (field->get_tag("var") == "quit-message")
          message_field = field;
      if (message_field)
        {
74
          const XmlNode* value = message_field->get_child("value", "jabber:x:data");
75 76 77 78 79 80
          if (value)
            quit_message = value->get_inner();
        }
      if (jids_field)
        {
          std::size_t num = 0;
81
          for (const XmlNode* value: jids_field->get_children("value", "jabber:x:data"))
82
            {
83
              Bridge* bridge = biboumi_component.find_user_bridge(value->get_inner());
84 85 86 87 88 89 90 91
              if (bridge)
                {
                  bridge->shutdown(quit_message);
                  num++;
                }
            }
          command_node.delete_all_children();

92
          XmlSubNode note(command_node, "note");
93 94 95 96 97 98 99
          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.");
100
          return;
101 102
        }
    }
103
  XmlSubNode error(command_node, ADHOC_NS":error");
104
  error["type"] = "modify";
105
  XmlSubNode condition(error, STANZA_NS":bad-request");
106 107
  session.terminate();
}
108 109

#ifdef USE_DATABASE
110 111 112 113 114 115 116 117

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());

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

125
  XmlSubNode max_histo_length(x, "field");
126 127 128 129 130
  max_histo_length["var"] = "max_history_length";
  max_histo_length["type"] = "text-single";
  max_histo_length["label"] = "Max history length";
  max_histo_length["desc"] = "The maximum number of lines in the history that the server sends when joining a channel";

131 132
  {
    XmlSubNode value(max_histo_length, "value");
133
    value.set_inner(std::to_string(options.col<Database::MaxHistoryLength>()));
134
  }
135

136
  XmlSubNode record_history(x, "field");
137 138 139 140 141
  record_history["var"] = "record_history";
  record_history["type"] = "boolean";
  record_history["label"] = "Record history";
  record_history["desc"] = "Whether to save the messages into the database, or not";

142 143 144
  {
    XmlSubNode value(record_history, "value");
    value.set_name("value");
145
    if (options.col<Database::RecordHistory>())
146 147 148 149
      value.set_inner("true");
    else
      value.set_inner("false");
  }
150 151
}

152
void ConfigureGlobalStep2(XmppComponent& xmpp_component, AdhocSession& session, XmlNode& command_node)
153
{
louiz’'s avatar
louiz’ committed
154
  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);
155

156 157 158 159 160 161 162 163 164 165 166
  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());
      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())
167
            options.col<Database::MaxHistoryLength>() = atoi(value->get_inner().data());
168 169 170
          else if (field->get_tag("var") == "record_history" &&
                   value && !value->get_inner().empty())
            {
171
              options.col<Database::RecordHistory>() = to_bool(value->get_inner());
172 173
              Bridge* bridge = biboumi_component.find_user_bridge(owner.bare());
              if (bridge)
174
                bridge->set_record_history(options.col<Database::RecordHistory>());
175
            }
176 177
        }

178
      options.save(Database::db);
179 180

      command_node.delete_all_children();
181
      XmlSubNode note(command_node, "note");
182 183 184 185
      note["type"] = "info";
      note.set_inner("Configuration successfully applied.");
      return;
    }
186
  XmlSubNode error(command_node, ADHOC_NS":error");
187
  error["type"] = "modify";
188
  XmlSubNode condition(error, STANZA_NS":bad-request");
189 190 191
  session.terminate();
}

192
void ConfigureIrcServerStep1(XmppComponent&, AdhocSession& session, XmlNode& command_node)
193 194 195
{
  const Jid owner(session.get_owner_jid());
  const Jid target(session.get_target_jid());
196 197 198
  std::string server_domain;
  if ((server_domain = Config::get("fixed_irc_server", "")).empty())
    server_domain = target.local;
199
  auto options = Database::get_irc_server_options(owner.local + "@" + owner.domain,
200
                                                  server_domain);
201

202
  XmlSubNode x(command_node, "jabber:x:data:x");
203
  x["type"] = "form";
204
  XmlSubNode title(x, "title");
205
  title.set_inner("Configure the IRC server "s + server_domain);
206
  XmlSubNode instructions(x, "instructions");
207
  instructions.set_inner("Edit the form, to configure the settings of the IRC server "s + server_domain);
208

209
  XmlSubNode ports(x, "field");
210 211 212 213
  ports["var"] = "ports";
  ports["type"] = "text-multi";
  ports["label"] = "Ports";
  ports["desc"] = "List of ports to try, without TLS. Defaults: 6667.";
214
  for (const auto& val: utils::split(options.col<Database::Ports>(), ';', false))
215
    {
216
      XmlSubNode ports_value(ports, "value");
217 218 219 220
      ports_value.set_inner(val);
    }

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

232
  XmlSubNode verify_cert(x, "field");
233 234 235 236
  verify_cert["var"] = "verify_cert";
  verify_cert["type"] = "boolean";
  verify_cert["label"] = "Verify certificate";
  verify_cert["desc"] = "Whether or not to abort the connection if the server’s TLS certificate is invalid";
237
  XmlSubNode verify_cert_value(verify_cert, "value");
238
  if (options.col<Database::VerifyCert>())
239 240 241
    verify_cert_value.set_inner("true");
  else
    verify_cert_value.set_inner("false");
242

243
  XmlSubNode fingerprint(x, "field");
244 245 246
  fingerprint["var"] = "fingerprint";
  fingerprint["type"] = "text-single";
  fingerprint["label"] = "SHA-1 fingerprint of the TLS certificate to trust.";
247
  if (!options.col<Database::TrustedFingerprint>().empty())
248
    {
249
      XmlSubNode fingerprint_value(fingerprint, "value");
250
      fingerprint_value.set_inner(options.col<Database::TrustedFingerprint>());
251
    }
252
#endif
253

254
  XmlSubNode pass(x, "field");
255 256
  pass["var"] = "pass";
  pass["type"] = "text-private";
louiz’'s avatar
louiz’ committed
257 258
  pass["label"] = "Server password";
  pass["desc"] = "Will be used in a PASS command when connecting";
259
  if (!options.col<Database::Pass>().empty())
260
    {
261
      XmlSubNode pass_value(pass, "value");
262
      pass_value.set_inner(options.col<Database::Pass>());
263 264
    }

265
  XmlSubNode after_cnt_cmd(x, "field");
266 267 268 269
  after_cnt_cmd["var"] = "after_connect_command";
  after_cnt_cmd["type"] = "text-single";
  after_cnt_cmd["desc"] = "Custom IRC command sent after the connection is established with the server.";
  after_cnt_cmd["label"] = "After-connection IRC command";
270
  if (!options.col<Database::AfterConnectionCommand>().empty())
271
    {
272
      XmlSubNode after_cnt_cmd_value(after_cnt_cmd, "value");
273
      after_cnt_cmd_value.set_inner(options.col<Database::AfterConnectionCommand>());
274 275
    }

276
  if (Config::get("realname_customization", "true") == "true")
277
    {
278
      XmlSubNode username(x, "field");
279 280 281
      username["var"] = "username";
      username["type"] = "text-single";
      username["label"] = "Username";
282
      if (!options.col<Database::Username>().empty())
283
        {
284
          XmlSubNode username_value(username, "value");
285
          username_value.set_inner(options.col<Database::Username>());
286 287
        }

288
      XmlSubNode realname(x, "field");
289 290 291
      realname["var"] = "realname";
      realname["type"] = "text-single";
      realname["label"] = "Realname";
292
      if (!options.col<Database::Realname>().empty())
293
        {
294
          XmlSubNode realname_value(realname, "value");
295
          realname_value.set_inner(options.col<Database::Realname>());
296
        }
297 298
    }

299
  XmlSubNode encoding_out(x, "field");
300 301 302 303
  encoding_out["var"] = "encoding_out";
  encoding_out["type"] = "text-single";
  encoding_out["desc"] = "The encoding used when sending messages to the IRC server.";
  encoding_out["label"] = "Out encoding";
304
  if (!options.col<Database::EncodingOut>().empty())
305
    {
306
      XmlSubNode encoding_out_value(encoding_out, "value");
307
      encoding_out_value.set_inner(options.col<Database::EncodingOut>());
308 309
    }

310
  XmlSubNode encoding_in(x, "field");
311 312 313 314
  encoding_in["var"] = "encoding_in";
  encoding_in["type"] = "text-single";
  encoding_in["desc"] = "The encoding used to decode message received from the IRC server.";
  encoding_in["label"] = "In encoding";
315
  if (!options.col<Database::EncodingIn>().empty())
316
    {
317
      XmlSubNode encoding_in_value(encoding_in, "value");
318
      encoding_in_value.set_inner(options.col<Database::EncodingIn>());
319
    }
320 321
}

322
void ConfigureIrcServerStep2(XmppComponent&, AdhocSession& session, XmlNode& command_node)
323 324 325 326 327 328
{
  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());
329 330 331
      std::string server_domain;
      if ((server_domain = Config::get("fixed_irc_server", "")).empty())
        server_domain = target.local;
332
      auto options = Database::get_irc_server_options(owner.local + "@" + owner.domain,
333
                                                      server_domain);
334 335 336
      for (const XmlNode* field: x->get_children("field", "jabber:x:data"))
        {
          const XmlNode* value = field->get_child("value", "jabber:x:data");
337 338 339 340 341 342
          const std::vector<const XmlNode*> values = field->get_children("value", "jabber:x:data");
          if (field->get_tag("var") == "ports")
            {
              std::string ports;
              for (const auto& val: values)
                ports += val->get_inner() + ";";
343
              options.col<Database::Ports>() = ports;
344 345 346 347 348 349 350 351
            }

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

355
          else if (field->get_tag("var") == "verify_cert" && value
356
                   && !value->get_inner().empty())
357 358
            {
              auto val = to_bool(value->get_inner());
359
              options.col<Database::VerifyCert>() = val;
360
            }
361 362 363 364

          else if (field->get_tag("var") == "fingerprint" && value &&
                   !value->get_inner().empty())
            {
365
              options.col<Database::TrustedFingerprint>() = value->get_inner();
366 367
            }

368
#endif // BOTAN_FOUND
369 370

          else if (field->get_tag("var") == "pass" &&
371
                   value && !value->get_inner().empty())
372
            options.col<Database::Pass>() = value->get_inner();
373 374 375

          else if (field->get_tag("var") == "after_connect_command" &&
                   value && !value->get_inner().empty())
376
            options.col<Database::AfterConnectionCommand>() = value->get_inner();
377 378 379

          else if (field->get_tag("var") == "username" &&
                   value && !value->get_inner().empty())
380 381 382
            {
              auto username = value->get_inner();
              // The username must not contain spaces
383
              std::replace(username.begin(), username.end(), ' ', '_');
384
              options.col<Database::Username>() = username;
385
            }
386 387 388

          else if (field->get_tag("var") == "realname" &&
                   value && !value->get_inner().empty())
389
            options.col<Database::Realname>() = value->get_inner();
390 391 392

          else if (field->get_tag("var") == "encoding_out" &&
                   value && !value->get_inner().empty())
393
            options.col<Database::EncodingOut>() = value->get_inner();
394 395 396

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

399 400
        }

401
      options.save(Database::db);
402 403

      command_node.delete_all_children();
404
      XmlSubNode note(command_node, "note");
405 406 407 408
      note["type"] = "info";
      note.set_inner("Configuration successfully applied.");
      return;
    }
409
  XmlSubNode error(command_node, ADHOC_NS":error");
410
  error["type"] = "modify";
411
  XmlSubNode condition(error, STANZA_NS":bad-request");
412 413 414 415 416 417 418
  session.terminate();
}

void ConfigureIrcChannelStep1(XmppComponent&, AdhocSession& session, XmlNode& command_node)
{
  const Jid owner(session.get_owner_jid());
  const Jid target(session.get_target_jid());
419 420 421 422 423 424

  insert_irc_channel_configuration_form(command_node, owner, target);
}

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

  auto options = Database::get_irc_channel_options_with_server_default(requester.local + "@" + requester.domain,
428 429
                                                                       iid.get_server(), iid.get_local());

430
  XmlSubNode x(node, "jabber:x:data:x");
431
  x["type"] = "form";
432
  XmlSubNode title(x, "title");
433
  title.set_inner("Configure the IRC channel "s + iid.get_local() + " on server "s + iid.get_server());
434
  XmlSubNode instructions(x, "instructions");
435 436
  instructions.set_inner("Edit the form, to configure the settings of the IRC channel "s + iid.get_local());

437
  XmlSubNode encoding_out(x, "field");
438 439 440 441
  encoding_out["var"] = "encoding_out";
  encoding_out["type"] = "text-single";
  encoding_out["desc"] = "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";
442
  if (!options.col<Database::EncodingOut>().empty())
443
    {
444
      XmlSubNode encoding_out_value(encoding_out, "value");
445
      encoding_out_value.set_inner(options.col<Database::EncodingOut>());
446 447
    }

448
  XmlSubNode encoding_in(x, "field");
449 450 451 452
  encoding_in["var"] = "encoding_in";
  encoding_in["type"] = "text-single";
  encoding_in["desc"] = "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";
453
  if (!options.col<Database::EncodingIn>().empty())
454
    {
455
      XmlSubNode encoding_in_value(encoding_in, "value");
456
      encoding_in_value.set_inner(options.col<Database::EncodingIn>());
457
    }
louiz’'s avatar
louiz’ committed
458 459 460 461 462 463 464 465 466

  XmlSubNode persistent(x, "field");
  persistent["var"] = "persistent";
  persistent["type"] = "boolean";
  persistent["desc"] = "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");
467
    if (options.col<Database::Persistent>())
louiz’'s avatar
louiz’ committed
468 469 470 471
      value.set_inner("true");
    else
      value.set_inner("false");
  }
472 473 474 475
}

void ConfigureIrcChannelStep2(XmppComponent&, AdhocSession& session, XmlNode& command_node)
{
476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497
  const Jid owner(session.get_owner_jid());
  const Jid target(session.get_target_jid());

  if (handle_irc_channel_configuration_form(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(const XmlNode& node, const Jid& requester, const Jid& target)
{
  const XmlNode* x = node.get_child("x", "jabber:x:data");
498 499
  if (x)
    {
500
      if (x->get_tag("type") == "submit")
501
        {
502 503 504 505 506 507
          const Iid iid(target.local, {});
          auto options = Database::get_irc_channel_options(requester.local + "@" + requester.domain,
                                                           iid.get_server(), iid.get_local());
          for (const XmlNode *field: x->get_children("field", "jabber:x:data"))
            {
              const XmlNode *value = field->get_child("value", "jabber:x:data");
508

509 510
              if (field->get_tag("var") == "encoding_out" &&
                  value && !value->get_inner().empty())
511
                options.col<Database::EncodingOut>() = value->get_inner();
louiz’'s avatar
louiz’ committed
512

513 514
              else if (field->get_tag("var") == "encoding_in" &&
                       value && !value->get_inner().empty())
515
                options.col<Database::EncodingIn>() = value->get_inner();
516

517 518
              else if (field->get_tag("var") == "persistent" &&
                       value)
519
                options.col<Database::Persistent>() = to_bool(value->get_inner());
520
            }
521

522
          options.save(Database::db);
523 524
        }
      return true;
525
    }
526
  return false;
527 528
}
#endif  // USE_DATABASE
529

530
void DisconnectUserFromServerStep1(XmppComponent& xmpp_component, AdhocSession& session, XmlNode& command_node)
531 532 533 534 535 536 537 538 539 540
{
  const Jid owner(session.get_owner_jid());
  if (owner.bare() != Config::get("admin", ""))
    { // 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
louiz’'s avatar
louiz’ committed
541
      auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);
542

543
      XmlSubNode x(command_node, "jabber:x:data:x");
544
      x["type"] = "form";
545
      XmlSubNode title(x, "title");
546
      title.set_inner("Disconnect a user from selected IRC servers");
547
      XmlSubNode instructions(x, "instructions");
548
      instructions.set_inner("Choose a user JID");
549
      XmlSubNode jids_field(x, "field");
550 551 552
      jids_field["var"] = "jid";
      jids_field["type"] = "list-single";
      jids_field["label"] = "The JID to disconnect";
553
      XmlSubNode required(jids_field, "required");
554
      for (Bridge* bridge: biboumi_component.get_bridges())
555
        {
556
          XmlSubNode option(jids_field, "option");
557
          option["label"] = bridge->get_jid();
558
          XmlSubNode value(option, "value");
559 560 561 562 563
          value.set_inner(bridge->get_jid());
        }
    }
}

564
void DisconnectUserFromServerStep2(XmppComponent& xmpp_component, AdhocSession& session, XmlNode& command_node)
565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585
{
  // 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();
louiz’'s avatar
louiz’ committed
586
  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);
587

588
  XmlSubNode x(command_node, "jabber:x:data:x");
589
  x["type"] = "form";
590
  XmlSubNode title(x, "title");
591
  title.set_inner("Disconnect a user from selected IRC servers");
592
  XmlSubNode instructions(x, "instructions");
593
  instructions.set_inner("Choose one or more servers to disconnect this JID from");
594
  XmlSubNode jids_field(x, "field");
595 596 597
  jids_field["var"] = "irc-servers";
  jids_field["type"] = "list-multi";
  jids_field["label"] = "The servers to disconnect from";
598
  XmlSubNode required(jids_field, "required");
599
  Bridge* bridge = biboumi_component.find_user_bridge(jid_to_disconnect);
600 601 602

  if (!bridge || bridge->get_irc_clients().empty())
    {
603
      XmlSubNode note(command_node, "note");
604 605 606 607 608 609 610 611
      note["type"] = "info";
      note.set_inner("User "s + jid_to_disconnect + " is not connected to any IRC server.");
      session.terminate();
      return ;
    }

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

618
  XmlSubNode message_field(x, "field");
619 620 621
  message_field["var"] = "quit-message";
  message_field["type"] = "text-single";
  message_field["label"] = "Quit message";
622
  XmlSubNode message_value(message_field, "value");
623 624 625
  message_value.set_inner("Killed by admin");
}

626
void DisconnectUserFromServerStep3(XmppComponent& xmpp_component, AdhocSession& session, XmlNode& command_node)
627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650
{
  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();
        }
    }

louiz’'s avatar
louiz’ committed
651
  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(xmpp_component);
652
  Bridge* bridge = biboumi_component.find_user_bridge(jid_to_disconnect);
653 654 655 656 657 658 659 660 661 662 663 664 665 666 667
  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();
668
  XmlSubNode note(command_node, "note");
669 670 671 672 673 674 675
  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);
}
676 677 678

void GetIrcConnectionInfoStep1(XmppComponent& component, AdhocSession& session, XmlNode& command_node)
{
louiz’'s avatar
louiz’ committed
679
  auto& biboumi_component = dynamic_cast<BiboumiComponent&>(component);
680 681 682 683 684 685 686 687 688 689

  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();
690
                         XmlSubNode note(command_node, "note");
691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717
                         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 "s + 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);
718
#ifdef HAS_PUT_TIME
719
  ss << " since " << std::put_time(std::localtime(&now_c), "%F %T");
720 721 722 723 724 725 726
#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", std::localtime(&now_c));
  if (res > 0)
    ss << " since " << buf;
#endif
727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745
  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();
}