1package eu.siacs.conversations.generator;
 2
 3import java.text.SimpleDateFormat;
 4import java.util.Date;
 5import java.util.TimeZone;
 6
 7import net.java.otr4j.OtrException;
 8import net.java.otr4j.session.Session;
 9import eu.siacs.conversations.entities.Account;
10import eu.siacs.conversations.entities.Conversation;
11import eu.siacs.conversations.entities.Message;
12import eu.siacs.conversations.xml.Element;
13import eu.siacs.conversations.xmpp.stanzas.MessagePacket;
14
15public class MessageGenerator {
16	private MessagePacket preparePacket(Message message, boolean addDelay) {
17		Conversation conversation = message.getConversation();
18		Account account = conversation.getAccount();
19		MessagePacket packet = new MessagePacket();
20		if (conversation.getMode() == Conversation.MODE_SINGLE) {
21			packet.setTo(message.getCounterpart());
22			packet.setType(MessagePacket.TYPE_CHAT);
23		} else {
24			packet.setTo(message.getCounterpart().split("/")[0]);
25			packet.setType(MessagePacket.TYPE_GROUPCHAT);
26		}
27		packet.setFrom(account.getFullJid());
28		packet.setId(message.getUuid());
29		if (addDelay) {
30			addDelay(packet,message.getTimeSent());
31		}
32		return packet;
33	}
34	
35	private void addDelay(MessagePacket packet, long timestamp) {
36		final SimpleDateFormat mDateFormat = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSS'Z'");
37		mDateFormat.setTimeZone(TimeZone.getTimeZone("UTC"));
38		Element delay = packet.addChild("delay", "urn:xmpp:delay");
39		Date date = new Date(timestamp);
40		delay.setAttribute("stamp", mDateFormat.format(date));
41	}
42	
43	public MessagePacket generateOtrChat(Message message) throws OtrException {
44		return generateOtrChat(message, false);
45	}
46	
47	public MessagePacket generateOtrChat(Message message, boolean addDelay) throws OtrException {
48		Session otrSession = message.getConversation().getOtrSession();
49		if (otrSession==null) {
50			throw new OtrException(null);
51		}
52		MessagePacket packet = preparePacket(message,addDelay);
53		packet.addChild("private", "urn:xmpp:carbons:2");
54		packet.addChild("no-copy", "urn:xmpp:hints");
55		packet.setBody(otrSession.transformSending(message
56				.getBody()));
57		return packet;
58	}
59	
60	public MessagePacket generateChat(Message message) {
61		return generateChat(message, false);
62	}
63	
64	public MessagePacket generateChat(Message message, boolean addDelay) {
65		MessagePacket packet = preparePacket(message,addDelay);
66		packet.setBody(message.getBody());
67		return packet;
68	}
69	
70	public MessagePacket generatePgpChat(Message message) {
71		return generatePgpChat(message, false);
72	}
73	
74	public MessagePacket generatePgpChat(Message message, boolean addDelay) {
75		MessagePacket packet = preparePacket(message,addDelay);
76		packet.setBody("This is an XEP-0027 encryted message");
77		if (message.getEncryption() == Message.ENCRYPTION_DECRYPTED) {
78			packet.addChild("x", "jabber:x:encrypted").setContent(
79					message.getEncryptedBody());
80		} else if (message.getEncryption() == Message.ENCRYPTION_PGP) {
81			packet.setBody(message.getBody());
82		}
83		return packet;
84	}
85}