diff --git a/briar-core/src/main/java/org/briarproject/briar/remotewipe/MessageEncoderImpl.java b/briar-core/src/main/java/org/briarproject/briar/remotewipe/MessageEncoderImpl.java index c9ef3ad1d..2463793f6 100644 --- a/briar-core/src/main/java/org/briarproject/briar/remotewipe/MessageEncoderImpl.java +++ b/briar-core/src/main/java/org/briarproject/briar/remotewipe/MessageEncoderImpl.java @@ -7,8 +7,8 @@ import org.briarproject.briar.api.remotewipe.MessageEncoder; import javax.inject.Inject; -import static org.briarproject.briar.remotewipe.MessageType.SETUP; -import static org.briarproject.briar.remotewipe.MessageType.WIPE; +import static org.briarproject.briar.api.remotewipe.MessageType.SETUP; +import static org.briarproject.briar.api.remotewipe.MessageType.WIPE; public class MessageEncoderImpl implements MessageEncoder { diff --git a/briar-core/src/main/java/org/briarproject/briar/remotewipe/RemoteWipeManagerImpl.java b/briar-core/src/main/java/org/briarproject/briar/remotewipe/RemoteWipeManagerImpl.java index 599f4422f..67e9be615 100644 --- a/briar-core/src/main/java/org/briarproject/briar/remotewipe/RemoteWipeManagerImpl.java +++ b/briar-core/src/main/java/org/briarproject/briar/remotewipe/RemoteWipeManagerImpl.java @@ -34,6 +34,7 @@ import org.briarproject.briar.api.remotewipe.RemoteWipeManager; import org.briarproject.briar.api.remotewipe.RemoteWipeMessageHeader; import org.briarproject.briar.api.remotewipe.RemoteWipeReceivedEvent; import org.briarproject.briar.client.ConversationClientImpl; +import org.briarproject.briar.api.remotewipe.MessageType; import java.util.ArrayList; import java.util.Collection; @@ -45,8 +46,8 @@ import javax.annotation.Nullable; import javax.inject.Inject; import static org.briarproject.briar.client.MessageTrackerConstants.MSG_KEY_READ; -import static org.briarproject.briar.remotewipe.MessageType.SETUP; -import static org.briarproject.briar.remotewipe.MessageType.WIPE; +import static org.briarproject.briar.api.remotewipe.MessageType.SETUP; +import static org.briarproject.briar.api.remotewipe.MessageType.WIPE; import static org.briarproject.briar.remotewipe.RemoteWipeConstants.GROUP_KEY_CONTACT_ID; import static org.briarproject.briar.remotewipe.RemoteWipeConstants.GROUP_KEY_RECEIVED_WIPE; import static org.briarproject.briar.remotewipe.RemoteWipeConstants.GROUP_KEY_WIPERS; @@ -121,7 +122,8 @@ public class RemoteWipeManagerImpl extends ConversationClientImpl protected boolean incomingMessage(Transaction txn, Message m, BdfList body, BdfDictionary meta) throws DbException, FormatException { System.out.println("Incoming remote wipe message"); - MessageType type = MessageType.fromValue(body.getLong(0).intValue()); + MessageType type = MessageType + .fromValue(body.getLong(0).intValue()); if (type == SETUP) { messageTracker.trackIncomingMessage(txn, m); ContactId contactId = getContactId(txn, m.getGroupId()); @@ -129,7 +131,7 @@ public class RemoteWipeManagerImpl extends ConversationClientImpl MessageStatus status = db.getMessageStatus(txn, contactId, m.getId()); txn.attach(new RemoteWipeReceivedEvent( - createMessageHeader(m, meta, status), contactId)); + createMessageHeader(m, meta, status, type), contactId)); } else if (type == WIPE) { if (!remoteWipeIsSetup(txn)) return false; if (clock.currentTimeMillis() - m.getTimestamp() > MAX_MESSAGE_AGE) @@ -308,7 +310,7 @@ public class RemoteWipeManagerImpl extends ConversationClientImpl MessageStatus status = db.getMessageStatus(txn, contactId, messageEntry.getKey()); headers.add( - createMessageHeader(message, meta, status)); + createMessageHeader(message, meta, status, SETUP)); } else if (meta.getLong(MSG_KEY_MESSAGE_TYPE).intValue() == WIPE.getValue()) { Message message = clientHelper @@ -316,7 +318,7 @@ public class RemoteWipeManagerImpl extends ConversationClientImpl MessageStatus status = db.getMessageStatus(txn, contactId, messageEntry.getKey()); headers.add( - createMessageHeader(message, meta, status)); + createMessageHeader(message, meta, status, WIPE)); } } return headers; @@ -326,7 +328,7 @@ public class RemoteWipeManagerImpl extends ConversationClientImpl } private RemoteWipeMessageHeader createMessageHeader( - Message message, BdfDictionary meta, MessageStatus status + Message message, BdfDictionary meta, MessageStatus status, MessageType type ) throws FormatException { @@ -343,7 +345,7 @@ public class RemoteWipeManagerImpl extends ConversationClientImpl return new RemoteWipeMessageHeader( message.getId(), message.getGroupId(), timestamp, isLocal, read, status.isSent(), status.isSeen(), - attachmentHeaders); + attachmentHeaders, type); } @Override @@ -414,7 +416,7 @@ public class RemoteWipeManagerImpl extends ConversationClientImpl @Nullable private Pair findMessage(Transaction txn, - GroupId g, MessageType type, boolean local) + GroupId g, org.briarproject.briar.api.remotewipe.MessageType type, boolean local) throws DbException, FormatException { BdfDictionary query = BdfDictionary.of( new BdfEntry(MSG_KEY_MESSAGE_TYPE, type.getValue()), diff --git a/briar-core/src/main/java/org/briarproject/briar/remotewipe/RemoteWipeValidator.java b/briar-core/src/main/java/org/briarproject/briar/remotewipe/RemoteWipeValidator.java index 2a52c9f6f..761991a21 100644 --- a/briar-core/src/main/java/org/briarproject/briar/remotewipe/RemoteWipeValidator.java +++ b/briar-core/src/main/java/org/briarproject/briar/remotewipe/RemoteWipeValidator.java @@ -17,8 +17,8 @@ import javax.annotation.concurrent.Immutable; import javax.inject.Inject; import static org.briarproject.bramble.util.ValidationUtils.checkSize; -import static org.briarproject.briar.remotewipe.MessageType.SETUP; -import static org.briarproject.briar.remotewipe.MessageType.WIPE; +import static org.briarproject.briar.api.remotewipe.MessageType.SETUP; +import static org.briarproject.briar.api.remotewipe.MessageType.WIPE; import static org.briarproject.briar.socialbackup.SocialBackupConstants.MSG_KEY_LOCAL; import static org.briarproject.briar.socialbackup.SocialBackupConstants.MSG_KEY_MESSAGE_TYPE; @@ -35,7 +35,8 @@ class RemoteWipeValidator extends BdfMessageValidator { @Override protected BdfMessageContext validateMessage(Message m, Group g, BdfList body) throws FormatException { - MessageType type = MessageType + org.briarproject.briar.api.remotewipe.MessageType + type = org.briarproject.briar.api.remotewipe.MessageType .fromValue(body.getLong(0).intValue()); if (type == SETUP) return validateSetupMessage(body); else if (type == WIPE) return validateWipeMessage(body);