external-link-icon.ts 1.2 KB

1234567891011121314151617181920212223242526272829303132
  1. import type MarkdownIt from 'markdown-it'
  2. import type Renderer from 'markdown-it/lib/renderer'
  3. export default (md: MarkdownIt): void => {
  4. const renderToken: Renderer.RenderRule = (tokens, idx, options, env, self) => self.renderToken(tokens, idx, options)
  5. const defaultLinkOpenRenderer = md.renderer.rules.link_open || renderToken
  6. const defaultLinkCloseRenderer = md.renderer.rules.link_close || renderToken
  7. let isExternalLink = false
  8. md.renderer.rules.link_open = (tokens, idx, options, env, self) => {
  9. const token = tokens[idx]
  10. const href = token.attrGet('href')
  11. if (href) {
  12. token.attrJoin('class', 'vp-link')
  13. if (/^((ht|f)tps?):\/\/?/.test(href)) {
  14. isExternalLink = true
  15. }
  16. }
  17. return defaultLinkOpenRenderer(tokens, idx, options, env, self)
  18. }
  19. md.renderer.rules.link_close = (tokens, idx, options, env, self) => {
  20. if (isExternalLink) {
  21. isExternalLink = false
  22. return `<i-ri-external-link-line class="link-icon" />${self.renderToken(tokens, idx, options)}`
  23. }
  24. return defaultLinkCloseRenderer(tokens, idx, options, env, self)
  25. }
  26. }