git.sesse.net Git - ultimatescore/blob - client/mainwindow.cpp

   1 #include "mainwindow.h"
   2 #include "post_to_main_thread.h"
   3 #include "ui_mainwindow.h"
   4
   5 #include <arpa/inet.h>
   6 #include <netinet/in.h>
   7 #include <netinet/tcp.h>
   8 #include <stdlib.h>
   9 #include <sys/types.h>
  10 #include <sys/socket.h>
  11
  12 using namespace std;
  13
  14 string escape_html(const string &str)
  15 {
  16         string s = "";
  17         for (char ch : str) {
  18                 if (ch == '<') {
  19                         s += "&lt;";
  20                 } else if (ch == '>') {
  21                         s += "&gt;";
  22                 } else if (ch == '&') {
  23                         s += "&amp;";
  24                 } else {
  25                         s += ch;
  26                 }
  27         }
  28         return s;
  29 }
  30
  31 string escape_unicode(const string &str)
  32 {
  33         string s = "";
  34         for (size_t pos = 0; pos < str.size(); ) {
  35                 wchar_t wc;
  36                 int len = mbtowc(&wc, str.data() + pos, str.size() - pos);
  37                 if (len == -1) {
  38                         wc = '?';
  39                         len = 1;
  40                 }
  41                 pos += len;
  42
  43                 if (wc == '\\') {
  44                         s += "\\\\";
  45                 } else if (isprint(wc)) {
  46                         s += wc;
  47                 } else {
  48                         char buf[16];
  49                         snprintf(buf, sizeof(buf), "\\u%04x", wc);
  50                         s += buf;
  51                 }
  52         }
  53         return s;
  54 }
  55
  56 string escape_quotes(const string &str)
  57 {
  58         string s = "";
  59         for (char ch : str) {
  60                 if (ch == '"') {
  61                         s += '\\';
  62                 }
  63                 s += ch;
  64         }
  65         return s;
  66 }
  67
  68 string serialize_as_json(const map<string, string> &param)
  69 {
  70         string s = "{";
  71
  72         bool first = true;
  73         for (const auto &key_value : param) {
  74                 if (!first) s += ", ";
  75                 first = false;
  76
  77                 s += '"';
  78                 s += escape_quotes(escape_unicode(key_value.first));
  79                 s += "\": \"";
  80                 s += escape_quotes(escape_unicode(key_value.second));
  81                 s += '"';
  82         }
  83         s += "}";
  84         return s;
  85 }
  86
  87 MainWindow::MainWindow(QWidget *parent) :
  88     QMainWindow(parent),
  89     ui(new Ui::MainWindow)
  90 {
  91         ui->setupUi(this);
  92         ws = new WSServer("127.0.0.1", 5250);
  93         ws->set_connection_callback([this](bool connected) {
  94                 string msg = connected ? "Connected" : "Not connected";
  95                 post_to_main_thread([this, msg]() {
  96                         ui->ws_connected_label->setText(QString::fromStdString(msg));
  97                 });
  98         });
  99
 100         udp_thread = std::thread(&MainWindow::udp_thread_func, this, 6000);
 101         udp_thread2 = std::thread(&MainWindow::udp_thread_func, this, 6001);
 102
 103         connect(ui->ws_disconnect_btn, &QPushButton::clicked, this, &MainWindow::ws_disconnect_clicked);
 104         connect(ui->set_initials_btn, &QPushButton::clicked, this, &MainWindow::set_initials_clicked);
 105         connect(ui->set_match_2_initials_btn, &QPushButton::clicked, this, &MainWindow::set_match_2_initials_clicked);
 106         connect(ui->set_color_btn, &QPushButton::clicked, this, &MainWindow::set_color_clicked);
 107         connect(ui->set_score_btn, &QPushButton::clicked, this, &MainWindow::set_score_clicked);
 108         connect(ui->set_all_scorebug_btn, &QPushButton::clicked, this, &MainWindow::set_all_scorebug_clicked);
 109         connect(ui->goal_1_btn, &QPushButton::clicked, this, [this]() { add_goal(ui->score_1_box, 1); });
 110         connect(ui->ungoal_1_btn, &QPushButton::clicked, this, [this]() { add_goal(ui->score_1_box, -1); });
 111         connect(ui->goal_2_btn, &QPushButton::clicked, this, [this]() { add_goal(ui->score_2_box, 1); });
 112         connect(ui->ungoal_2_btn, &QPushButton::clicked, this, [this]() { add_goal(ui->score_2_box, -1); });
 113
 114         connect(ui->set_clock_btn, &QPushButton::clicked, this, &MainWindow::set_clock_clicked);
 115         connect(ui->set_clock_limit_btn, &QPushButton::clicked, this, &MainWindow::set_clock_limit_clicked);
 116         connect(ui->start_and_show_clock_btn, &QPushButton::clicked, this, &MainWindow::start_and_show_clock_clicked);
 117         connect(ui->stop_clock_btn, &QPushButton::clicked, this, &MainWindow::stop_clock_clicked);
 118         connect(ui->show_clock_btn, &QPushButton::clicked, this, &MainWindow::show_clock_clicked);
 119         connect(ui->hide_clock_btn, &QPushButton::clicked, this, &MainWindow::hide_clock_clicked);
 120         connect(ui->show_match_2_btn, &QPushButton::clicked, this, &MainWindow::show_match_2_clicked);
 121         connect(ui->hide_match_2_btn, &QPushButton::clicked, this, &MainWindow::hide_match_2_clicked);
 122
 123         connect(ui->set_comment_btn, &QPushButton::clicked, this, &MainWindow::set_comment_clicked);
 124         connect(ui->set_and_show_comment_btn, &QPushButton::clicked, this, &MainWindow::set_and_show_comment_clicked);
 125         connect(ui->hide_comment_btn, &QPushButton::clicked, this, &MainWindow::hide_comment_clicked);
 126         connect(ui->set_and_show_autocomment_btn, &QPushButton::clicked, this, &MainWindow::set_and_show_autocomment_clicked);
 127         connect(ui->autoshow_autocomment, &QCheckBox::stateChanged, this, &MainWindow::autocomment_update);
 128
 129         connect(ui->show_lower_third_btn, &QPushButton::clicked, this, &MainWindow::show_lower_third_clicked);
 130         connect(ui->hide_lower_third_btn, &QPushButton::clicked, this, &MainWindow::hide_lower_third_clicked);
 131
 132         connect(ui->quick_lower_third_edit, &QLineEdit::returnPressed, this, &MainWindow::quick_lower_third_activate);
 133         connect(ui->show_quick_lower_third_btn, &QPushButton::clicked, this, &MainWindow::quick_lower_third_activate);
 134
 135         connect(ui->show_scorebug_btn, &QPushButton::clicked, this, &MainWindow::show_scorebug_clicked);
 136         connect(ui->show_group_a_btn, &QPushButton::clicked, this, [this]() { show_group_clicked("Group A"); });
 137         connect(ui->show_group_b_btn, &QPushButton::clicked, this, [this]() { show_group_clicked("Group B"); });
 138         connect(ui->show_group_c_btn, &QPushButton::clicked, this, [this]() { show_group_clicked("Group C"); });
 139         connect(ui->show_schedule_btn, &QPushButton::clicked, this, &MainWindow::show_schedule_clicked);
 140         connect(ui->show_carousel_btn, &QPushButton::clicked, this, &MainWindow::show_carousel_clicked);
 141         connect(ui->show_nothing_btn, &QPushButton::clicked, this, &MainWindow::show_nothing_clicked);
 142         connect(ui->show_roster_1_btn, &QPushButton::clicked, this, [this]() { show_roster_clicked(ui->initials_1_edit->text().toStdString()); });
 143         connect(ui->show_roster_2_btn, &QPushButton::clicked, this, [this]() { show_roster_clicked(ui->initials_2_edit->text().toStdString()); });
 144         connect(ui->show_roster_carousel_btn, &QPushButton::clicked, this, &MainWindow::show_roster_carousel_clicked);
 145
 146         autocomment_update();
 147
 148         const set<pair<unsigned, unsigned>> usb{{ 0x0e8f, 0x0041 }};
 149         event_device = new EventDevice(usb, ui->quick_lower_third_edit);
 150         event_device->start_thread();
 151 }
 152
 153 MainWindow::~MainWindow()
 154 {
 155         delete ui;
 156 }
 157
 158 void MainWindow::ws_disconnect_clicked()
 159 {
 160         ws->change_port(stoi(ui->ws_port_box->text().toStdString()));
 161 }
 162
 163 void MainWindow::set_initials_clicked()
 164 {
 165         map<string, string> param;
 166         param["team1"] = escape_html(ui->initials_1_edit->text().toStdString());
 167         param["team2"] = escape_html(ui->initials_2_edit->text().toStdString());
 168         ws->send_command("update " + serialize_as_json(param));
 169         ws->send_command("eval setteams()");
 170 }
 171
 172 void MainWindow::set_match_2_initials_clicked()
 173 {
 174         map<string, string> param;
 175         param["team1"] = escape_html(ui->match_2_initials_1_edit->text().toStdString());
 176         param["team2"] = escape_html(ui->match_2_initials_2_edit->text().toStdString());
 177         ws->send_command("update " + serialize_as_json(param));
 178         ws->send_command("eval setteams2()");
 179 }
 180
 181 void MainWindow::set_color_clicked()
 182 {
 183         map<string, string> param;
 184         param["team1color"] = ui->color_1_edit->text().toStdString();  // Should maybe be escaped, but meh.
 185         param["team2color"] = ui->color_2_edit->text().toStdString();
 186         ws->send_command("update " + serialize_as_json(param));
 187         ws->send_command("eval setcolors()");
 188 }
 189
 190 void MainWindow::set_score_clicked()
 191 {
 192         map<string, string> param;
 193         param["score1"] = to_string(ui->score_1_box->value());
 194         param["score2"] = to_string(ui->score_2_box->value());
 195         ws->send_command("update " + serialize_as_json(param));
 196         ws->send_command("eval setscore()");
 197         autocomment_update();
 198 }
 199
 200 void MainWindow::set_all_scorebug_clicked()
 201 {
 202         set_initials_clicked();
 203         set_color_clicked();
 204         set_score_clicked();
 205 }
 206
 207 void MainWindow::add_goal(QSpinBox *box, int delta)
 208 {
 209         box->setValue(box->value() + delta);
 210         set_score_clicked();
 211 }
 212
 213 void MainWindow::set_clock_clicked()
 214 {
 215         map<string, string> param;
 216         param["clock_min"] = to_string(ui->clock_min_box->value());
 217         param["clock_sec"] = to_string(ui->clock_sec_box->value());
 218         ws->send_command("update " + serialize_as_json(param));
 219         ws->send_command("eval setclockfromstate()");
 220 }
 221
 222 void MainWindow::set_clock_limit_clicked()
 223 {
 224         map<string, string> param;
 225         param["clock_limit_min"] = to_string(ui->clock_limit_min_box->value());
 226         param["clock_limit_sec"] = to_string(ui->clock_limit_sec_box->value());
 227         ws->send_command("update " + serialize_as_json(param));
 228         ws->send_command("eval setclocklimitfromstate()");
 229 }
 230
 231 void MainWindow::start_and_show_clock_clicked()
 232 {
 233         ws->send_command("eval startclock()");  // Also shows.
 234 }
 235
 236 void MainWindow::stop_clock_clicked()
 237 {
 238         ws->send_command("eval stopclock()");
 239 }
 240
 241 void MainWindow::show_clock_clicked()
 242 {
 243         ws->send_command("eval showclock()");
 244 }
 245
 246 void MainWindow::hide_clock_clicked()
 247 {
 248         ws->send_command("eval hideclock()");
 249 }
 250
 251 void MainWindow::show_match_2_clicked()
 252 {
 253         ws->send_command("eval showmatch2()");
 254 }
 255
 256 void MainWindow::hide_match_2_clicked()
 257 {
 258         ws->send_command("eval hidematch2()");
 259 }
 260
 261 void MainWindow::set_comment_clicked()
 262 {
 263         map<string, string> param;
 264         param["comment"] = ui->comment_edit->text().toStdString();
 265         ws->send_command("update " + serialize_as_json(param));
 266         ws->send_command("eval setcomment()");
 267 }
 268
 269 void MainWindow::set_and_show_comment_clicked()
 270 {
 271         set_comment_clicked();
 272         ws->send_command("eval showcomment()");
 273 }
 274
 275 void MainWindow::set_and_show_autocomment_clicked()
 276 {
 277         ui->comment_edit->setText(ui->autocomment_edit->text());
 278         set_and_show_comment_clicked();
 279 }
 280
 281 void MainWindow::hide_comment_clicked()
 282 {
 283         ws->send_command("eval hidecomment()");
 284 }
 285
 286 void MainWindow::show_lower_third_clicked()
 287 {
 288         map<string, string> param;
 289         param["text1"] = ui->lowerthird_heading_edit->text().toStdString();
 290         param["text2"] = ui->lowerthird_subheading_edit->text().toStdString();
 291         ws->send_command("update " + serialize_as_json(param));
 292         ws->send_command("eval setandshowlowerthird()");
 293 }
 294
 295 void MainWindow::hide_lower_third_clicked()
 296 {
 297         ws->send_command("eval hidelowerthird()");
 298 }
 299
 300 void MainWindow::quick_lower_third_activate()
 301 {
 302         string code = ui->quick_lower_third_edit->text().toUpper().toStdString();
 303         if (code == "A") {
 304                 add_goal(ui->score_1_box, 1);
 305         } else if (code == "B") {
 306                 add_goal(ui->score_2_box, 1);
 307         } else if (code == "C") {
 308                 ws->send_command("eval hidelowerthird()");
 309         } else {
 310                 map<string, string> param;
 311                 param["code"] = code;
 312                 ws->send_command("update " + serialize_as_json(param));
 313                 ws->send_command("eval quicklowerthird()");
 314         }
 315         ui->quick_lower_third_edit->clear();
 316 }
 317
 318 void MainWindow::autocomment_update()
 319 {
 320         int score1 = ui->score_1_box->value();
 321         int score2 = ui->score_2_box->value();
 322         string msg;
 323         if (abs(score1 - score2) >= 3) {
 324                 msg = "Game ends after this point";
 325         } else {
 326                 int cap = max(score1, score2) + 1;
 327                 if (score1 == score2) ++cap;
 328
 329                 if (cap >= 13) {
 330                         msg = "Point cap: First to 13";
 331                 } else {
 332                         char buf[32];
 333                         snprintf(buf, sizeof(buf), "Pagacap: First to %d", cap);
 334                         msg = buf;
 335                 }
 336         }
 337         ui->autocomment_edit->setText(QString::fromStdString(msg));
 338
 339         map<string, string> param;
 340         param["autocomment_on_clock_limit"] = ui->autoshow_autocomment->isChecked() ? "1" : "0";
 341         param["autocomment"] = msg;
 342         ws->send_command("update " + serialize_as_json(param));
 343 }
 344
 345 void MainWindow::show_scorebug_clicked()
 346 {
 347         ws->send_command("eval stopcarousel()");
 348         ws->send_command("eval hidetable()");
 349         ws->send_command("eval showscorebug()");
 350 }
 351
 352 void MainWindow::show_group_clicked(const std::string &group_name)
 353 {
 354         map<string, string> param;
 355         param["group_name"] = group_name;
 356         ws->send_command("eval stopcarousel()");
 357         ws->send_command("update " + serialize_as_json(param));
 358         ws->send_command("eval showgroup_from_state()");
 359 }
 360
 361 void MainWindow::show_roster_clicked(const std::string &team_code)
 362 {
 363         map<string, string> param;
 364         param["team_code"] = team_code;
 365         ws->send_command("eval stopcarousel()");
 366         ws->send_command("update " + serialize_as_json(param));
 367         ws->send_command("eval showroster_from_state()");
 368 }
 369
 370 void MainWindow::show_schedule_clicked()
 371 {
 372         ws->send_command("eval stopcarousel()");
 373         ws->send_command("eval showschedule()");
 374 }
 375
 376 void MainWindow::show_carousel_clicked()
 377 {
 378         ws->send_command("eval stopcarousel()");
 379         ws->send_command("eval showcarousel()");
 380 }
 381
 382 void MainWindow::show_roster_carousel_clicked()
 383 {
 384         map<string, string> param;
 385         param["team1"] = escape_html(ui->initials_1_edit->text().toStdString());
 386         param["team2"] = escape_html(ui->initials_2_edit->text().toStdString());
 387         ws->send_command("eval stopcarousel()");
 388         ws->send_command("update " + serialize_as_json(param));
 389         ws->send_command("eval showrostercarousel_from_state()");
 390 }
 391
 392 void MainWindow::show_nothing_clicked()
 393 {
 394         ws->send_command("eval hidescorebug()");
 395         ws->send_command("eval stopcarousel()");
 396         ws->send_command("eval hidetable()");
 397 }
 398
 399 void MainWindow::udp_thread_func(int port)
 400 {
 401         int sock = socket(PF_INET6, SOCK_DGRAM, IPPROTO_UDP);
 402         if (sock == -1) {
 403                 perror("socket");
 404                 exit(1);
 405         }
 406
 407         int one = 1;
 408         if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one)) == -1) {
 409                 perror("setsockopt");
 410                 exit(1);
 411         }
 412
 413         sockaddr_in6 saddr6;
 414         memset(&saddr6, 0, sizeof(saddr6));
 415         saddr6.sin6_family = AF_INET6;
 416         inet_pton(AF_INET6, "::", &saddr6.sin6_addr);
 417         saddr6.sin6_port = htons(port);
 418         if (bind(sock, (sockaddr *)&saddr6, sizeof(saddr6)) == -1) {
 419                 perror("bind");
 420                 exit(1);
 421         }
 422
 423         for ( ;; ) {
 424                 char buf[4096];
 425                 int err = recv(sock, buf, sizeof(buf), 0);
 426                 if (err == -1) {
 427                         perror("recv");
 428                         exit(1);
 429                 }
 430
 431                 post_to_main_thread([buf, err, port, this] {
 432                         bt6000_message_received(string(buf, err), port);
 433                 });
 434         }
 435 }
 436
 437 int parse_digit(char ch)
 438 {
 439         if (ch >= '0' && ch <= '9') {
 440                 return ch - '0';
 441         }
 442         return 0;
 443 }
 444
 445 int parse_clock(char ch1, char ch2)
 446 {
 447         int s1 = parse_digit(ch1);
 448         int s2 = parse_digit(ch2);
 449         return s1 * 10 + s2;
 450 }
 451
 452 int parse_score(char ch1, char ch2, char ch3)
 453 {
 454         int s1 = parse_digit(ch1);
 455         int s2 = parse_digit(ch2);
 456         int s3 = parse_digit(ch3);
 457         return s1 * 100 + s2 * 10 + s3;
 458 }
 459
 460 void MainWindow::bt6000_message_received(const string &msg, int port)
 461 {
 462         fprintf(stderr, "BT6000 message: '%s' (port %d)\n", msg.c_str(), port);
 463         if (port == 6001) {
 464                 if (!ui->bt6000_2_enable->isChecked()) {
 465                         return;
 466                 }
 467         } else {
 468                 if (!ui->bt6000_enable->isChecked()) {
 469                         return;
 470                 }
 471         }
 472
 473         if (msg.size() >= 9 && msg[0] == 'G' && msg[1] == '0' && msg[2] == '1') {
 474                 // G01: Game clock, period number, and number of time-outs.
 475                 bool clock_running = !(msg[3] & 0x02);
 476 //              bool klaxon = (msg[3] & 0x04);
 477                 int minutes = parse_clock(msg[5], msg[6]);
 478                 int seconds = parse_clock(msg[7], msg[8]);
 479
 480                 map<string, string> param;
 481                 param["clock_min"] = to_string(minutes);
 482                 param["clock_sec"] = to_string(seconds);
 483                 ws->send_command("update " + serialize_as_json(param));
 484
 485                 if (port == 6001) {
 486                         ws->send_command("eval adjustclock2fromstate()");
 487                         if (clock_running) {
 488                                 ws->send_command("eval startclock2()");
 489                         } else {
 490                                 ws->send_command("eval stopclock2()");
 491                         }
 492                 } else {
 493                         ws->send_command("eval adjustclockfromstate()");
 494                         if (clock_running) {
 495                                 ws->send_command("eval startclock()");
 496                         } else {
 497                                 ws->send_command("eval stopclock()");
 498                         }
 499                 }
 500         }
 501         if (msg.size() >= 10 && msg[0] == 'G' && msg[1] == '0' && msg[2] == '2') {
 502                 int score1 = parse_score(msg[4], msg[5], msg[6]);
 503                 int score2 = parse_score(msg[7], msg[8], msg[9]);
 504                 if (port == 6001) {
 505                         map<string, string> param;
 506                         param["score1"] = to_string(score1);
 507                         param["score2"] = to_string(score2);
 508                         ws->send_command("update " + serialize_as_json(param));
 509                         ws->send_command("eval setscore2()");
 510                 } else {
 511                         ui->score_1_box->setValue(score1);
 512                         ui->score_2_box->setValue(score2);
 513                         set_score_clicked();
 514                 }
 515         }
 516
 517         // Ignore type 3 (penalties) and type 4 (timeouts).
 518 }