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