// 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);
}