X-Git-Url: https://git.nmode.ca/signal-cli/blobdiff_plain/55d485de88bf2f67f64be3978cf7c5649f1cc36f..a44034a79e6a98e7d77bbde0af1483d8c30a22c6:/src/main/java/org/asamk/signal/TrustLevel.java?ds=inline

diff --git a/src/main/java/org/asamk/signal/TrustLevel.java b/src/main/java/org/asamk/signal/TrustLevel.java
index e9e7796d..5eaf960a 100644
--- 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);
+    }
 }