MessageGenerator.java

  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("yyyy-MM-dd'T'HH:mm:ss.SSS'Z'",Locale.US);
 42		mDateFormat.setTimeZone(TimeZone.getTimeZone("UTC"));
 43		Element delay = packet.addChild("delay", "urn:xmpp:delay");
 44		Date date = new Date(timestamp);
 45		delay.setAttribute("stamp", mDateFormat.format(date));
 46	}
 47	
 48	public MessagePacket generateOtrChat(Message message) {
 49		return generateOtrChat(message, false);
 50	}
 51	
 52	public MessagePacket generateOtrChat(Message message, boolean addDelay) {
 53		Session otrSession = message.getConversation().getOtrSession();
 54		if (otrSession==null) {
 55			return null;
 56		}
 57		MessagePacket packet = preparePacket(message,addDelay);
 58		packet.addChild("private", "urn:xmpp:carbons:2");
 59		packet.addChild("no-copy", "urn:xmpp:hints");
 60		try {
 61			packet.setBody(otrSession.transformSending(message
 62					.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,String subject) {
125		MessagePacket packet = new MessagePacket();
126		packet.setType(MessagePacket.TYPE_GROUPCHAT);
127		packet.setTo(conversation.getContactJid().split("/")[0]);
128		Element subjectChild = new Element("subject");
129		subjectChild.setContent(subject);
130		packet.addChild(subjectChild);
131		packet.setFrom(conversation.getAccount().getJid());
132		return packet;
133	}
134	
135	public MessagePacket directInvite(Conversation conversation, String contact) {
136		MessagePacket packet = new MessagePacket();
137		packet.setType(MessagePacket.TYPE_NORMAL);
138		packet.setTo(contact);
139		packet.setFrom(conversation.getAccount().getFullJid());
140		Element x = packet.addChild("x", "jabber:x:conference");
141		x.setAttribute("jid", conversation.getContactJid().split("/")[0]);
142		return packet;
143	}
144	
145	public MessagePacket invite(Conversation conversation, String contact) {
146		MessagePacket packet = new MessagePacket();
147		packet.setTo(conversation.getContactJid().split("/")[0]);
148		packet.setFrom(conversation.getAccount().getFullJid());
149		Element x = new Element("x");
150		x.setAttribute("xmlns", "http://jabber.org/protocol/muc#user");
151		Element invite = new Element("invite");
152		invite.setAttribute("to", contact);
153		x.addChild(invite);
154		packet.addChild(x);
155		return packet;
156	}
157	
158	public MessagePacket received(Account account, MessagePacket originalMessage, String namespace) {
159		MessagePacket receivedPacket = new MessagePacket();
160		receivedPacket.setType(MessagePacket.TYPE_NORMAL);
161		receivedPacket.setTo(originalMessage.getFrom());
162		receivedPacket.setFrom(account.getFullJid());
163		Element received = receivedPacket.addChild("received",namespace);
164		received.setAttribute("id", originalMessage.getId());
165		return receivedPacket;
166	}
167}