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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
# Source this in your ~/.config/nushell/config.nu
# minimum supported version = 0.93.0
module compat {
export def --wrapped "random uuid -v 7" [...rest] { atuin uuid }
}
use (if not (
(version).major > 0 or
(version).minor >= 103
) { "compat" }) *
if 'ATUIN_SESSION' not-in $env or ('ATUIN_SHLVL' not-in $env) or ($env.ATUIN_SHLVL != ($env.SHLVL? | default "")) {
$env.ATUIN_SESSION = (random uuid -v 7 | str replace -a "-" "")
$env.ATUIN_SHLVL = ($env.SHLVL? | default "")
}
hide-env -i ATUIN_HISTORY_ID
def _atuin_osc133_command_executed [] {
if 'ATUIN_PTY_PROXY_ACTIVE' not-in $env {
return
}
if 'ATUIN_HISTORY_ID' not-in $env or ($env.ATUIN_HISTORY_ID | is-empty) {
return
}
print -n $"(char esc)]133;C(char bel)"
}
def _atuin_osc133_command_finished [exit_code: int] {
if 'ATUIN_PTY_PROXY_ACTIVE' not-in $env {
return
}
if 'ATUIN_HISTORY_ID' not-in $env or ($env.ATUIN_HISTORY_ID | is-empty) {
return
}
print -n $"(char esc)]133;D;($exit_code);history_id=($env.ATUIN_HISTORY_ID);session_id=($env.ATUIN_SESSION)(char bel)"
}
# Magic token to make sure we don't record commands run by keybindings
let ATUIN_KEYBINDING_TOKEN = $"# (random uuid)"
let _atuin_pre_execution = {||
if ($nu | get history-enabled?) == false {
return
}
let cmd = (commandline)
if ($cmd | is-empty) {
return
}
if not ($cmd | str starts-with $ATUIN_KEYBINDING_TOKEN) {
$env.ATUIN_HISTORY_ID = (atuin history start -- $cmd | complete | get stdout | str trim)
_atuin_osc133_command_executed
}
}
let _atuin_pre_prompt = {||
let last_exit = $env.LAST_EXIT_CODE
if 'ATUIN_HISTORY_ID' not-in $env {
return
}
_atuin_osc133_command_finished $last_exit
with-env { ATUIN_LOG: error } {
if (version).minor >= 104 or (version).major > 0 {
job spawn {
^atuin history end $'--exit=($env.LAST_EXIT_CODE)' -- $env.ATUIN_HISTORY_ID | complete
} | ignore
} else {
do { atuin history end $'--exit=($last_exit)' -- $env.ATUIN_HISTORY_ID } | complete
}
}
hide-env ATUIN_HISTORY_ID
}
def _atuin_search_cmd [...flags: string] {
if (version).minor >= 106 or (version).major > 0 {
[
$ATUIN_KEYBINDING_TOKEN,
([
`with-env { ATUIN_LOG: error, ATUIN_QUERY: (commandline), ATUIN_SHELL: nu } {`,
([
'let output = (run-external atuin search',
($flags | append [--interactive] | each {|e| $'"($e)"'}),
'e>| str trim)',
] | flatten | str join ' '),
'if ($output | str starts-with "__atuin_accept__:") {',
'commandline edit --accept ($output | str replace "__atuin_accept__:" "")',
'} else {',
'commandline edit $output',
'}',
`}`,
] | flatten | str join "\n"),
]
} else {
[
$ATUIN_KEYBINDING_TOKEN,
([
`with-env { ATUIN_LOG: error, ATUIN_QUERY: (commandline) } {`,
'commandline edit',
'(run-external atuin search',
($flags | append [--interactive] | each {|e| $'"($e)"'}),
' e>| str trim)',
`}`,
] | flatten | str join ' '),
]
} | str join "\n"
}
$env.config = ($env | default {} config).config
$env.config = ($env.config | default {} hooks)
$env.config = (
$env.config | upsert hooks (
$env.config.hooks
| upsert pre_execution (
$env.config.hooks | get pre_execution? | default [] | append $_atuin_pre_execution)
| upsert pre_prompt (
$env.config.hooks | get pre_prompt? | default [] | append $_atuin_pre_prompt)
)
)
$env.config = ($env.config | default [] keybindings)
|
