[2.8.6] Demo Recording/Playback

include/aoapplication.h

@@ -3,6 +3,7 @@
 
 #include "aopacket.h"
 #include "datatypes.h"
+#include "demoserver.h"
 #include "discord_rich_presence.h"
 
 #include "bass.h"
@@ -27,6 +28,8 @@
 #include <QStringList>
 #include <QTextStream>
 
+#include <QElapsedTimer>
+
 class NetworkManager;
 class Lobby;
 class Courtroom;
@@ -261,6 +264,9 @@ class AOApplication : public QApplication {
   // directory if it doesn't exist.
   bool append_to_file(QString p_text, QString p_file, bool make_dir = false);
 
+  // Append to the currently open demo file if there is one
+  void append_to_demofile(QString packet_string);
+
   // Appends the argument string to serverlist.txt
   void write_to_serverlist_txt(QString p_line);
 
@@ -458,6 +464,9 @@ class AOApplication : public QApplication {
                                  void *user);
   static void doBASSreset();
 
+  QElapsedTimer demo_timer;
+  DemoServer* demo_server = nullptr;
+
 private:
   const int RELEASE = 2;
   const int MAJOR_VERSION = 8;

include/aopacket.h

@@ -12,7 +12,7 @@ class AOPacket {
 
   QString get_header() { return m_header; }
   QStringList &get_contents() { return m_contents; }
-  QString to_string();
+  QString to_string(bool encoded = false);
 
   void net_encode();
   void net_decode();

include/demoserver.h

@@ -0,0 +1,55 @@
+#ifndef DEMOSERVER_H
+#define DEMOSERVER_H
+
+#include "aopacket.h"
+
+#include <QDebug>
+#include <QObject>
+#include <QQueue>
+#include <QTcpServer>
+#include <QTcpSocket>
+#include <QTimer>
+#include <QFileDialog>
+
+class DemoServer : public QObject
+{
+    Q_OBJECT
+public:
+    explicit DemoServer(QObject *parent = nullptr);
+
+    bool server_started = false;
+    int port = 27088;
+    int max_wait = -1;
+    int min_wait = -1;
+
+private:
+    void handle_packet(AOPacket packet);
+    void load_demo(QString filename);
+
+    QTcpServer* tcp_server;
+    QTcpSocket* client_sock = nullptr;
+    bool client_connected = false;
+    bool partial_packet = false;
+    QString temp_packet = "";
+    QQueue<QString> demo_data;
+    QString sc_packet;
+    int num_chars = 0;
+    QString p_path;
+    QTimer *timer;
+    int elapsed_time = 0;
+
+private slots:
+    void accept_connection();
+    void destroy_connection();
+    void recv_data();
+    void client_disconnect();
+    void playback();
+
+public slots:
+    void start_server();
+
+signals:
+
+};
+
+#endif // DEMOSERVER_H

src/aoapplication.cpp

@@ -45,6 +45,10 @@ void AOApplication::construct_lobby()
   if (is_discord_enabled())
     discord->state_lobby();
 
+  if (demo_server)
+      demo_server->deleteLater();
+  demo_server = new DemoServer();
+
   w_lobby->show();
 }
 

src/aopacket.cpp

@@ -8,9 +8,15 @@ AOPacket::AOPacket(QString p_packet_string)
   m_contents = packet_contents.mid(1, packet_contents.size()-2); // trims %
 }
 
-QString AOPacket::to_string()
+QString AOPacket::to_string(bool encoded)
 {
-  return m_header + "#" + m_contents.join("#") + "#%";
+  QStringList contents = m_contents;
+  if (encoded)
+    contents.replaceInStrings("#", "<num>")
+      .replaceInStrings("%", "<percent>")
+      .replaceInStrings("$", "<dollar>")
+      .replaceInStrings("&", "<and>");
+  return m_header + "#" + contents.join("#") + "#%";
 }
 
 void AOPacket::net_encode()

src/courtroom.cpp

@@ -1147,13 +1147,20 @@ void Courtroom::done_received()
   objection_player->set_volume(0);
   blip_player->set_volume(0);
 
-  set_char_select_page();
+  if (char_list.size() > 0)
+  {
+    set_char_select_page();
+    set_char_select();
+  }
+  else
+  {
+    update_character(m_cid);
+    enter_courtroom();
+  }
 
   set_mute_list();
   set_pair_list();
 
-  set_char_select();
-
   show();
 
   ui_spectator->show();
@@ -1267,8 +1274,6 @@ void Courtroom::set_pos_dropdown(QStringList pos_dropdowns)
   ui_pos_dropdown->addItems(pos_dropdown_list);
   // Unblock the signals so the element can be used for setting pos again
   ui_pos_dropdown->blockSignals(false);
-
-  qDebug() << pos_dropdown_list;
 }
 
 void Courtroom::update_character(int p_cid)
@@ -1311,7 +1316,6 @@ void Courtroom::update_character(int p_cid)
   set_sfx_dropdown();
   set_effects_dropdown();
 
-  qDebug() << "update_character called";
   if (newchar) // Avoid infinite loop of death and suffering
     set_iniswap_dropdown();
 
@@ -2507,7 +2511,7 @@ void Courtroom::play_char_sfx(QString sfx_name)
 void Courtroom::initialize_chatbox()
 {
   int f_charid = m_chatmessage[CHAR_ID].toInt();
-  if (f_charid >= 0 &&
+  if (f_charid >= 0 && f_charid < char_list.size() &&
       (m_chatmessage[SHOWNAME].isEmpty() || !ui_showname_enable->isChecked())) {
     QString real_name = char_list.at(f_charid).name;
 
@@ -2955,7 +2959,7 @@ void Courtroom::log_ic_text(QString p_name, QString p_showname,
 {
   chatlogpiece log_entry(p_name, p_showname, p_message, p_action, p_color);
   ic_chatlog_history.append(log_entry);
-  if (ao_app->get_auto_logging_enabled())
+  if (ao_app->get_auto_logging_enabled() && !ao_app->log_filename.isEmpty())
     ao_app->append_to_file(log_entry.get_full(), ao_app->log_filename, true);
 
   while (ic_chatlog_history.size() > log_maximum_blocks &&
@@ -3120,7 +3124,7 @@ void Courtroom::play_preanim(bool immediate)
     else
       anim_state = 1;
     preanim_done();
-    qDebug() << "could not find " + anim_to_find;
+    qDebug() << "W: could not find " + anim_to_find;
     return;
   }
 
@@ -3782,9 +3786,6 @@ void Courtroom::on_ooc_return_pressed()
 {
   QString ooc_message = ui_ooc_chat_message->text();
 
-  if (ooc_message == "" || ui_ooc_chat_name->text() == "")
-    return;
-
   if (ooc_message.startsWith("/pos")) {
     if (ooc_message == "/pos jud") {
       toggle_judge_buttons(true);
@@ -4737,7 +4738,6 @@ void Courtroom::on_area_list_double_clicked(QTreeWidgetItem *p_item, int column)
   QStringList packet_contents;
   packet_contents.append(p_area);
   packet_contents.append(QString::number(m_cid));
-  qDebug() << packet_contents;
   ao_app->send_server_packet(new AOPacket("MC", packet_contents), false);
 }