completion: only apply leading indentation for ';;text;' prompts; not for ';text;'

1 files changed, 36 insertions, 0 deletions

diff --git a/internal/lsp/handlers.go b/internal/lsp/handlers.go
index 8c4fd51..63b32d0 100644
--- a/internal/lsp/handlers.go
+++ b/internal/lsp/handlers.go
@@ -837,6 +837,15 @@ func (s *Server) tryLLMCompletion(p CompletionParams, above, current, below, fun
     }
     if cleaned != "" { cleaned = stripDuplicateAssignmentPrefix(current[:p.Position.Character], cleaned) }
     if cleaned != "" { cleaned = stripDuplicateGeneralPrefix(current[:p.Position.Character], cleaned) }
+    // Preserve the current line's leading indentation only for double-semicolon
+    // inline prompts (";;text;"), since strict ";text;" replacements already
+    // occur in-place without affecting leading indentation.
+    if cleaned != "" && hasDoubleSemicolonTrigger(current) {
+        indent := leadingIndent(current)
+        if indent != "" {
+            cleaned = applyIndent(indent, cleaned)
+        }
+    }
     if cleaned == "" {
         return nil, false, false
     }
@@ -1225,6 +1234,33 @@ func lineHasInlinePrompt(line string) bool {
     return hasDoubleSemicolonTrigger(line)
 }
 
+// leadingIndent returns the run of leading spaces/tabs from the provided line.
+func leadingIndent(line string) string {
+    i := 0
+    for i < len(line) {
+        if line[i] == ' ' || line[i] == '\t' {
+            i++
+            continue
+        }
+        break
+    }
+    if i == 0 { return "" }
+    return line[:i]
+}
+
+// applyIndent prefixes each non-empty line of suggestion with the given indent
+// unless it already starts with that indent.
+func applyIndent(indent, suggestion string) string {
+    if indent == "" || suggestion == "" { return suggestion }
+    lines := splitLines(suggestion)
+    for i, ln := range lines {
+        if strings.TrimSpace(ln) == "" { continue }
+        if strings.HasPrefix(ln, indent) { continue }
+        lines[i] = indent + ln
+    }
+    return strings.Join(lines, "\n")
+}
+
 // isBareDoubleSemicolon reports whether the line contains a standalone
 // double-semicolon marker with no inline content (";;" possibly with only
 // whitespace after it). It explicitly excludes the valid form ";;text;".