blob: bf5c625624555efb2c157436d3d6b8f22d242bc3 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
pub async fn run() -> eyre::Result<()> {
let zsh_function = generate_zsh_integration();
println!("{}", zsh_function);
Ok(())
}
/// Generate the zsh integration function - pure function for easy testing
pub fn generate_zsh_integration() -> &'static str {
r#"
# TUI uses an alternate screen, so no explicit cleanup is needed.
_atuin_ai_cleanup() {
true
}
# Question mark at start of line - natural language mode
_atuin_ai_question_mark() {
# If buffer is empty or just contains '?', trigger natural language mode
if [[ -z "$BUFFER" || "$BUFFER" == "?" ]]; then
BUFFER=""
local output
output=$(atuin-ai inline --natural-language 3>&1 1>&2 2>&3)
# Clean up the inline viewport
_atuin_ai_cleanup
if [[ $output == __atuin_ai_cancel__ ]]; then
zle reset-prompt
elif [[ $output == __atuin_ai_execute__:* ]]; then
RBUFFER=""
LBUFFER=${output#__atuin_ai_execute__:}
zle reset-prompt
zle accept-line
elif [[ $output == __atuin_ai_insert__:* ]]; then
RBUFFER=""
LBUFFER=${output#__atuin_ai_insert__:}
zle reset-prompt
elif [[ -n $output ]]; then
RBUFFER=""
LBUFFER=$output
zle reset-prompt
else
zle reset-prompt
fi
else
LBUFFER="${LBUFFER}?"
fi
}
# Set up keybindings
zle -N _atuin_ai_question_mark
bindkey '?' _atuin_ai_question_mark # Question mark
"#
.trim()
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_generate_zsh_integration() {
let result = generate_zsh_integration();
assert!(result.contains("_atuin_ai_question_mark"));
assert!(result.contains("bindkey"));
assert!(result.contains("atuin-ai inline"));
assert!(result.contains("__atuin_ai_cancel__"));
assert!(result.contains("__atuin_ai_execute__"));
assert!(result.contains("__atuin_ai_insert__"));
}
}
|
