blob: ea55a78c39e84c296569755985c0ee1d323d9d94 (
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
|
/*
* Copyright (c) 2010 Kevin Smith
* Licensed under the GNU General Public License v3.
* See Documentation/Licenses/GPLv3.txt for more information.
*/
#pragma once
#include <vector>
#include <set>
#include <Swiften/Base/API.h>
#include <boost/shared_ptr.hpp>
#include <Swiften/Base/boost_bsignals.h>
#include <string>
#include <Swiften/JID/JID.h>
#include <Swiften/Elements/DiscoInfo.h>
#include <Swiften/Elements/DiscoItems.h>
#include <Swiften/Elements/ErrorPayload.h>
#include <Swiften/Disco/GetDiscoInfoRequest.h>
#include <Swiften/Disco/GetDiscoItemsRequest.h>
namespace Swift {
class IQRouter;
/**
* Recursively walk service discovery trees to find all services offered.
* This stops on any disco item that's not reporting itself as a server.
*/
class SWIFTEN_API DiscoServiceWalker {
public:
DiscoServiceWalker(const JID& service, IQRouter* iqRouter, size_t maxSteps = 200);
/**
* Start the walk.
*
* Call this exactly once.
*/
void beginWalk();
/**
* End the walk.
*/
void endWalk();
bool isActive() const {
return active_;
}
/** Emitted for each service found. */
boost::signal<void(const JID&, boost::shared_ptr<DiscoInfo>)> onServiceFound;
/** Emitted when walking is aborted. */
boost::signal<void()> onWalkAborted;
/** Emitted when walking is complete.*/
boost::signal<void()> onWalkComplete;
private:
void walkNode(const JID& jid);
void markNodeCompleted(const JID& jid);
void handleDiscoInfoResponse(boost::shared_ptr<DiscoInfo> info, ErrorPayload::ref error, GetDiscoInfoRequest::ref request);
void handleDiscoItemsResponse(boost::shared_ptr<DiscoItems> items, ErrorPayload::ref error, GetDiscoItemsRequest::ref request);
void handleDiscoError(const JID& jid, ErrorPayload::ref error);
private:
JID service_;
IQRouter* iqRouter_;
size_t maxSteps_;
bool active_;
std::set<JID> servicesBeingSearched_;
std::set<JID> searchedServices_;
std::set<GetDiscoInfoRequest::ref> pendingDiscoInfoRequests_;
std::set<GetDiscoItemsRequest::ref> pendingDiscoItemsRequests_;
};
}
|
Swift