All files / client/util CrowiRenderer.ts

0% Statements 0/54
0% Branches 0/20
0% Functions 0/7
0% Lines 0/52

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161                                                                                                                                                                                                                                                                                                                                 
import Crowi from './Crowi'
 
import marked from 'marked'
import hljs from 'highlight.js'
 
import MarkdownFixer from './PreProcessor/MarkdownFixer'
import Linker from './PreProcessor/Linker'
import ImageExpander from './PreProcessor/ImageExpander'
 
import Emoji from './PostProcessor/Emoji'
import Mathjax from './PostProcessor/Mathjax'
 
import Tsv2Table from './LangProcessor/Tsv2Table'
import Template from './LangProcessor/Template'
import PlantUML from './LangProcessor/PlantUML'
 
interface PreProcessor {
  process: (markdown: string, dom: HTMLElement | undefined) => string
}
 
interface PostProcessor {
  process: (markdown: string, dom: HTMLElement | undefined) => string
}
 
interface LangProcessor {
  process: (code: string, lang: string) => string
}
 
export default class CrowiRenderer {
  crowi: Crowi
 
  preProcessors: PreProcessor[]
 
  postProcessors: PostProcessor[]
 
  langProcessors: { [key: string]: LangProcessor }
 
  constructor(crowi: Crowi) {
    this.crowi = crowi
 
    this.preProcessors = [new MarkdownFixer(), new Linker(), new ImageExpander()]
    this.postProcessors = [new Emoji(), new Mathjax(crowi)]
 
    this.langProcessors = {
      tsv: new Tsv2Table(crowi),
      'tsv-h': new Tsv2Table(crowi, { header: true }),
      template: new Template(crowi),
      plantuml: new PlantUML(crowi),
    }
 
    this.parseMarkdown = this.parseMarkdown.bind(this)
    this.codeRenderer = this.codeRenderer.bind(this)
  }
 
  preProcess(markdown: string, dom: HTMLElement | undefined = undefined) {
    for (let i = 0; i < this.preProcessors.length; i++) {
      if (!this.preProcessors[i].process) {
        continue
      }
      markdown = this.preProcessors[i].process(markdown, dom)
    }
    return markdown
  }
 
  postProcess(html: string, dom: HTMLElement | undefined = undefined) {
    for (let i = 0; i < this.postProcessors.length; i++) {
      if (!this.postProcessors[i].process) {
        continue
      }
      html = this.postProcessors[i].process(html, dom)
    }
 
    return html
  }
 
  codeRenderer(code: string, lang: string, escaped = false) {
    let result = ''
    let hl
 
    if (lang) {
      const langAndFn = lang.split(':')
      const langPattern = langAndFn[0]
      const langFn = langAndFn[1] || null
      if (this.langProcessors[langPattern]) {
        return this.langProcessors[langPattern].process(code, lang)
      }
 
      try {
        hl = hljs.highlight(langPattern, code)
        result = hl.value
        escaped = true
      } catch (e) {
        result = code
      }
 
      result = escape ? result : Crowi.escape(result, true)
 
      let citeTag = ''
      if (langFn) {
        citeTag = `<cite>${langFn}</cite>`
      }
      return `<pre class="wiki-code wiki-lang">${citeTag}<code class="lang-${lang}">${result}\n</code></pre>\n`
    }
 
    // no lang specified
    return `<pre class="wiki-code"><code>${Crowi.escape(code, true)}\n</code></pre>`
  }
 
  parseMarkdown(markdown: string) {
    let parsed = ''
 
    const markedRenderer = new marked.Renderer()
    markedRenderer.code = this.codeRenderer
 
    try {
      // TODO
      marked.setOptions({
        gfm: true,
        tables: true,
        breaks: true,
        pedantic: false,
        sanitize: false,
        smartLists: true,
        smartypants: false,
        renderer: markedRenderer,
      })
 
      // override
      // @ts-ignore
      marked.Lexer.lex = function(src: string, options: marked.MarkedOptions) {
        var lexer = new marked.Lexer(options)
 
        // this is maybe not an official way
        // @ts-ignore: Unofficial hack
        if (lexer.rules) {
          // @ts-ignore: Unofficial hack
          lexer.rules.fences = /^ *(`{3,}|~{3,})[ .]*([^\r\n]+)? *\n([\s\S]*?)\s*\1 *(?:\n+|$)/
        }
 
        return lexer.lex(src)
      }
 
      parsed = marked(markdown)
    } catch (e) {
      console.log(e, e.stack)
    }
 
    return parsed
  }
 
  render(markdown: string, dom: HTMLElement | undefined = undefined) {
    let html = ''
 
    markdown = this.preProcess(markdown, dom)
    html = this.parseMarkdown(markdown)
    html = this.postProcess(html, dom)
 
    return html
  }
}