summaryrefslogtreecommitdiffstats
blob: ab2ab09703052df8c3948c58ae7dc0cf28e8cf56 (plain)
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();
}

}