Files
welshman/packages/util/src/List.ts
T
2025-04-25 10:19:56 -07:00

101 lines
2.8 KiB
TypeScript

import {parseJson, uniq, nthEq} from "@welshman/lib"
import {Address} from "./Address.js"
import {uniqTags, getRelayTags} from "./Tags.js"
import {isRelayUrl, RelayMode, normalizeRelayUrl} from "./Relay.js"
import {Encryptable, DecryptedEvent} from "./Encryptable.js"
import type {EncryptableUpdates} from "./Encryptable.js"
export type ListParams = {
kind: number
}
export type List = ListParams & {
publicTags: string[][]
privateTags: string[][]
event?: DecryptedEvent
}
export type PublishedList = Omit<List, "event"> & {
event: DecryptedEvent
}
export const makeList = (list: ListParams & Partial<List>): List => ({
publicTags: [],
privateTags: [],
...list,
})
const isValidTag = (tag: string[]) => {
if (tag[0] === "p") return tag[1]?.length === 64
if (tag[0] === "e") return tag[1]?.length === 64
if (tag[0] === "a") return Address.isAddress(tag[1] || "")
if (tag[0] === "t") return tag[1]?.length > 0
if (tag[0] === "r") return isRelayUrl(tag[1])
if (tag[0] === "relay") return isRelayUrl(tag[1])
return true
}
export const readList = (event: DecryptedEvent): PublishedList => {
const getTags = (tags: string[][]) => (Array.isArray(tags) ? tags.filter(isValidTag) : [])
const privateTags = getTags(parseJson(event.plaintext?.content))
const publicTags = getTags(event.tags)
return {event, kind: event.kind, publicTags, privateTags}
}
export const getListTags = (list: List | undefined) => [
...(list?.publicTags || []),
...(list?.privateTags || []),
]
export const removeFromListByPredicate = (list: List, pred: (t: string[]) => boolean) => {
const plaintext: EncryptableUpdates = {}
const template = {
kind: list.kind,
content: list.event?.content || "",
tags: list.publicTags.filter(t => !pred(t)),
}
// Avoid redundant encrypt calls if possible
if (list.privateTags.some(t => pred(t))) {
plaintext.content = JSON.stringify(list.privateTags.filter(t => !pred(t)))
}
return new Encryptable(template, plaintext)
}
export const removeFromList = (list: List, value: string) =>
removeFromListByPredicate(list, nthEq(1, value))
export const addToListPublicly = (list: List, ...tags: string[][]) => {
const template = {
kind: list.kind,
content: list.event?.content || "",
tags: uniqTags([...list.publicTags, ...tags]),
}
return new Encryptable(template, {})
}
export const addToListPrivately = (list: List, ...tags: string[][]) => {
const template = {
kind: list.kind,
tags: list.publicTags,
}
return new Encryptable(template, {
content: JSON.stringify(uniqTags([...list.privateTags, ...tags])),
})
}
export const getRelaysFromList = (list?: List, mode?: RelayMode): string[] => {
let tags = getRelayTags(getListTags(list))
if (mode) {
tags = tags.filter((t: string[]) => !t[2] || t[2] === mode)
}
return uniq(tags.map(t => normalizeRelayUrl(t[1])))
}