Simplify getting data from ipns mapping

Peergos · Jan 31, 2025 · 9d40be0 · 9d40be0
1 parent 5f6960b
commit 9d40be0
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 3 deletions.
diff --git a/src/main/java/org/peergos/net/APIHandler.java b/src/main/java/org/peergos/net/APIHandler.java
@@ -295,10 +295,10 @@ public void handleCallToAPI(HttpExchange httpExchange) {
                     if (records.isEmpty())
                         throw new IllegalStateException("Couldn't resolve " + signer);
                     IpnsRecord latest = records.get(records.size() - 1);
-                    Ipns.IpnsEntry entry = Ipns.IpnsEntry.parseFrom(ByteBuffer.wrap(latest.raw));
+                    IpnsMapping mapping = new IpnsMapping(signer, latest);
                     Map<String,  Object> res = new HashMap<>();
-                    res.put("sig", ArrayOps.bytesToHex(entry.getSignatureV2().toByteArray()));
-                    res.put("data", ArrayOps.bytesToHex(entry.getData().toByteArray()));
+                    res.put("sig", ArrayOps.bytesToHex(mapping.getSignature()));
+                    res.put("data", ArrayOps.bytesToHex(mapping.getData()));
                     String json = JSONParser.toString(res);
                     replyJson(httpExchange, json);
                     break;

diff --git a/src/main/java/org/peergos/protocol/ipns/IpnsMapping.java b/src/main/java/org/peergos/protocol/ipns/IpnsMapping.java
@@ -1,6 +1,10 @@
 package org.peergos.protocol.ipns;
 
+import com.google.protobuf.InvalidProtocolBufferException;
 import io.ipfs.multihash.*;
+import org.peergos.protocol.ipns.pb.Ipns;
+
+import java.nio.ByteBuffer;
 
 public class IpnsMapping {
     public final Multihash publisher;
@@ -10,4 +14,22 @@ public IpnsMapping(Multihash publisher, IpnsRecord value) {
         this.publisher = publisher;
         this.value = value;
     }
+
+    public byte[] getData() {
+        try {
+            Ipns.IpnsEntry entry = Ipns.IpnsEntry.parseFrom(ByteBuffer.wrap(value.raw));
+            return entry.getData().toByteArray();
+        } catch (InvalidProtocolBufferException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    public byte[] getSignature() {
+        try {
+        Ipns.IpnsEntry entry = Ipns.IpnsEntry.parseFrom(ByteBuffer.wrap(value.raw));
+            return entry.getSignatureV2().toByteArray();
+        } catch (InvalidProtocolBufferException e) {
+            throw new RuntimeException(e);
+        }
+    }
 }