packages/muyajs/lib/parser/render/renderBlock/renderIcon.js - develop - githubmirror/marktext

packages/muyajs/lib/parser/render/renderBlock/renderIcon.js
163 строки4 KB
Ran Luo
chore: convert repo to pnpm monorepo (packages/desktop, muyajs, website) (#4302)
29 май 2026, 05:25
29 май 2026, 05:25565bfcd
import { h } from '../snabbdom'
import { CLASS_OR_ID } from '../../../config'
import paragraphIcon from '../../../assets/pngicon/paragraph/2.png'
import htmlIcon from '../../../assets/pngicon/html/2.png'
import hrIcon from '../../../assets/pngicon/horizontal_line/2.png'
import frontMatterIcon from '../../../assets/pngicon/front_matter/2.png'
import header1Icon from '../../../assets/pngicon/heading_1/2.png'
import header2Icon from '../../../assets/pngicon/heading_2/2.png'
import header3Icon from '../../../assets/pngicon/heading_3/2.png'
import header4Icon from '../../../assets/pngicon/heading_4/2.png'
import header5Icon from '../../../assets/pngicon/heading_5/2.png'
import header6Icon from '../../../assets/pngicon/heading_6/2.png'
import newTableIcon from '../../../assets/pngicon/new_table/2.png'
import bulletListIcon from '../../../assets/pngicon/bullet_list/2.png'
import codeIcon from '../../../assets/pngicon/code/2.png'
import quoteIcon from '../../../assets/pngicon/quote_block/2.png'
import todoListIcon from '../../../assets/pngicon/todolist/2.png'
import mathblockIcon from '../../../assets/pngicon/math/2.png'
import orderListIcon from '../../../assets/pngicon/order_list/2.png'
import flowchartIcon from '../../../assets/pngicon/flowchart/2.png'
import sequenceIcon from '../../../assets/pngicon/sequence/2.png'
import plantumlIcon from '../../../assets/pngicon/plantuml/2.png'
import mermaidIcon from '../../../assets/pngicon/mermaid/2.png'
import vegaIcon from '../../../assets/pngicon/chart/2.png'
import footnoteIcon from '../../../assets/pngicon/footnote/2.png'
import formatLink from '../../../assets/pngicon/format_link/2.png'

const FUNCTION_TYPE_HASH = {
  mermaid: mermaidIcon,
  flowchart: flowchartIcon,
  sequence: sequenceIcon,
  plantuml: plantumlIcon,
  'vega-lite': vegaIcon,
  table: newTableIcon,
  html: htmlIcon,
  multiplemath: mathblockIcon,
  fencecode: codeIcon,
  indentcode: codeIcon,
  frontmatter: frontMatterIcon,
  footnote: footnoteIcon
}

export default function renderIcon(block, t) {
  if (block.parent) {
    console.error(t('editor.onlyTopBlockCanRenderIcon'))
  }
  const { type, functionType, listType } = block
  const selector = `a.${CLASS_OR_ID.AG_FRONT_ICON}`
  let icon = null
  let isCopyLink = false

  switch (type) {
    case 'p': {
      icon = paragraphIcon
      break
    }
    case 'figure':
    case 'pre': {
      icon = FUNCTION_TYPE_HASH[functionType]
      if (!icon) {
        console.warn(t('editor.unhandledFunctionType', { functionType }))
        icon = paragraphIcon
      }
      break
    }
    case 'ul': {
      if (listType === 'task') {
        icon = todoListIcon
      } else {
        icon = bulletListIcon
      }
      break
    }
    case 'ol': {
      icon = orderListIcon
      break
    }
    case 'blockquote': {
      icon = quoteIcon
      break
    }
    case 'h1': {
      icon = header1Icon
      isCopyLink = true
      break
    }
    case 'h2': {
      icon = header2Icon
      isCopyLink = true
      break
    }
    case 'h3': {
      icon = header3Icon
      isCopyLink = true
      break
    }
    case 'h4': {
      icon = header4Icon
      isCopyLink = true
      break
    }
    case 'h5': {
      icon = header5Icon
      isCopyLink = true
      break
    }
    case 'h6': {
      icon = header6Icon
      isCopyLink = true
      break
    }
    case 'hr': {
      icon = hrIcon
      break
    }
    default:
      icon = paragraphIcon
      break
  }

  const iconVnode = h(
    'i.icon.ag-front-icon-button',
    h(
      'img.icon-inner',
      {
        attrs: {
          src: icon
        }
      },
      ''
    )
  )

  const linkCopyIcon = h(
    'i.icon.ag-copy-header-link',
    {
      style: {
        left: '-30px'
      }
    },
    h(
      'img.icon-inner',
      {
        attrs: {
          src: formatLink
        }
      },
      ''
    )
  )

  const iconsToRender = isCopyLink ? [iconVnode, linkCopyIcon] : [iconVnode]

  return h(
    selector,
    {
      attrs: {
        contenteditable: 'false'
      }
    },
    iconsToRender
  )
}