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.muc()) {
111 packet.setTo(mam.getWith());
112 } else if (mam.getWith()!=null) {
113 data.put("with", mam.getWith().toString());
114 }
115 data.put("start",getTimestamp(mam.getStart()));
116 data.put("end",getTimestamp(mam.getEnd()));
117 query.addChild(data);
118 if (mam.getPagingOrder() == MessageArchiveService.PagingOrder.REVERSE) {
119 query.addChild("set", "http://jabber.org/protocol/rsm").addChild("before").setContent(mam.getReference());
120 } else if (mam.getReference() != null) {
121 query.addChild("set", "http://jabber.org/protocol/rsm").addChild("after").setContent(mam.getReference());
122 }
123 return packet;
124 }
125 public IqPacket generateGetBlockList() {
126 final IqPacket iq = new IqPacket(IqPacket.TYPE.GET);
127 iq.addChild("blocklist", Xmlns.BLOCKING);
128
129 return iq;
130 }
131
132 public IqPacket generateSetBlockRequest(final Jid jid) {
133 final IqPacket iq = new IqPacket(IqPacket.TYPE.SET);
134 final Element block = iq.addChild("block", Xmlns.BLOCKING);
135 block.addChild("item").setAttribute("jid", jid.toBareJid().toString());
136 return iq;
137 }
138
139 public IqPacket generateSetUnblockRequest(final Jid jid) {
140 final IqPacket iq = new IqPacket(IqPacket.TYPE.SET);
141 final Element block = iq.addChild("unblock", Xmlns.BLOCKING);
142 block.addChild("item").setAttribute("jid", jid.toBareJid().toString());
143 return iq;
144 }
145
146 public IqPacket generateSetPassword(final Account account, final String newPassword) {
147 final IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
148 packet.setTo(account.getServer());
149 final Element query = packet.addChild("query", Xmlns.REGISTER);
150 final Jid jid = account.getJid();
151 query.addChild("username").setContent(jid.getLocalpart());
152 query.addChild("password").setContent(newPassword);
153 return packet;
154 }
155
156 public IqPacket changeAffiliation(Conversation conference, Jid jid, String affiliation) {
157 List<Jid> jids = new ArrayList<>();
158 jids.add(jid);
159 return changeAffiliation(conference,jids,affiliation);
160 }
161
162 public IqPacket changeAffiliation(Conversation conference, List<Jid> jids, String affiliation) {
163 IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
164 packet.setTo(conference.getJid().toBareJid());
165 packet.setFrom(conference.getAccount().getJid());
166 Element query = packet.query("http://jabber.org/protocol/muc#admin");
167 for(Jid jid : jids) {
168 Element item = query.addChild("item");
169 item.setAttribute("jid", jid.toString());
170 item.setAttribute("affiliation", affiliation);
171 }
172 return packet;
173 }
174
175 public IqPacket changeRole(Conversation conference, String nick, String role) {
176 IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
177 packet.setTo(conference.getJid().toBareJid());
178 packet.setFrom(conference.getAccount().getJid());
179 Element item = packet.query("http://jabber.org/protocol/muc#admin").addChild("item");
180 item.setAttribute("nick", nick);
181 item.setAttribute("role", role);
182 return packet;
183 }
184}