Add expand line selection logic

Author: paxos

src/index.ts

@@ -433,7 +433,22 @@ function styleSelectedText(textarea: HTMLTextAreaElement, styleArgs: StyleArgs)
   insertText(textarea, result)
 }
 
-export function selectionIndexForLine(lines: string[], line: number): SelectionRange | null {
+function expandSelectionToLine(textarea: HTMLTextAreaElement) {
+  const lines = textarea.value.split('\n')
+  let counter = 0
+  for (let index = 0; index < lines.length; index++) {
+    const lineLength = lines[index].length + 1
+    if (textarea.selectionStart >= counter && textarea.selectionStart <= counter + lineLength) {
+      textarea.selectionStart = counter
+    }
+    if (textarea.selectionEnd > counter && textarea.selectionEnd <= counter + lineLength) {
+      textarea.selectionEnd = counter + lineLength
+    }
+    counter += lineLength
+  }
+}
+
+function selectionIndexForLine(lines: string[], line: number): SelectionRange | null {
   let counter = 0
   for (let index = 0; index < lines.length; index++) {
     if (index === line) {
@@ -636,7 +651,9 @@ function listStyle(textarea: HTMLTextAreaElement, style: StyleArgs): SelectionRa
 
   const prefix = '- '
 
-  //let selectedText = expandSelectedText(textarea, prefix, '', style.multiline)
+  // Expand selection to full lines
+
+  expandSelectionToLine(textarea)
 
   // Style only the selected line
   if (noInitialSelection) {
@@ -648,10 +665,10 @@ function listStyle(textarea: HTMLTextAreaElement, style: StyleArgs): SelectionRa
 
     // Select whole line
     const range = selectionIndexForLine(lines, currentLine)
-    if (range) {
-      textarea.selectionStart = range.selectionStart ?? 0
-      textarea.selectionEnd = range.selectionEnd ?? 0
-    }
+    // if (range) {
+    //   textarea.selectionStart = range.selectionStart ?? 0
+    //   textarea.selectionEnd = range.selectionEnd ?? 0
+    // }
 
     // line with caret
     //lines[linesBefore.length - 1] = prefix + linesBefore[linesBefore.length - 1]
@@ -661,7 +678,7 @@ function listStyle(textarea: HTMLTextAreaElement, style: StyleArgs): SelectionRa
     // if (style.surroundWithNewlines) {
     const {newlinesToAppend, newlinesToPrepend} = newlinesToSurroundSelectedText(textarea)
     selectionStart = selectionStart + prefix.length + 1
-    selectionEnd = selectionStart
+    selectionEnd = selectionEnd + prefix.length + 1
     text = newlinesToAppend + text + newlinesToPrepend
 
     return {text, selectionStart, selectionEnd}