Function enhanceCodeBlockCopy

packages/app-core/src/lib/code-block-copy.ts:18–45

(
  root: ParentNode,
  options: CodeBlockEnhanceOptions = {}
)

}

export function enhanceCodeBlockCopy(
  root: ParentNode,
  options: CodeBlockEnhanceOptions = {}
): void {
  const blocks = Array.from(root.querySelectorAll<HTMLPreElement>('pre'))
  const storageKey = options.notePath ? codeBlockFoldStorageKey(options.notePath) : null
  let blockIndex = 0

  for (const pre of blocks) {
    const code = pre.firstElementChild
    if (!(code instanceof HTMLElement) || code.tagName.toLowerCase() !== 'code') {
      continue
    }

    const index = blockIndex
    blockIndex += 1
    const wrapper = ensureCodeBlockWrapper(pre)
    wrapper.setAttribute(CODE_BLOCK_INDEX_ATTR, String(index))
    if (storageKey) wrapper.setAttribute(CODE_BLOCK_STORAGE_KEY_ATTR, storageKey)
    else wrapper.removeAttribute(CODE_BLOCK_STORAGE_KEY_ATTR)

    ensureCodeBlockToolbar(wrapper, pre)
    ensureCodeBlockSummary(wrapper, code)

    const persisted = storageKey ? readPersistedFoldState(storageKey, index) : null
    applyCodeBlockFoldState(wrapper, persisted ?? false)
  }
}

export function getCodeBlockTextForCopyButton(button: Element): string | null {
  const block = button.closest(`.${CODE_BLOCK_CLASS}`)