1 files changed, 71 insertions, 0 deletions
diff --git a/src/lib/nip51.ts b/src/lib/nip51.ts
new file mode 100644
index 00000000..b1b0b613
--- /dev/null
+++ b/src/lib/nip51.ts
@@ -0,0 +1,71 @@
+import { SimplePool } from 'nostr-tools/pool';
+import { verifyEvent } from 'nostr-tools/pure';
+import { signEvent, encrypt, decrypt } from '$lib/nip07';
+const pool = new SimplePool();
+
+import { PUBLIC_DM_RELAYS } from '$env/static/public';
+const DM_RELAYS_LIST = PUBLIC_DM_RELAYS.split(',');
+
+// Mute the provided pubkey, as the provided user.
+export const mute = async (user: object, pubkey: string, hide: bool) => {
+    let { shown, hidden } = await getMuteLists(user);
+    if (hide) {
+        hidden.push(["p", pubkey]);
+    } else {
+        shown.push(["p", pubkey]);
+    }
+    await publishMuteList(user, shown, hidden);
+}
+
+const tagMatches = (tag: string[], pubkey: string): bool =>
+    tag.length >= 2 && tag[0] === "p" && tag[1] === pubkey;
+
+// Unmute the provided pubkey, as the provided user.
+export const unmute = async (user: object, pubkey: string) => {
+    let { shown, hidden } = await getMuteLists(user);
+    shown = shown.filter(tag => !tagMatches(tag, pubkey));
+    hidden = hidden.filter(tag => !tagMatches(tag, pubkey));
+    await publishMuteList(user, shown, hidden);
+}
+
+// Returns a set of the provided user's muted accounts.
+export const mutedAccounts = async (user: object): Set<string> => {
+    const { shown, hidden } = await getMuteLists(user);
+    let muted = new Set<string>();
+    for (const tag of shown) {
+        if (tag.length >= 2 && tag[0] === "p") {
+            muted.add(tag[1]);
+        }
+    }
+    for (const tag of hidden) {
+        if (tag.length >= 2 && tag[0] === "p") {
+            muted.add(tag[1]);
+        }
+    }
+    return muted;
+}
+
+const getMuteLists = async (user: object): object => {
+    const listEvents = await pool.querySync(
+        DM_RELAYS_LIST, { kinds: [10000], "authors": [user.pubkey], limit: 1 }
+    );
+    if (listEvents.length == 0) {
+        return { shown: [], hidden: [] };
+    }
+
+    return {
+        shown: listEvents[0].tags,
+        hidden: listEvents[0].content == "" ? [] :
+            await decrypt(listEvents[0].content, user, user.pubkey)
+    };
+}
+
+const publishMuteList = async (user: object, shown: string[], hidden: string[]) => {
+    const event = await signEvent({
+        kind: 10000,
+        created_at: Math.floor(Date.now() / 1000),
+        tags: shown,
+        content: await encrypt(hidden, user, user.pubkey)
+    }, user);
+    return Promise.any(pool.publish(DM_RELAYS_LIST, event));
+}