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 {
 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) {
 46		return generateOtrChat(message, false);
 47	}
 48	
 49	public MessagePacket generateOtrChat(Message message, boolean addDelay) {
 50		Session otrSession = message.getConversation().getOtrSession();
 51		if (otrSession==null) {
 52			return 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		try {
 58			packet.setBody(otrSession.transformSending(message
 59					.getBody()));
 60			return packet;
 61		} catch (OtrException e) {
 62			return null;
 63		}
 64	}
 65	
 66	public MessagePacket generateChat(Message message) {
 67		return generateChat(message, false);
 68	}
 69	
 70	public MessagePacket generateChat(Message message, boolean addDelay) {
 71		MessagePacket packet = preparePacket(message,addDelay);
 72		packet.setBody(message.getBody());
 73		return packet;
 74	}
 75	
 76	public MessagePacket generatePgpChat(Message message) {
 77		return generatePgpChat(message, false);
 78	}
 79	
 80	public MessagePacket generatePgpChat(Message message, boolean addDelay) {
 81		MessagePacket packet = preparePacket(message,addDelay);
 82		packet.setBody("This is an XEP-0027 encryted message");
 83		if (message.getEncryption() == Message.ENCRYPTION_DECRYPTED) {
 84			packet.addChild("x", "jabber:x:encrypted").setContent(
 85					message.getEncryptedBody());
 86		} else if (message.getEncryption() == Message.ENCRYPTION_PGP) {
 87			packet.setBody(message.getBody());
 88		}
 89		return packet;
 90	}
 91
 92	public MessagePacket generateNotAcceptable(MessagePacket origin) {
 93		MessagePacket packet = generateError(origin);
 94		Element error = packet.addChild("error");
 95		error.setAttribute("type", "modify");
 96		error.setAttribute("code", "406");
 97		error.addChild("not-acceptable");
 98		return packet;
 99	}
100	
101	private MessagePacket generateError(MessagePacket origin) {
102		MessagePacket packet = new MessagePacket();
103		packet.setId(origin.getId());
104		packet.setTo(origin.getFrom());
105		packet.setBody(origin.getBody());
106		packet.setType(MessagePacket.TYPE_ERROR);
107		return packet;
108	}
109	
110	public MessagePacket confirm(Account account, String to, String id) {
111		MessagePacket packet = new MessagePacket();
112		packet.setType(MessagePacket.TYPE_NORMAL);
113		packet.setTo(to);
114		packet.setFrom(account.getFullJid());
115		Element received = packet.addChild("displayed",
116				"urn:xmpp:chat-markers:0");
117		received.setAttribute("id", id);
118		return packet;
119	}
120	
121	public MessagePacket conferenceSubject(Conversation conversation,String subject) {
122		MessagePacket packet = new MessagePacket();
123		packet.setType(MessagePacket.TYPE_GROUPCHAT);
124		packet.setTo(conversation.getContactJid().split("/")[0]);
125		Element subjectChild = new Element("subject");
126		subjectChild.setContent(subject);
127		packet.addChild(subjectChild);
128		packet.setFrom(conversation.getAccount().getJid());
129		return packet;
130	}
131	
132	public MessagePacket invite(Conversation conversation, String contact) {
133		MessagePacket packet = new MessagePacket();
134		packet.setType(MessagePacket.TYPE_NORMAL);
135		packet.setTo(contact);
136		packet.setFrom(conversation.getAccount().getFullJid());
137		Element x = packet.addChild("x", "jabber:x:conference");
138		x.setAttribute("jid", conversation.getContactJid().split("/")[0]);
139		return packet;
140	}
141}