Refactor RecipientAddress

[signal-cli] / lib / src / main / java / org / asamk / signal / manager / api / RecipientAddress.java

diff --git a/lib/src/main/java/org/asamk/signal/manager/api/RecipientAddress.java b/lib/src/main/java/org/asamk/signal/manager/api/RecipientAddress.java
new file mode 100644 (file)
index 0000000..f477b9e
--- /dev/null
+++ b/lib/src/main/java/org/asamk/signal/manager/api/RecipientAddress.java
@@ -0,0 +1,71 @@
+package org.asamk.signal.manager.api;
+
+import org.whispersystems.signalservice.api.push.ServiceId;
+import org.whispersystems.signalservice.api.push.SignalServiceAddress;
+
+import java.util.Optional;
+import java.util.UUID;
+
+public record RecipientAddress(Optional<UUID> uuid, Optional<String> number) {
+
+    public static final UUID UNKNOWN_UUID = ServiceId.UNKNOWN.uuid();
+
+    /**
+     * Construct a RecipientAddress.
+     *
+     * @param uuid   The UUID of the user, if available.
+     * @param number The phone number of the user, if available.
+     */
+    public RecipientAddress {
+        uuid = uuid.isPresent() && uuid.get().equals(UNKNOWN_UUID) ? Optional.empty() : uuid;
+        if (uuid.isEmpty() && number.isEmpty()) {
+            throw new AssertionError("Must have either a UUID or E164 number!");
+        }
+    }
+
+    public RecipientAddress(UUID uuid, String e164) {
+        this(Optional.ofNullable(uuid), Optional.ofNullable(e164));
+    }
+
+    public RecipientAddress(SignalServiceAddress address) {
+        this(Optional.of(address.getServiceId().uuid()), address.getNumber());
+    }
+
+    public RecipientAddress(UUID uuid) {
+        this(Optional.of(uuid), Optional.empty());
+    }
+
+    public ServiceId getServiceId() {
+        return ServiceId.from(uuid.orElse(UNKNOWN_UUID));
+    }
+
+    public String getIdentifier() {
+        if (uuid.isPresent()) {
+            return uuid.get().toString();
+        } else if (number.isPresent()) {
+            return number.get();
+        } else {
+            throw new AssertionError("Given the checks in the constructor, this should not be possible.");
+        }
+    }
+
+    public String getLegacyIdentifier() {
+        if (number.isPresent()) {
+            return number.get();
+        } else if (uuid.isPresent()) {
+            return uuid.get().toString();
+        } else {
+            throw new AssertionError("Given the checks in the constructor, this should not be possible.");
+        }
+    }
+
+    public boolean matches(RecipientAddress other) {
+        return (uuid.isPresent() && other.uuid.isPresent() && uuid.get().equals(other.uuid.get())) || (
+                number.isPresent() && other.number.isPresent() && number.get().equals(other.number.get())
+        );
+    }
+
+    public SignalServiceAddress toSignalServiceAddress() {
+        return new SignalServiceAddress(getServiceId(), number);
+    }
+}