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
|
#include <boost/bind.hpp>
#include <iostream>
#include "Swiften/LinkLocal/LinkLocalRoster.h"
#include "Swiften/Network/HostAddress.h"
namespace Swift {
LinkLocalRoster::LinkLocalRoster(boost::shared_ptr<DNSSDService> service) : dnsSDService(service) {
dnsSDService->onStopped.connect(boost::bind(&LinkLocalRoster::handleStopped, this, _1));
dnsSDService->onServiceRegistered.connect(boost::bind(&LinkLocalRoster::handleServiceRegistered, this, _1));
dnsSDService->onServiceAdded.connect(boost::bind(&LinkLocalRoster::handleServiceAdded, this, _1));
dnsSDService->onServiceRemoved.connect(boost::bind(&LinkLocalRoster::handleServiceRemoved, this, _1));
dnsSDService->onServiceResolved.connect(boost::bind(&LinkLocalRoster::handleServiceResolved, this, _1, _2));
}
boost::shared_ptr<RosterPayload> LinkLocalRoster::getRoster() const {
boost::shared_ptr<RosterPayload> roster(new RosterPayload());
for(ServiceMap::const_iterator i = services.begin(); i != services.end(); ++i) {
roster->addItem(getRosterItem(i->first, i->second));
}
return roster;
}
std::vector<boost::shared_ptr<Presence> > LinkLocalRoster::getAllPresence() const {
std::vector<boost::shared_ptr<Presence> > result;
for(ServiceMap::const_iterator i = services.begin(); i != services.end(); ++i) {
result.push_back(getPresence(i->first, i->second));
}
return result;
}
RosterItemPayload LinkLocalRoster::getRosterItem(const DNSSDService::Service& service, const DNSSDService::ResolveResult& resolveResult) const {
return RosterItemPayload(getJIDForService(service), getRosterName(service, resolveResult), RosterItemPayload::Both);
}
String LinkLocalRoster::getRosterName(const DNSSDService::Service& service, const DNSSDService::ResolveResult& resolveResult) const {
if (!resolveResult.info.getNick().isEmpty()) {
return resolveResult.info.getNick();
}
else if (!resolveResult.info.getFirstName().isEmpty()) {
String result = resolveResult.info.getFirstName();
if (!resolveResult.info.getLastName().isEmpty()) {
result += " " + resolveResult.info.getLastName();
}
return result;
}
else if (!resolveResult.info.getLastName().isEmpty()) {
return resolveResult.info.getLastName();
}
return service.name;
}
JID LinkLocalRoster::getJIDForService(const DNSSDService::Service& service) const {
return JID(service.name);
}
boost::shared_ptr<Presence> LinkLocalRoster::getPresence(const DNSSDService::Service& service, const DNSSDService::ResolveResult& resolveResult) const {
boost::shared_ptr<Presence> presence(new Presence());
presence->setFrom(getJIDForService(service));
switch (resolveResult.info.getStatus()) {
case LinkLocalServiceInfo::Available:
presence->setShow(StatusShow::Online);
break;
case LinkLocalServiceInfo::Away:
presence->setShow(StatusShow::Away);
break;
case LinkLocalServiceInfo::DND:
presence->setShow(StatusShow::DND);
break;
}
presence->setStatus(resolveResult.info.getMessage());
return presence;
}
void LinkLocalRoster::handleServiceAdded(const DNSSDService::Service& service) {
if (selfService && *selfService == service) {
return;
}
dnsSDService->startResolvingService(service);
}
void LinkLocalRoster::handleServiceRemoved(const DNSSDService::Service& service) {
dnsSDService->stopResolvingService(service);
services.erase(service);
boost::shared_ptr<RosterPayload> roster(new RosterPayload());
roster->addItem(RosterItemPayload(getJIDForService(service), "", RosterItemPayload::Remove));
onRosterChanged(roster);
}
void LinkLocalRoster::handleServiceResolved(const DNSSDService::Service& service, const DNSSDService::ResolveResult& result) {
services.insert(std::make_pair(service, result));
dnsSDService->resolveHostname(result.host);
boost::shared_ptr<RosterPayload> roster(new RosterPayload());
roster->addItem(getRosterItem(service, result));
onRosterChanged(roster);
onPresenceChanged(getPresence(service, result));
}
void LinkLocalRoster::handleServiceRegistered(const DNSSDService::Service& service) {
selfService = service;
}
void LinkLocalRoster::handleStopped(bool error) {
std::cout << "DNSSDService stopped: " << error << std::endl;
}
}
|