602 lines
29 KiB
Python
602 lines
29 KiB
Python
import re
|
|
import hashlib
|
|
import os
|
|
import subprocess
|
|
import shutil
|
|
import sys
|
|
import yaml
|
|
import argparse
|
|
|
|
# Define configuration file paths.
|
|
DEFAULT_CONFIG_PATH = os.path.join("config", "defaults.yaml")
|
|
USER_CONFIG_PATH = os.path.join("config", "config.yaml")
|
|
BIN_DIR = os.path.expanduser("~/.local/bin")
|
|
|
|
def load_config():
|
|
"""Load configuration from defaults and merge in user config if present."""
|
|
if not os.path.exists(DEFAULT_CONFIG_PATH):
|
|
print(f"Default configuration file '{DEFAULT_CONFIG_PATH}' not found.")
|
|
sys.exit(1)
|
|
with open(DEFAULT_CONFIG_PATH, 'r') as f:
|
|
config = yaml.safe_load(f)
|
|
if "base" not in config or "repos" not in config:
|
|
print("Default config file must contain 'base' and 'repos' keys.")
|
|
sys.exit(1)
|
|
if os.path.exists(USER_CONFIG_PATH):
|
|
with open(USER_CONFIG_PATH, 'r') as f:
|
|
user_config = yaml.safe_load(f)
|
|
if user_config:
|
|
if "base" in user_config:
|
|
config["base"] = user_config["base"]
|
|
if "repos" in user_config:
|
|
config["repos"].extend(user_config["repos"])
|
|
return config
|
|
|
|
def save_user_config(user_config):
|
|
"""Save the user configuration to USER_CONFIG_PATH."""
|
|
os.makedirs(os.path.dirname(USER_CONFIG_PATH), exist_ok=True)
|
|
with open(USER_CONFIG_PATH, 'w') as f:
|
|
yaml.dump(user_config, f)
|
|
print(f"User configuration updated in {USER_CONFIG_PATH}.")
|
|
|
|
def run_command(command, cwd=None, preview=False):
|
|
"""Run a shell command in a given directory, or print it in preview mode."""
|
|
if preview:
|
|
print(f"[Preview] In '{cwd or os.getcwd()}': {command}")
|
|
else:
|
|
print(f"Running in '{cwd or os.getcwd()}': {command}")
|
|
subprocess.run(command, cwd=cwd, shell=True, check=False)
|
|
|
|
def get_repo_identifier(repo, all_repos):
|
|
"""
|
|
Return a unique identifier for the repository.
|
|
If the repository name is unique among all_repos, return repository name;
|
|
otherwise, return 'provider/account/repository'.
|
|
"""
|
|
repo_name = repo.get("repository")
|
|
count = sum(1 for r in all_repos if r.get("repository") == repo_name)
|
|
if count == 1:
|
|
return repo_name
|
|
else:
|
|
return f'{repo.get("provider")}/{repo.get("account")}/{repo.get("repository")}'
|
|
|
|
def resolve_repos(identifiers, all_repos):
|
|
"""
|
|
Given a list of identifier strings, return a list of repository configs.
|
|
The identifier can be:
|
|
- the full identifier "provider/account/repository"
|
|
- the repository name (if unique among all_repos)
|
|
- the alias (if defined)
|
|
"""
|
|
selected = []
|
|
for ident in identifiers:
|
|
matches = []
|
|
for repo in all_repos:
|
|
full_id = f'{repo.get("provider")}/{repo.get("account")}/{repo.get("repository")}'
|
|
if ident == full_id:
|
|
matches.append(repo)
|
|
elif ident == repo.get("alias"):
|
|
matches.append(repo)
|
|
elif ident == repo.get("repository"):
|
|
# Only match if repository name is unique among all_repos.
|
|
if sum(1 for r in all_repos if r.get("repository") == ident) == 1:
|
|
matches.append(repo)
|
|
if not matches:
|
|
print(f"Identifier '{ident}' did not match any repository in config.")
|
|
else:
|
|
selected.extend(matches)
|
|
return selected
|
|
|
|
def filter_ignored(repos):
|
|
"""Filter out repositories that have 'ignore' set to True."""
|
|
return [r for r in repos if not r.get("ignore", False)]
|
|
|
|
def generate_alias(repo, bin_dir, existing_aliases):
|
|
"""
|
|
Generate an alias for a repository based on its repository name.
|
|
|
|
Steps:
|
|
1. Keep only consonants from the repository name (letters from BCDFGHJKLMNPQRSTVWXYZ).
|
|
2. Collapse consecutive identical consonants.
|
|
3. Truncate to at most 12 characters.
|
|
4. If that alias conflicts (already in existing_aliases or a file exists in bin_dir),
|
|
then prefix with the first letter of provider and account.
|
|
5. If still conflicting, append a three-character hash until the alias is unique.
|
|
"""
|
|
repo_name = repo.get("repository")
|
|
# Keep only consonants.
|
|
consonants = re.sub(r"[^bcdfghjklmnpqrstvwxyzBCDFGHJKLMNPQRSTVWXYZ]", "", repo_name)
|
|
# Collapse consecutive identical consonants.
|
|
collapsed = re.sub(r"(.)\1+", r"\1", consonants)
|
|
base_alias = collapsed[:12] if len(collapsed) > 12 else collapsed
|
|
candidate = base_alias.lower()
|
|
|
|
def conflict(alias):
|
|
alias_path = os.path.join(bin_dir, alias)
|
|
return alias in existing_aliases or os.path.exists(alias_path)
|
|
|
|
if not conflict(candidate):
|
|
return candidate
|
|
|
|
prefix = (repo.get("provider", "")[0] + repo.get("account", "")[0]).lower()
|
|
candidate2 = (prefix + candidate)[:12]
|
|
if not conflict(candidate2):
|
|
return candidate2
|
|
|
|
h = hashlib.md5(repo_name.encode("utf-8")).hexdigest()[:3]
|
|
candidate3 = (candidate2 + h)[:12]
|
|
while conflict(candidate3):
|
|
candidate3 += "x"
|
|
candidate3 = candidate3[:12]
|
|
return candidate3
|
|
|
|
def create_executable(repo, base_dir, bin_dir, all_repos, preview=False):
|
|
"""Create an executable bash wrapper for the repository.
|
|
|
|
If 'verified' is set, the wrapper will checkout that commit and warn in orange if it does not match.
|
|
If no verified commit is set, a warning in orange is printed.
|
|
If an 'alias' field is provided, a symlink is created in bin_dir with that alias.
|
|
"""
|
|
repo_identifier = get_repo_identifier(repo, all_repos)
|
|
repo_dir = os.path.join(base_dir, repo.get("provider"), repo.get("account"), repo.get("repository"))
|
|
command = repo.get("command")
|
|
if not command:
|
|
main_sh = os.path.join(repo_dir, "main.sh")
|
|
main_py = os.path.join(repo_dir, "main.py")
|
|
if os.path.exists(main_sh):
|
|
command = "bash main.sh"
|
|
elif os.path.exists(main_py):
|
|
command = "python3 main.py"
|
|
else:
|
|
print(f"No command defined and no main.sh/main.py found in {repo_dir}. Skipping alias creation.")
|
|
return
|
|
|
|
ORANGE = r"\033[38;5;208m"
|
|
RESET = r"\033[0m"
|
|
|
|
verified = repo.get("verified")
|
|
if verified:
|
|
preamble = f"""\
|
|
git checkout {verified} || echo -e "{ORANGE}Warning: Failed to checkout commit {verified}.{RESET}"
|
|
CURRENT=$(git rev-parse HEAD)
|
|
if [ "$CURRENT" != "{verified}" ]; then
|
|
echo -e "{ORANGE}Warning: Current commit ($CURRENT) does not match verified commit ({verified}).{RESET}"
|
|
fi
|
|
"""
|
|
else:
|
|
preamble = f'echo -e "{ORANGE}Warning: No verified commit set for this repository.{RESET}"'
|
|
|
|
script_content = f"""#!/bin/bash
|
|
cd "{repo_dir}"
|
|
{preamble}
|
|
{command} "$@"
|
|
"""
|
|
alias_path = os.path.join(bin_dir, repo_identifier)
|
|
if preview:
|
|
print(f"[Preview] Would create executable '{alias_path}' with content:\n{script_content}")
|
|
else:
|
|
os.makedirs(bin_dir, exist_ok=True)
|
|
with open(alias_path, "w") as f:
|
|
f.write(script_content)
|
|
os.chmod(alias_path, 0o755)
|
|
print(f"Installed executable for {repo_identifier} at {alias_path}")
|
|
|
|
alias_name = repo.get("alias")
|
|
if alias_name:
|
|
alias_link_path = os.path.join(bin_dir, alias_name)
|
|
try:
|
|
if os.path.exists(alias_link_path) or os.path.islink(alias_link_path):
|
|
os.remove(alias_link_path)
|
|
os.symlink(alias_path, alias_link_path)
|
|
print(f"Created alias '{alias_name}' pointing to {repo_identifier}")
|
|
except Exception as e:
|
|
print(f"Error creating alias '{alias_name}': {e}")
|
|
|
|
def install_repos(selected_repos, base_dir, bin_dir, all_repos, preview=False):
|
|
"""Install repositories by creating executable wrappers and running setup."""
|
|
for repo in selected_repos:
|
|
repo_identifier = get_repo_identifier(repo, all_repos)
|
|
repo_dir = os.path.join(base_dir, repo.get("provider"), repo.get("account"), repo.get("repository"))
|
|
if not os.path.exists(repo_dir):
|
|
print(f"Repository directory '{repo_dir}' does not exist. Clone it first.")
|
|
continue
|
|
create_executable(repo, base_dir, bin_dir, all_repos, preview=preview)
|
|
setup_cmd = repo.get("setup")
|
|
if setup_cmd:
|
|
run_command(setup_cmd, cwd=repo_dir, preview=preview)
|
|
|
|
def exec_git_command(selected_repos, base_dir, all_repos, git_cmd, extra_args, preview=False):
|
|
"""Execute a given git command with extra arguments for each repository."""
|
|
for repo in selected_repos:
|
|
repo_identifier = get_repo_identifier(repo, all_repos)
|
|
repo_dir = os.path.join(base_dir, repo.get("provider"), repo.get("account"), repo.get("repository"))
|
|
if os.path.exists(repo_dir):
|
|
full_cmd = f"git {git_cmd} {' '.join(extra_args)}"
|
|
run_command(full_cmd, cwd=repo_dir, preview=preview)
|
|
else:
|
|
print(f"Repository directory '{repo_dir}' not found for {repo_identifier}.")
|
|
|
|
def pull_repos(selected_repos, base_dir, all_repos, extra_args, preview=False):
|
|
exec_git_command(selected_repos, base_dir, all_repos, "pull", extra_args, preview)
|
|
|
|
def push_repos(selected_repos, base_dir, all_repos, extra_args, preview=False):
|
|
exec_git_command(selected_repos, base_dir, all_repos, "push", extra_args, preview)
|
|
|
|
def status_repos(selected_repos, base_dir, all_repos, extra_args, list_only=False, system_status=False, preview=False):
|
|
if system_status:
|
|
print("System status:")
|
|
run_command("yay -Qu", preview=preview)
|
|
if list_only:
|
|
for repo in selected_repos:
|
|
print(get_repo_identifier(repo, all_repos))
|
|
else:
|
|
exec_git_command(selected_repos, base_dir, all_repos, "status", extra_args, preview)
|
|
|
|
def clone_repos(selected_repos, base_dir, all_repos, preview=False):
|
|
for repo in selected_repos:
|
|
repo_identifier = get_repo_identifier(repo, all_repos)
|
|
repo_dir = os.path.join(base_dir, repo.get("provider"), repo.get("account"), repo.get("repository"))
|
|
if os.path.exists(repo_dir):
|
|
print(f"Repository '{repo_identifier}' already exists at '{repo_dir}'.")
|
|
continue
|
|
target = repo.get("replacement") if repo.get("replacement") else f"{repo.get('provider')}/{repo.get('account')}/{repo.get('repository')}"
|
|
clone_url = f"https://{target}.git"
|
|
parent_dir = os.path.dirname(repo_dir)
|
|
os.makedirs(parent_dir, exist_ok=True)
|
|
run_command(f"git clone {clone_url} {repo_dir}", cwd=parent_dir, preview=preview)
|
|
|
|
def deinstall_repos(selected_repos, base_dir, bin_dir, all_repos, preview=False):
|
|
for repo in selected_repos:
|
|
repo_identifier = get_repo_identifier(repo, all_repos)
|
|
alias_path = os.path.join(bin_dir, repo_identifier)
|
|
if os.path.exists(alias_path):
|
|
if preview:
|
|
print(f"[Preview] Would remove executable '{alias_path}'.")
|
|
else:
|
|
os.remove(alias_path)
|
|
print(f"Removed executable for {repo_identifier}.")
|
|
else:
|
|
print(f"No executable found for {repo_identifier} in {bin_dir}.")
|
|
teardown_cmd = repo.get("teardown")
|
|
repo_dir = os.path.join(base_dir, repo.get("provider"), repo.get("account"), repo.get("repository"))
|
|
if teardown_cmd and os.path.exists(repo_dir):
|
|
run_command(teardown_cmd, cwd=repo_dir, preview=preview)
|
|
|
|
def delete_repos(selected_repos, base_dir, all_repos, preview=False):
|
|
for repo in selected_repos:
|
|
repo_identifier = get_repo_identifier(repo, all_repos)
|
|
repo_dir = os.path.join(base_dir, repo.get("provider"), repo.get("account"), repo.get("repository"))
|
|
if os.path.exists(repo_dir):
|
|
if preview:
|
|
print(f"[Preview] Would delete directory '{repo_dir}' for {repo_identifier}.")
|
|
else:
|
|
shutil.rmtree(repo_dir)
|
|
print(f"Deleted repository directory '{repo_dir}' for {repo_identifier}.")
|
|
else:
|
|
print(f"Repository directory '{repo_dir}' not found for {repo_identifier}.")
|
|
|
|
def update_repos(selected_repos, base_dir, bin_dir, all_repos, system_update=False, preview=False):
|
|
pull_repos(selected_repos, base_dir, all_repos, extra_args=[], preview=preview)
|
|
install_repos(selected_repos, base_dir, bin_dir, all_repos, preview=preview)
|
|
if system_update:
|
|
run_command("yay -S", preview=preview)
|
|
run_command("sudo pacman -Syyu", preview=preview)
|
|
|
|
def diff_repos(selected_repos, base_dir, all_repos, extra_args, preview=False):
|
|
exec_git_command(selected_repos, base_dir, all_repos, "diff", extra_args, preview)
|
|
|
|
def gitadd_repos(selected_repos, base_dir, all_repos, extra_args, preview=False):
|
|
exec_git_command(selected_repos, base_dir, all_repos, "add", extra_args, preview)
|
|
|
|
def show_repos(selected_repos, base_dir, all_repos, extra_args, preview=False):
|
|
exec_git_command(selected_repos, base_dir, all_repos, "show", extra_args, preview)
|
|
|
|
def checkout_repos(selected_repos, base_dir, all_repos, extra_args, preview=False):
|
|
exec_git_command(selected_repos, base_dir, all_repos, "checkout", extra_args, preview)
|
|
|
|
def show_config(selected_repos, full_config=False):
|
|
"""Display configuration for one or more repositories, or the entire merged config."""
|
|
if full_config:
|
|
merged = load_config()
|
|
print(yaml.dump(merged, default_flow_style=False))
|
|
else:
|
|
for repo in selected_repos:
|
|
identifier = f'{repo.get("provider")}/{repo.get("account")}/{repo.get("repository")}'
|
|
print(f"Repository: {identifier}")
|
|
for key, value in repo.items():
|
|
print(f" {key}: {value}")
|
|
print("-" * 40)
|
|
|
|
def interactive_add(config):
|
|
"""Interactively prompt the user to add a new repository entry to the user config."""
|
|
print("Adding a new repository configuration entry.")
|
|
new_entry = {}
|
|
new_entry["provider"] = input("Provider (e.g., github.com): ").strip()
|
|
new_entry["account"] = input("Account (e.g., yourusername): ").strip()
|
|
new_entry["repository"] = input("Repository name (e.g., mytool): ").strip()
|
|
new_entry["verified"] = input("Verified commit id: ").strip()
|
|
new_entry["command"] = input("Command (optional, leave blank to auto-detect): ").strip()
|
|
new_entry["description"] = input("Description (optional): ").strip()
|
|
new_entry["replacement"] = input("Replacement (optional): ").strip()
|
|
new_entry["setup"] = input("Setup command (optional): ").strip()
|
|
new_entry["teardown"] = input("Teardown command (optional): ").strip()
|
|
new_entry["alias"] = input("Alias (optional): ").strip()
|
|
# Allow the user to mark this entry as ignored.
|
|
ignore_val = input("Ignore this entry? (y/N): ").strip().lower()
|
|
if ignore_val == "y":
|
|
new_entry["ignore"] = True
|
|
|
|
print("\nNew entry:")
|
|
for key, value in new_entry.items():
|
|
if value:
|
|
print(f"{key}: {value}")
|
|
confirm = input("Add this entry to user config? (y/N): ").strip().lower()
|
|
if confirm == "y":
|
|
if os.path.exists(USER_CONFIG_PATH):
|
|
with open(USER_CONFIG_PATH, 'r') as f:
|
|
user_config = yaml.safe_load(f) or {}
|
|
else:
|
|
user_config = {"repos": []}
|
|
user_config.setdefault("repos", [])
|
|
user_config["repos"].append(new_entry)
|
|
save_user_config(user_config)
|
|
else:
|
|
print("Entry not added.")
|
|
|
|
def edit_config():
|
|
"""Open the user configuration file in nano."""
|
|
run_command(f"nano {USER_CONFIG_PATH}")
|
|
|
|
def config_init(user_config, defaults_config, bin_dir):
|
|
"""
|
|
Scan the base directory (defaults_config["base"]) for repositories.
|
|
The folder structure is assumed to be:
|
|
{base}/{provider}/{account}/{repository}
|
|
For each repository found, automatically determine:
|
|
- provider, account, repository from folder names.
|
|
- verified: the latest commit (via 'git log -1 --format=%H').
|
|
- alias: generated from the repository name using generate_alias().
|
|
Repositories already defined in defaults_config["repos"] or user_config["repos"] are skipped.
|
|
"""
|
|
base_dir = os.path.expanduser(defaults_config["base"])
|
|
if not os.path.isdir(base_dir):
|
|
print(f"Base directory '{base_dir}' does not exist.")
|
|
return
|
|
|
|
default_keys = {(entry.get("provider"), entry.get("account"), entry.get("repository"))
|
|
for entry in defaults_config.get("repos", [])}
|
|
existing_keys = {(entry.get("provider"), entry.get("account"), entry.get("repository"))
|
|
for entry in user_config.get("repos", [])}
|
|
existing_aliases = {entry.get("alias") for entry in user_config.get("repos", []) if entry.get("alias")}
|
|
|
|
new_entries = []
|
|
for provider in os.listdir(base_dir):
|
|
provider_path = os.path.join(base_dir, provider)
|
|
if not os.path.isdir(provider_path):
|
|
continue
|
|
for account in os.listdir(provider_path):
|
|
account_path = os.path.join(provider_path, account)
|
|
if not os.path.isdir(account_path):
|
|
continue
|
|
for repo_name in os.listdir(account_path):
|
|
repo_path = os.path.join(account_path, repo_name)
|
|
if not os.path.isdir(repo_path):
|
|
continue
|
|
key = (provider, account, repo_name)
|
|
if key in default_keys or key in existing_keys:
|
|
continue
|
|
try:
|
|
result = subprocess.run(
|
|
["git", "log", "-1", "--format=%H"],
|
|
cwd=repo_path,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
text=True,
|
|
check=True,
|
|
)
|
|
verified = result.stdout.strip()
|
|
except Exception as e:
|
|
verified = ""
|
|
print(f"Could not determine latest commit for {repo_name} ({provider}/{account}): {e}")
|
|
|
|
entry = {
|
|
"provider": provider,
|
|
"account": account,
|
|
"repository": repo_name,
|
|
"verified": verified,
|
|
"ignore": True
|
|
}
|
|
alias = generate_alias({"repository": repo_name, "provider": provider, "account": account}, bin_dir, existing_aliases)
|
|
entry["alias"] = alias
|
|
existing_aliases.add(alias)
|
|
new_entries.append(entry)
|
|
print(f"Adding new repo entry: {entry}")
|
|
|
|
if new_entries:
|
|
user_config.setdefault("repos", []).extend(new_entries)
|
|
save_user_config(user_config)
|
|
else:
|
|
print("No new repositories found.")
|
|
|
|
# Main program.
|
|
if __name__ == "__main__":
|
|
config_merged = load_config()
|
|
base_dir = os.path.expanduser(config_merged["base"])
|
|
all_repos_list = config_merged["repos"]
|
|
|
|
parser = argparse.ArgumentParser(description="Package Manager")
|
|
subparsers = parser.add_subparsers(dest="command", help="Subcommands")
|
|
|
|
def add_identifier_arguments(subparser):
|
|
subparser.add_argument("identifiers", nargs="*", help="Identifier(s) for repositories")
|
|
subparser.add_argument("--all", action="store_true", help="Apply to all repositories in the config")
|
|
subparser.add_argument("--preview", action="store_true", help="Preview changes without executing commands")
|
|
subparser.add_argument("--list", action="store_true", help="List affected repositories (with preview or status)")
|
|
subparser.add_argument("extra_args", nargs=argparse.REMAINDER, help="Extra arguments for the git command")
|
|
|
|
install_parser = subparsers.add_parser("install", help="Install repository/repositories")
|
|
add_identifier_arguments(install_parser)
|
|
|
|
pull_parser = subparsers.add_parser("pull", help="Pull updates for repository/repositories")
|
|
add_identifier_arguments(pull_parser)
|
|
|
|
clone_parser = subparsers.add_parser("clone", help="Clone repository/repositories")
|
|
add_identifier_arguments(clone_parser)
|
|
|
|
push_parser = subparsers.add_parser("push", help="Push changes for repository/repositories")
|
|
add_identifier_arguments(push_parser)
|
|
|
|
deinstall_parser = subparsers.add_parser("deinstall", help="Deinstall repository/repositories")
|
|
add_identifier_arguments(deinstall_parser)
|
|
|
|
delete_parser = subparsers.add_parser("delete", help="Delete repository directory for repository/repositories")
|
|
add_identifier_arguments(delete_parser)
|
|
|
|
update_parser = subparsers.add_parser("update", help="Update (pull + install) repository/repositories")
|
|
add_identifier_arguments(update_parser)
|
|
update_parser.add_argument("--system", action="store_true", help="Include system update commands")
|
|
|
|
status_parser = subparsers.add_parser("status", help="Show status for repository/repositories or system")
|
|
add_identifier_arguments(status_parser)
|
|
status_parser.add_argument("--system", action="store_true", help="Show system status")
|
|
|
|
diff_parser = subparsers.add_parser("diff", help="Execute 'git diff' for repository/repositories")
|
|
add_identifier_arguments(diff_parser)
|
|
|
|
gitadd_parser = subparsers.add_parser("add", help="Execute 'git add' for repository/repositories")
|
|
add_identifier_arguments(gitadd_parser)
|
|
|
|
show_parser = subparsers.add_parser("show", help="Execute 'git show' for repository/repositories")
|
|
add_identifier_arguments(show_parser)
|
|
|
|
commit_parser = subparsers.add_parser("commit", help="Execute 'git commit' for repository/repositories")
|
|
add_identifier_arguments(commit_parser)
|
|
|
|
checkout_parser = subparsers.add_parser("checkout", help="Execute 'git checkout' for repository/repositories")
|
|
add_identifier_arguments(checkout_parser)
|
|
|
|
config_parser = subparsers.add_parser("config", help="Manage configuration")
|
|
config_subparsers = config_parser.add_subparsers(dest="subcommand", help="Config subcommands", required=True)
|
|
config_show = config_subparsers.add_parser("show", help="Show configuration")
|
|
add_identifier_arguments(config_show)
|
|
config_add = config_subparsers.add_parser("add", help="Interactively add a new repository entry")
|
|
config_edit = config_subparsers.add_parser("edit", help="Edit configuration file with nano")
|
|
config_init_parser = config_subparsers.add_parser("init", help="Initialize user configuration by scanning the base directory")
|
|
config_delete = config_subparsers.add_parser("delete", help="Delete repository entry from user config")
|
|
add_identifier_arguments(config_delete)
|
|
config_ignore = config_subparsers.add_parser("ignore", help="Set ignore flag for repository entries in user config")
|
|
add_identifier_arguments(config_ignore)
|
|
config_ignore.add_argument("--set", choices=["true", "false"], required=True, help="Set ignore to true or false")
|
|
|
|
|
|
args = parser.parse_args()
|
|
|
|
# Dispatch commands.
|
|
if args.command == "install":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
install_repos(selected, base_dir, BIN_DIR, all_repos_list, preview=args.preview)
|
|
elif args.command == "pull":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
pull_repos(selected, base_dir, all_repos_list, args.extra_args, preview=args.preview)
|
|
elif args.command == "clone":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
clone_repos(selected, base_dir, all_repos_list, preview=args.preview)
|
|
elif args.command == "push":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
push_repos(selected, base_dir, all_repos_list, args.extra_args, preview=args.preview)
|
|
elif args.command == "deinstall":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
deinstall_repos(selected, base_dir, BIN_DIR, all_repos_list, preview=args.preview)
|
|
elif args.command == "delete":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
delete_repos(selected, base_dir, all_repos_list, preview=args.preview)
|
|
elif args.command == "commit":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
exec_git_command(selected, base_dir, all_repos_list, "commit", args.extra_args, preview=args.preview)
|
|
elif args.command == "update":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
update_repos(selected, base_dir, BIN_DIR, all_repos_list, system_update=args.system, preview=args.preview)
|
|
elif args.command == "status":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
status_repos(selected, base_dir, all_repos_list, args.extra_args, list_only=args.list, system_status=args.system, preview=args.preview)
|
|
elif args.command == "diff":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
diff_repos(selected, base_dir, all_repos_list, args.extra_args, preview=args.preview)
|
|
elif args.command == "add":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
gitadd_repos(selected, base_dir, all_repos_list, args.extra_args, preview=args.preview)
|
|
elif args.command == "show":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
show_repos(selected, base_dir, all_repos_list, args.extra_args, preview=args.preview)
|
|
elif args.command == "checkout":
|
|
selected = all_repos_list if args.all or (not args.identifiers) else resolve_repos(args.identifiers, all_repos_list)
|
|
selected = filter_ignored(selected)
|
|
checkout_repos(selected, base_dir, all_repos_list, args.extra_args, preview=args.preview)
|
|
elif args.command == "config":
|
|
if args.subcommand == "show":
|
|
if args.all or (not args.identifiers):
|
|
show_config([], full_config=True)
|
|
else:
|
|
selected = resolve_repos(args.identifiers, all_repos_list)
|
|
if selected:
|
|
show_config(selected, full_config=False)
|
|
elif args.subcommand == "add":
|
|
interactive_add(config_merged)
|
|
elif args.subcommand == "edit":
|
|
edit_config()
|
|
elif args.subcommand == "init":
|
|
if os.path.exists(USER_CONFIG_PATH):
|
|
with open(USER_CONFIG_PATH, 'r') as f:
|
|
user_config = yaml.safe_load(f) or {}
|
|
else:
|
|
user_config = {"repos": []}
|
|
config_init(user_config, config_merged, BIN_DIR)
|
|
elif args.subcommand == "delete":
|
|
# Load user config from USER_CONFIG_PATH.
|
|
if os.path.exists(USER_CONFIG_PATH):
|
|
with open(USER_CONFIG_PATH, 'r') as f:
|
|
user_config = yaml.safe_load(f) or {"repos": []}
|
|
else:
|
|
user_config = {"repos": []}
|
|
if args.all or not args.identifiers:
|
|
print("You must specify identifiers to delete.")
|
|
else:
|
|
to_delete = resolve_repos(args.identifiers, user_config.get("repos", []))
|
|
new_repos = [entry for entry in user_config.get("repos", []) if entry not in to_delete]
|
|
user_config["repos"] = new_repos
|
|
save_user_config(user_config)
|
|
print(f"Deleted {len(to_delete)} entries from user config.")
|
|
elif args.subcommand == "ignore":
|
|
# Load user config from USER_CONFIG_PATH.
|
|
if os.path.exists(USER_CONFIG_PATH):
|
|
with open(USER_CONFIG_PATH, 'r') as f:
|
|
user_config = yaml.safe_load(f) or {"repos": []}
|
|
else:
|
|
user_config = {"repos": []}
|
|
if args.all or not args.identifiers:
|
|
print("You must specify identifiers to modify ignore flag.")
|
|
else:
|
|
to_modify = resolve_repos(args.identifiers, user_config.get("repos", []))
|
|
for entry in user_config["repos"]:
|
|
key = (entry.get("provider"), entry.get("account"), entry.get("repository"))
|
|
for mod in to_modify:
|
|
mod_key = (mod.get("provider"), mod.get("account"), mod.get("repository"))
|
|
if key == mod_key:
|
|
entry["ignore"] = (args.set == "true")
|
|
print(f"Set ignore for {key} to {entry['ignore']}")
|
|
save_user_config(user_config)
|
|
else:
|
|
parser.print_help()
|