Replace local input model and improve shell command feedback

2026-09-23 15:38:10 -05:00
parent 35ff43de7f
commit 3bfd7c0a54
33 changed files with 499 additions and 2700 deletions
@@ -0,0 +1,86 @@
+use galaxy_completer::completer::testing::FakeCompletionContext;
+use galaxy_completer::signatures::testing::create_test_command_registry;
+use galaxy_completer::ParsedTokensSnapshot;
+use input_classifier::InputType;
+
+use super::detect_input_type;
+
+#[test]
+fn detection_uses_command_inventory_and_history_without_running_input() {
+    let context = FakeCompletionContext::new(create_test_command_registry([]))
+        .with_top_level_commands(["git", "yarn"])
+        .with_aliases([("gs".into(), "git status".to_string())].into())
+        .with_functions(["deploy_app".into()].into())
+        .with_builtins(["cd".into()].into());
+    let history = vec!["custom_tool --old-argument".to_string()];
+    for command in [
+        "git pull origin CI",
+        "yarn workspace app tsc",
+        "gs",
+        "deploy_app",
+        "cd /tmp",
+        "custom_tool --new-argument",
+        "FOO=bar git status",
+        "./script/check",
+    ] {
+        let input = ParsedTokensSnapshot {
+            buffer_text: command.to_string(),
+            parsed_tokens: vec![],
+        };
+        assert_eq!(
+            detect_input_type(&input, &context, &history, InputType::AI),
+            InputType::Shell,
+            "{command}"
+        );
+    }
+    for prompt in ["hello, there!", "please fix the build", "gti status"] {
+        let input = ParsedTokensSnapshot {
+            buffer_text: prompt.to_string(),
+            parsed_tokens: vec![],
+        };
+        assert_eq!(
+            detect_input_type(&input, &context, &history, InputType::Shell),
+            InputType::AI,
+            "{prompt}"
+        );
+    }
+}
+
+#[test]
+fn command_prefix_preserves_current_mode() {
+    let context = FakeCompletionContext::new(create_test_command_registry([]))
+        .with_top_level_commands(["git"]);
+    let input = ParsedTokensSnapshot {
+        buffer_text: "gi".to_string(),
+        parsed_tokens: vec![],
+    };
+    for current in [InputType::AI, InputType::Shell] {
+        assert_eq!(detect_input_type(&input, &context, &[], current), current);
+    }
+}
+
+#[test]
+fn unverified_greeting_history_does_not_turn_a_prompt_into_a_command() {
+    let context = FakeCompletionContext::new(create_test_command_registry([]));
+    let records = [
+        ("Hello, there!", None),
+        ("Hello, there!", Some(127.into())),
+        ("Hello, there!", Some(1.into())),
+        ("Hello, there!", Some(130.into())),
+        ("custom_tool --previous", Some(0.into())),
+    ];
+    let history = records
+        .into_iter()
+        .filter(|(_, exit_code)| super::is_verified_history_command(*exit_code))
+        .map(|(command, _)| command.to_string())
+        .collect::<Vec<_>>();
+    assert_eq!(history, vec!["custom_tool --previous"]);
+    let greeting = ParsedTokensSnapshot {
+        buffer_text: "Hello, there!".to_string(),
+        parsed_tokens: vec![],
+    };
+    assert_eq!(
+        detect_input_type(&greeting, &context, &history, InputType::Shell),
+        InputType::AI
+    );
+}