summaryrefslogtreecommitdiffstats
blob: d3587e9770dca451f72d0dc65ab2a0ff2f527eee (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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/*
 * Copyright (c) 2010 Remko Tronçon
 * Licensed under the GNU General Public License v3.
 * See Documentation/Licenses/GPLv3.txt for more information.
 */

//...
#include <iostream>
#include <boost/bind.hpp>
#include <boost/smart_ptr/make_shared.hpp>

#include <Swiften/Swiften.h>

using namespace Swift;
using namespace boost;
//...
#include "EchoPayload.h"
#include "EchoPayloadParserFactory.h"
#include "EchoPayloadSerializer.h"

class EchoBot {
	public:
		EchoBot(NetworkFactories* networkFactories) {
			//...
			client = new Client("echobot@wonderland.lit", "mypass", networkFactories);
			client->onConnected.connect(bind(&EchoBot::handleConnected, this));
			client->onMessageReceived.connect(
					bind(&EchoBot::handleMessageReceived, this, _1));
			client->onPresenceReceived.connect(
					bind(&EchoBot::handlePresenceReceived, this, _1));
			tracer = new ClientXMLTracer(client);

			softwareVersionResponder = new SoftwareVersionResponder(client->getIQRouter());
			softwareVersionResponder->setVersion("EchoBot", "1.0");
			softwareVersionResponder->start();
			//...
			client->addPayloadParserFactory(&echoPayloadParserFactory);
			client->addPayloadSerializer(&echoPayloadSerializer);
			//...
			client->connect();
			//...
		}

		~EchoBot() {
			client->removePayloadSerializer(&echoPayloadSerializer);
			client->removePayloadParserFactory(&echoPayloadParserFactory);
			//...
			softwareVersionResponder->stop();
			delete softwareVersionResponder;
			delete tracer;
			delete client;
			//...
		}
		//...
	
	private:
		void handlePresenceReceived(Presence::ref presence) {
			// Automatically approve subscription requests
			if (presence->getType() == Presence::Subscribe) {
				Presence::ref response = Presence::create();
				response->setTo(presence->getFrom());
				response->setType(Presence::Subscribed);
				client->sendPresence(response);
			}
		}

		void handleConnected() {
			// Request the roster
			GetRosterRequest::ref rosterRequest = 
					GetRosterRequest::create(client->getIQRouter());
			rosterRequest->onResponse.connect(
					bind(&EchoBot::handleRosterReceived, this, _2));
			rosterRequest->send();
		}

		void handleRosterReceived(ErrorPayload::ref error) {
			if (error) {
				std::cerr << "Error receiving roster. Continuing anyway.";
			}
			// Send initial available presence
			client->sendPresence(Presence::create("Send me a message"));
		}

		//...
		void handleMessageReceived(Message::ref message) {
			//...
			// Echo back the incoming message
			message->setTo(message->getFrom());
			message->setFrom(JID());
			//...
			if (!message->getPayload<EchoPayload>()) {
				boost::shared_ptr<EchoPayload> echoPayload = boost::make_shared<EchoPayload>();
				echoPayload->setMessage("This is an echoed message");
				message->addPayload(echoPayload);
				client->sendMessage(message);
			}
		}
		//...

		//...
	private:
		//...
		Client* client;
		ClientXMLTracer* tracer;
		SoftwareVersionResponder* softwareVersionResponder;
		//...
		EchoPayloadParserFactory echoPayloadParserFactory;
		EchoPayloadSerializer echoPayloadSerializer;
};
//...

int main(int, char**) {
	SimpleEventLoop eventLoop;
	BoostNetworkFactories networkFactories(&eventLoop);

	EchoBot bot(&networkFactories);

	eventLoop.run();
	return 0;
}
//...