Update wallet backup blueprint to use wallet event

Author: hzrd149

.changeset/nasty-mirrors-push.md

@@ -0,0 +1,5 @@
+---
+"applesauce-core": minor
+---
+
+Add `setEventContentEncryptionMethod` method

packages/core/src/helpers/hidden-content.ts

@@ -47,6 +47,11 @@ export const EventContentEncryptionMethod: Record<number, "nip04" | "nip44"> = {
   [kinds.Interestsets]: "nip04",
 };
 
+/** Sets the encryption method that is used for the contents of a specific event kind */
+export function setEventContentEncryptionMethod(kind: number, method: "nip04" | "nip44") {
+  EventContentEncryptionMethod[kind] = method;
+}
+
 /** Checks if an event can have hidden content */
 export function canHaveHiddenContent(kind: number): boolean {
   return EventContentEncryptionMethod[kind] !== undefined;

packages/wallet/src/actions/__tests__/wallet.test.ts

@@ -5,7 +5,7 @@ import { ActionHub } from "applesauce-actions";
 
 import { FakeUser } from "../../__tests__/fake-user.js";
 import { CreateWallet } from "../wallet.js";
-import { WALLET_BACKUP_KIND, WALLET_KIND } from "../../helpers/wallet.js";
+import { WALLET_BACKUP_KIND } from "../../helpers/wallet.js";
 import { NostrEvent } from "nostr-tools";
 import { unlockHiddenTags } from "applesauce-core/helpers";
 

packages/wallet/src/actions/wallet.ts

@@ -9,10 +9,10 @@ export function CreateWallet(mints: string[], privateKey = generateSecretKey()):
     const existing = events.getReplaceable(WALLET_KIND, self);
     if (existing) throw new Error("Wallet already exists");
 
-    const backup = await factory.create(WalletBackupBlueprint, privateKey, mints);
-    const wallet = await factory.create(WalletBlueprint, privateKey, mints);
+    const wallet = await factory.sign(await factory.create(WalletBlueprint, privateKey, mints));
+    const backup = await factory.sign(await factory.create(WalletBackupBlueprint, wallet));
 
-    await publish("Wallet backup", await factory.sign(backup));
-    await publish("New wallet", await factory.sign(wallet));
+    await publish("Wallet backup", backup);
+    await publish("Create wallet", wallet);
   };
 }

packages/wallet/src/blueprints/wallet.ts

@@ -3,6 +3,8 @@ import { modifyHiddenTags } from "applesauce-factory/operations/event";
 
 import { WALLET_BACKUP_KIND, WALLET_KIND } from "../helpers/wallet.js";
 import { setMintTags, setPrivateKeyTag } from "../operations/index.js";
+import { NostrEvent } from "nostr-tools";
+import { setWalletBackupContent } from "../operations/wallet.js";
 
 /** A blueprint to create a new 17375 wallet */
 export function WalletBlueprint(privateKey: Uint8Array, mints: string[]): EventBlueprint {
@@ -15,11 +17,6 @@ export function WalletBlueprint(privateKey: Uint8Array, mints: string[]): EventB
 }
 
 /** A blueprint that creates a new 375 wallet backup event */
-export function WalletBackupBlueprint(privateKey: Uint8Array, mints: string[]): EventBlueprint {
-  return (ctx) =>
-    EventFactory.runProcess(
-      { kind: WALLET_BACKUP_KIND },
-      ctx,
-      modifyHiddenTags(setPrivateKeyTag(privateKey), setMintTags(mints)),
-    );
+export function WalletBackupBlueprint(wallet: NostrEvent): EventBlueprint {
+  return (ctx) => EventFactory.runProcess({ kind: WALLET_BACKUP_KIND }, ctx, setWalletBackupContent(wallet));
 }

packages/wallet/src/operations/__tests__/wallet.test.ts

@@ -0,0 +1,47 @@
+import { describe, expect, it } from "vitest";
+import { setWalletBackupContent } from "../wallet.js";
+import { FakeUser } from "../../__tests__/fake-user.js";
+import { EventFactory } from "applesauce-factory";
+import { WalletBlueprint } from "../../blueprints/wallet.js";
+import { generateSecretKey } from "nostr-tools";
+import { WALLET_BACKUP_KIND } from "../../helpers/wallet.js";
+import { unixNow } from "applesauce-core/helpers";
+
+const user = new FakeUser();
+const factory = new EventFactory({ signer: user });
+
+describe("setWalletBackupContent", () => {
+  it("should throw if kind is not wallet kind", async () => {
+    const note = user.note();
+
+    await expect(
+      setWalletBackupContent(note)(
+        { kind: WALLET_BACKUP_KIND, tags: [], created_at: unixNow(), content: "" },
+        factory.context,
+      ),
+    ).rejects.toThrow();
+  });
+
+  it("should throw if pubkey does not match", async () => {
+    const wallet = await factory.sign(await factory.create(WalletBlueprint, generateSecretKey(), []));
+    const user2 = new FakeUser();
+
+    await expect(
+      setWalletBackupContent(wallet)(
+        { kind: WALLET_BACKUP_KIND, tags: [], created_at: unixNow(), content: "" },
+        { signer: user2 },
+      ),
+    ).rejects.toThrow();
+  });
+
+  it("should copy the content of the wallet event", async () => {
+    const wallet = await factory.sign(await factory.create(WalletBlueprint, generateSecretKey(), []));
+
+    expect(
+      await setWalletBackupContent(wallet)(
+        { kind: WALLET_BACKUP_KIND, tags: [], created_at: unixNow(), content: "" },
+        factory.context,
+      ),
+    ).toEqual(expect.objectContaining({ content: wallet.content }));
+  });
+});

packages/wallet/src/operations/wallet.ts

@@ -0,0 +1,16 @@
+import { EventOperation } from "applesauce-factory";
+import { NostrEvent } from "nostr-tools";
+import { WALLET_KIND } from "../helpers/wallet.js";
+
+/** Sets the content of a kind 375 wallet backup event */
+export function setWalletBackupContent(wallet: NostrEvent): EventOperation {
+  return async (draft, ctx) => {
+    if (wallet.kind !== WALLET_KIND) throw new Error(`Cant create a wallet backup from kind ${wallet.kind}`);
+    if (!wallet.content) throw new Error("Wallet missing content");
+
+    const pubkey = await ctx.signer?.getPublicKey();
+    if (wallet.pubkey !== pubkey) throw new Error("Wallet pubkey dose not match signer pubkey");
+
+    return { ...draft, content: wallet.content };
+  };
+}