client/src/app/core/renderer/markdown.service.ts

   1 import * as MarkdownIt from 'markdown-it'
   2 import { buildVideoLink } from 'src/assets/player/utils'
   3 import { Injectable } from '@angular/core'
   4 import { HtmlRendererService } from './html-renderer.service'
   5
   6 type MarkdownParsers = {
   7   textMarkdownIt: MarkdownIt
   8   textWithHTMLMarkdownIt: MarkdownIt
   9
  10   enhancedMarkdownIt: MarkdownIt
  11   enhancedWithHTMLMarkdownIt: MarkdownIt
  12
  13   completeMarkdownIt: MarkdownIt
  14 }
  15
  16 type MarkdownConfig = {
  17   rules: string[]
  18   html: boolean
  19   escape?: boolean
  20 }
  21
  22 type MarkdownParserConfigs = {
  23   [id in keyof MarkdownParsers]: MarkdownConfig
  24 }
  25
  26 @Injectable()
  27 export class MarkdownService {
  28   static TEXT_RULES = [
  29     'linkify',
  30     'autolink',
  31     'emphasis',
  32     'link',
  33     'newline',
  34     'list'
  35   ]
  36   static TEXT_WITH_HTML_RULES = MarkdownService.TEXT_RULES.concat([ 'html_inline', 'html_block' ])
  37
  38   static ENHANCED_RULES = MarkdownService.TEXT_RULES.concat([ 'image' ])
  39   static ENHANCED_WITH_HTML_RULES = MarkdownService.TEXT_WITH_HTML_RULES.concat([ 'image' ])
  40
  41   static COMPLETE_RULES = MarkdownService.ENHANCED_WITH_HTML_RULES.concat([ 'block', 'inline', 'heading', 'paragraph' ])
  42
  43   private markdownParsers: MarkdownParsers = {
  44     textMarkdownIt: null,
  45     textWithHTMLMarkdownIt: null,
  46     enhancedMarkdownIt: null,
  47     enhancedWithHTMLMarkdownIt: null,
  48     completeMarkdownIt: null
  49   }
  50   private parsersConfig: MarkdownParserConfigs = {
  51     textMarkdownIt: { rules: MarkdownService.TEXT_RULES, html: false },
  52     textWithHTMLMarkdownIt: { rules: MarkdownService.TEXT_WITH_HTML_RULES, html: true, escape: true },
  53
  54     enhancedMarkdownIt: { rules: MarkdownService.ENHANCED_RULES, html: false },
  55     enhancedWithHTMLMarkdownIt: { rules: MarkdownService.ENHANCED_WITH_HTML_RULES, html: true, escape: true },
  56
  57     completeMarkdownIt: { rules: MarkdownService.COMPLETE_RULES, html: true }
  58   }
  59
  60   constructor (private htmlRenderer: HtmlRendererService) {}
  61
  62   textMarkdownToHTML (markdown: string, withHtml = false, withEmoji = false) {
  63     if (withHtml) return this.render('textWithHTMLMarkdownIt', markdown, withEmoji)
  64
  65     return this.render('textMarkdownIt', markdown, withEmoji)
  66   }
  67
  68   enhancedMarkdownToHTML (markdown: string, withHtml = false, withEmoji = false) {
  69     if (withHtml) return this.render('enhancedWithHTMLMarkdownIt', markdown, withEmoji)
  70
  71     return this.render('enhancedMarkdownIt', markdown, withEmoji)
  72   }
  73
  74   completeMarkdownToHTML (markdown: string) {
  75     return this.render('completeMarkdownIt', markdown, true)
  76   }
  77
  78   async processVideoTimestamps (html: string) {
  79     return html.replace(/((\d{1,2}):)?(\d{1,2}):(\d{1,2})/g, function (str, _, h, m, s) {
  80       const t = (3600 * +(h || 0)) + (60 * +(m || 0)) + (+(s || 0))
  81       const url = buildVideoLink({ startTime: t })
  82       return `<a class="video-timestamp" href="${url}">${str}</a>`
  83     })
  84   }
  85
  86   private async render (name: keyof MarkdownParsers, markdown: string, withEmoji = false) {
  87     if (!markdown) return ''
  88
  89     const config = this.parsersConfig[ name ]
  90     if (!this.markdownParsers[ name ]) {
  91       this.markdownParsers[ name ] = await this.createMarkdownIt(config)
  92
  93       if (withEmoji) {
  94         // TODO: write types
  95         const emoji = require('markdown-it-emoji/light')
  96         this.markdownParsers[ name ].use(emoji)
  97       }
  98     }
  99
 100     let html = this.markdownParsers[ name ].render(markdown)
 101     html = this.avoidTruncatedTags(html)
 102
 103     if (config.escape) return this.htmlRenderer.toSafeHtml(html)
 104
 105     return html
 106   }
 107
 108   private async createMarkdownIt (config: MarkdownConfig) {
 109     // FIXME: import('...') returns a struct module, containing a "default" field
 110     const MarkdownItClass: typeof import ('markdown-it') = (await import('markdown-it') as any).default
 111
 112     const markdownIt = new MarkdownItClass('zero', { linkify: true, breaks: true, html: config.html })
 113
 114     for (const rule of config.rules) {
 115       markdownIt.enable(rule)
 116     }
 117
 118     this.setTargetToLinks(markdownIt)
 119
 120     return markdownIt
 121   }
 122
 123   private setTargetToLinks (markdownIt: MarkdownIt) {
 124     // Snippet from markdown-it documentation: https://github.com/markdown-it/markdown-it/blob/master/docs/architecture.md#renderer
 125     const defaultRender = markdownIt.renderer.rules.link_open || function (tokens, idx, options, env, self) {
 126       return self.renderToken(tokens, idx, options)
 127     }
 128
 129     markdownIt.renderer.rules.link_open = function (tokens, index, options, env, self) {
 130       const token = tokens[index]
 131
 132       const targetIndex = token.attrIndex('target')
 133       if (targetIndex < 0) token.attrPush([ 'target', '_blank' ])
 134       else token.attrs[targetIndex][1] = '_blank'
 135
 136       const relIndex = token.attrIndex('rel')
 137       if (relIndex < 0) token.attrPush([ 'rel', 'noopener noreferrer' ])
 138       else token.attrs[relIndex][1] = 'noopener noreferrer'
 139
 140       // pass token to default renderer.
 141       return defaultRender(tokens, index, options, env, self)
 142     }
 143   }
 144
 145   private avoidTruncatedTags (html: string) {
 146     return html.replace(/\*\*?([^*]+)$/, '$1')
 147       .replace(/<a[^>]+>([^<]+)<\/a>\s*...((<\/p>)|(<\/li>)|(<\/strong>))?$/mi, '$1...')
 148       .replace(/\[[^\]]+\]\(([^\)]+)$/m, '$1')
 149       .replace(/\s?\[[^\]]+\]?[.]{3}<\/p>$/m, '...</p>')
 150   }
 151 }