feat: add CLI entry point, command modules, and zsh completion

- CLI with context detection, config merging, VM blocking
- Command modules: dotfiles, packages, setup, remote, dev, projects
- Zsh completion with declarative command/subcommand/flag structure

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
2026-03-16 05:06:31 +02:00
parent f79154d86f
commit 6ea23e02df
15 changed files with 466 additions and 1717 deletions

View File

@@ -1,130 +1,43 @@
"""Tests for flow.commands.completion dynamic suggestions."""
"""Tests for zsh completion."""
from flow.commands import completion
from flow.commands.completion import complete
def test_complete_top_level():
result = complete(["flow", ""], 1)
assert "dotfiles" in result
assert "packages" in result
assert "setup" in result
assert "remote" in result
assert "dev" in result
assert "projects" in result
def test_complete_top_level_prefix():
out = completion.complete(["flow", "do"], 2)
assert "dotfiles" in out
assert "dot" in out
result = complete(["flow", "do"], 1)
assert result == ["dotfiles"]
def test_complete_bootstrap_profiles(monkeypatch):
monkeypatch.setattr(completion, "_list_bootstrap_profiles", lambda: ["linux-vm", "macos-host"])
out = completion.complete(["flow", "bootstrap", "show", "li"], 4)
assert out == ["linux-vm"]
def test_complete_dotfiles_subcommands():
result = complete(["flow", "dotfiles", ""], 2)
assert "link" in result
assert "unlink" in result
assert "status" in result
def test_complete_bootstrap_packages_options(monkeypatch):
monkeypatch.setattr(completion, "_list_bootstrap_profiles", lambda: ["linux-vm", "macos-host"])
out = completion.complete(["flow", "bootstrap", "packages", "--p"], 4)
assert out == ["--profile"]
out = completion.complete(["flow", "bootstrap", "packages", "--profile", "m"], 5)
assert out == ["macos-host"]
def test_complete_dotfiles_link_flags():
result = complete(["flow", "dotfiles", "link", "--"], 3)
assert "--profile" in result
assert "--dry-run" in result
def test_complete_package_install(monkeypatch):
monkeypatch.setattr(completion, "_list_manifest_packages", lambda: ["neovim", "fzf"])
out = completion.complete(["flow", "package", "install", "n"], 4)
assert out == ["neovim"]
def test_complete_unknown_command():
result = complete(["flow", "unknown", ""], 2)
assert result == []
def test_complete_package_remove(monkeypatch):
monkeypatch.setattr(completion, "_list_installed_packages", lambda: ["hello", "jq"])
out = completion.complete(["flow", "package", "remove", "h"], 4)
assert out == ["hello"]
def test_list_manifest_packages_is_consistent_for_list_and_dict_forms(monkeypatch):
manifests = [
{
"packages": [
{"name": "neovim", "type": "binary"},
{"name": "ripgrep", "type": "pkg"},
{"name": "fzf", "type": "binary"},
]
},
{
"packages": {
"neovim": {"type": "binary"},
"ripgrep": {"type": "pkg"},
"fzf": {"type": "binary"},
}
},
]
monkeypatch.setattr(completion, "_safe_manifest", lambda: manifests.pop(0))
from_list = completion._list_manifest_packages()
from_dict = completion._list_manifest_packages()
assert from_list == ["fzf", "neovim"]
assert from_dict == ["fzf", "neovim"]
def test_list_manifest_packages_uses_mapping_key_when_name_missing(monkeypatch):
monkeypatch.setattr(
completion,
"_safe_manifest",
lambda: {"packages": {"bat": {"type": "binary"}, "git": {"type": "pkg"}}},
)
assert completion._list_manifest_packages() == ["bat"]
def test_complete_dotfiles_profile_value(monkeypatch):
monkeypatch.setattr(completion, "_list_dotfiles_profiles", lambda: ["work", "personal"])
out = completion.complete(["flow", "dotfiles", "link", "--profile", "w"], 5)
assert out == ["work"]
def test_complete_dotfiles_repo_subcommands():
out = completion.complete(["flow", "dotfiles", "repo", "p"], 4)
assert out == ["pull", "push"]
def test_complete_dotfiles_top_level_includes_undo():
out = completion.complete(["flow", "dotfiles", "u"], 3)
assert out == ["undo", "unlink"]
def test_complete_dotfiles_modules_subcommands():
out = completion.complete(["flow", "dotfiles", "modules", "s"], 4)
assert out == ["sync"]
def test_complete_dotfiles_modules_profile_value(monkeypatch):
monkeypatch.setattr(completion, "_list_dotfiles_profiles", lambda: ["work", "personal"])
out = completion.complete(["flow", "dotfiles", "modules", "list", "--profile", "w"], 6)
assert out == ["work"]
def test_complete_enter_targets(monkeypatch):
monkeypatch.setattr(completion, "_list_targets", lambda: ["personal@orb", "work@ec2"])
out = completion.complete(["flow", "enter", "p"], 3)
assert out == ["personal@orb"]
def test_complete_dev_subcommands():
out = completion.complete(["flow", "dev", "c"], 3)
assert out == ["connect", "create"]
def test_complete_completion_subcommands():
out = completion.complete(["flow", "completion", "i"], 3)
assert out == ["install-zsh"]
def test_rc_snippet_is_idempotent(tmp_path):
rc_path = tmp_path / ".zshrc"
completion_dir = tmp_path / "completions"
first = completion._ensure_rc_snippet(rc_path, completion_dir)
second = completion._ensure_rc_snippet(rc_path, completion_dir)
assert first is True
assert second is False
text = rc_path.read_text()
assert text.count(completion.ZSH_RC_START) == 1
assert text.count(completion.ZSH_RC_END) == 1
def test_complete_packages_subcommands():
result = complete(["flow", "packages", ""], 2)
assert "install" in result
assert "remove" in result
assert "list" in result