flotilla/src/app/notifications.ts

import {writable, derived} from "svelte/store"
import {page} from "$app/stores"
import {deriveEvents} from "@welshman/store"
import {repository, pubkey} from "@welshman/app"
import {prop, max, sortBy, assoc, lt, now} from "@welshman/lib"
import type {Filter, TrustedEvent} from "@welshman/util"
import {DIRECT_MESSAGE, MESSAGE, THREAD, COMMENT} from "@welshman/util"
import {makeSpacePath} from "@app/routes"
import {LEGACY_THREAD, deriveEventsForUrl, getMembershipUrls, userMembership} from "@app/state"

// Checked state

export const checked = writable<Record<string, number>>({})

checked.subscribe(v => console.log("====== checked", v))

export const deriveChecked = (key: string) => derived(checked, prop(key))

export const setChecked = (key: string, ts = now()) =>
  Boolean(console.trace("====== setChecked", key))||
  checked.update(state => ({...state, [key]: ts}))

// Filters for various routes

export const CHAT_FILTERS: Filter[] = [{kinds: [DIRECT_MESSAGE]}]

export const SPACE_FILTERS: Filter[] = [{kinds: [THREAD, MESSAGE, COMMENT]}]

export const ROOM_FILTERS: Filter[] = [{kinds: [MESSAGE]}]

export const THREAD_FILTERS: Filter[] = [
  {kinds: [THREAD, LEGACY_THREAD]},
  {kinds: [COMMENT], "#K": [String(THREAD), String(LEGACY_THREAD)]},
]

export const getNotificationFilters = (since: number): Filter[] =>
  [...CHAT_FILTERS, ...SPACE_FILTERS, ...THREAD_FILTERS].map(assoc("since", since))

export const getRoomFilters = (room: string): Filter[] => ROOM_FILTERS.map(assoc("#h", [room]))

// Notification derivation

export const getNotification = (
  pubkey: string | null,
  lastChecked: number,
  events: TrustedEvent[],
) => {
  const [latestEvent] = sortBy($e => -$e.created_at, events)

  return latestEvent?.pubkey !== pubkey && lt(lastChecked, latestEvent?.created_at)
}

export const deriveNotification = (path: string, filters: Filter[], url?: string) => {
  const events = url ? deriveEventsForUrl(url, filters) : deriveEvents(repository, {filters})

  return derived(
    [pubkey, deriveChecked("*"), deriveChecked(path), events],
    ([$pubkey, $allChecked, $checked, $events]) => {
      return getNotification($pubkey, max([$allChecked, $checked]), $events)
    },
  )
}

export const spacesNotifications = derived(
  [pubkey, checked, userMembership, deriveEvents(repository, {filters: SPACE_FILTERS})],
  ([$pubkey, $checked, $userMembership, $events]) => {
    return getMembershipUrls($userMembership).filter(url => {
      const path = makeSpacePath(url)
      const lastChecked = max([$checked["*"], $checked[path]])
      const [latestEvent] = sortBy($e => -$e.created_at, $events)

      return latestEvent?.pubkey !== $pubkey && lt(lastChecked, latestEvent?.created_at)
    })
  },
)

export const inactiveSpacesNotifications = derived(
  [page, spacesNotifications],
  ([$page, $spacesNotifications]) =>
    $spacesNotifications.filter(url => !$page.url.pathname.startsWith(makeSpacePath(url))),
)