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 {
26			packet.setTo(message.getCounterpart().split("/")[0]);
27			packet.setType(MessagePacket.TYPE_GROUPCHAT);
28		}
29		packet.setFrom(account.getFullJid());
30		packet.setId(message.getUuid());
31		if (addDelay) {
32			addDelay(packet,message.getTimeSent());
33		}
34		return packet;
35	}
36	
37	private void addDelay(MessagePacket packet, long timestamp) {
38		final SimpleDateFormat mDateFormat = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSS'Z'",Locale.US);
39		mDateFormat.setTimeZone(TimeZone.getTimeZone("UTC"));
40		Element delay = packet.addChild("delay", "urn:xmpp:delay");
41		Date date = new Date(timestamp);
42		delay.setAttribute("stamp", mDateFormat.format(date));
43	}
44	
45	public MessagePacket generateOtrChat(Message message) throws OtrException {
46		return generateOtrChat(message, false);
47	}
48	
49	public MessagePacket generateOtrChat(Message message, boolean addDelay) throws OtrException {
50		Session otrSession = message.getConversation().getOtrSession();
51		if (otrSession==null) {
52			throw new OtrException(null);
53		}
54		MessagePacket packet = preparePacket(message,addDelay);
55		packet.addChild("private", "urn:xmpp:carbons:2");
56		packet.addChild("no-copy", "urn:xmpp:hints");
57		packet.setBody(otrSession.transformSending(message
58				.getBody()));
59		return packet;
60	}
61	
62	public MessagePacket generateChat(Message message) {
63		return generateChat(message, false);
64	}
65	
66	public MessagePacket generateChat(Message message, boolean addDelay) {
67		MessagePacket packet = preparePacket(message,addDelay);
68		packet.setBody(message.getBody());
69		return packet;
70	}
71	
72	public MessagePacket generatePgpChat(Message message) {
73		return generatePgpChat(message, false);
74	}
75	
76	public MessagePacket generatePgpChat(Message message, boolean addDelay) {
77		MessagePacket packet = preparePacket(message,addDelay);
78		packet.setBody("This is an XEP-0027 encryted message");
79		if (message.getEncryption() == Message.ENCRYPTION_DECRYPTED) {
80			packet.addChild("x", "jabber:x:encrypted").setContent(
81					message.getEncryptedBody());
82		} else if (message.getEncryption() == Message.ENCRYPTION_PGP) {
83			packet.setBody(message.getBody());
84		}
85		return packet;
86	}
87}