platform/services/github/pod-github/src/sync/guest.ts

//
// Copyright © 2023 Hardcore Engineering Inc.
//

import { Branding, generateId, TxOperations, WorkspaceIdWithUrl } from '@hcengineering/core'
import { MarkupMarkType, MarkupNode, MarkupNodeType, traverseMarkupNode } from '@hcengineering/text'
import { getPublicLink } from '@hcengineering/server-guest-resources'
import { Task } from '@hcengineering/task'
import { generateToken } from '@hcengineering/server-token'

const githubLinkText = process.env.LINK_TEXT ?? 'Huly®:'

const githubLinkTextOld = 'View in Huly'

export function hasHulyLinkText (text: string): boolean {
  return text.includes(githubLinkText) || text.includes(githubLinkTextOld)
}

export function hasHulyLink (href: string, guestLink: string): boolean {
  return href.includes(guestLink)
}

export async function stripGuestLink (markdown: MarkupNode): Promise<void> {
  const toRemove: MarkupNode[] = []

  traverseMarkupNode(markdown, (node) => {
    if (node.content === undefined) {
      return
    }
    const oldLength = node.content.length
    node.content = node.content.filter((it) => it.type !== MarkupNodeType.subLink)

    // sub is an inline node hence tiptap wraps it with a paragraph
    // so we need to remove the parent paragraph node if it is empty
    if (node.content.length !== oldLength && node.type === MarkupNodeType.paragraph) {
      toRemove.push(node)
    }
  })

  // traverse nodes once again and remove empty parent node
  traverseMarkupNode(markdown, (node) => {
    if (node.content === undefined) {
      return
    }
    node.content = node.content.filter((it) => !toRemove.includes(it))
  })
}
export async function appendGuestLink (
  client: TxOperations,
  doc: Task,
  markdown: MarkupNode,
  workspace: WorkspaceIdWithUrl,
  branding: Branding | null
): Promise<void> {
  const publicLink = await getPublicLink(doc, client, workspace, false, branding)
  await stripGuestLink(markdown)
  appendGuestLinkToModel(markdown, publicLink, doc.identifier)
  appendGuestLinkToImage(markdown, workspace)
}

export function appendGuestLinkToModel (markdown: MarkupNode, publicLink: string, identifier: string): void {
  markdown.content = [
    ...(markdown.content ?? []),
    {
      type: MarkupNodeType.paragraph,
      content: [
        {
          type: MarkupNodeType.subLink,
          content: [
            {
              type: MarkupNodeType.text,
              text: githubLinkText.trim() + ' <b>' + identifier + '</b>',
              marks: [{ type: MarkupMarkType.link, attrs: { href: publicLink, _target: '_blank' } }]
            }
          ]
        }
      ]
    }
  ]
}

function findImageTags (node: MarkupNode): MarkupNode[] {
  if (node.type === MarkupNodeType.paragraph) {
    return node.content?.flatMap(findImageTags) ?? []
  }
  if (node.type === MarkupNodeType.image) {
    return [node]
  }
  return []
}

export function appendGuestLinkToImage (markdown: MarkupNode, workspace: WorkspaceIdWithUrl): void {
  const imageTags: MarkupNode[] = markdown.content?.flatMap(findImageTags) ?? []

  if (imageTags.length === 0) {
    return
  }

  const id = generateId()
  const token = generateToken(id, workspace, { linkId: id, guest: 'true' })

  for (const imageTag of imageTags) {
    const src = imageTag.attrs?.src
    if (src !== undefined && imageTag.attrs !== undefined) {
      imageTag.attrs.token = token
    }
  }
}