TipTap JSONContent mapLoops

tiptap-map-loops.ts

export type PdfMap = { [key: string]: string | boolean | PdfMap[] }

export enum JSONContentType {
  Heading = "heading",
  Paragraph = "paragraph",
  Text = "text",
  BulletList = "bulletList",
  OrderedList = "orderedList",
  ListItem = "listItem",
  HardBreak = "hardBreak",
  HorizontalRule = "horizontalRule",
  Section = "section"
}

export class PdfService {
  static mapLoops(json: JSONContent, map: PdfMap, result?: JSONContent): JSONContent {
    if (!result) {
      result = {
        type: "doc",
        content: []
      }
    }

    json.content?.forEach((c) => {
      const varName = c.attrs?.varName

      if ((c.type === JSONContentType.ListItem || c.type === JSONContentType.Section) && varName && map[varName] && Array.isArray(map[varName])) {
        // Alter content with current demultiplied
        result!.content?.push(
          ...(map[varName] as PdfMap[]).map(
            (li) =>
              ({
                ...c,
                attrs: c.attrs ? this.removeAttr(c.attrs, "varName") : undefined,
                content: this.mapLoops(c, { ...map, ...li }, { ...c, content: [] }).content
              } as JSONContent)
          )
        )
      } else if (c.type === JSONContentType.Section && varName) {
        if (map[varName] === undefined) {
          result!.content?.push(this.mapLoops(c, map, { ...c, content: [] }))
        } else if (map[varName]) {
          // Add section
          result!.content?.push(this.mapLoops(c, map, { ...c, attrs: c.attrs ? this.removeAttr(c.attrs, "varName") : undefined, content: [] }))
        } // else no add
      } else if (c.type === JSONContentType.Text) {
        // Content replacement
        let varText = c.text + ""
        if (c.text) {
          varText = this.mapReplace(varText, map)
        }

        // Add node
        result!.content?.push({ ...c, text: varText })
      } else {
        // All others, add contents in a fresh node
        result!.content?.push(this.mapLoops(c, map, { ...c, content: [] }))
      }
    })

    return result
  }

  static MapReplace(str: string, map: PdfMap): string {
    Object.keys(map).forEach((k) => {
      if (!Array.isArray(map[k]) && map[k]) {
        // Only non empty text nodes are allowed (keep placeholder for empty nodes)
        str = str.replace(new RegExp("\\[" + k + "\\]", "g"), map[k] as string)
      }
    })

    return str
  }

  static removeAttr(attrs: Record<string, any>, attrToRemove: string): Record<string, any> {
    const newAttrs: Record<string, any> = {}
    for (const key in attrs) {
      if (key !== attrToRemove) {
        newAttrs[key] = attrs[key]
      }
    }
    return newAttrs
  }
}