feat(scoring): transition workflow learning and fix history UX (#39)

- add transition scoring engine to learn sequential workflows per directory using command skeletons - prioritize current active git branch over older branches in suggestions - restore strict chronological order for history navigation and bypass AI re-ranking - add in-memory session history for instant access to just-run commands - fix PTY prompt not syncing when using arrow keys in the history menu - fix history tie-breaker to properly prioritize newer commands by assigning larger IDs - fix potential mutex deadlock between bufferMu and pty write during history navigation - fix workspace git detection to remove depth limits and correctly identify repositories with detached HEADs - fix SQLite connection leaks in frecency queries - fix global state leaks in history tests by snapshotting and restoring registry states - modernize string prefix checks using strings.CutPrefix
2026-07-26 19:10:56 +07:00
parent 85ac76e85a
commit 6ffb9f83f2
20 changed files with 1004 additions and 72 deletions
@@ -0,0 +1,97 @@
+package spec
+
+import (
+	"testing"
+)
+
+func TestTryExtractSkeleton(t *testing.T) {
+	ResetRegistry()
+	Register(&Spec{
+		Name: "git",
+		Subcommands: []Subcommand{
+			{
+				Name:    "checkout",
+				Aliases: []string{"co"},
+			},
+			{
+				Name: "remote",
+				Subcommands: []Subcommand{
+					{Name: "add"},
+				},
+			},
+		},
+	})
+	Register(&Spec{
+		Name: "docker",
+		Subcommands: []Subcommand{
+			{
+				Name: "compose",
+				Subcommands: []Subcommand{
+					{Name: "up"},
+				},
+			},
+		},
+	})
+
+	tests := []struct {
+		name         string
+		input        string
+		wantSkeleton string
+		wantOk       bool
+	}{
+		{
+			name:         "git checkout with argument",
+			input:        "git checkout feature-x -b",
+			wantSkeleton: "git checkout",
+			wantOk:       true,
+		},
+		{
+			name:         "git co alias with argument",
+			input:        "git co feature-y",
+			wantSkeleton: "git checkout",
+			wantOk:       true,
+		},
+		{
+			name:         "git remote add with arguments",
+			input:        "git remote add origin https://github.com/test/test.git",
+			wantSkeleton: "git remote add",
+			wantOk:       true,
+		},
+		{
+			name:         "git global flag before subcommand",
+			input:        "git -C /tmp checkout main",
+			wantSkeleton: "git checkout",
+			wantOk:       true,
+		},
+		{
+			name:         "docker compose up with flags",
+			input:        "docker compose up -d",
+			wantSkeleton: "docker compose up",
+			wantOk:       true,
+		},
+		{
+			name:         "unregistered spec",
+			input:        "cargo build --release",
+			wantSkeleton: "",
+			wantOk:       false,
+		},
+		{
+			name:         "empty input",
+			input:        "   ",
+			wantSkeleton: "",
+			wantOk:       false,
+		},
+	}
+
+	for _, tc := range tests {
+		t.Run(tc.name, func(t *testing.T) {
+			got, ok := TryExtractSkeleton(tc.input)
+			if ok != tc.wantOk {
+				t.Fatalf("TryExtractSkeleton(%q) ok = %v, want %v", tc.input, ok, tc.wantOk)
+			}
+			if got != tc.wantSkeleton {
+				t.Errorf("TryExtractSkeleton(%q) = %q, want %q", tc.input, got, tc.wantSkeleton)
+			}
+		})
+	}
+}