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
|
// Hardcoded dictionary of well-known commands and their arity
// (number of tokens that form the "human-understandable" prefix)
const ARITY: Record<string, number> = {
"git": 2, // "git checkout", "git commit", etc.
"npm": 3, // "npm run dev", "npm install -g"
"docker": 2, // "docker compose", "docker build"
"kubectl": 2, // "kubectl get", "kubectl apply"
"bun": 2, // "bun install", "bun test"
"cargo": 2, // "cargo build", "cargo test"
"go": 2, // "go build", "go test"
"python": 2, // "python -m", "python script.py"
"python3": 2,
"pip": 2,
"pip3": 2,
"brew": 2,
"apt": 2,
"apt-get": 2,
"dnf": 2,
"yum": 2,
"pacman": 2,
"systemctl": 2,
"journalctl": 2,
"ssh": 2,
"scp": 2,
"rsync": 2,
"curl": 2, // "curl -X", "curl https://"
"wget": 2,
"tar": 2, // "tar -xzf", "tar -czf"
"zip": 2,
"unzip": 2,
"chown": 2,
"chmod": 2,
"mount": 2,
"umount": 2,
// Default: all other commands are arity 1
};
// Get the normalized prefix for a list of tokens
export function prefix(tokens: string[]): string[] {
if (tokens.length === 0) return [];
const first = tokens[0];
if (!first) return [];
const arity = ARITY[first.toLowerCase()];
if (arity === undefined) {
return [first];
}
return tokens.slice(0, arity);
}
|
