amis/scripts/markdownPlugin.ts

import type {Plugin} from 'vite';
import * as marked from 'marked';
import * as prism from 'prismjs';
import * as yaml from 'js-yaml';
let loadLanguages = require('prismjs/components/');
loadLanguages([
  'bash',
  'javascript',
  'java',
  'python',
  'jsx',
  'tsx',
  'css',
  'markup',
  'json'
]);
const rYml = /^\s*---([\s\S]*?)---\s/;
const renderer = new marked.Renderer();
(marked as any).setOptions({
  renderer: renderer,
  gfm: true,
  tables: true,
  breaks: false,
  pedantic: false,
  smartLists: true,
  smartypants: false
});

// Synchronous highlighting with prism.js
(marked as any).setOptions({
  highlight: function (code: string, lang: string) {
    if (lang) {
      return prism.highlight(code, prism.languages[lang], lang);
    } else {
      return code;
    }
  }
});

// renderer.table = function(header, body) {
//   return '<table class="table table-striped">\n'
//     + '<thead>\n'
//     + header
//     + '</thead>\n'
//     + '<tbody>\n'
//     + body
//     + '</tbody>\n'
//     + '</table>\n';
// };

renderer.link = function (href: string, title: string, text: string) {
  if ((this as any).options.sanitize) {
    try {
      var prot = decodeURIComponent(unescape(href))
        .replace(/[^\w:]/g, '')
        .toLowerCase();
    } catch (e) {
      return '';
    }
    if (prot.indexOf('javascript:') === 0 || prot.indexOf('vbscript:') === 0) {
      return '';
    }
  }
  if (href && href[0] === '#') {
    href =
      '#' +
      encodeURIComponent(
        href
          .substring(1)
          .toLowerCase()
          .replace(/[^\u4e00-\u9fa5_a-zA-Z0-9]+/g, '-')
      );
  }

  var out = '<a href="' + href + '"';
  if (title) {
    out += ' title="' + title + '"';
  }
  out += '>' + text + '</a>';
  return out;
};

function markdown2js(content: string, file: string) {
  var m = rYml.exec(content);
  var info: any = {};
  if (m && m[1]) {
    info = (yaml as any).load(m[1]);
    content = content.substring(m[0].length);
  }

  var toc: any = {
    label: '目录',
    type: 'toc',
    children: [],
    level: 0
  };
  var stack = [toc];

  renderer.heading = function (text, level) {
    var escapedText = encodeURIComponent(
      text.toLowerCase().replace(/[^\u4e00-\u9fa5_a-zA-Z0-9]+/g, '-')
    );

    if (level < 5) {
      var menu = {
        label: text,
        fragment: escapedText,
        fullPath: '#' + escapedText,
        level: level
      };

      while (stack.length && stack[0].level >= level) {
        stack.shift();
      }

      stack[0].children = stack[0].children || [];
      stack[0].children.push(menu);

      stack.unshift(menu);
    }

    var anchor =
      '<a class="anchor" name="' +
      escapedText +
      '" href="#' +
      escapedText +
      '" aria-hidden="true"><svg aria-hidden="true" class="octicon octicon-link" height="16" version="1.1" viewBox="0 0 16 16" width="16"><path d="M4 9h1v1H4c-1.5 0-3-1.69-3-3.5S2.55 3 4 3h4c1.45 0 3 1.69 3 3.5 0 1.41-.91 2.72-2 3.25V8.59c.58-.45 1-1.27 1-2.09C10 5.22 8.98 4 8 4H4c-.98 0-2 1.22-2 2.5S3 9 4 9zm9-3h-1v1h1c1 0 2 1.22 2 2.5S13.98 12 13 12H9c-.98 0-2-1.22-2-2.5 0-.83.42-1.64 1-2.09V6.25c-1.09.53-2 1.84-2 3.25C6 11.31 7.55 13 9 13h4c1.45 0 3-1.69 3-3.5S14.5 6 13 6z"></path></svg></a>';

    return '<h' + level + '>' + anchor + text + '</h' + level + '>';

    // return '<h' + level + '><a name="' +
    //   escapedText +
    //   '" class="anchor" href="#' +
    //   escapedText +
    //   '"><span class="header-link"></span></a>' +
    //   text + '</h' + level + '>';
  };

  const placeholder: any = {};
  let index = 1;

  content = content
    .replace(/\!\!\!include\s*\(([^\)]+?)\)\!\!\!/g, (_, val) => {
      // todo
      // const result =  null;

      // if (result) {
      //   // 暂时不支持嵌套 include
      //   return result.file.getContent();
      // }

      return _;
    })
    .replace(
      /```(schema|html)(?::(.*?))?\n([\s\S]*?)```/g,
      function (_, lang, attr, code) {
        const setting: any = {};
        attr &&
          attr.split(/\s+/).forEach(function (item: string) {
            var parts = item.split('=');

            if (parts[1] && /^('|").*\1/.test(parts[1])) {
              parts[1] = parts[1].substring(1, parts[1].length - 1);
            }

            setting[parts[0]] = parts[1] ? decodeURIComponent(parts[1]) : '';

            if (parts[0] === 'height') {
              setting.height = parseInt(setting.height, 10) /*编辑器的高度*/;
              attr = attr.replace(item, `height="${setting.height}"`);
            }
          });

        // placeholder[index] = `<iframe class="doc-iframe" width="100%" height="${setting.height || 200}px" frameBorder="0" src="/play?code=${encodeURIComponent(code)}&scope=${encodeURIComponent(setting.scope)}"></iframe>`;
        if (lang === 'html') {
          if (~code.indexOf('<html') || ~code.indexOf('<link')) {
            return _;
          }

          placeholder[
            index
          ] = `<!--amis-preview-start--><div class="amis-doc"><div class="preview">${code}</div><pre><code class="lang-html">${prism.highlight(
            code
              .replace(/"data:(\w+\/\w+);.*?"/g, '"data:$1; ..."')
              .replace(/<svg([^>]*)>[\s\S]*?<\/svg>/g, '<svg$1>...</svg>')
              .replace(/class="([^"]*?)\.\.\.([^"]*?)"/g, 'class="$1..."'),
            prism.languages[lang],
            lang
          )}</code></pre></div><!--amis-preview-end-->`;
        } else {
          placeholder[
            index
          ] = `<!--amis-preview-start--><div class="amis-preview" style="min-height: ${setting.height}px"><script type="text/schema" ${attr}>${code}</script></div><!--amis-preview-end-->`;
        }

        return `[[${index++}]]`;
      }
    );

  content = (marked as any)
    .parse(content)
    .replace(/<p>\[\[(\d+)\]\]<\/p>/g, function (_: string, id: string) {
      return placeholder[id] || '';
    });

  // content = global.fis ? fis.compile.partial(content, file, 'html') : content;
  // + `\n\n<div class="m-t-lg b-l b-info b-3x wrapper bg-light dk">文档内容有误？欢迎大家一起来编写，文档地址：<i class="fa fa-github"></i><a href="https://github.com/baidu/amis/tree/master${file.subpath}">${file.subpath}</a>。</div>`;
  info.html =
    '<div class="markdown-body">' +
    content.replace(
      /<\!\-\-amis\-preview\-(start|end)\-\-\>/g,
      function (_, type) {
        return type === 'start' ? '</div>' : '<div class="markdown-body">';
      }
    ) +
    '</div>';
  info.toc = toc;

  return 'export default ' + JSON.stringify(info, null, 2) + ';';
}

export default function markdownPlugin(options: {} = {}): Plugin {
  return {
    name: 'markdown-as-js',
    enforce: 'pre',
    apply: 'serve',
    transform(code: string, id: string) {
      // if (id.endsWith('.scss') && /\/_[^\/]+\.scss$/.test(id)) {
      //   const markdowns: Array<string> = [];

      //   code.replace(
      //     /\/\*\!markdown\n([\s\S]+?)\*\//g,
      //     (_: string, md: string) => {
      //       markdowns.push(md.trim());
      //       return _;
      //     }
      //   );

      //   if (markdowns.length) {
      //     return {code: markdown2js(markdowns.join('\n'), id) as string};
      //   }

      //   return null;
      // }

      if (!id.endsWith('.md')) return null;

      return {code: markdown2js(code, id) as string};
    }
  };
}