From 7ce52cbff7e622a9046e6a36d4e664d211043e1e Mon Sep 17 00:00:00 2001
From: Derrick Hammer <derrick@derrickhammer.com>
Date: Sun, 3 Sep 2023 20:21:35 -0400
Subject: [PATCH] feat: add encodeRegistryCid, encodeRegistryValue,
 decodeRegistryValue, decodeRegistryCid

---
 src/cid.ts | 91 +++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 86 insertions(+), 5 deletions(-)

diff --git a/src/cid.ts b/src/cid.ts
index 2fd80c7..c7c7438 100644
--- a/src/cid.ts
+++ b/src/cid.ts
@@ -1,10 +1,12 @@
 import { ErrTuple } from "#types.js";
+import type { CID } from "@lumeweb/libportal";
 import {
   decodeCid as decodeCidPortal,
   encodeCid as encodeCidPortal,
 } from "@lumeweb/libportal";
 import { addContextToErr } from "#err.js";
-import type { CID } from "@lumeweb/libportal";
+import { CID_HASH_TYPES, CID_TYPES, REGISTRY_TYPES } from "@lumeweb/libs5";
+import { concatBytes } from "@noble/hashes/utils";
 
 export function encodeCid(
   hash: Uint8Array,
@@ -12,21 +14,21 @@ export function encodeCid(
   type?: number,
   hashType?: number,
   raw?: boolean,
-): ErrTuple<string>;
+): ErrTuple<string | Uint8Array>;
 export function encodeCid(
   hash: string,
   size: bigint,
   type?: number,
   hashType?: number,
   raw?: boolean,
-): ErrTuple<string>;
+): ErrTuple<string | Uint8Array>;
 export function encodeCid(
   hash: any,
   size: bigint,
   type?: number,
   hashType?: number,
   raw: boolean = false,
-): ErrTuple<string> {
+): ErrTuple<string | Uint8Array> {
   try {
     return [encodeCidPortal(hash, size, type, hashType, raw), null];
   } catch (e) {
@@ -34,7 +36,7 @@ export function encodeCid(
   }
 }
 
-export function decodeCid(cid: string): ErrTuple<CID> {
+export function decodeCid(cid: string | Uint8Array): ErrTuple<CID> {
   try {
     return [decodeCidPortal(cid), null];
   } catch (e) {
@@ -51,4 +53,83 @@ export function verifyCid(cid: string): boolean {
   }
 }
 
+export function encodeRegistryCid(
+  hash: Uint8Array,
+  size?: bigint,
+  type?: number,
+  hashType?: number,
+  raw?: boolean,
+): ErrTuple<string | Uint8Array>;
+export function encodeRegistryCid(
+  hash: string,
+  size?: bigint,
+  type?: number,
+  hashType?: number,
+  raw?: boolean,
+): ErrTuple<string | Uint8Array>;
+export function encodeRegistryCid(
+  hash: any,
+  size = BigInt(0),
+  type = CID_TYPES.RESOLVER,
+  hashType = CID_HASH_TYPES.ED25519,
+  raw: boolean = false,
+): ErrTuple<string | Uint8Array> {
+  return encodeCid(hash, size, type, hashType, raw);
+}
+
+export function encodeRegistryValue(
+  cid: CID | string,
+  type: number = REGISTRY_TYPES.CID,
+  hashType = CID_HASH_TYPES.BLAKE3,
+): ErrTuple<Uint8Array> {
+  if (typeof cid === "string") {
+    let err;
+    [cid, err] = decodeCid(cid);
+
+    if (err) {
+      return [new Uint8Array(), err];
+    }
+  }
+
+  const [ret, err] = encodeCid(
+    cid.hash,
+    cid.size,
+    CID_TYPES.RESOLVER,
+    CID_HASH_TYPES.BLAKE3,
+    true,
+  );
+
+  if (err) {
+    return [new Uint8Array(), err];
+  }
+
+  return [
+    concatBytes(Uint8Array.from([REGISTRY_TYPES.CID]), ret as Uint8Array),
+    null,
+  ];
+}
+
+export function decodeRegistryValue(hash: Uint8Array): ErrTuple<CID> {
+  if (!Object.values(REGISTRY_TYPES).includes(hash[0])) {
+    return [null, "invalid registry type"];
+  }
+
+  hash = hash.slice(1);
+
+  return decodeCid(hash);
+}
+
+export function decodeRegistryCid(cid: string | Uint8Array): ErrTuple<CID> {
+  const [ret, err] = decodeCid(cid);
+  if (err) {
+    return [null, err];
+  }
+
+  if (ret.type !== CID_TYPES.RESOLVER) {
+    return [null, "not a valid registry cid"];
+  }
+
+  return [cid, null];
+}
+
 export type { CID } from "@lumeweb/libportal";