blob: 38750c62b0f03cddbdc6a8939ba2d6c8aea5d37a [file] [log] [blame]
Zhenkai Zhu6d589aa2012-05-29 17:34:35 -07001#include <QtGui>
2#include "chatdialog.h"
Zhenkai Zhu85845d22012-06-01 23:10:43 -07003#include "settingdialog.h"
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -07004#include <ctime>
Zhenkai Zhub6338822012-05-31 13:27:24 -07005#include <iostream>
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -07006#include <QTimer>
Zhenkai Zhu10ccb5a2012-06-04 21:55:14 -07007#include <QMetaType>
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -07008#include <QMessageBox>
Zhenkai Zhu59245aa2012-09-26 16:07:04 -07009#include <boost/random/random_device.hpp>
10#include <boost/random/uniform_int_distribution.hpp>
Zhenkai Zhu6d589aa2012-05-29 17:34:35 -070011
Zhenkai Zhu0b3fa332012-09-27 21:58:43 -070012#define BROADCAST_PREFIX_FOR_SYNC_DEMO "/ndn/broadcast/chronos"
Zhenkai Zhu82a62752012-06-04 17:11:04 -070013
Zhenkai Zhuee5c90f2012-09-27 14:05:41 -070014static const int HELLO_INTERVAL = 90; // seconds
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -070015
Zhenkai Zhu6d589aa2012-05-29 17:34:35 -070016ChatDialog::ChatDialog(QWidget *parent)
Zhenkai Zhu25e33e52012-09-28 13:00:07 -070017 : QDialog(parent), m_sock(NULL), m_lastMsgTime(0)
Zhenkai Zhu6d589aa2012-05-29 17:34:35 -070018{
Zhenkai Zhu10ccb5a2012-06-04 21:55:14 -070019 // have to register this, otherwise
20 // the signal-slot system won't recognize this type
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -070021 qRegisterMetaType<std::vector<Sync::MissingDataInfo> >("std::vector<Sync::MissingDataInfo>");
22 qRegisterMetaType<size_t>("size_t");
Zhenkai Zhu6d589aa2012-05-29 17:34:35 -070023 setupUi(this);
Zhenkai Zhu82a62752012-06-04 17:11:04 -070024 m_session = time(NULL);
Zhenkai Zhuee5c90f2012-09-27 14:05:41 -070025 boost::random::random_device rng;
26 boost::random::uniform_int_distribution<> uniform(1, 29000);
27 m_randomizedInterval = HELLO_INTERVAL * 1000 + uniform(rng);
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -070028
29 readSettings();
Zhenkai Zhu82a62752012-06-04 17:11:04 -070030
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -070031 updateLabels();
32
Zhenkai Zhu6d589aa2012-05-29 17:34:35 -070033 lineEdit->setFocusPolicy(Qt::StrongFocus);
Zhenkai Zhu82a62752012-06-04 17:11:04 -070034 m_scene = new DigestTreeScene(this);
Zhenkai Zhub45e38a2012-06-01 15:44:36 -070035
Zhenkai Zhu82a62752012-06-04 17:11:04 -070036 treeViewer->setScene(m_scene);
37 m_scene->plot("Empty");
38 QRectF rect = m_scene->itemsBoundingRect();
39 m_scene->setSceneRect(rect);
40
Zhenkai Zhu9036e032012-09-27 20:59:33 -070041 listView->setStyleSheet("QListView { alternate-background-color: white; background: #F0F0F0; color: darkGreen; font: bold large; }");
Zhenkai Zhuf55f4382012-09-28 10:58:54 -070042 listView->setEditTriggers(QAbstractItemView::NoEditTriggers);
43 listView->setDragDropMode(QAbstractItemView::NoDragDrop);
44 listView->setSelectionMode(QAbstractItemView::NoSelection);
Zhenkai Zhu6082ede2012-09-27 17:28:46 -070045 m_rosterModel = new QStringListModel(this);
46 listView->setModel(m_rosterModel);
47
Zhenkai Zhu86df7412012-09-27 16:30:20 -070048 createActions();
49 createTrayIcon();
50 m_timer = new QTimer(this);
51 connect(lineEdit, SIGNAL(returnPressed()), this, SLOT(returnPressed()));
52 connect(setButton, SIGNAL(pressed()), this, SLOT(buttonPressed()));
Zhenkai Zhub60b7e12012-09-28 11:34:21 -070053 connect(treeButton, SIGNAL(pressed()), this, SLOT(treeButtonPressed()));
Zhenkai Zhu86df7412012-09-27 16:30:20 -070054 connect(this, SIGNAL(dataReceived(QString, const char *, size_t, bool)), this, SLOT(processData(QString, const char *, size_t, bool)));
55 connect(this, SIGNAL(treeUpdated(const std::vector<Sync::MissingDataInfo>)), this, SLOT(processTreeUpdate(const std::vector<Sync::MissingDataInfo>)));
Zhenkai Zhu86df7412012-09-27 16:30:20 -070056 connect(m_timer, SIGNAL(timeout()), this, SLOT(replot()));
57 connect(m_scene, SIGNAL(replot()), this, SLOT(replot()));
58 connect(trayIcon, SIGNAL(messageClicked()), this, SLOT(showNormal()));
59 connect(trayIcon, SIGNAL(activated(QSystemTrayIcon::ActivationReason)), this, SLOT(iconActivated(QSystemTrayIcon::ActivationReason)));
Zhenkai Zhu25e33e52012-09-28 13:00:07 -070060 connect(m_scene, SIGNAL(rosterChanged(QStringList)), this, SLOT(updateRosterList(QStringList)));
Zhenkai Zhu86df7412012-09-27 16:30:20 -070061
Zhenkai Zhu82a62752012-06-04 17:11:04 -070062 // create sync socket
63 if(!m_user.getChatroom().isEmpty()) {
64 std::string syncPrefix = BROADCAST_PREFIX_FOR_SYNC_DEMO;
65 syncPrefix += "/";
66 syncPrefix += m_user.getChatroom().toStdString();
Zhenkai Zhua4fb1242012-06-05 20:26:05 -070067 try
68 {
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -070069 m_sock = new Sync::SyncAppSocket(syncPrefix, bind(&ChatDialog::processTreeUpdateWrapper, this, _1, _2), bind(&ChatDialog::processRemoveWrapper, this, _1));
Zhenkai Zhu25e33e52012-09-28 13:00:07 -070070 sendJoin();
Zhenkai Zhu86df7412012-09-27 16:30:20 -070071 m_timer->start(FRESHNESS * 2000);
Zhenkai Zhua4fb1242012-06-05 20:26:05 -070072 }
73 catch (Sync::CcnxOperationException ex)
74 {
Zhenkai Zhu0b3fa332012-09-27 21:58:43 -070075 QMessageBox::critical(this, tr("Chronos"), tr("Canno connect to ccnd.\n Have you started your ccnd?"), QMessageBox::Ok);
Zhenkai Zhua4fb1242012-06-05 20:26:05 -070076 std::exit(1);
77 }
Zhenkai Zhu82a62752012-06-04 17:11:04 -070078 }
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -070079
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -070080}
81
Zhenkai Zhu82a62752012-06-04 17:11:04 -070082ChatDialog::~ChatDialog()
83{
84 if (m_sock != NULL)
85 {
Zhenkai Zhu25e33e52012-09-28 13:00:07 -070086 SyncDemo::ChatMessage msg;
Zhenkai Zhub5b78462012-09-28 14:10:37 -070087 formControlMessage(msg, SyncDemo::ChatMessage::LEAVE);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -070088 sendMsg(msg);
Zhenkai Zhu3859e192012-09-28 14:04:26 -070089 sleep(1);
Zhenkai Zhu591e8c32012-09-26 11:57:50 -070090 m_sock->remove(m_user.getPrefix().toStdString());
Zhenkai Zhu3859e192012-09-28 14:04:26 -070091 sleep(1);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -070092#ifdef __DEBUG
93 std::cout << "Sync REMOVE signal sent" << std::endl;
94#endif
Zhenkai Zhu82a62752012-06-04 17:11:04 -070095 delete m_sock;
96 m_sock = NULL;
97 }
98}
99
Zhenkai Zhu86df7412012-09-27 16:30:20 -0700100void
101ChatDialog::replot()
102{
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700103 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu86df7412012-09-27 16:30:20 -0700104 m_scene->plot(m_sock->getRootDigest().c_str());
105}
106
Zhenkai Zhu6082ede2012-09-27 17:28:46 -0700107void
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700108ChatDialog::updateRosterList(QStringList staleUserList)
Zhenkai Zhu6082ede2012-09-27 17:28:46 -0700109{
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700110 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu6082ede2012-09-27 17:28:46 -0700111 QStringList rosterList = m_scene->getRosterList();
112 m_rosterModel->setStringList(rosterList);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700113 QString user;
114 QStringListIterator it(staleUserList);
115 while(it.hasNext())
116 {
117 SyncDemo::ChatMessage msg;
Zhenkai Zhub5b78462012-09-28 14:10:37 -0700118 formControlMessage(msg, SyncDemo::ChatMessage::LEAVE);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700119 msg.set_from(it.next().toStdString());
120 appendMessage(msg);
121 }
Zhenkai Zhu6082ede2012-09-27 17:28:46 -0700122}
123
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -0700124void
125ChatDialog::setVisible(bool visible)
126{
127 minimizeAction->setEnabled(visible);
128 maximizeAction->setEnabled(!isMaximized());
129 restoreAction->setEnabled(isMaximized() || !visible);
130 QDialog::setVisible(visible);
131}
132
133void
134ChatDialog::closeEvent(QCloseEvent *e)
135{
136 if (trayIcon->isVisible())
137 {
Zhenkai Zhu0b3fa332012-09-27 21:58:43 -0700138 QMessageBox::information(this, tr("Chronos"),
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -0700139 tr("The program will keep running in the "
140 "system tray. To terminate the program"
141 "choose <b>Quit</b> in the context memu"
142 "of the system tray entry."));
143 hide();
144 e->ignore();
145 }
146}
147
148void
149ChatDialog::changeEvent(QEvent *e)
150{
151 switch(e->type())
152 {
153 case QEvent::ActivationChange:
154 if (isActiveWindow())
155 {
156 trayIcon->setIcon(QIcon(":/images/icon_small.png"));
157 }
158 break;
159 default:
160 break;
161 }
162}
163
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700164void
Zhenkai Zhu10ccb5a2012-06-04 21:55:14 -0700165ChatDialog::appendMessage(const SyncDemo::ChatMessage msg)
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700166{
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700167 boost::recursive_mutex::scoped_lock lock(m_msgMutex);
Zhenkai Zhub6338822012-05-31 13:27:24 -0700168
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700169 if (msg.type() == SyncDemo::ChatMessage::CHAT)
Zhenkai Zhub98e6022012-09-27 13:48:23 -0700170 {
Zhenkai Zhub98e6022012-09-27 13:48:23 -0700171
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700172 if (!msg.has_data())
Zhenkai Zhub98e6022012-09-27 13:48:23 -0700173 {
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700174 return;
Zhenkai Zhub98e6022012-09-27 13:48:23 -0700175 }
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700176
177 if (msg.from().empty() || msg.data().empty())
178 {
179 return;
180 }
181
182 if (!msg.has_timestamp())
183 {
184 return;
185 }
186
187 QTextCharFormat nickFormat;
188 nickFormat.setForeground(Qt::darkGreen);
189 nickFormat.setFontWeight(QFont::Bold);
190 nickFormat.setFontUnderline(true);
191 nickFormat.setUnderlineColor(Qt::gray);
192 QTextCharFormat timeFormat;
193 timeFormat.setForeground(Qt::gray);
194 timeFormat.setFontUnderline(true);
195 timeFormat.setUnderlineColor(Qt::gray);
196
197 QTextCursor cursor(textEdit->textCursor());
198 cursor.movePosition(QTextCursor::End);
199 QTextTableFormat tableFormat;
200 tableFormat.setBorder(0);
201 QTextTable *table = cursor.insertTable(1, 2, tableFormat);
202 QString from = QString("%1 ").arg(msg.from().c_str());
203 QTextTableCell fromCell = table->cellAt(0, 0);
204 fromCell.setFormat(nickFormat);
205 fromCell.firstCursorPosition().insertText(from);
206 QTextTableCell timeCell = table->cellAt(0, 1);
207 timeCell.setFormat(timeFormat);
208 time_t timestamp = msg.timestamp();
209 struct tm *tm_time = localtime(&timestamp);
210 int hour = tm_time->tm_hour;
211 QString amOrPM;
212 if (hour > 12)
213 {
214 hour -= 12;
215 amOrPM = "PM";
216 }
217 else
218 {
219 amOrPM = "AM";
220 if (hour == 0)
221 {
222 hour = 12;
223 }
224 }
225
226 char textTime[12];
227 sprintf(textTime, "%d:%02d:%02d %s", hour, tm_time->tm_min, tm_time->tm_sec, amOrPM.toStdString().c_str());
228 timeCell.firstCursorPosition().insertText(textTime);
229
230
231 QTextCursor nextCursor(textEdit->textCursor());
232 nextCursor.movePosition(QTextCursor::End);
233 table = nextCursor.insertTable(1, 1, tableFormat);
234 table->cellAt(0, 0).firstCursorPosition().insertText(msg.data().c_str());
235 showMessage(from, msg.data().c_str());
Zhenkai Zhub98e6022012-09-27 13:48:23 -0700236 }
237
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700238 if (msg.type() == SyncDemo::ChatMessage::JOIN || msg.type() == SyncDemo::ChatMessage::LEAVE)
239 {
240 QTextCharFormat nickFormat;
241 nickFormat.setForeground(Qt::gray);
242 nickFormat.setFontWeight(QFont::Bold);
243 nickFormat.setFontUnderline(true);
244 nickFormat.setUnderlineColor(Qt::gray);
245 QTextCharFormat timeFormat;
246 timeFormat.setForeground(Qt::gray);
247 timeFormat.setFontUnderline(true);
248 timeFormat.setUnderlineColor(Qt::gray);
Zhenkai Zhub98e6022012-09-27 13:48:23 -0700249
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700250 QTextCursor cursor(textEdit->textCursor());
251 cursor.movePosition(QTextCursor::End);
252 QTextTableFormat tableFormat;
253 tableFormat.setBorder(0);
254 QTextTable *table = cursor.insertTable(1, 2, tableFormat);
255 QString action;
256 if (msg.type() == SyncDemo::ChatMessage::JOIN)
257 {
258 action = "enters room";
259 }
260 else
261 {
262 action = "leaves room";
263 }
264
265 QString from = QString("%1 %2 ").arg(msg.from().c_str()).arg(action);
266 QTextTableCell fromCell = table->cellAt(0, 0);
267 fromCell.setFormat(nickFormat);
268 fromCell.firstCursorPosition().insertText(from);
269 QTextTableCell timeCell = table->cellAt(0, 1);
270 timeCell.setFormat(timeFormat);
271 time_t timestamp = msg.timestamp();
272 struct tm *tm_time = localtime(&timestamp);
273 int hour = tm_time->tm_hour;
274 QString amOrPM;
275 if (hour > 12)
276 {
277 hour -= 12;
278 amOrPM = "PM";
279 }
280 else
281 {
282 amOrPM = "AM";
283 if (hour == 0)
284 {
285 hour = 12;
286 }
287 }
288
289 char textTime[12];
290 sprintf(textTime, "%d:%02d:%02d %s", hour, tm_time->tm_min, tm_time->tm_sec, amOrPM.toStdString().c_str());
291 timeCell.firstCursorPosition().insertText(textTime);
292 }
293
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700294 QScrollBar *bar = textEdit->verticalScrollBar();
295 bar->setValue(bar->maximum());
296}
297
Zhenkai Zhu85845d22012-06-01 23:10:43 -0700298void
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700299ChatDialog::processTreeUpdateWrapper(const std::vector<Sync::MissingDataInfo> v, Sync::SyncAppSocket *sock)
300{
301 emit treeUpdated(v);
Zhenkai Zhud1c5a972012-06-05 14:07:41 -0700302#ifdef __DEBUG
303 std::cout << "<<< Tree update signal emitted" << std::endl;
304#endif
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700305}
306
307void
308ChatDialog::processTreeUpdate(const std::vector<Sync::MissingDataInfo> v)
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700309{
Zhenkai Zhud1c5a972012-06-05 14:07:41 -0700310#ifdef __DEBUG
311 std::cout << "<<< processing Tree Update" << std::endl;
312#endif
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700313 if (v.empty())
314 {
315 return;
316 }
317
318 // reflect the changes on digest tree
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700319 {
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700320 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700321 m_scene->processUpdate(v, m_sock->getRootDigest().c_str());
322 }
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700323
324 int n = v.size();
325 int totalMissingPackets = 0;
326 for (int i = 0; i < n; i++)
327 {
328 totalMissingPackets += v[i].high.getSeq() - v[i].low.getSeq() + 1;
329 }
330
Zhenkai Zhubb198112012-09-27 11:31:42 -0700331 for (int i = 0; i < n; i++)
332 {
333 if (totalMissingPackets < 4)
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700334 {
335 for (Sync::SeqNo seq = v[i].low; seq <= v[i].high; ++seq)
336 {
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700337 m_sock->fetchRaw(v[i].prefix, seq, bind(&ChatDialog::processDataWrapper, this, _1, _2, _3), 2);
Zhenkai Zhud1c5a972012-06-05 14:07:41 -0700338#ifdef __DEBUG
339 std::cout << "<<< Fetching " << v[i].prefix << "/" <<seq.getSession() <<"/" << seq.getSeq() << std::endl;
340#endif
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700341 }
342 }
Zhenkai Zhubb198112012-09-27 11:31:42 -0700343 else
344 {
345 m_sock->fetchRaw(v[i].prefix, v[i].high, bind(&ChatDialog::processDataNoShowWrapper, this, _1, _2, _3), 2);
346 }
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700347 }
348
349 // adjust the view
350 fitView();
351
352}
353
354void
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700355ChatDialog::processDataWrapper(std::string name, const char *buf, size_t len)
356{
Zhenkai Zhubb198112012-09-27 11:31:42 -0700357 emit dataReceived(name.c_str(), buf, len, true);
Zhenkai Zhud1c5a972012-06-05 14:07:41 -0700358#ifdef __DEBUG
359 std::cout <<"<<< " << name << " fetched" << std::endl;
360#endif
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700361}
362
363void
Zhenkai Zhubb198112012-09-27 11:31:42 -0700364ChatDialog::processDataNoShowWrapper(std::string name, const char *buf, size_t len)
365{
366 emit dataReceived(name.c_str(), buf, len, false);
367}
368
369void
370ChatDialog::processData(QString name, const char *buf, size_t len, bool show)
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700371{
372 SyncDemo::ChatMessage msg;
373 if (!msg.ParseFromArray(buf, len))
374 {
Zhenkai Zhubb198112012-09-27 11:31:42 -0700375 std::cerr << "Errrrr.. Can not parse msg with name: " << name.toStdString() << ". what is happening?" << std::endl;
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700376 }
377
Zhenkai Zhu10ccb5a2012-06-04 21:55:14 -0700378 // display msg received from network
379 // we have to do so; this function is called by ccnd thread
380 // so if we call appendMsg directly
381 // Qt crash as "QObject: Cannot create children for a parent that is in a different thread"
382 // the "cannonical" way to is use signal-slot
Zhenkai Zhubb198112012-09-27 11:31:42 -0700383 if (show)
384 {
385 appendMessage(msg);
386 }
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700387
388 // update the tree view
Zhenkai Zhu097bfe72012-06-05 14:30:17 -0700389 std::string stdStrName = name.toStdString();
390 std::string stdStrNameWithoutSeq = stdStrName.substr(0, stdStrName.find_last_of('/'));
391 std::string prefix = stdStrNameWithoutSeq.substr(0, stdStrNameWithoutSeq.find_last_of('/'));
392#ifdef __DEBUG
393 std::cout <<"<<< updating scene for" << prefix << ": " << msg.from() << std::endl;
394#endif
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700395 if (msg.type() == SyncDemo::ChatMessage::LEAVE)
396 {
397 processRemove(prefix.c_str());
398 }
399 else
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700400 {
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700401 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700402 m_scene->msgReceived(prefix.c_str(), msg.from().c_str());
403 }
Zhenkai Zhuc5470612012-06-05 12:28:59 -0700404 fitView();
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700405}
406
407void
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700408ChatDialog::processRemoveWrapper(std::string prefix)
409{
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700410#ifdef __DEBUG
411 std::cout << "Sync REMOVE signal received for prefix: " << prefix << std::endl;
412#endif
413 //emit removeReceived(prefix.c_str());
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700414}
415
416void
417ChatDialog::processRemove(QString prefix)
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700418{
Zhenkai Zhu591e8c32012-09-26 11:57:50 -0700419#ifdef __DEBUG
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700420 std::cout << "<<< remove node for prefix" << prefix.toStdString() << std::endl;
Zhenkai Zhu591e8c32012-09-26 11:57:50 -0700421#endif
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700422 bool removed = m_scene->removeNode(prefix);
Zhenkai Zhu591e8c32012-09-26 11:57:50 -0700423 if (removed)
424 {
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700425 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu591e8c32012-09-26 11:57:50 -0700426 m_scene->plot(m_sock->getRootDigest().c_str());
427 }
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700428}
429
430void
Zhenkai Zhu85845d22012-06-01 23:10:43 -0700431ChatDialog::formChatMessage(const QString &text, SyncDemo::ChatMessage &msg) {
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700432 msg.set_from(m_user.getNick().toStdString());
433 msg.set_to(m_user.getChatroom().toStdString());
Zhenkai Zhu85845d22012-06-01 23:10:43 -0700434 msg.set_data(text.toStdString());
435 time_t seconds = time(NULL);
436 msg.set_timestamp(seconds);
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700437 msg.set_type(SyncDemo::ChatMessage::CHAT);
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700438}
439
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700440void
Zhenkai Zhub5b78462012-09-28 14:10:37 -0700441ChatDialog::formControlMessage(SyncDemo::ChatMessage &msg, SyncDemo::ChatMessage::ChatMessageType type)
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700442{
443 msg.set_from(m_user.getNick().toStdString());
444 msg.set_to(m_user.getChatroom().toStdString());
445 time_t seconds = time(NULL);
446 msg.set_timestamp(seconds);
Zhenkai Zhub5b78462012-09-28 14:10:37 -0700447 msg.set_type(type);
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700448}
449
Zhenkai Zhu59245aa2012-09-26 16:07:04 -0700450static std::string chars("qwertyuiopasdfghjklzxcvbnmQWERTYUIOPASDFGHJKLZXCVBNM0123456789");
451
452QString
453ChatDialog::getRandomString()
454{
455 std::string randStr;
456 boost::random::random_device rng;
457 boost::random::uniform_int_distribution<> index_dist(0, chars.size() - 1);
458 for (int i = 0; i < 10; i ++)
459 {
460 randStr += chars[index_dist(rng)];
461 }
462 return randStr.c_str();
463}
464
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700465bool
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700466ChatDialog::readSettings()
467{
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700468#ifndef __DEBUG
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700469 QSettings s(ORGANIZATION, APPLICATION);
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700470 QString nick = s.value("nick", "").toString();
471 QString chatroom = s.value("chatroom", "").toString();
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700472 QString originPrefix = s.value("originPrefix", "").toString();
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700473 if (nick == "" || chatroom == "" || prefix == "") {
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700474 QTimer::singleShot(500, this, SLOT(buttonPressed()));
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700475 return false;
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700476 }
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700477 else {
478 m_user.setNick(nick);
479 m_user.setChatroom(chatroom);
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700480 m_user.setOriginPrefix(originPrefix);
481 m_user.setPrefix(origin_prefix + "/" + chatroom + "/" + getRandomString());
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700482 return true;
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700483 }
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700484#else
485 QTimer::singleShot(500, this, SLOT(buttonPressed()));
486 return false;
487#endif
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700488}
489
490void
491ChatDialog::writeSettings()
492{
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700493#ifndef __DEBUG
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700494 QSettings s(ORGANIZATION, APPLICATION);
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700495 s.setValue("nick", m_user.getNick());
496 s.setValue("chatroom", m_user.getChatroom());
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700497 s.setValue("originPrefix", m_user.getOriginPrefix());
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700498#endif
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700499}
500
501void
502ChatDialog::updateLabels()
503{
Zhenkai Zhu76ff02b2012-09-27 21:11:03 -0700504 QString settingDisp = QString("Chatroom: %1").arg(m_user.getChatroom());
505 infoLabel->setStyleSheet("QLabel {color: #630; font-size: 16px; font: bold \"Verdana\";}");
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700506 infoLabel->setText(settingDisp);
Zhenkai Zhub60b7e12012-09-28 11:34:21 -0700507 //QString prefixDisp = QString("<Prefix: %1>").arg(m_user.getPrefix());
508 //prefixLabel->setText(prefixDisp);
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700509}
510
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700511void
512ChatDialog::returnPressed()
513{
514 QString text = lineEdit->text();
515 if (text.isEmpty())
516 return;
517
Zhenkai Zhub6338822012-05-31 13:27:24 -0700518 lineEdit->clear();
519
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700520 SyncDemo::ChatMessage msg;
521 formChatMessage(text, msg);
522
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700523 appendMessage(msg);
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700524
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700525 sendMsg(msg);
526
527 fitView();
528}
529
530void
531ChatDialog::sendMsg(SyncDemo::ChatMessage &msg)
532{
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700533 // send msg
534 size_t size = msg.ByteSize();
535 char *buf = new char[size];
536 msg.SerializeToArray(buf, size);
537 if (!msg.IsInitialized())
538 {
539 std::cerr << "Errrrr.. msg was not probally initialized "<<__FILE__ <<":"<<__LINE__<<". what is happening?" << std::endl;
540 abort();
541 }
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700542 m_sock->publishRaw(m_user.getPrefix().toStdString(), m_session, buf, size, FRESHNESS);
Zhenkai Zhuc5470612012-06-05 12:28:59 -0700543
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700544 delete buf;
545
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700546 m_lastMsgTime = time(NULL);
547
Zhenkai Zhuc5470612012-06-05 12:28:59 -0700548 int nextSequence = m_sock->getNextSeq(m_user.getPrefix().toStdString(), m_session);
Zhenkai Zhud1c5a972012-06-05 14:07:41 -0700549 Sync::MissingDataInfo mdi = {m_user.getPrefix().toStdString(), Sync::SeqNo(0), Sync::SeqNo(nextSequence - 1)};
Zhenkai Zhuc5470612012-06-05 12:28:59 -0700550 std::vector<Sync::MissingDataInfo> v;
551 v.push_back(mdi);
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700552 {
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700553 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700554 m_scene->processUpdate(v, m_sock->getRootDigest().c_str());
555 m_scene->msgReceived(m_user.getPrefix(), m_user.getNick());
556 }
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700557}
558
559void
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700560ChatDialog::sendJoin()
561{
562 SyncDemo::ChatMessage msg;
Zhenkai Zhub5b78462012-09-28 14:10:37 -0700563 formControlMessage(msg, SyncDemo::ChatMessage::JOIN);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700564 sendMsg(msg);
565 QTimer::singleShot(m_randomizedInterval, this, SLOT(sendHello()));
566}
567
568void
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700569ChatDialog::sendHello()
570{
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700571 time_t now = time(NULL);
572 int elapsed = now - m_lastMsgTime;
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700573 if (elapsed >= m_randomizedInterval / 1000)
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700574 {
575 SyncDemo::ChatMessage msg;
Zhenkai Zhub5b78462012-09-28 14:10:37 -0700576 formControlMessage(msg, SyncDemo::ChatMessage::HELLO);
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700577 sendMsg(msg);
Zhenkai Zhuee5c90f2012-09-27 14:05:41 -0700578 QTimer::singleShot(m_randomizedInterval, this, SLOT(sendHello()));
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700579 }
580 else
581 {
Zhenkai Zhuee5c90f2012-09-27 14:05:41 -0700582 QTimer::singleShot((m_randomizedInterval - elapsed * 1000), this, SLOT(sendHello()));
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700583 }
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700584}
585
586void
Zhenkai Zhu85845d22012-06-01 23:10:43 -0700587ChatDialog::buttonPressed()
588{
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700589 SettingDialog dialog(this, m_user.getNick(), m_user.getChatroom(), m_user.getOriginPrefix());
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700590 connect(&dialog, SIGNAL(updated(QString, QString, QString)), this, SLOT(settingUpdated(QString, QString, QString)));
Zhenkai Zhu85845d22012-06-01 23:10:43 -0700591 dialog.exec();
Zhenkai Zhue837f792012-06-05 20:47:54 -0700592 QTimer::singleShot(100, this, SLOT(checkSetting()));
593}
594
Zhenkai Zhub60b7e12012-09-28 11:34:21 -0700595void ChatDialog::treeButtonPressed()
596{
597 if (treeViewer->isVisible())
598 {
599 treeViewer->hide();
600 treeButton->setText("Show Sync Tree");
601 }
602 else
603 {
604 treeViewer->show();
605 treeButton->setText("Hide Sync Tree");
606 }
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700607
608 fitView();
Zhenkai Zhub60b7e12012-09-28 11:34:21 -0700609}
610
Zhenkai Zhue837f792012-06-05 20:47:54 -0700611void
612ChatDialog::checkSetting()
613{
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700614 if (m_user.getOriginPrefix().isEmpty() || m_user.getNick().isEmpty() || m_user.getChatroom().isEmpty())
Zhenkai Zhue837f792012-06-05 20:47:54 -0700615 {
616 buttonPressed();
617 }
Zhenkai Zhu6d589aa2012-05-29 17:34:35 -0700618}
Zhenkai Zhue08afe02012-05-31 15:49:07 -0700619
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700620void
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700621ChatDialog::settingUpdated(QString nick, QString chatroom, QString originPrefix)
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700622{
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700623 QString randString = getRandomString();
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700624 bool needWrite = false;
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700625 bool needFresh = false;
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700626 if (!nick.isEmpty() && nick != m_user.getNick()) {
627 m_user.setNick(nick);
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700628 needWrite = true;
629 }
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700630 if (!originPrefix.isEmpty() && originPrefix != m_user.getOriginPrefix()) {
631 m_user.setOriginPrefix(originPrefix);
632 m_user.setPrefix(originPrefix + "/" + m_user.getChatroom() + "/" + randString);
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700633 needWrite = true;
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700634 needFresh = true;
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700635 }
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700636 if (!chatroom.isEmpty() && chatroom != m_user.getChatroom()) {
637 m_user.setChatroom(chatroom);
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700638 m_user.setPrefix(m_user.getOriginPrefix() + "/" + chatroom + "/" + randString);
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700639 needWrite = true;
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700640 needFresh = true;
641 }
642
643 if (needFresh)
644 {
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700645
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700646 {
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700647 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700648 m_scene->clearAll();
649 m_scene->plot("Empty");
650 }
Zhenkai Zhucc4c2c02012-09-27 21:24:37 -0700651
652 textEdit->clear();
653
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700654 // TODO: perhaps need to do a lot. e.g. use a new SyncAppSokcet
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700655 if (m_sock != NULL)
656 {
657 delete m_sock;
658 m_sock = NULL;
659 }
660 std::string syncPrefix = BROADCAST_PREFIX_FOR_SYNC_DEMO;
661 syncPrefix += "/";
662 syncPrefix += m_user.getChatroom().toStdString();
Zhenkai Zhua4fb1242012-06-05 20:26:05 -0700663 try
664 {
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700665 m_sock = new Sync::SyncAppSocket(syncPrefix, bind(&ChatDialog::processTreeUpdateWrapper, this, _1, _2), bind(&ChatDialog::processRemoveWrapper, this, _1));
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700666 sendJoin();
Zhenkai Zhu86df7412012-09-27 16:30:20 -0700667 m_timer->start(FRESHNESS * 2000);
Zhenkai Zhua4fb1242012-06-05 20:26:05 -0700668 }
669 catch (Sync::CcnxOperationException ex)
670 {
Zhenkai Zhu0b3fa332012-09-27 21:58:43 -0700671 QMessageBox::critical(this, tr("Chronos"), tr("Canno connect to ccnd.\n Have you started your ccnd?"), QMessageBox::Ok);
Zhenkai Zhua4fb1242012-06-05 20:26:05 -0700672 std::exit(1);
673 }
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700674
675 fitView();
676
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700677 }
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700678
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700679 if (needWrite) {
680 writeSettings();
681 updateLabels();
682 }
683}
Zhenkai Zhud13acd02012-06-04 15:25:20 -0700684
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -0700685void
686ChatDialog::iconActivated(QSystemTrayIcon::ActivationReason reason)
687{
688 switch (reason)
689 {
690 case QSystemTrayIcon::Trigger:
691 case QSystemTrayIcon::DoubleClick:
692 break;
693 case QSystemTrayIcon::MiddleClick:
694 // showMessage();
695 break;
696 default:;
697 }
698}
699
700void
701ChatDialog::showMessage(QString from, QString data)
702{
703 // std::cout <<"Showing Message: " << from.toStdString() << ": " << data.toStdString() << std::endl;
704 if (!isActiveWindow())
705 {
706 trayIcon->showMessage(QString("Chatroom %1 has a new message").arg(m_user.getChatroom()), QString("<%1>: %2").arg(from).arg(data), QSystemTrayIcon::Information, 20000);
707 trayIcon->setIcon(QIcon(":/images/note.png"));
708 }
709}
710
711void
712ChatDialog::messageClicked()
713{
714 this->showMaximized();
715}
716
717void
718ChatDialog::createActions()
719{
720 minimizeAction = new QAction(tr("Mi&nimize"), this);
721 connect(minimizeAction, SIGNAL(triggered()), this, SLOT(hide()));
722
723 maximizeAction = new QAction(tr("Ma&ximize"), this);
724 connect(maximizeAction, SIGNAL(triggered()), this, SLOT(showMaximized()));
725
726 restoreAction = new QAction(tr("&Restore"), this);
727 connect(restoreAction, SIGNAL(triggered()), this, SLOT(showNormal()));
728
729 quitAction = new QAction(tr("Quit"), this);
730 connect(quitAction, SIGNAL(triggered()), qApp, SLOT(quit()));
731}
732
733void
734ChatDialog::createTrayIcon()
735{
736 trayIconMenu = new QMenu(this);
737 trayIconMenu->addAction(minimizeAction);
738 trayIconMenu->addAction(maximizeAction);
739 trayIconMenu->addAction(restoreAction);
740 trayIconMenu->addSeparator();
741 trayIconMenu->addAction(quitAction);
742
743 trayIcon = new QSystemTrayIcon(this);
744 trayIcon->setContextMenu(trayIconMenu);
745
746 QIcon icon(":/images/icon_small.png");
747 trayIcon->setIcon(icon);
748 setWindowIcon(icon);
Zhenkai Zhu0b3fa332012-09-27 21:58:43 -0700749 trayIcon->setToolTip("Chronos System Tray Icon");
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -0700750 trayIcon->setVisible(true);
751}
752
Zhenkai Zhud13acd02012-06-04 15:25:20 -0700753void
754ChatDialog::resizeEvent(QResizeEvent *e)
755{
756 fitView();
757}
758
759void
760ChatDialog::showEvent(QShowEvent *e)
761{
762 fitView();
763}
764
765void
766ChatDialog::fitView()
767{
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700768 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu21d75f92012-06-04 21:23:34 -0700769 QRectF rect = m_scene->itemsBoundingRect();
770 m_scene->setSceneRect(rect);
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700771 treeViewer->fitInView(m_scene->itemsBoundingRect(), Qt::KeepAspectRatio);
Zhenkai Zhud13acd02012-06-04 15:25:20 -0700772}