Reformat imports

[signal-cli] / src / main / java / org / asamk / signal / TrustLevel.java

diff --git a/src/main/java/org/asamk/signal/TrustLevel.java b/src/main/java/org/asamk/signal/TrustLevel.java
index e9e7796de2821bdca74676342172c1e2279757cd..5eaf960af924fcea0bd7fb3256c97bbc444e1370 100644 (file)
--- a/src/main/java/org/asamk/signal/TrustLevel.java
+++ b/src/main/java/org/asamk/signal/TrustLevel.java
@@ -1,5 +1,7 @@
 package org.asamk.signal;
 
+import org.whispersystems.signalservice.api.messages.multidevice.VerifiedMessage;
+
 public enum TrustLevel {
     UNTRUSTED,
     TRUSTED_UNVERIFIED,
@@ -13,4 +15,28 @@ public enum TrustLevel {
         }
         return TrustLevel.cachedValues[i];
     }
+
+    public static TrustLevel fromVerifiedState(VerifiedMessage.VerifiedState verifiedState) {
+        switch (verifiedState) {
+            case DEFAULT:
+                return TRUSTED_UNVERIFIED;
+            case UNVERIFIED:
+                return UNTRUSTED;
+            case VERIFIED:
+                return TRUSTED_VERIFIED;
+        }
+        throw new RuntimeException("Unknown verified state: " + verifiedState);
+    }
+
+    public VerifiedMessage.VerifiedState toVerifiedState() {
+        switch (this) {
+            case TRUSTED_UNVERIFIED:
+                return VerifiedMessage.VerifiedState.DEFAULT;
+            case UNTRUSTED:
+                return VerifiedMessage.VerifiedState.UNVERIFIED;
+            case TRUSTED_VERIFIED:
+                return VerifiedMessage.VerifiedState.VERIFIED;
+        }
+        throw new RuntimeException("Unknown verified state: " + this);
+    }
 }