package lsp

import (
    "strings"
    "testing"
)

func TestHasDoubleSemicolonTrigger(t *testing.T) {
    cases := []struct{
        line string
        want bool
    }{
        {";;todo; remove this", true},
        {"prefix ;;x; suffix", true},
        {";; spaced ;", false},
        {"no markers", false},
        {";;x ; space before close", false},
    }
    for _, tc := range cases {
        got := hasDoubleSemicolonTrigger(tc.line)
        if got != tc.want {
            t.Fatalf("hasDoubleSemicolonTrigger(%q)=%v want %v", tc.line, got, tc.want)
        }
    }
}

func TestCollectSemicolonMarkers(t *testing.T) {
    line := "keep ;ok; this and ;another; that"
    edits := collectSemicolonMarkers(line, 7)
    if len(edits) != 2 {
        t.Fatalf("expected 2 edits, got %d", len(edits))
    }
    // Validate the first edit aligns with ;ok;
    start := strings.Index(line, ";ok;")
    if start < 0 { t.Fatalf("test setup: missing ;ok;") }
    if edits[0].Range.Start.Line != 7 || edits[0].Range.Start.Character != start {
        t.Fatalf("first edit start got line=%d char=%d want line=7 char=%d", edits[0].Range.Start.Line, edits[0].Range.Start.Character, start)
    }
}

func TestPromptRemovalEditsForLine_WholeLine(t *testing.T) {
    line := ";;todo; remove this whole line"
    edits := promptRemovalEditsForLine(line, 3)
    if len(edits) != 1 {
        t.Fatalf("expected 1 whole-line edit, got %d", len(edits))
    }
    e := edits[0]
    if e.Range.Start.Line != 3 || e.Range.End.Line != 3 || e.Range.Start.Character != 0 || e.Range.End.Character != len(line) {
        t.Fatalf("unexpected range for whole-line removal: %+v", e.Range)
    }
}