VERSE
860b475b53
Enhance lookup mechanics and priority propagation (`spec/lookup.go`) Remove the requirement of typing a leading `-` when searching for options, allowing flag suggestions to surface naturally during input Accurately forward the `Priority` field from registry specs into the `Suggestion` struct so the scoring engine evaluates relevance properly Integrate scoring into the main suggestion pipeline (`root/suggestions.go`) Collect context signals using `scoring.CollectSignals(cwd, query, rootCmd, store)` with the live working directory retrieved from `spec.GetCWD()` Pass the entire deduplicated list across spec, history and AI items through `scoring.Score(deduped, signals)` to compute scores and sort by descending order before rendering on screen Record execution history in Frecency Store (`root/wrapper.go`) Save the last executed command to `lastSubmittedCommand` whenever Enter is pressed or a suggestion is selected Trigger `store.Record(...)` asynchronously within the `IRIS_CMD_STOP` hook using a dedicated goroutine guarded by panic recovery and timeout limits, preventing blocks on the main shell thread Optimize workspace identification for AI (`internal/ai/context_provider.go`) Use `workspace.DetectCached(cwd)` to instantly fetch ecosystem context across Git, Node, Go, Rust, Python and Docker alongside active git branch state This caching mechanism avoids redundant subprocess execution when outside git repositories, significantly speeding up prompt context generation
118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
package workspace
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestDetect_GitAndGoProject(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
|
|
_ = os.Mkdir(filepath.Join(tmp, ".git"), 0755)
|
|
_ = os.WriteFile(filepath.Join(tmp, "go.mod"), []byte("module test"), 0644)
|
|
|
|
info := Detect(tmp)
|
|
|
|
if !info.HasGit {
|
|
t.Error("expected HasGit to be true")
|
|
}
|
|
if !info.HasGoProject {
|
|
t.Error("expected HasGoProject to be true")
|
|
}
|
|
if info.HasNodeProject {
|
|
t.Error("expected HasNodeProject to be false")
|
|
}
|
|
if info.HasRustProject {
|
|
t.Error("expected HasRustProject to be false")
|
|
}
|
|
if len(info.SignatureFiles) != 2 {
|
|
t.Errorf("expected 2 signature files, got %d: %v", len(info.SignatureFiles), info.SignatureFiles)
|
|
}
|
|
}
|
|
|
|
func TestDetect_EmptyDirectory(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
info := Detect(tmp)
|
|
|
|
if info.HasGit || info.HasNodeProject || info.HasGoProject || info.HasRustProject || info.HasDockerfile || info.HasMakefile {
|
|
t.Error("expected all flags to be false for empty directory")
|
|
}
|
|
if len(info.SignatureFiles) != 0 {
|
|
t.Errorf("expected 0 signature files, got %d", len(info.SignatureFiles))
|
|
}
|
|
}
|
|
|
|
func TestDetect_NodeProject(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
_ = os.WriteFile(filepath.Join(tmp, "package.json"), []byte("{}"), 0644)
|
|
_ = os.WriteFile(filepath.Join(tmp, "Dockerfile"), []byte("FROM node"), 0644)
|
|
|
|
info := Detect(tmp)
|
|
|
|
if !info.HasNodeProject {
|
|
t.Error("expected HasNodeProject to be true")
|
|
}
|
|
if !info.HasDockerfile {
|
|
t.Error("expected HasDockerfile to be true")
|
|
}
|
|
if info.HasGit {
|
|
t.Error("expected HasGit to be false")
|
|
}
|
|
}
|
|
|
|
func TestDetectCached_MidSessionFileCreation(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
|
|
// first call: no go.mod exists
|
|
info1 := DetectCached(tmp)
|
|
if info1.HasGoProject {
|
|
t.Fatal("expected HasGoProject to be false before creating go.mod")
|
|
}
|
|
|
|
// create go.mod mid-session (same cwd, no cd)
|
|
_ = os.WriteFile(filepath.Join(tmp, "go.mod"), []byte("module test"), 0644)
|
|
|
|
// second call: cache should invalidate because directory modtime changed
|
|
info2 := DetectCached(tmp)
|
|
if !info2.HasGoProject {
|
|
t.Fatal("expected HasGoProject to be true after creating go.mod in same cwd")
|
|
}
|
|
}
|
|
|
|
func TestDetectCached_CwdChange(t *testing.T) {
|
|
dir1 := t.TempDir()
|
|
dir2 := t.TempDir()
|
|
|
|
_ = os.Mkdir(filepath.Join(dir1, ".git"), 0755)
|
|
|
|
info1 := DetectCached(dir1)
|
|
if !info1.HasGit {
|
|
t.Fatal("expected HasGit for dir1")
|
|
}
|
|
|
|
info2 := DetectCached(dir2)
|
|
if info2.HasGit {
|
|
t.Fatal("expected no HasGit for dir2")
|
|
}
|
|
}
|
|
|
|
func TestDetect_MultiEcosystems(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
_ = os.WriteFile(filepath.Join(tmp, "justfile"), []byte("build:"), 0644)
|
|
_ = os.WriteFile(filepath.Join(tmp, "pyproject.toml"), []byte(""), 0644)
|
|
_ = os.WriteFile(filepath.Join(tmp, "Chart.yaml"), []byte("apiVersion: v2"), 0644)
|
|
|
|
info := Detect(tmp)
|
|
|
|
if !info.HasJustfile {
|
|
t.Error("expected HasJustfile to be true")
|
|
}
|
|
if !info.HasPythonProject {
|
|
t.Error("expected HasPythonProject to be true")
|
|
}
|
|
if !info.HasK8s {
|
|
t.Error("expected HasK8s to be true")
|
|
}
|
|
}
|