blob: 2299738ccaf086ade22e5baa720afa1d0d9d2ff5 [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 {
Alexander Afanasyev288edd82012-10-04 17:07:56 -070069 m_sock = new Sync::SyncAppSocket(syncPrefix,
70 bind(&ChatDialog::processTreeUpdateWrapper, this, _1, _2),
71 bind(&ChatDialog::processRemoveWrapper, this, _1));
72 QTimer::singleShot(1000, this, SLOT(sendJoin()));
Zhenkai Zhu86df7412012-09-27 16:30:20 -070073 m_timer->start(FRESHNESS * 2000);
Zhenkai Zhua4fb1242012-06-05 20:26:05 -070074 }
75 catch (Sync::CcnxOperationException ex)
76 {
Zhenkai Zhu0b3fa332012-09-27 21:58:43 -070077 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 -070078 std::exit(1);
79 }
Zhenkai Zhu82a62752012-06-04 17:11:04 -070080 }
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -070081
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -070082}
83
Zhenkai Zhu82a62752012-06-04 17:11:04 -070084ChatDialog::~ChatDialog()
85{
86 if (m_sock != NULL)
87 {
Zhenkai Zhu25e33e52012-09-28 13:00:07 -070088 SyncDemo::ChatMessage msg;
Zhenkai Zhub5b78462012-09-28 14:10:37 -070089 formControlMessage(msg, SyncDemo::ChatMessage::LEAVE);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -070090 sendMsg(msg);
Zhenkai Zhu3974a492012-09-28 14:39:45 -070091 usleep(500000);
Zhenkai Zhu591e8c32012-09-26 11:57:50 -070092 m_sock->remove(m_user.getPrefix().toStdString());
Zhenkai Zhu3974a492012-09-28 14:39:45 -070093 usleep(5000);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -070094#ifdef __DEBUG
95 std::cout << "Sync REMOVE signal sent" << std::endl;
96#endif
Zhenkai Zhu82a62752012-06-04 17:11:04 -070097 delete m_sock;
98 m_sock = NULL;
99 }
100}
101
Zhenkai Zhu86df7412012-09-27 16:30:20 -0700102void
103ChatDialog::replot()
104{
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700105 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu86df7412012-09-27 16:30:20 -0700106 m_scene->plot(m_sock->getRootDigest().c_str());
Zhenkai Zhuc9e4e3c2012-10-02 11:47:31 -0700107 fitView();
Zhenkai Zhu86df7412012-09-27 16:30:20 -0700108}
109
Zhenkai Zhu6082ede2012-09-27 17:28:46 -0700110void
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700111ChatDialog::updateRosterList(QStringList staleUserList)
Zhenkai Zhu6082ede2012-09-27 17:28:46 -0700112{
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700113 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu6082ede2012-09-27 17:28:46 -0700114 QStringList rosterList = m_scene->getRosterList();
115 m_rosterModel->setStringList(rosterList);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700116 QString user;
117 QStringListIterator it(staleUserList);
118 while(it.hasNext())
119 {
Zhenkai Zhu32cbdce2012-10-02 11:09:38 -0700120 std::string nick = it.next().toStdString();
121 if (nick.empty())
122 continue;
123
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700124 SyncDemo::ChatMessage msg;
Zhenkai Zhub5b78462012-09-28 14:10:37 -0700125 formControlMessage(msg, SyncDemo::ChatMessage::LEAVE);
Zhenkai Zhu32cbdce2012-10-02 11:09:38 -0700126 msg.set_from(nick);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700127 appendMessage(msg);
128 }
Zhenkai Zhu6082ede2012-09-27 17:28:46 -0700129}
130
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -0700131void
132ChatDialog::setVisible(bool visible)
133{
134 minimizeAction->setEnabled(visible);
135 maximizeAction->setEnabled(!isMaximized());
136 restoreAction->setEnabled(isMaximized() || !visible);
137 QDialog::setVisible(visible);
138}
139
140void
141ChatDialog::closeEvent(QCloseEvent *e)
142{
Zhenkai Zhu3974a492012-09-28 14:39:45 -0700143 if (trayIcon->isVisible() && !m_minimaniho)
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -0700144 {
Zhenkai Zhu0b3fa332012-09-27 21:58:43 -0700145 QMessageBox::information(this, tr("Chronos"),
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -0700146 tr("The program will keep running in the "
147 "system tray. To terminate the program"
148 "choose <b>Quit</b> in the context memu"
149 "of the system tray entry."));
150 hide();
151 e->ignore();
Zhenkai Zhu3974a492012-09-28 14:39:45 -0700152 m_minimaniho = true;
153 writeSettings();
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -0700154 }
155}
156
157void
158ChatDialog::changeEvent(QEvent *e)
159{
160 switch(e->type())
161 {
162 case QEvent::ActivationChange:
163 if (isActiveWindow())
164 {
165 trayIcon->setIcon(QIcon(":/images/icon_small.png"));
166 }
167 break;
168 default:
169 break;
170 }
171}
172
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700173void
Zhenkai Zhu10ccb5a2012-06-04 21:55:14 -0700174ChatDialog::appendMessage(const SyncDemo::ChatMessage msg)
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700175{
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700176 boost::recursive_mutex::scoped_lock lock(m_msgMutex);
Zhenkai Zhub6338822012-05-31 13:27:24 -0700177
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700178 if (msg.type() == SyncDemo::ChatMessage::CHAT)
Zhenkai Zhub98e6022012-09-27 13:48:23 -0700179 {
Zhenkai Zhub98e6022012-09-27 13:48:23 -0700180
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700181 if (!msg.has_data())
Zhenkai Zhub98e6022012-09-27 13:48:23 -0700182 {
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700183 return;
Zhenkai Zhub98e6022012-09-27 13:48:23 -0700184 }
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700185
186 if (msg.from().empty() || msg.data().empty())
187 {
188 return;
189 }
190
191 if (!msg.has_timestamp())
192 {
193 return;
194 }
195
196 QTextCharFormat nickFormat;
197 nickFormat.setForeground(Qt::darkGreen);
198 nickFormat.setFontWeight(QFont::Bold);
199 nickFormat.setFontUnderline(true);
200 nickFormat.setUnderlineColor(Qt::gray);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700201
202 QTextCursor cursor(textEdit->textCursor());
203 cursor.movePosition(QTextCursor::End);
204 QTextTableFormat tableFormat;
205 tableFormat.setBorder(0);
206 QTextTable *table = cursor.insertTable(1, 2, tableFormat);
207 QString from = QString("%1 ").arg(msg.from().c_str());
208 QTextTableCell fromCell = table->cellAt(0, 0);
209 fromCell.setFormat(nickFormat);
210 fromCell.firstCursorPosition().insertText(from);
Zhenkai Zhu560ef1b2012-09-28 14:23:33 -0700211
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700212 time_t timestamp = msg.timestamp();
Zhenkai Zhu560ef1b2012-09-28 14:23:33 -0700213 printTimeInCell(table, timestamp);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700214
215 QTextCursor nextCursor(textEdit->textCursor());
216 nextCursor.movePosition(QTextCursor::End);
217 table = nextCursor.insertTable(1, 1, tableFormat);
218 table->cellAt(0, 0).firstCursorPosition().insertText(msg.data().c_str());
219 showMessage(from, msg.data().c_str());
Zhenkai Zhub98e6022012-09-27 13:48:23 -0700220 }
221
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700222 if (msg.type() == SyncDemo::ChatMessage::JOIN || msg.type() == SyncDemo::ChatMessage::LEAVE)
223 {
224 QTextCharFormat nickFormat;
225 nickFormat.setForeground(Qt::gray);
226 nickFormat.setFontWeight(QFont::Bold);
227 nickFormat.setFontUnderline(true);
228 nickFormat.setUnderlineColor(Qt::gray);
Zhenkai Zhub98e6022012-09-27 13:48:23 -0700229
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700230 QTextCursor cursor(textEdit->textCursor());
231 cursor.movePosition(QTextCursor::End);
232 QTextTableFormat tableFormat;
233 tableFormat.setBorder(0);
234 QTextTable *table = cursor.insertTable(1, 2, tableFormat);
235 QString action;
236 if (msg.type() == SyncDemo::ChatMessage::JOIN)
237 {
238 action = "enters room";
239 }
240 else
241 {
242 action = "leaves room";
243 }
244
245 QString from = QString("%1 %2 ").arg(msg.from().c_str()).arg(action);
246 QTextTableCell fromCell = table->cellAt(0, 0);
247 fromCell.setFormat(nickFormat);
248 fromCell.firstCursorPosition().insertText(from);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700249
Zhenkai Zhu560ef1b2012-09-28 14:23:33 -0700250 time_t timestamp = msg.timestamp();
251 printTimeInCell(table, timestamp);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700252 }
253
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700254 QScrollBar *bar = textEdit->verticalScrollBar();
255 bar->setValue(bar->maximum());
256}
257
Zhenkai Zhu560ef1b2012-09-28 14:23:33 -0700258void
259ChatDialog::printTimeInCell(QTextTable *table, time_t timestamp)
260{
261 QTextCharFormat timeFormat;
262 timeFormat.setForeground(Qt::gray);
263 timeFormat.setFontUnderline(true);
264 timeFormat.setUnderlineColor(Qt::gray);
265 QTextTableCell timeCell = table->cellAt(0, 1);
266 timeCell.setFormat(timeFormat);
267 timeCell.firstCursorPosition().insertText(formatTime(timestamp));
268}
269
270QString
271ChatDialog::formatTime(time_t timestamp)
272{
273 struct tm *tm_time = localtime(&timestamp);
274 int hour = tm_time->tm_hour;
275 QString amOrPM;
276 if (hour > 12)
277 {
278 hour -= 12;
279 amOrPM = "PM";
280 }
281 else
282 {
283 amOrPM = "AM";
284 if (hour == 0)
285 {
286 hour = 12;
287 }
288 }
289
290 char textTime[12];
291 sprintf(textTime, "%d:%02d:%02d %s", hour, tm_time->tm_min, tm_time->tm_sec, amOrPM.toStdString().c_str());
292 return QString(textTime);
293}
294
Zhenkai Zhu85845d22012-06-01 23:10:43 -0700295void
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700296ChatDialog::processTreeUpdateWrapper(const std::vector<Sync::MissingDataInfo> v, Sync::SyncAppSocket *sock)
297{
298 emit treeUpdated(v);
Zhenkai Zhud1c5a972012-06-05 14:07:41 -0700299#ifdef __DEBUG
300 std::cout << "<<< Tree update signal emitted" << std::endl;
301#endif
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700302}
303
304void
305ChatDialog::processTreeUpdate(const std::vector<Sync::MissingDataInfo> v)
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700306{
Zhenkai Zhud1c5a972012-06-05 14:07:41 -0700307#ifdef __DEBUG
308 std::cout << "<<< processing Tree Update" << std::endl;
309#endif
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700310 if (v.empty())
311 {
312 return;
313 }
314
315 // reflect the changes on digest tree
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700316 {
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700317 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700318 m_scene->processUpdate(v, m_sock->getRootDigest().c_str());
319 }
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700320
321 int n = v.size();
322 int totalMissingPackets = 0;
323 for (int i = 0; i < n; i++)
324 {
325 totalMissingPackets += v[i].high.getSeq() - v[i].low.getSeq() + 1;
326 }
327
Zhenkai Zhubb198112012-09-27 11:31:42 -0700328 for (int i = 0; i < n; i++)
329 {
330 if (totalMissingPackets < 4)
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700331 {
332 for (Sync::SeqNo seq = v[i].low; seq <= v[i].high; ++seq)
333 {
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700334 m_sock->fetchRaw(v[i].prefix, seq, bind(&ChatDialog::processDataWrapper, this, _1, _2, _3), 2);
Zhenkai Zhud1c5a972012-06-05 14:07:41 -0700335#ifdef __DEBUG
336 std::cout << "<<< Fetching " << v[i].prefix << "/" <<seq.getSession() <<"/" << seq.getSeq() << std::endl;
337#endif
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700338 }
339 }
Zhenkai Zhubb198112012-09-27 11:31:42 -0700340 else
341 {
342 m_sock->fetchRaw(v[i].prefix, v[i].high, bind(&ChatDialog::processDataNoShowWrapper, this, _1, _2, _3), 2);
343 }
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700344 }
345
346 // adjust the view
347 fitView();
348
349}
350
351void
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700352ChatDialog::processDataWrapper(std::string name, const char *buf, size_t len)
353{
Zhenkai Zhubb198112012-09-27 11:31:42 -0700354 emit dataReceived(name.c_str(), buf, len, true);
Zhenkai Zhud1c5a972012-06-05 14:07:41 -0700355#ifdef __DEBUG
356 std::cout <<"<<< " << name << " fetched" << std::endl;
357#endif
Zhenkai Zhu64f9ede2012-06-05 11:32:00 -0700358}
359
360void
Zhenkai Zhubb198112012-09-27 11:31:42 -0700361ChatDialog::processDataNoShowWrapper(std::string name, const char *buf, size_t len)
362{
363 emit dataReceived(name.c_str(), buf, len, false);
364}
365
366void
367ChatDialog::processData(QString name, const char *buf, size_t len, bool show)
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700368{
369 SyncDemo::ChatMessage msg;
370 if (!msg.ParseFromArray(buf, len))
371 {
Zhenkai Zhubb198112012-09-27 11:31:42 -0700372 std::cerr << "Errrrr.. Can not parse msg with name: " << name.toStdString() << ". what is happening?" << std::endl;
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700373 }
374
Zhenkai Zhu10ccb5a2012-06-04 21:55:14 -0700375 // display msg received from network
376 // we have to do so; this function is called by ccnd thread
377 // so if we call appendMsg directly
378 // Qt crash as "QObject: Cannot create children for a parent that is in a different thread"
379 // the "cannonical" way to is use signal-slot
Zhenkai Zhubb198112012-09-27 11:31:42 -0700380 if (show)
381 {
382 appendMessage(msg);
383 }
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700384
385 // update the tree view
Zhenkai Zhu097bfe72012-06-05 14:30:17 -0700386 std::string stdStrName = name.toStdString();
387 std::string stdStrNameWithoutSeq = stdStrName.substr(0, stdStrName.find_last_of('/'));
388 std::string prefix = stdStrNameWithoutSeq.substr(0, stdStrNameWithoutSeq.find_last_of('/'));
389#ifdef __DEBUG
390 std::cout <<"<<< updating scene for" << prefix << ": " << msg.from() << std::endl;
391#endif
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700392 if (msg.type() == SyncDemo::ChatMessage::LEAVE)
393 {
394 processRemove(prefix.c_str());
395 }
396 else
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700397 {
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700398 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700399 m_scene->msgReceived(prefix.c_str(), msg.from().c_str());
400 }
Zhenkai Zhuc5470612012-06-05 12:28:59 -0700401 fitView();
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700402}
403
404void
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700405ChatDialog::processRemoveWrapper(std::string prefix)
406{
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700407#ifdef __DEBUG
408 std::cout << "Sync REMOVE signal received for prefix: " << prefix << std::endl;
409#endif
410 //emit removeReceived(prefix.c_str());
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700411}
412
413void
414ChatDialog::processRemove(QString prefix)
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700415{
Zhenkai Zhu591e8c32012-09-26 11:57:50 -0700416#ifdef __DEBUG
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700417 std::cout << "<<< remove node for prefix" << prefix.toStdString() << std::endl;
Zhenkai Zhu591e8c32012-09-26 11:57:50 -0700418#endif
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700419 bool removed = m_scene->removeNode(prefix);
Zhenkai Zhu591e8c32012-09-26 11:57:50 -0700420 if (removed)
421 {
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700422 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu591e8c32012-09-26 11:57:50 -0700423 m_scene->plot(m_sock->getRootDigest().c_str());
424 }
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700425}
426
427void
Zhenkai Zhu85845d22012-06-01 23:10:43 -0700428ChatDialog::formChatMessage(const QString &text, SyncDemo::ChatMessage &msg) {
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700429 msg.set_from(m_user.getNick().toStdString());
430 msg.set_to(m_user.getChatroom().toStdString());
Zhenkai Zhu85845d22012-06-01 23:10:43 -0700431 msg.set_data(text.toStdString());
432 time_t seconds = time(NULL);
433 msg.set_timestamp(seconds);
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700434 msg.set_type(SyncDemo::ChatMessage::CHAT);
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700435}
436
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700437void
Zhenkai Zhub5b78462012-09-28 14:10:37 -0700438ChatDialog::formControlMessage(SyncDemo::ChatMessage &msg, SyncDemo::ChatMessage::ChatMessageType type)
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700439{
440 msg.set_from(m_user.getNick().toStdString());
441 msg.set_to(m_user.getChatroom().toStdString());
442 time_t seconds = time(NULL);
443 msg.set_timestamp(seconds);
Zhenkai Zhub5b78462012-09-28 14:10:37 -0700444 msg.set_type(type);
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700445}
446
Zhenkai Zhu59245aa2012-09-26 16:07:04 -0700447static std::string chars("qwertyuiopasdfghjklzxcvbnmQWERTYUIOPASDFGHJKLZXCVBNM0123456789");
448
449QString
450ChatDialog::getRandomString()
451{
452 std::string randStr;
453 boost::random::random_device rng;
454 boost::random::uniform_int_distribution<> index_dist(0, chars.size() - 1);
455 for (int i = 0; i < 10; i ++)
456 {
457 randStr += chars[index_dist(rng)];
458 }
459 return randStr.c_str();
460}
461
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700462bool
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700463ChatDialog::readSettings()
464{
465 QSettings s(ORGANIZATION, APPLICATION);
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700466 QString nick = s.value("nick", "").toString();
467 QString chatroom = s.value("chatroom", "").toString();
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700468 QString originPrefix = s.value("originPrefix", "").toString();
Zhenkai Zhu3974a492012-09-28 14:39:45 -0700469 m_minimaniho = s.value("minimaniho", false).toBool();
470 if (nick == "" || chatroom == "" || originPrefix == "") {
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700471 QTimer::singleShot(500, this, SLOT(buttonPressed()));
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700472 return false;
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700473 }
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700474 else {
475 m_user.setNick(nick);
476 m_user.setChatroom(chatroom);
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700477 m_user.setOriginPrefix(originPrefix);
Zhenkai Zhu3974a492012-09-28 14:39:45 -0700478 m_user.setPrefix(originPrefix + "/" + chatroom + "/" + getRandomString());
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700479 return true;
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700480 }
Zhenkai Zhu3974a492012-09-28 14:39:45 -0700481
482// QTimer::singleShot(500, this, SLOT(buttonPressed()));
483 // return false;
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700484}
485
486void
487ChatDialog::writeSettings()
488{
489 QSettings s(ORGANIZATION, APPLICATION);
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700490 s.setValue("nick", m_user.getNick());
491 s.setValue("chatroom", m_user.getChatroom());
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700492 s.setValue("originPrefix", m_user.getOriginPrefix());
Zhenkai Zhu3974a492012-09-28 14:39:45 -0700493 s.setValue("minimaniho", m_minimaniho);
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700494}
495
496void
497ChatDialog::updateLabels()
498{
Zhenkai Zhu76ff02b2012-09-27 21:11:03 -0700499 QString settingDisp = QString("Chatroom: %1").arg(m_user.getChatroom());
500 infoLabel->setStyleSheet("QLabel {color: #630; font-size: 16px; font: bold \"Verdana\";}");
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700501 infoLabel->setText(settingDisp);
Zhenkai Zhub60b7e12012-09-28 11:34:21 -0700502 //QString prefixDisp = QString("<Prefix: %1>").arg(m_user.getPrefix());
503 //prefixLabel->setText(prefixDisp);
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700504}
505
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700506void
507ChatDialog::returnPressed()
508{
509 QString text = lineEdit->text();
510 if (text.isEmpty())
511 return;
512
Zhenkai Zhub6338822012-05-31 13:27:24 -0700513 lineEdit->clear();
514
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700515 SyncDemo::ChatMessage msg;
516 formChatMessage(text, msg);
517
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700518 appendMessage(msg);
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700519
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700520 sendMsg(msg);
521
522 fitView();
523}
524
525void
526ChatDialog::sendMsg(SyncDemo::ChatMessage &msg)
527{
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700528 // send msg
529 size_t size = msg.ByteSize();
530 char *buf = new char[size];
531 msg.SerializeToArray(buf, size);
532 if (!msg.IsInitialized())
533 {
534 std::cerr << "Errrrr.. msg was not probally initialized "<<__FILE__ <<":"<<__LINE__<<". what is happening?" << std::endl;
535 abort();
536 }
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700537 m_sock->publishRaw(m_user.getPrefix().toStdString(), m_session, buf, size, FRESHNESS);
Zhenkai Zhuc5470612012-06-05 12:28:59 -0700538
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700539 delete buf;
540
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700541 m_lastMsgTime = time(NULL);
542
Zhenkai Zhuc5470612012-06-05 12:28:59 -0700543 int nextSequence = m_sock->getNextSeq(m_user.getPrefix().toStdString(), m_session);
Zhenkai Zhud1c5a972012-06-05 14:07:41 -0700544 Sync::MissingDataInfo mdi = {m_user.getPrefix().toStdString(), Sync::SeqNo(0), Sync::SeqNo(nextSequence - 1)};
Zhenkai Zhuc5470612012-06-05 12:28:59 -0700545 std::vector<Sync::MissingDataInfo> v;
546 v.push_back(mdi);
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700547 {
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700548 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700549 m_scene->processUpdate(v, m_sock->getRootDigest().c_str());
550 m_scene->msgReceived(m_user.getPrefix(), m_user.getNick());
551 }
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700552}
553
554void
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700555ChatDialog::sendJoin()
556{
557 SyncDemo::ChatMessage msg;
Zhenkai Zhub5b78462012-09-28 14:10:37 -0700558 formControlMessage(msg, SyncDemo::ChatMessage::JOIN);
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700559 sendMsg(msg);
560 QTimer::singleShot(m_randomizedInterval, this, SLOT(sendHello()));
561}
562
563void
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700564ChatDialog::sendHello()
565{
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700566 time_t now = time(NULL);
567 int elapsed = now - m_lastMsgTime;
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700568 if (elapsed >= m_randomizedInterval / 1000)
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700569 {
570 SyncDemo::ChatMessage msg;
Zhenkai Zhub5b78462012-09-28 14:10:37 -0700571 formControlMessage(msg, SyncDemo::ChatMessage::HELLO);
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700572 sendMsg(msg);
Zhenkai Zhuee5c90f2012-09-27 14:05:41 -0700573 QTimer::singleShot(m_randomizedInterval, this, SLOT(sendHello()));
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700574 }
575 else
576 {
Zhenkai Zhuee5c90f2012-09-27 14:05:41 -0700577 QTimer::singleShot((m_randomizedInterval - elapsed * 1000), this, SLOT(sendHello()));
Zhenkai Zhu3e26bb42012-09-27 11:04:09 -0700578 }
Zhenkai Zhu5a8d5aa2012-05-30 21:25:23 -0700579}
580
581void
Zhenkai Zhu85845d22012-06-01 23:10:43 -0700582ChatDialog::buttonPressed()
583{
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700584 SettingDialog dialog(this, m_user.getNick(), m_user.getChatroom(), m_user.getOriginPrefix());
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700585 connect(&dialog, SIGNAL(updated(QString, QString, QString)), this, SLOT(settingUpdated(QString, QString, QString)));
Zhenkai Zhu85845d22012-06-01 23:10:43 -0700586 dialog.exec();
Zhenkai Zhue837f792012-06-05 20:47:54 -0700587 QTimer::singleShot(100, this, SLOT(checkSetting()));
588}
589
Zhenkai Zhub60b7e12012-09-28 11:34:21 -0700590void ChatDialog::treeButtonPressed()
591{
592 if (treeViewer->isVisible())
593 {
594 treeViewer->hide();
595 treeButton->setText("Show Sync Tree");
596 }
597 else
598 {
599 treeViewer->show();
600 treeButton->setText("Hide Sync Tree");
601 }
Zhenkai Zhu25e33e52012-09-28 13:00:07 -0700602
603 fitView();
Zhenkai Zhub60b7e12012-09-28 11:34:21 -0700604}
605
Zhenkai Zhue837f792012-06-05 20:47:54 -0700606void
607ChatDialog::checkSetting()
608{
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700609 if (m_user.getOriginPrefix().isEmpty() || m_user.getNick().isEmpty() || m_user.getChatroom().isEmpty())
Zhenkai Zhue837f792012-06-05 20:47:54 -0700610 {
611 buttonPressed();
612 }
Zhenkai Zhu6d589aa2012-05-29 17:34:35 -0700613}
Zhenkai Zhue08afe02012-05-31 15:49:07 -0700614
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700615void
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700616ChatDialog::settingUpdated(QString nick, QString chatroom, QString originPrefix)
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700617{
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700618 QString randString = getRandomString();
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700619 bool needWrite = false;
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700620 bool needFresh = false;
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700621 if (!nick.isEmpty() && nick != m_user.getNick()) {
622 m_user.setNick(nick);
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700623 needWrite = true;
624 }
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700625 if (!originPrefix.isEmpty() && originPrefix != m_user.getOriginPrefix()) {
626 m_user.setOriginPrefix(originPrefix);
627 m_user.setPrefix(originPrefix + "/" + m_user.getChatroom() + "/" + randString);
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700628 needWrite = true;
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700629 needFresh = true;
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700630 }
Zhenkai Zhu71b42cb2012-06-04 09:42:53 -0700631 if (!chatroom.isEmpty() && chatroom != m_user.getChatroom()) {
632 m_user.setChatroom(chatroom);
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700633 m_user.setPrefix(m_user.getOriginPrefix() + "/" + chatroom + "/" + randString);
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700634 needWrite = true;
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700635 needFresh = true;
636 }
637
638 if (needFresh)
639 {
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700640
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700641 {
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700642 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu44b43752012-06-05 21:18:37 -0700643 m_scene->clearAll();
644 m_scene->plot("Empty");
645 }
Zhenkai Zhucc4c2c02012-09-27 21:24:37 -0700646
647 textEdit->clear();
648
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700649 // TODO: perhaps need to do a lot. e.g. use a new SyncAppSokcet
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700650 if (m_sock != NULL)
651 {
652 delete m_sock;
653 m_sock = NULL;
654 }
655 std::string syncPrefix = BROADCAST_PREFIX_FOR_SYNC_DEMO;
656 syncPrefix += "/";
657 syncPrefix += m_user.getChatroom().toStdString();
Zhenkai Zhua4fb1242012-06-05 20:26:05 -0700658 try
659 {
Zhenkai Zhu2425b6a2012-09-26 17:11:44 -0700660 m_sock = new Sync::SyncAppSocket(syncPrefix, bind(&ChatDialog::processTreeUpdateWrapper, this, _1, _2), bind(&ChatDialog::processRemoveWrapper, this, _1));
Alexander Afanasyev288edd82012-10-04 17:07:56 -0700661 QTimer::singleShot(1000, this, SLOT(sendJoin()));
Zhenkai Zhu86df7412012-09-27 16:30:20 -0700662 m_timer->start(FRESHNESS * 2000);
Zhenkai Zhua4fb1242012-06-05 20:26:05 -0700663 }
664 catch (Sync::CcnxOperationException ex)
665 {
Zhenkai Zhu0b3fa332012-09-27 21:58:43 -0700666 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 -0700667 std::exit(1);
668 }
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700669
670 fitView();
671
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700672 }
Zhenkai Zhue95c64a2012-09-27 21:46:44 -0700673
Zhenkai Zhu7e9b06d2012-06-02 00:44:42 -0700674 if (needWrite) {
675 writeSettings();
676 updateLabels();
677 }
678}
Zhenkai Zhud13acd02012-06-04 15:25:20 -0700679
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -0700680void
681ChatDialog::iconActivated(QSystemTrayIcon::ActivationReason reason)
682{
683 switch (reason)
684 {
685 case QSystemTrayIcon::Trigger:
686 case QSystemTrayIcon::DoubleClick:
687 break;
688 case QSystemTrayIcon::MiddleClick:
689 // showMessage();
690 break;
691 default:;
692 }
693}
694
695void
696ChatDialog::showMessage(QString from, QString data)
697{
698 // std::cout <<"Showing Message: " << from.toStdString() << ": " << data.toStdString() << std::endl;
699 if (!isActiveWindow())
700 {
701 trayIcon->showMessage(QString("Chatroom %1 has a new message").arg(m_user.getChatroom()), QString("<%1>: %2").arg(from).arg(data), QSystemTrayIcon::Information, 20000);
702 trayIcon->setIcon(QIcon(":/images/note.png"));
703 }
704}
705
706void
707ChatDialog::messageClicked()
708{
709 this->showMaximized();
710}
711
712void
713ChatDialog::createActions()
714{
715 minimizeAction = new QAction(tr("Mi&nimize"), this);
716 connect(minimizeAction, SIGNAL(triggered()), this, SLOT(hide()));
717
718 maximizeAction = new QAction(tr("Ma&ximize"), this);
719 connect(maximizeAction, SIGNAL(triggered()), this, SLOT(showMaximized()));
720
721 restoreAction = new QAction(tr("&Restore"), this);
722 connect(restoreAction, SIGNAL(triggered()), this, SLOT(showNormal()));
723
724 quitAction = new QAction(tr("Quit"), this);
725 connect(quitAction, SIGNAL(triggered()), qApp, SLOT(quit()));
726}
727
728void
729ChatDialog::createTrayIcon()
730{
731 trayIconMenu = new QMenu(this);
732 trayIconMenu->addAction(minimizeAction);
733 trayIconMenu->addAction(maximizeAction);
734 trayIconMenu->addAction(restoreAction);
735 trayIconMenu->addSeparator();
736 trayIconMenu->addAction(quitAction);
737
738 trayIcon = new QSystemTrayIcon(this);
739 trayIcon->setContextMenu(trayIconMenu);
740
741 QIcon icon(":/images/icon_small.png");
742 trayIcon->setIcon(icon);
743 setWindowIcon(icon);
Zhenkai Zhu0b3fa332012-09-27 21:58:43 -0700744 trayIcon->setToolTip("Chronos System Tray Icon");
Zhenkai Zhu3a008fc2012-06-08 17:36:39 -0700745 trayIcon->setVisible(true);
746}
747
Zhenkai Zhud13acd02012-06-04 15:25:20 -0700748void
749ChatDialog::resizeEvent(QResizeEvent *e)
750{
751 fitView();
752}
753
754void
755ChatDialog::showEvent(QShowEvent *e)
756{
757 fitView();
758}
759
760void
761ChatDialog::fitView()
762{
Zhenkai Zhu9036e032012-09-27 20:59:33 -0700763 boost::recursive_mutex::scoped_lock lock(m_sceneMutex);
Zhenkai Zhu21d75f92012-06-04 21:23:34 -0700764 QRectF rect = m_scene->itemsBoundingRect();
765 m_scene->setSceneRect(rect);
Zhenkai Zhu82a62752012-06-04 17:11:04 -0700766 treeViewer->fitInView(m_scene->itemsBoundingRect(), Qt::KeepAspectRatio);
Zhenkai Zhud13acd02012-06-04 15:25:20 -0700767}