chore: add loop directives now showup only in loops

Author: SerhiiCho

.gitignore

@@ -1,2 +1,3 @@
 log.txt
 main
+Session.vim

CHANGELOG.md

@@ -1,14 +1,17 @@
 # Release Notes
 
+## v0.4.0 (2025-11-07)
+- ✨ Loop directives like `@break`, `@continue`, `@breakIf`, `@continueIf` are now getting suggested only inside of loops.
+
 ## v0.3.1 (2025-06-14)
-- 🐛 Update LSP textwire to the latest version which will fix `@break($1)` snippet to `@break`
+- 🐛 Updated LSP textwire to the latest version which will fix `@break($1)` snippet to `@break`
 
 ## v0.3.0 (2025-06-14)
-- 🐛 Update LSP textwire to the latest version which will fix crashing LSP when you make a syntax error in your Textwire code
+- 🐛 Updated LSP textwire to the latest version which will fix crashing LSP when you make a syntax error in your Textwire code
 - ✨ Autocomplete snippets that appear after you hit enter are now more complex. Instead of simple autocomple like `@if` you know get the full if statement and the cursor placed inside condition. It allows you to hit tab to move to the next place in a snippet
 
 ## v0.2.0 (2025-05-30)
-- ✨ Add autocompletion for loop object. Now if you type `loop.` inside of a loop, it will show available properties on the object
+- ✨ Added autocompletion for loop object. Now if you type `loop.` inside of a loop, it will show available properties on the object
 
 ## v0.1.4 (2025-05-17)
 - ✨ Autocomplete suggestions show code example with syntax highlighting. Before, it was just displayed as text

analysis/completion.go

@@ -2,6 +2,7 @@ package analysis
 
 import (
 	"regexp"
+	"slices"
 	"strings"
 
 	"github.com/textwire/lsp/internal/logger"
@@ -35,18 +36,9 @@ func (s *State) Completion(id int, uri string, pos lsp.Position) (lsp.Completion
 	var err error
 
 	if directiveMatch != nil {
-		completionItems, err = completions.GetDirectives(locale)
+		completionItems, err = handleDirectivesAutocomplete(doc, pos, uri)
 	} else if strings.HasSuffix(textBeforeCursor, "loop.") {
-		doc = removeTrailingChar(doc, pos.Line, pos.Character, '.')
-
-		isInsideLoop, errors := twLsp.IsInLoop(doc, uri, pos.Line, pos.Character)
-		if len(errors) > 0 {
-			logger.Error.Println(errors[0])
-		}
-
-		if isInsideLoop {
-			completionItems, err = completions.GetLoopObjFields(locale)
-		}
+		completionItems, err = handleLoopObjectAutocomplete(doc, pos, uri)
 	}
 
 	if err != nil {
@@ -57,6 +49,55 @@ func (s *State) Completion(id int, uri string, pos lsp.Position) (lsp.Completion
 	return s.completionResponse(id, s.makeCompletions(completionItems)), nil
 }
 
+func handleDirectivesAutocomplete(doc string, pos lsp.Position, uri string) ([]completions.Completion, error) {
+	dirs, err := completions.GetDirectives(locale)
+	if err != nil {
+		return []completions.Completion{}, err
+	}
+
+	isInsideLoop, errors := twLsp.IsInLoop(doc, uri, pos.Line, pos.Character)
+	if len(errors) > 0 {
+		logger.Error.Println(errors[0])
+	}
+
+	if isInsideLoop {
+		return dirs, nil
+	}
+
+	filteredDirs := make([]completions.Completion, 0, len(dirs))
+	loopDirs := []string{"@break", "@breakIf", "@continue", "@continueIf"}
+
+	for _, dir := range dirs {
+		if slices.Contains(loopDirs, dir.Label) {
+			continue
+		}
+
+		filteredDirs = append(filteredDirs, dir)
+	}
+
+	return filteredDirs, nil
+}
+
+func handleLoopObjectAutocomplete(doc string, pos lsp.Position, uri string) ([]completions.Completion, error) {
+	doc = removeTrailingChar(doc, pos.Line, pos.Character, '.')
+
+	isInsideLoop, errors := twLsp.IsInLoop(doc, uri, pos.Line, pos.Character)
+	if len(errors) > 0 {
+		logger.Error.Println(errors[0])
+	}
+
+	if !isInsideLoop {
+		return []completions.Completion{}, nil
+	}
+
+	fields, err := completions.GetLoopObjFields(locale)
+	if err != nil {
+		return []completions.Completion{}, err
+	}
+
+	return fields, nil
+}
+
 func removeTrailingChar(doc string, line, col uint, char byte) string {
 	lines := strings.Split(doc, "\n")
 	if int(line) >= len(lines) {