flotilla/src/lib/editor/util.ts

import type {JSONContent, PasteRuleMatch, InputRuleMatch} from "@tiptap/core"
import {Editor} from "@tiptap/core"
import {choice} from "@welshman/lib"
import {Address} from "@welshman/util"

export const asInline = (extend: Record<string, any>) => ({
  inline: true,
  group: "inline",
  ...extend,
})

export const createInputRuleMatch = <T extends Record<string, unknown>>(
  match: RegExpMatchArray,
  data: T,
): InputRuleMatch => ({index: match.index!, text: match[0], match, data})

export const createPasteRuleMatch = <T extends Record<string, unknown>>(
  match: RegExpMatchArray,
  data: T,
): PasteRuleMatch => ({index: match.index!, text: match[0], match, data})

export const findNodes = (type: string, json: JSONContent) => {
  const results: JSONContent[] = []

  for (const node of json.content || []) {
    if (node.type === type) {
      results.push(node)
    }

    for (const result of findNodes(type, node)) {
      results.push(result)
    }
  }

  return results
}

export const findMarks = (type: string, json: JSONContent) => {
  const results: JSONContent[] = []

  for (const node of json.content || []) {
    for (const mark of node.marks || []) {
      if (mark.type === type) {
        results.push(mark)
      }
    }

    for (const result of findMarks(type, node)) {
      results.push(result)
    }
  }

  return results
}

export const getEditorTags = (editor: Editor) => {
  const json = editor.getJSON()

  const topicTags = findMarks("tag", json).map(
    ({attrs}: any) => ["t", attrs.tag.replace(/^#/, '').toLowerCase()],
  )

  const naddrTags = findNodes("naddr", json).map(
    ({kind, pubkey, identifier, relays}: any) => {
      const address = new Address(kind, pubkey, identifier).toString()

      return ["q", address, choice(relays) || "", pubkey]
    },
  )

  const neventTags = findNodes("nevent", json).map(
    ({id, author, relays}: any) => ["q", id, choice(relays) || "", author || ""],
  )

  const mentionTags = findNodes("nprofile", json).map(
    ({pubkey, relays}: any) => ["p", pubkey, choice(relays) || "", ""],
  )

  const imetaTags = findNodes("image", json).map(
    ({src, sha256}: any) => ["imeta", `url ${src}`, `x ${sha256}`, `ox ${sha256}`],
  )

  return [...topicTags, ...naddrTags, ...neventTags, ...mentionTags, ...imetaTags]
}

export const addFile = (editor: Editor) => editor.chain().selectFiles().run()

export const uploadFiles = (editor: Editor) => editor.chain().uploadFiles().run()