@waku/enr 0.0.1

package/src/multiaddrs_codec.ts

@@ -0,0 +1,50 @@
+import { multiaddr } from "@multiformats/multiaddr";
+import type { Multiaddr } from "@multiformats/multiaddr";
+
+import { MULTIADDR_LENGTH_SIZE } from "./constants.js";
+
+export function decodeMultiaddrs(bytes: Uint8Array): Multiaddr[] {
+  const multiaddrs = [];
+
+  let index = 0;
+
+  while (index < bytes.length) {
+    const sizeDataView = new DataView(
+      bytes.buffer,
+      index,
+      MULTIADDR_LENGTH_SIZE
+    );
+    const size = sizeDataView.getUint16(0);
+    index += MULTIADDR_LENGTH_SIZE;
+
+    const multiaddrBytes = bytes.slice(index, index + size);
+    index += size;
+
+    multiaddrs.push(multiaddr(multiaddrBytes));
+  }
+  return multiaddrs;
+}
+
+export function encodeMultiaddrs(multiaddrs: Multiaddr[]): Uint8Array {
+  const totalLength = multiaddrs.reduce(
+    (acc, ma) => acc + MULTIADDR_LENGTH_SIZE + ma.bytes.length,
+    0
+  );
+  const bytes = new Uint8Array(totalLength);
+  const dataView = new DataView(bytes.buffer);
+
+  let index = 0;
+  multiaddrs.forEach((multiaddr) => {
+    if (multiaddr.getPeerId())
+      throw new Error("`multiaddr` field MUST not contain peer id");
+
+    // Prepend the size of the next entry
+    dataView.setUint16(index, multiaddr.bytes.length);
+    index += MULTIADDR_LENGTH_SIZE;
+
+    bytes.set(multiaddr.bytes, index);
+    index += multiaddr.bytes.length;
+  });
+
+  return bytes;
+}

package/src/types.ts

@@ -0,0 +1,11 @@
+// Custom and aliased types for ENRs
+
+/**
+ * We represent NodeId as a hex string, since node equality is used very heavily
+ * and it is convenient to index data by NodeId
+ */
+export type NodeId = string;
+export type SequenceNumber = bigint;
+
+export type ENRKey = string;
+export type ENRValue = Uint8Array;

package/src/v4.ts

@@ -0,0 +1,21 @@
+import * as secp from "@noble/secp256k1";
+import { bytesToHex } from "@waku/byte-utils";
+
+import { keccak256 } from "./crypto.js";
+import { NodeId } from "./types.js";
+
+export async function sign(
+  privKey: Uint8Array,
+  msg: Uint8Array
+): Promise<Uint8Array> {
+  return secp.sign(keccak256(msg), privKey, {
+    der: false,
+  });
+}
+
+export function nodeId(pubKey: Uint8Array): NodeId {
+  const publicKey = secp.Point.fromHex(pubKey);
+  const uncompressedPubkey = publicKey.toRawBytes(false);
+
+  return bytesToHex(keccak256(uncompressedPubkey.slice(1)));
+}

package/src/waku2_codec.ts

@@ -0,0 +1,39 @@
+export interface Waku2 {
+  relay: boolean;
+  store: boolean;
+  filter: boolean;
+  lightPush: boolean;
+}
+
+export function encodeWaku2(protocols: Waku2): number {
+  let byte = 0;
+
+  if (protocols.lightPush) byte += 1;
+  byte = byte << 1;
+  if (protocols.filter) byte += 1;
+  byte = byte << 1;
+  if (protocols.store) byte += 1;
+  byte = byte << 1;
+  if (protocols.relay) byte += 1;
+
+  return byte;
+}
+
+export function decodeWaku2(byte: number): Waku2 {
+  const waku2 = {
+    relay: false,
+    store: false,
+    filter: false,
+    lightPush: false,
+  };
+
+  if (byte % 2) waku2.relay = true;
+  byte = byte >> 1;
+  if (byte % 2) waku2.store = true;
+  byte = byte >> 1;
+  if (byte % 2) waku2.filter = true;
+  byte = byte >> 1;
+  if (byte % 2) waku2.lightPush = true;
+
+  return waku2;
+}