## Feat * feat(ai): add AI suggestion engine, context gathering, and ghost text overlay (`v0.3.0`) ## Security & Perf * fix(ai): restrict background `--help` execution to a hardcoded command allowlist to prevent RCE * fix(root): defer context cancellation in goroutine to prevent resource leaks * perf(ai): implement LRU eviction and maximum size limit for `ProviderCache` * perf(ai): truncate command context output (`CommandContextProvider.Gather`) to 1000 characters to save tokens ## Bug Fixes * fix(ai): add mutex synchronization and thread-safe snapshotting to `AIEngine.RegisterProvider` and `GatherDynamicContext` * fix(ai): return a copy of fresh `Suggestion` in `AIEngine.Suggest` to prevent cache mutation * fix(ai): use length-prefixed encoding in `EnvSnapshot.Hash` to prevent delimiter collisions * fix(ai): skip variable assignment lines containing `=` when extracting Makefile targets * fix(runner): check `scanner.Err()` and return `nil` on scan errors in justfile generator * test(ci): rename `git commit` to `git checkout` in overlay test to resolve CI `typos` false positive
86 lines
1.9 KiB
Go
86 lines
1.9 KiB
Go
package runner
|
|
|
|
import (
|
|
"bufio"
|
|
"os"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/versenilvis/iris/spec"
|
|
)
|
|
|
|
func init() {
|
|
spec.Register(&spec.Spec{
|
|
Name: "just",
|
|
Description: "command runner",
|
|
MaxArgs: 1,
|
|
Generator: func(tokens []string, prefix string, partial string) []spec.Suggestion {
|
|
cwd := spec.GetCWD()
|
|
file, err := os.Open(filepath.Join(cwd, "justfile"))
|
|
if err != nil {
|
|
// try uppercase
|
|
file, err = os.Open(filepath.Join(cwd, "Justfile"))
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
}
|
|
defer func() { _ = file.Close() }()
|
|
|
|
var suggestions []spec.Suggestion
|
|
seen := make(map[string]bool)
|
|
scanner := bufio.NewScanner(file)
|
|
// matching recipes like `build:`
|
|
recipeRegex := regexp.MustCompile(`^([a-zA-Z0-9_-]+):`)
|
|
|
|
lastComment := ""
|
|
for scanner.Scan() {
|
|
line := strings.TrimSpace(scanner.Text())
|
|
|
|
// Handle comments
|
|
if after, ok := strings.CutPrefix(line, "#"); ok {
|
|
lastComment = strings.TrimSpace(after)
|
|
continue
|
|
}
|
|
|
|
// Skip blank lines or group attributes
|
|
if line == "" || strings.HasPrefix(line, "[") {
|
|
continue
|
|
}
|
|
|
|
if matches := recipeRegex.FindStringSubmatch(line); len(matches) > 1 {
|
|
recipe := matches[1]
|
|
if seen[recipe] {
|
|
lastComment = ""
|
|
continue
|
|
}
|
|
seen[recipe] = true
|
|
|
|
cmd := recipe
|
|
|
|
desc := "just recipe"
|
|
if lastComment != "" {
|
|
desc = lastComment
|
|
}
|
|
|
|
suggestions = append(suggestions, spec.Suggestion{
|
|
Cmd: cmd,
|
|
Desc: desc,
|
|
})
|
|
lastComment = ""
|
|
} else {
|
|
// If it's something else (like a command line), reset comment
|
|
// unless it's just indented (which we already skip or handle)
|
|
if !strings.HasPrefix(scanner.Text(), " ") && !strings.HasPrefix(scanner.Text(), "\t") {
|
|
lastComment = ""
|
|
}
|
|
}
|
|
}
|
|
if err := scanner.Err(); err != nil {
|
|
return nil
|
|
}
|
|
return suggestions
|
|
},
|
|
})
|
|
}
|