1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
/*
* Copyright (c) 2010 Remko Tronçon
* Licensed under the GNU General Public License v3.
* See Documentation/Licenses/GPLv3.txt for more information.
*/
#include "SwifTools/Notifier/GNTPNotifier.h"
#include <cassert>
#include <iostream>
#include <boost/bind.hpp>
#include <sstream>
#include "Swiften/Base/foreach.h"
#include "Swiften/Network/ConnectionFactory.h"
namespace Swift {
GNTPNotifier::GNTPNotifier(const String& name, const boost::filesystem::path& icon, ConnectionFactory* connectionFactory) : name(name), icon(icon), connectionFactory(connectionFactory), initialized(false), registered(false) {
// Registration message
std::ostringstream message;
message << "GNTP/1.0 REGISTER NONE\r\n";
message << "Application-Name: " << name << "\r\n";
message << "Application-Icon: file://" << icon.string() << "\r\n";
message << "Notifications-Count: " << getAllTypes().size() << "\r\n";
std::vector<Notifier::Type> defaultTypes = getDefaultTypes();
std::vector<Notifier::Type> allTypes = getAllTypes();
foreach(Notifier::Type type, allTypes) {
message << "\r\n";
message << "Notification-Name: " << typeToString(type) << "\r\n";
message << "Notification-Enabled: " << (std::find(defaultTypes.begin(), defaultTypes.end(), type) == defaultTypes.end() ? "false" : "true") << "\r\n";
}
message << "\r\n";
send(message.str());
}
GNTPNotifier::~GNTPNotifier() {
}
void GNTPNotifier::send(const std::string& message) {
if (currentConnection) {
return;
}
currentMessage = message;
currentConnection = connectionFactory->createConnection();
currentConnection->onConnectFinished.connect(boost::bind(&GNTPNotifier::handleConnectFinished, this, _1));
currentConnection->onDataRead.connect(boost::bind(&GNTPNotifier::handleDataRead, this, _1));
currentConnection->connect(HostAddressPort(HostAddress("127.0.0.1"), 23053));
}
void GNTPNotifier::showMessage(Type type, const String& subject, const String& description, const boost::filesystem::path& picture, boost::function<void()> callback) {
if (registered) {
std::ostringstream message;
message << "GNTP/1.0 NOTIFY NONE\r\n";
message << "Application-Name: " << name << "\r\n";
message << "Notification-Name: " << typeToString(type) << "\r\n";
message << "Notification-Title: " << subject << "\r\n";
message << "Notification-Text: " << description << "\r\n";
message << "Notification-Icon: " << picture.string() << "\r\n";
message << "\r\n";
send(message.str());
}
}
void GNTPNotifier::handleConnectFinished(bool error) {
if (!initialized) {
initialized = true;
registered = !error;
}
if (!error) {
currentConnection->write(currentMessage.c_str());
}
}
void GNTPNotifier::handleDataRead(const ByteArray& data) {
currentConnection->onDataRead.disconnect(boost::bind(&GNTPNotifier::handleDataRead, this, _1));
currentConnection->onConnectFinished.disconnect(boost::bind(&GNTPNotifier::handleConnectFinished, this, _1));
currentConnection.reset();
}
}
|
Swift