1package eu.siacs.conversations.generator;
2
3import java.text.SimpleDateFormat;
4import java.util.Date;
5import java.util.Locale;
6import java.util.TimeZone;
7
8import net.java.otr4j.OtrException;
9import net.java.otr4j.session.Session;
10import eu.siacs.conversations.entities.Account;
11import eu.siacs.conversations.entities.Conversation;
12import eu.siacs.conversations.entities.Message;
13import eu.siacs.conversations.services.XmppConnectionService;
14import eu.siacs.conversations.xml.Element;
15import eu.siacs.conversations.xmpp.chatstate.ChatState;
16import eu.siacs.conversations.xmpp.jid.Jid;
17import eu.siacs.conversations.xmpp.stanzas.MessagePacket;
18
19public class MessageGenerator extends AbstractGenerator {
20 public MessageGenerator(XmppConnectionService service) {
21 super(service);
22 }
23
24 private MessagePacket preparePacket(Message message, boolean addDelay) {
25 Conversation conversation = message.getConversation();
26 Account account = conversation.getAccount();
27 MessagePacket packet = new MessagePacket();
28 if (conversation.getMode() == Conversation.MODE_SINGLE) {
29 packet.setTo(message.getCounterpart());
30 packet.setType(MessagePacket.TYPE_CHAT);
31 packet.addChild("markable", "urn:xmpp:chat-markers:0");
32 if (this.mXmppConnectionService.indicateReceived()) {
33 packet.addChild("request", "urn:xmpp:receipts");
34 }
35 } else if (message.getType() == Message.TYPE_PRIVATE) {
36 packet.setTo(message.getCounterpart());
37 packet.setType(MessagePacket.TYPE_CHAT);
38 if (this.mXmppConnectionService.indicateReceived()) {
39 packet.addChild("request", "urn:xmpp:receipts");
40 }
41 } else {
42 packet.setTo(message.getCounterpart().toBareJid());
43 packet.setType(MessagePacket.TYPE_GROUPCHAT);
44 }
45 packet.setFrom(account.getJid());
46 packet.setId(message.getUuid());
47 if (addDelay) {
48 addDelay(packet, message.getTimeSent());
49 }
50 return packet;
51 }
52
53 private void addDelay(MessagePacket packet, long timestamp) {
54 final SimpleDateFormat mDateFormat = new SimpleDateFormat(
55 "yyyy-MM-dd'T'HH:mm:ss.SSS'Z'", Locale.US);
56 mDateFormat.setTimeZone(TimeZone.getTimeZone("UTC"));
57 Element delay = packet.addChild("delay", "urn:xmpp:delay");
58 Date date = new Date(timestamp);
59 delay.setAttribute("stamp", mDateFormat.format(date));
60 }
61
62 public MessagePacket generateOtrChat(Message message) {
63 return generateOtrChat(message, false);
64 }
65
66 public MessagePacket generateOtrChat(Message message, boolean addDelay) {
67 Session otrSession = message.getConversation().getOtrSession();
68 if (otrSession == null) {
69 return null;
70 }
71 MessagePacket packet = preparePacket(message, addDelay);
72 packet.addChild("private", "urn:xmpp:carbons:2");
73 packet.addChild("no-copy", "urn:xmpp:hints");
74 packet.addChild("no-permanent-store", "urn:xmpp:hints");
75 try {
76 String content;
77 if (message.hasFileOnRemoteHost()) {
78 content = message.getFileParams().url.toString();
79 } else {
80 content = message.getBody();
81 }
82 packet.setBody(otrSession.transformSending(content)[0]);
83 return packet;
84 } catch (OtrException e) {
85 return null;
86 }
87 }
88
89 public MessagePacket generateChat(Message message) {
90 return generateChat(message, false);
91 }
92
93 public MessagePacket generateChat(Message message, boolean addDelay) {
94 MessagePacket packet = preparePacket(message, addDelay);
95 if (message.hasFileOnRemoteHost()) {
96 packet.setBody(message.getFileParams().url.toString());
97 } else {
98 packet.setBody(message.getBody());
99 }
100 return packet;
101 }
102
103 public MessagePacket generatePgpChat(Message message) {
104 return generatePgpChat(message, false);
105 }
106
107 public MessagePacket generatePgpChat(Message message, boolean addDelay) {
108 MessagePacket packet = preparePacket(message, addDelay);
109 packet.setBody("This is an XEP-0027 encrypted message");
110 if (message.getEncryption() == Message.ENCRYPTION_DECRYPTED) {
111 packet.addChild("x", "jabber:x:encrypted").setContent(message.getEncryptedBody());
112 } else if (message.getEncryption() == Message.ENCRYPTION_PGP) {
113 packet.addChild("x", "jabber:x:encrypted").setContent(message.getBody());
114 }
115 return packet;
116 }
117
118 public MessagePacket generateChatState(Conversation conversation) {
119 final Account account = conversation.getAccount();
120 MessagePacket packet = new MessagePacket();
121 packet.setTo(conversation.getJid().toBareJid());
122 packet.setFrom(account.getJid());
123 packet.addChild(ChatState.toElement(conversation.getOutgoingChatState()));
124 return packet;
125 }
126
127 public MessagePacket confirm(final Account account, final Jid to, final String id) {
128 MessagePacket packet = new MessagePacket();
129 packet.setType(MessagePacket.TYPE_NORMAL);
130 packet.setTo(to);
131 packet.setFrom(account.getJid());
132 Element received = packet.addChild("displayed",
133 "urn:xmpp:chat-markers:0");
134 received.setAttribute("id", id);
135 return packet;
136 }
137
138 public MessagePacket conferenceSubject(Conversation conversation,
139 String subject) {
140 MessagePacket packet = new MessagePacket();
141 packet.setType(MessagePacket.TYPE_GROUPCHAT);
142 packet.setTo(conversation.getJid().toBareJid());
143 Element subjectChild = new Element("subject");
144 subjectChild.setContent(subject);
145 packet.addChild(subjectChild);
146 packet.setFrom(conversation.getAccount().getJid().toBareJid());
147 return packet;
148 }
149
150 public MessagePacket directInvite(final Conversation conversation, final Jid contact) {
151 MessagePacket packet = new MessagePacket();
152 packet.setType(MessagePacket.TYPE_NORMAL);
153 packet.setTo(contact);
154 packet.setFrom(conversation.getAccount().getJid());
155 Element x = packet.addChild("x", "jabber:x:conference");
156 x.setAttribute("jid", conversation.getJid().toBareJid().toString());
157 return packet;
158 }
159
160 public MessagePacket invite(Conversation conversation, Jid contact) {
161 MessagePacket packet = new MessagePacket();
162 packet.setTo(conversation.getJid().toBareJid());
163 packet.setFrom(conversation.getAccount().getJid());
164 Element x = new Element("x");
165 x.setAttribute("xmlns", "http://jabber.org/protocol/muc#user");
166 Element invite = new Element("invite");
167 invite.setAttribute("to", contact.toBareJid().toString());
168 x.addChild(invite);
169 packet.addChild(x);
170 return packet;
171 }
172
173 public MessagePacket received(Account account,
174 MessagePacket originalMessage, String namespace) {
175 MessagePacket receivedPacket = new MessagePacket();
176 receivedPacket.setType(MessagePacket.TYPE_NORMAL);
177 receivedPacket.setTo(originalMessage.getFrom());
178 receivedPacket.setFrom(account.getJid());
179 Element received = receivedPacket.addChild("received", namespace);
180 received.setAttribute("id", originalMessage.getId());
181 return receivedPacket;
182 }
183
184 public MessagePacket generateOtrError(Jid to, String id, String errorText) {
185 MessagePacket packet = new MessagePacket();
186 packet.setType(MessagePacket.TYPE_ERROR);
187 packet.setAttribute("id",id);
188 packet.setTo(to);
189 Element error = packet.addChild("error");
190 error.setAttribute("code","406");
191 error.setAttribute("type","modify");
192 error.addChild("not-acceptable","urn:ietf:params:xml:ns:xmpp-stanzas");
193 error.addChild("text").setContent("?OTR Error:" + errorText);
194 return packet;
195 }
196}