package eu.siacs.conversations.entities; import java.net.MalformedURLException; import java.net.URL; import java.util.Arrays; import eu.siacs.conversations.Config; import eu.siacs.conversations.R; import android.content.ContentValues; import android.content.Context; import android.database.Cursor; public class Message extends AbstractEntity { public static final String TABLENAME = "messages"; public static final int STATUS_RECEIVED = 0; public static final int STATUS_UNSEND = 1; public static final int STATUS_SEND = 2; public static final int STATUS_SEND_FAILED = 3; public static final int STATUS_SEND_REJECTED = 4; public static final int STATUS_WAITING = 5; public static final int STATUS_OFFERED = 6; public static final int STATUS_SEND_RECEIVED = 7; public static final int STATUS_SEND_DISPLAYED = 8; public static final int ENCRYPTION_NONE = 0; public static final int ENCRYPTION_PGP = 1; public static final int ENCRYPTION_OTR = 2; public static final int ENCRYPTION_DECRYPTED = 3; public static final int ENCRYPTION_DECRYPTION_FAILED = 4; public static final int TYPE_TEXT = 0; public static final int TYPE_IMAGE = 1; public static final int TYPE_AUDIO = 2; public static final int TYPE_STATUS = 3; public static final int TYPE_PRIVATE = 4; public static String CONVERSATION = "conversationUuid"; public static String COUNTERPART = "counterpart"; public static String TRUE_COUNTERPART = "trueCounterpart"; public static String BODY = "body"; public static String TIME_SENT = "timeSent"; public static String ENCRYPTION = "encryption"; public static String STATUS = "status"; public static String TYPE = "type"; public static String REMOTE_MSG_ID = "remoteMsgId"; protected String conversationUuid; protected String counterpart; protected String trueCounterpart; protected String body; protected String encryptedBody; protected long timeSent; protected int encryption; protected int status; protected int type; protected boolean read = true; protected String remoteMsgId = null; protected Conversation conversation = null; protected Downloadable downloadable = null; public boolean markable = false; private Message() { } public Message(Conversation conversation, String body, int encryption) { this(java.util.UUID.randomUUID().toString(), conversation.getUuid(), conversation.getContactJid(), null, body, System .currentTimeMillis(), encryption, Message.STATUS_UNSEND, TYPE_TEXT, null); this.conversation = conversation; } public Message(Conversation conversation, String counterpart, String body, int encryption, int status) { this(java.util.UUID.randomUUID().toString(), conversation.getUuid(), counterpart, null, body, System.currentTimeMillis(), encryption, status, TYPE_TEXT, null); this.conversation = conversation; } public Message(String uuid, String conversationUUid, String counterpart, String trueCounterpart, String body, long timeSent, int encryption, int status, int type, String remoteMsgId) { this.uuid = uuid; this.conversationUuid = conversationUUid; this.counterpart = counterpart; this.trueCounterpart = trueCounterpart; this.body = body; this.timeSent = timeSent; this.encryption = encryption; this.status = status; this.type = type; this.remoteMsgId = remoteMsgId; } @Override public ContentValues getContentValues() { ContentValues values = new ContentValues(); values.put(UUID, uuid); values.put(CONVERSATION, conversationUuid); values.put(COUNTERPART, counterpart); values.put(TRUE_COUNTERPART, trueCounterpart); values.put(BODY, body); values.put(TIME_SENT, timeSent); values.put(ENCRYPTION, encryption); values.put(STATUS, status); values.put(TYPE, type); values.put(REMOTE_MSG_ID, remoteMsgId); return values; } public String getConversationUuid() { return conversationUuid; } public Conversation getConversation() { return this.conversation; } public String getCounterpart() { return counterpart; } public Contact getContact() { if (this.conversation.getMode() == Conversation.MODE_SINGLE) { return this.conversation.getContact(); } else { if (this.trueCounterpart == null) { return null; } else { return this.conversation.getAccount().getRoster() .getContactFromRoster(this.trueCounterpart); } } } public String getBody() { return body; } public String getReadableBody(Context context) { if (encryption == ENCRYPTION_PGP) { return context.getText(R.string.encrypted_message_received) .toString(); } else if (encryption == ENCRYPTION_DECRYPTION_FAILED) { return context.getText(R.string.decryption_failed).toString(); } else if (type == TYPE_IMAGE) { return context.getText(R.string.image_file).toString(); } else { return body.trim(); } } public long getTimeSent() { return timeSent; } public int getEncryption() { return encryption; } public int getStatus() { return status; } public String getRemoteMsgId() { return this.remoteMsgId; } public void setRemoteMsgId(String id) { this.remoteMsgId = id; } public static Message fromCursor(Cursor cursor) { return new Message(cursor.getString(cursor.getColumnIndex(UUID)), cursor.getString(cursor.getColumnIndex(CONVERSATION)), cursor.getString(cursor.getColumnIndex(COUNTERPART)), cursor.getString(cursor.getColumnIndex(TRUE_COUNTERPART)), cursor.getString(cursor.getColumnIndex(BODY)), cursor.getLong(cursor.getColumnIndex(TIME_SENT)), cursor.getInt(cursor.getColumnIndex(ENCRYPTION)), cursor.getInt(cursor.getColumnIndex(STATUS)), cursor.getInt(cursor.getColumnIndex(TYPE)), cursor.getString(cursor.getColumnIndex(REMOTE_MSG_ID))); } public void setConversation(Conversation conv) { this.conversation = conv; } public void setStatus(int status) { this.status = status; } public boolean isRead() { return this.read; } public void markRead() { this.read = true; } public void markUnread() { this.read = false; } public void setTime(long time) { this.timeSent = time; } public void setEncryption(int encryption) { this.encryption = encryption; } public void setBody(String body) { this.body = body; } public String getEncryptedBody() { return this.encryptedBody; } public void setEncryptedBody(String body) { this.encryptedBody = body; } public void setType(int type) { this.type = type; } public int getType() { return this.type; } public void setPresence(String presence) { if (presence == null) { this.counterpart = this.counterpart.split("/", 2)[0]; } else { this.counterpart = this.counterpart.split("/", 2)[0] + "/" + presence; } } public void setTrueCounterpart(String trueCounterpart) { this.trueCounterpart = trueCounterpart; } public String getPresence() { String[] counterparts = this.counterpart.split("/", 2); if (counterparts.length == 2) { return counterparts[1]; } else { if (this.counterpart.contains("/")) { return ""; } else { return null; } } } public void setDownloadable(Downloadable downloadable) { this.downloadable = downloadable; } public Downloadable getDownloadable() { return this.downloadable; } public static Message createStatusMessage(Conversation conversation) { Message message = new Message(); message.setType(Message.TYPE_STATUS); message.setConversation(conversation); return message; } public void setCounterpart(String counterpart) { this.counterpart = counterpart; } public boolean equals(Message message) { if ((this.remoteMsgId != null) && (this.body != null) && (this.counterpart != null)) { return this.remoteMsgId.equals(message.getRemoteMsgId()) && this.body.equals(message.getBody()) && this.counterpart.equals(message.getCounterpart()); } else { return false; } } public Message next() { int index = this.conversation.getMessages().indexOf(this); if (index < 0 || index >= this.conversation.getMessages().size() - 1) { return null; } else { return this.conversation.getMessages().get(index + 1); } } public Message prev() { int index = this.conversation.getMessages().indexOf(this); if (index <= 0 || index > this.conversation.getMessages().size()) { return null; } else { return this.conversation.getMessages().get(index - 1); } } public boolean mergable(Message message) { if (message == null) { return false; } return (message.getType() == Message.TYPE_TEXT && this.getDownloadable() == null && message.getDownloadable() == null && message.getEncryption() != Message.ENCRYPTION_PGP && this.getType() == message.getType() && this.getEncryption() == message.getEncryption() && this.getCounterpart().equals(message.getCounterpart()) && (message.getTimeSent() - this.getTimeSent()) <= (Config.MESSAGE_MERGE_WINDOW * 1000) && ((this .getStatus() == message.getStatus() || ((this.getStatus() == Message.STATUS_SEND || this .getStatus() == Message.STATUS_SEND_RECEIVED) && (message .getStatus() == Message.STATUS_UNSEND || message.getStatus() == Message.STATUS_SEND || message .getStatus() == Message.STATUS_SEND_DISPLAYED))))); } public String getMergedBody() { Message next = this.next(); if (this.mergable(next)) { return body.trim() + '\n' + next.getMergedBody(); } return body.trim(); } public int getMergedStatus() { Message next = this.next(); if (this.mergable(next)) { return next.getMergedStatus(); } else { return getStatus(); } } public long getMergedTimeSent() { Message next = this.next(); if (this.mergable(next)) { return next.getMergedTimeSent(); } else { return getTimeSent(); } } public boolean wasMergedIntoPrevious() { Message prev = this.prev(); if (prev == null) { return false; } else { return prev.mergable(this); } } public boolean bodyContainsDownloadable() { Contact contact = this.getContact(); if (status <= STATUS_RECEIVED && (contact == null || !contact.trusted())) { return false; } try { URL url = new URL(this.getBody()); if (!url.getProtocol().equalsIgnoreCase("http") && !url.getProtocol().equalsIgnoreCase("https")) { return false; } if (url.getPath() == null) { return false; } String[] pathParts = url.getPath().split("/"); String filename = pathParts[pathParts.length - 1]; String[] extensionParts = filename.split("\\."); if (extensionParts.length == 2 && Arrays.asList(Downloadable.VALID_EXTENSIONS).contains( extensionParts[extensionParts.length - 1])) { return true; } else if (extensionParts.length == 3 && Arrays.asList(Downloadable.VALID_CRYPTO_EXTENSIONS) .contains(extensionParts.length - 1) && Arrays.asList(Downloadable.VALID_EXTENSIONS).contains( extensionParts[extensionParts.length - 2])) { return true; } else { return false; } } catch (MalformedURLException e) { return false; } } public ImageParams getImageParams() { ImageParams params = new ImageParams(); if (this.downloadable != null) { params.size = this.downloadable.getFileSize(); } if (body == null) { return params; } String parts[] = body.split(","); if (parts.length == 1) { try { params.size = Long.parseLong(parts[0]); } catch (NumberFormatException e) { params.origin = parts[0]; } } else if (parts.length == 3) { try { params.size = Long.parseLong(parts[0]); } catch (NumberFormatException e) { params.size = 0; } try { params.width = Integer.parseInt(parts[1]); } catch (NumberFormatException e) { params.width = 0; } try { params.height = Integer.parseInt(parts[2]); } catch (NumberFormatException e) { params.height = 0; } } else if (parts.length == 4) { params.origin = parts[0]; try { params.size = Long.parseLong(parts[1]); } catch (NumberFormatException e) { params.size = 0; } try { params.width = Integer.parseInt(parts[2]); } catch (NumberFormatException e) { params.width = 0; } try { params.height = Integer.parseInt(parts[3]); } catch (NumberFormatException e) { params.height = 0; } } return params; } public class ImageParams { public long size = 0; public int width = 0; public int height = 0; public String origin; } }