Retain text around tables

Author: koddsson

src/paste-markdown-table.ts

@@ -15,18 +15,17 @@ export function uninstall(el: HTMLElement): void {
 function onDrop(event: DragEvent) {
   const transfer = event.dataTransfer
   if (!transfer) return
-
   if (hasFile(transfer)) return
 
-  const table = getTable(transfer)
-  if (!table) return
+  const textToPaste = generateText(transfer)
+  if (!textToPaste) return
 
   event.stopPropagation()
   event.preventDefault()
 
   const field = event.currentTarget
   if (field instanceof HTMLTextAreaElement) {
-    insertText(field, tableMarkdown(table))
+    insertText(field, textToPaste)
   }
 }
 
@@ -38,15 +37,15 @@ function onDragover(event: DragEvent) {
 function onPaste(event: ClipboardEvent) {
   if (!event.clipboardData) return
 
-  const table = getTable(event.clipboardData)
-  if (!table) return
+  const textToPaste = generateText(event.clipboardData)
+  if (!textToPaste) return
 
   event.stopPropagation()
   event.preventDefault()
 
   const field = event.currentTarget
   if (field instanceof HTMLTextAreaElement) {
-    insertText(field, tableMarkdown(table))
+    insertText(field, textToPaste)
   }
 }
 
@@ -82,27 +81,19 @@ function tableMarkdown(node: Element): string {
   return `\n${header}${body}\n\n`
 }
 
-function parseTable(html: string): HTMLElement | null {
-  const el = document.createElement('div')
-  el.innerHTML = html
-
-  const table = el.querySelector('table')
-  table?.remove()
-
-  // Reject tables that have extra text around them.
-  if (el?.textContent?.trim()) {
-    return null
-  }
+function generateText(transfer: DataTransfer): string | undefined {
+  if (Array.from(transfer.types).indexOf('text/html') === -1) return
 
-  return table
-}
+  const html = transfer.getData('text/html')
+  if (!/<table/i.test(html)) return
 
-function getTable(transfer: DataTransfer): HTMLElement | null {
-  if (Array.from(transfer.types).indexOf('text/html') === -1) return null
+  const el = document.createElement('div')
+  el.innerHTML = html
+  let table = el.querySelector('table')
+  table = !table || table.closest('[data-paste-markdown-skip]') ? null : table
+  if (!table) return
 
-  const html = transfer.getData('text/html')
-  if (!/<table/i.test(html)) return null
+  const formattedTable = tableMarkdown(table)
 
-  const table = parseTable(html)
-  return !table || table.closest('[data-paste-markdown-skip]') ? null : table
+  return html.replace(/<meta.*?>/, '').replace(/<table.*<\/table>/, `\n${formattedTable}`)
 }