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.xml.Element;
14import eu.siacs.conversations.xmpp.stanzas.MessagePacket;
15
16public class MessageGenerator {
17 private MessagePacket preparePacket(Message message, boolean addDelay) {
18 Conversation conversation = message.getConversation();
19 Account account = conversation.getAccount();
20 MessagePacket packet = new MessagePacket();
21 if (conversation.getMode() == Conversation.MODE_SINGLE) {
22 packet.setTo(message.getCounterpart());
23 packet.setType(MessagePacket.TYPE_CHAT);
24 packet.addChild("markable", "urn:xmpp:chat-markers:0");
25 } else if (message.getType() == Message.TYPE_PRIVATE) {
26 packet.setTo(message.getCounterpart());
27 packet.setType(MessagePacket.TYPE_CHAT);
28 } else {
29 packet.setTo(message.getCounterpart().split("/")[0]);
30 packet.setType(MessagePacket.TYPE_GROUPCHAT);
31 }
32 packet.setFrom(account.getFullJid());
33 packet.setId(message.getUuid());
34 if (addDelay) {
35 addDelay(packet, message.getTimeSent());
36 }
37 return packet;
38 }
39
40 private void addDelay(MessagePacket packet, long timestamp) {
41 final SimpleDateFormat mDateFormat = new SimpleDateFormat(
42 "yyyy-MM-dd'T'HH:mm:ss.SSS'Z'", Locale.US);
43 mDateFormat.setTimeZone(TimeZone.getTimeZone("UTC"));
44 Element delay = packet.addChild("delay", "urn:xmpp:delay");
45 Date date = new Date(timestamp);
46 delay.setAttribute("stamp", mDateFormat.format(date));
47 }
48
49 public MessagePacket generateOtrChat(Message message) {
50 return generateOtrChat(message, false);
51 }
52
53 public MessagePacket generateOtrChat(Message message, boolean addDelay) {
54 Session otrSession = message.getConversation().getOtrSession();
55 if (otrSession == null) {
56 return null;
57 }
58 MessagePacket packet = preparePacket(message, addDelay);
59 packet.addChild("private", "urn:xmpp:carbons:2");
60 packet.addChild("no-copy", "urn:xmpp:hints");
61 try {
62 packet.setBody(otrSession.transformSending(message.getBody()));
63 return packet;
64 } catch (OtrException e) {
65 return null;
66 }
67 }
68
69 public MessagePacket generateChat(Message message) {
70 return generateChat(message, false);
71 }
72
73 public MessagePacket generateChat(Message message, boolean addDelay) {
74 MessagePacket packet = preparePacket(message, addDelay);
75 packet.setBody(message.getBody());
76 return packet;
77 }
78
79 public MessagePacket generatePgpChat(Message message) {
80 return generatePgpChat(message, false);
81 }
82
83 public MessagePacket generatePgpChat(Message message, boolean addDelay) {
84 MessagePacket packet = preparePacket(message, addDelay);
85 packet.setBody("This is an XEP-0027 encryted message");
86 if (message.getEncryption() == Message.ENCRYPTION_DECRYPTED) {
87 packet.addChild("x", "jabber:x:encrypted").setContent(
88 message.getEncryptedBody());
89 } else if (message.getEncryption() == Message.ENCRYPTION_PGP) {
90 packet.setBody(message.getBody());
91 }
92 return packet;
93 }
94
95 public MessagePacket generateNotAcceptable(MessagePacket origin) {
96 MessagePacket packet = generateError(origin);
97 Element error = packet.addChild("error");
98 error.setAttribute("type", "modify");
99 error.setAttribute("code", "406");
100 error.addChild("not-acceptable");
101 return packet;
102 }
103
104 private MessagePacket generateError(MessagePacket origin) {
105 MessagePacket packet = new MessagePacket();
106 packet.setId(origin.getId());
107 packet.setTo(origin.getFrom());
108 packet.setBody(origin.getBody());
109 packet.setType(MessagePacket.TYPE_ERROR);
110 return packet;
111 }
112
113 public MessagePacket confirm(Account account, String to, String id) {
114 MessagePacket packet = new MessagePacket();
115 packet.setType(MessagePacket.TYPE_NORMAL);
116 packet.setTo(to);
117 packet.setFrom(account.getFullJid());
118 Element received = packet.addChild("displayed",
119 "urn:xmpp:chat-markers:0");
120 received.setAttribute("id", id);
121 return packet;
122 }
123
124 public MessagePacket conferenceSubject(Conversation conversation,
125 String subject) {
126 MessagePacket packet = new MessagePacket();
127 packet.setType(MessagePacket.TYPE_GROUPCHAT);
128 packet.setTo(conversation.getContactJid().split("/")[0]);
129 Element subjectChild = new Element("subject");
130 subjectChild.setContent(subject);
131 packet.addChild(subjectChild);
132 packet.setFrom(conversation.getAccount().getJid());
133 return packet;
134 }
135
136 public MessagePacket directInvite(Conversation conversation, String contact) {
137 MessagePacket packet = new MessagePacket();
138 packet.setType(MessagePacket.TYPE_NORMAL);
139 packet.setTo(contact);
140 packet.setFrom(conversation.getAccount().getFullJid());
141 Element x = packet.addChild("x", "jabber:x:conference");
142 x.setAttribute("jid", conversation.getContactJid().split("/")[0]);
143 return packet;
144 }
145
146 public MessagePacket invite(Conversation conversation, String contact) {
147 MessagePacket packet = new MessagePacket();
148 packet.setTo(conversation.getContactJid().split("/")[0]);
149 packet.setFrom(conversation.getAccount().getFullJid());
150 Element x = new Element("x");
151 x.setAttribute("xmlns", "http://jabber.org/protocol/muc#user");
152 Element invite = new Element("invite");
153 invite.setAttribute("to", contact);
154 x.addChild(invite);
155 packet.addChild(x);
156 return packet;
157 }
158
159 public MessagePacket received(Account account,
160 MessagePacket originalMessage, String namespace) {
161 MessagePacket receivedPacket = new MessagePacket();
162 receivedPacket.setType(MessagePacket.TYPE_NORMAL);
163 receivedPacket.setTo(originalMessage.getFrom());
164 receivedPacket.setFrom(account.getFullJid());
165 Element received = receivedPacket.addChild("received", namespace);
166 received.setAttribute("id", originalMessage.getId());
167 return receivedPacket;
168 }
169}