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