markdown-actions 1.0.3 → 1.0.4

package/src/index.js

@@ -161,113 +161,77 @@ export function insertHeader(textarea, level = 1, toggle = false) {
   if (!textarea || textarea.disabled || textarea.readOnly) return
   if (level < 1 || level > 6) level = 1
   
-  // Save original cursor position before any expansion
-  const originalStart = textarea.selectionStart
-  const originalEnd = textarea.selectionEnd
-  const noInitialSelection = originalStart === originalEnd
-  
   const headerKey = `header${level === 1 ? '1' : level}`
   const style = mergeWithDefaults(FORMATS[headerKey] || FORMATS.header1)
   
-  // Select current line
-  const value = textarea.value
-  let lineStart = originalStart
-  let lineEnd = originalStart
-  
-  // Find line boundaries
-  while (lineStart > 0 && value[lineStart - 1] !== '\n') {
-    lineStart--
-  }
-  while (lineEnd < value.length && value[lineEnd] !== '\n') {
-    lineEnd++
-  }
-  
-  textarea.selectionStart = lineStart
-  textarea.selectionEnd = lineEnd
-  
-  const currentLine = value.slice(lineStart, lineEnd)
-  
-  // Check if we're removing or changing existing header
-  const existingHeaderMatch = currentLine.match(/^(#{1,6})\s*/)
-  const existingPrefixLength = existingHeaderMatch ? existingHeaderMatch[0].length : 0
-  const existingLevel = existingHeaderMatch ? existingHeaderMatch[1].length : 0
-  
-  // Remove any existing header formatting
-  const cleanedLine = currentLine.replace(/^#{1,6}\s*/, '')
-  
-  let newLine
-  let isRemoving = false
-  
-  // If toggle mode and current line has the same header level, remove it
-  if (toggle && existingLevel === level) {
-    newLine = cleanedLine
-    isRemoving = true
-  } else {
-    // Apply new header
-    newLine = style.prefix + cleanedLine
-  }
-  
-  let selectionStart, selectionEnd
+  // Save original positions before any operations
+  const originalStart = textarea.selectionStart
+  const originalEnd = textarea.selectionEnd
   
-  if (noInitialSelection) {
-    // No selection - just restore cursor position
-    const cursorOffsetInLine = originalStart - lineStart
-    
-    if (isRemoving) {
-      // We're removing the header
-      if (cursorOffsetInLine < existingPrefixLength) {
-        // Cursor was in the header prefix, move to start of line
-        selectionStart = lineStart
+  // Use applyLineOperation for consistent behavior
+  const result = applyLineOperation(
+    textarea,
+    (ta) => {
+      const currentLine = ta.value.slice(ta.selectionStart, ta.selectionEnd)
+      
+      // Check if we're removing or changing existing header
+      const existingHeaderMatch = currentLine.match(/^(#{1,6})\s*/)
+      const existingLevel = existingHeaderMatch ? existingHeaderMatch[1].length : 0
+      
+      // Remove any existing header formatting
+      const cleanedLine = currentLine.replace(/^#{1,6}\s*/, '')
+      
+      let newLine
+      
+      // If toggle mode and current line has the same header level, remove it
+      if (toggle && existingLevel === level) {
+        newLine = cleanedLine
       } else {
-        // Cursor was in the text, move back by prefix length
-        selectionStart = originalStart - existingPrefixLength
+        // Apply new header
+        newLine = style.prefix + cleanedLine
       }
-    } else if (existingPrefixLength > 0) {
-      // We're replacing an existing header
-      if (cursorOffsetInLine < existingPrefixLength) {
-        // Cursor was in the header prefix, keep it at the same position
-        selectionStart = lineStart + Math.min(cursorOffsetInLine, style.prefix.length)
-      } else {
-        // Cursor was in the text, adjust by the difference in prefix lengths
-        const prefixDiff = style.prefix.length - existingPrefixLength
-        selectionStart = originalStart + prefixDiff
+      
+      return {
+        text: newLine,
+        selectionStart: ta.selectionStart,
+        selectionEnd: ta.selectionEnd
+      }
+    },
+    {
+      prefix: style.prefix,
+      // Custom handler for header-specific logic
+      adjustSelection: (isRemoving, selStart, selEnd, lineStart) => {
+        const currentLine = textarea.value.slice(lineStart, textarea.selectionEnd)
+        const existingHeaderMatch = currentLine.match(/^(#{1,6})\s*/)
+        const existingPrefixLength = existingHeaderMatch ? existingHeaderMatch[0].length : 0
+        const existingLevel = existingHeaderMatch ? existingHeaderMatch[1].length : 0
+        
+        // Check if we're actually removing (toggle off)
+        const actuallyRemoving = toggle && existingLevel === level
+        
+        if (actuallyRemoving) {
+          // Removing the header entirely
+          return {
+            start: Math.max(selStart - existingPrefixLength, lineStart),
+            end: Math.max(selEnd - existingPrefixLength, lineStart)
+          }
+        } else if (existingPrefixLength > 0) {
+          // Replacing existing header with new one
+          const prefixDiff = style.prefix.length - existingPrefixLength
+          return {
+            start: selStart + prefixDiff,
+            end: selEnd + prefixDiff
+          }
+        } else {
+          // Adding new header
+          return {
+            start: selStart + style.prefix.length,
+            end: selEnd + style.prefix.length
+          }
+        }
       }
-    } else {
-      // We're adding a new header
-      // Move cursor by the length of the new prefix
-      selectionStart = originalStart + style.prefix.length
-    }
-    
-    // Ensure cursor doesn't go past end of line
-    selectionStart = Math.min(selectionStart, lineStart + newLine.length)
-    selectionEnd = selectionStart
-  } else {
-    // Had a selection - preserve it with adjustment
-    if (isRemoving) {
-      // When removing header, shift selection back
-      selectionStart = Math.max(originalStart - existingPrefixLength, lineStart)
-      selectionEnd = Math.max(originalEnd - existingPrefixLength, lineStart)
-    } else if (existingPrefixLength > 0) {
-      // We're replacing an existing header
-      const prefixDiff = style.prefix.length - existingPrefixLength
-      selectionStart = originalStart + prefixDiff
-      selectionEnd = originalEnd + prefixDiff
-    } else {
-      // We're adding a new header
-      selectionStart = originalStart + style.prefix.length
-      selectionEnd = originalEnd + style.prefix.length
     }
-    
-    // Ensure selection doesn't go past end of line
-    selectionStart = Math.min(selectionStart, lineStart + newLine.length)
-    selectionEnd = Math.min(selectionEnd, lineStart + newLine.length)
-  }
-  
-  const result = {
-    text: newLine,
-    selectionStart,
-    selectionEnd
-  }
+  )
   
   insertText(textarea, result)
 }