Files
iris-context/commands/runner/justfile.go
VERSE 94c8af7b1d feat: AI suggestion (#34)
## 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
2026-07-11 15:17:14 +07:00

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
},
})
}