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