1package eu.siacs.conversations.generator;
2
3import java.util.ArrayList;
4import java.util.Arrays;
5import java.util.Collections;
6import java.util.List;
7
8import eu.siacs.conversations.entities.Account;
9import eu.siacs.conversations.entities.Conversation;
10import eu.siacs.conversations.services.MessageArchiveService;
11import eu.siacs.conversations.services.XmppConnectionService;
12import eu.siacs.conversations.utils.Xmlns;
13import eu.siacs.conversations.xml.Element;
14import eu.siacs.conversations.xmpp.forms.Data;
15import eu.siacs.conversations.xmpp.jid.Jid;
16import eu.siacs.conversations.xmpp.pep.Avatar;
17import eu.siacs.conversations.xmpp.stanzas.IqPacket;
18
19public class IqGenerator extends AbstractGenerator {
20
21 public IqGenerator(final XmppConnectionService service) {
22 super(service);
23 }
24
25 public IqPacket discoResponse(final IqPacket request) {
26 final IqPacket packet = new IqPacket(IqPacket.TYPE.RESULT);
27 packet.setId(request.getId());
28 packet.setTo(request.getFrom());
29 final Element query = packet.addChild("query",
30 "http://jabber.org/protocol/disco#info");
31 query.setAttribute("node", request.query().getAttribute("node"));
32 final Element identity = query.addChild("identity");
33 identity.setAttribute("category", "client");
34 identity.setAttribute("type", this.IDENTITY_TYPE);
35 identity.setAttribute("name", IDENTITY_NAME);
36 final List<String> features = Arrays.asList(FEATURES);
37 Collections.sort(features);
38 for (final String feature : features) {
39 query.addChild("feature").setAttribute("var", feature);
40 }
41 return packet;
42 }
43
44 protected IqPacket publish(final String node, final Element item) {
45 final IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
46 final Element pubsub = packet.addChild("pubsub",
47 "http://jabber.org/protocol/pubsub");
48 final Element publish = pubsub.addChild("publish");
49 publish.setAttribute("node", node);
50 publish.addChild(item);
51 return packet;
52 }
53
54 protected IqPacket retrieve(String node, Element item) {
55 final IqPacket packet = new IqPacket(IqPacket.TYPE.GET);
56 final Element pubsub = packet.addChild("pubsub",
57 "http://jabber.org/protocol/pubsub");
58 final Element items = pubsub.addChild("items");
59 items.setAttribute("node", node);
60 if (item != null) {
61 items.addChild(item);
62 }
63 return packet;
64 }
65
66 public IqPacket publishAvatar(Avatar avatar) {
67 final Element item = new Element("item");
68 item.setAttribute("id", avatar.sha1sum);
69 final Element data = item.addChild("data", "urn:xmpp:avatar:data");
70 data.setContent(avatar.image);
71 return publish("urn:xmpp:avatar:data", item);
72 }
73
74 public IqPacket publishAvatarMetadata(final Avatar avatar) {
75 final Element item = new Element("item");
76 item.setAttribute("id", avatar.sha1sum);
77 final Element metadata = item
78 .addChild("metadata", "urn:xmpp:avatar:metadata");
79 final Element info = metadata.addChild("info");
80 info.setAttribute("bytes", avatar.size);
81 info.setAttribute("id", avatar.sha1sum);
82 info.setAttribute("height", avatar.height);
83 info.setAttribute("width", avatar.height);
84 info.setAttribute("type", avatar.type);
85 return publish("urn:xmpp:avatar:metadata", item);
86 }
87
88 public IqPacket retrieveAvatar(final Avatar avatar) {
89 final Element item = new Element("item");
90 item.setAttribute("id", avatar.sha1sum);
91 final IqPacket packet = retrieve("urn:xmpp:avatar:data", item);
92 packet.setTo(avatar.owner);
93 return packet;
94 }
95
96 public IqPacket retrieveAvatarMetaData(final Jid to) {
97 final IqPacket packet = retrieve("urn:xmpp:avatar:metadata", null);
98 if (to != null) {
99 packet.setTo(to);
100 }
101 return packet;
102 }
103
104 public IqPacket queryMessageArchiveManagement(final MessageArchiveService.Query mam) {
105 final IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
106 final Element query = packet.query("urn:xmpp:mam:0");
107 query.setAttribute("queryid",mam.getQueryId());
108 final Data data = new Data();
109 data.setFormType("urn:xmpp:mam:0");
110 if (mam.getWith()!=null) {
111 data.put("with", mam.getWith().toString());
112 }
113 data.put("start",getTimestamp(mam.getStart()));
114 data.put("end",getTimestamp(mam.getEnd()));
115 query.addChild(data);
116 if (mam.getPagingOrder() == MessageArchiveService.PagingOrder.REVERSE) {
117 query.addChild("set", "http://jabber.org/protocol/rsm").addChild("before").setContent(mam.getReference());
118 } else if (mam.getReference() != null) {
119 query.addChild("set", "http://jabber.org/protocol/rsm").addChild("after").setContent(mam.getReference());
120 }
121 return packet;
122 }
123 public IqPacket generateGetBlockList() {
124 final IqPacket iq = new IqPacket(IqPacket.TYPE.GET);
125 iq.addChild("blocklist", Xmlns.BLOCKING);
126
127 return iq;
128 }
129
130 public IqPacket generateSetBlockRequest(final Jid jid) {
131 final IqPacket iq = new IqPacket(IqPacket.TYPE.SET);
132 final Element block = iq.addChild("block", Xmlns.BLOCKING);
133 block.addChild("item").setAttribute("jid", jid.toBareJid().toString());
134 return iq;
135 }
136
137 public IqPacket generateSetUnblockRequest(final Jid jid) {
138 final IqPacket iq = new IqPacket(IqPacket.TYPE.SET);
139 final Element block = iq.addChild("unblock", Xmlns.BLOCKING);
140 block.addChild("item").setAttribute("jid", jid.toBareJid().toString());
141 return iq;
142 }
143
144 public IqPacket generateSetPassword(final Account account, final String newPassword) {
145 final IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
146 packet.setTo(account.getServer());
147 final Element query = packet.addChild("query", Xmlns.REGISTER);
148 final Jid jid = account.getJid();
149 query.addChild("username").setContent(jid.getLocalpart());
150 query.addChild("password").setContent(newPassword);
151 return packet;
152 }
153
154 public IqPacket changeAffiliation(Conversation conference, Jid jid, String affiliation) {
155 List<Jid> jids = new ArrayList<>();
156 jids.add(jid);
157 return changeAffiliation(conference,jids,affiliation);
158 }
159
160 public IqPacket changeAffiliation(Conversation conference, List<Jid> jids, String affiliation) {
161 IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
162 packet.setTo(conference.getJid().toBareJid());
163 packet.setFrom(conference.getAccount().getJid());
164 Element query = packet.query("http://jabber.org/protocol/muc#admin");
165 for(Jid jid : jids) {
166 Element item = query.addChild("item");
167 item.setAttribute("jid", jid.toString());
168 item.setAttribute("affiliation", affiliation);
169 }
170 return packet;
171 }
172
173 public IqPacket changeRole(Conversation conference, String nick, String role) {
174 IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
175 packet.setTo(conference.getJid().toBareJid());
176 packet.setFrom(conference.getAccount().getJid());
177 Element item = packet.query("http://jabber.org/protocol/muc#admin").addChild("item");
178 item.setAttribute("nick", nick);
179 item.setAttribute("role", role);
180 return packet;
181 }
182}