mirror of
https://github.com/theniceboy/.config.git
synced 2025-12-26 14:44:57 +08:00
improve upgrade-all
This commit is contained in:
David Chen
parent
beec4f71f4
commit
60468c72cb
1 changed files with 231 additions and 42 deletions
273
bin/upgrade-all
273
bin/upgrade-all
|
|
@ -1,5 +1,7 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
import json
|
||||
import shutil
|
||||
import subprocess
|
||||
import threading
|
||||
import time
|
||||
|
|
@ -20,9 +22,11 @@ class ParallelTaskRunner:
|
|||
self.task_positions = {}
|
||||
self.lock = threading.Lock()
|
||||
|
||||
def add_task(self, name, command):
|
||||
def add_task(self, name, command=None, action=None):
|
||||
"""Add a task to the execution queue."""
|
||||
self.tasks.append({'name': name, 'command': command})
|
||||
if command is None and action is None:
|
||||
raise ValueError("Task must define a command or action")
|
||||
self.tasks.append({'name': name, 'command': command, 'action': action})
|
||||
|
||||
def _update_task_line(self, task_id, text):
|
||||
"""Update a specific task's display line with thread-safe cursor movement."""
|
||||
|
|
@ -36,47 +40,120 @@ class ParallelTaskRunner:
|
|||
|
||||
def _execute_task(self, task):
|
||||
"""Execute a single task with real-time status updates."""
|
||||
name, command = task['name'], task['command']
|
||||
name = task['name']
|
||||
command = task.get('command')
|
||||
action = task.get('action')
|
||||
task_id = threading.get_ident()
|
||||
|
||||
self._update_task_line(task_id, f"{Colors.YELLOW}[STARTING]{Colors.RESET} {name}")
|
||||
|
||||
try:
|
||||
process = subprocess.Popen(
|
||||
command,
|
||||
shell=True,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.STDOUT,
|
||||
text=True
|
||||
)
|
||||
|
||||
# Animated progress indicator
|
||||
spinner_chars = "|/-\\"
|
||||
spinner_idx = 0
|
||||
|
||||
while process.poll() is None:
|
||||
spinner = f"{Colors.BLUE}[RUNNING]{Colors.RESET} {name} [{spinner_chars[spinner_idx]}]"
|
||||
self._update_task_line(task_id, spinner)
|
||||
spinner_idx = (spinner_idx + 1) % len(spinner_chars)
|
||||
time.sleep(0.1)
|
||||
|
||||
stdout, _ = process.communicate()
|
||||
|
||||
if process.returncode == 0:
|
||||
self._update_task_line(task_id, f"{Colors.GREEN}[COMPLETED]{Colors.RESET} {name}")
|
||||
self.results.put({'name': name, 'success': True, 'output': stdout})
|
||||
else:
|
||||
if command is not None:
|
||||
try:
|
||||
process = subprocess.Popen(
|
||||
command,
|
||||
shell=True,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.STDOUT,
|
||||
text=True
|
||||
)
|
||||
|
||||
# Animated progress indicator
|
||||
spinner_chars = "|/-\\"
|
||||
spinner_idx = 0
|
||||
|
||||
while process.poll() is None:
|
||||
spinner = f"{Colors.BLUE}[RUNNING]{Colors.RESET} {name} [{spinner_chars[spinner_idx]}]"
|
||||
self._update_task_line(task_id, spinner)
|
||||
spinner_idx = (spinner_idx + 1) % len(spinner_chars)
|
||||
time.sleep(0.1)
|
||||
|
||||
stdout, _ = process.communicate()
|
||||
|
||||
if process.returncode == 0:
|
||||
self._update_task_line(task_id, f"{Colors.GREEN}[COMPLETED]{Colors.RESET} {name}")
|
||||
self.results.put({'name': name, 'success': True, 'output': stdout})
|
||||
else:
|
||||
self._update_task_line(task_id, f"{Colors.RED}[FAILED]{Colors.RESET} {name}")
|
||||
if stdout:
|
||||
with self.lock:
|
||||
print(f"\nError output for {name}:\n{stdout}")
|
||||
self.results.put({'name': name, 'success': False, 'output': stdout})
|
||||
|
||||
except Exception as e:
|
||||
self._update_task_line(task_id, f"{Colors.RED}[FAILED]{Colors.RESET} {name}")
|
||||
if stdout:
|
||||
with self.lock:
|
||||
print(f"\nError output for {name}:\n{stdout}")
|
||||
self.results.put({'name': name, 'success': False, 'output': stdout})
|
||||
|
||||
except Exception as e:
|
||||
with self.lock:
|
||||
print(f"\nException for {name}: {e}")
|
||||
self.results.put({'name': name, 'success': False, 'output': str(e)})
|
||||
return
|
||||
|
||||
if action is None:
|
||||
return
|
||||
|
||||
# Animated progress for callable actions
|
||||
spinner_chars = "|/-\\"
|
||||
spinner_idx = 0
|
||||
result_container = {}
|
||||
exception_container = {}
|
||||
done_event = threading.Event()
|
||||
|
||||
def run_action():
|
||||
try:
|
||||
result_container['result'] = action()
|
||||
except Exception as exc:
|
||||
exception_container['exception'] = exc
|
||||
finally:
|
||||
done_event.set()
|
||||
|
||||
worker = threading.Thread(target=run_action)
|
||||
worker.start()
|
||||
|
||||
while not done_event.is_set():
|
||||
spinner = f"{Colors.BLUE}[RUNNING]{Colors.RESET} {name} [{spinner_chars[spinner_idx]}]"
|
||||
self._update_task_line(task_id, spinner)
|
||||
spinner_idx = (spinner_idx + 1) % len(spinner_chars)
|
||||
time.sleep(0.1)
|
||||
|
||||
worker.join()
|
||||
|
||||
if exception_container:
|
||||
exception = exception_container['exception']
|
||||
self._update_task_line(task_id, f"{Colors.RED}[FAILED]{Colors.RESET} {name}")
|
||||
with self.lock:
|
||||
print(f"\nException for {name}: {e}")
|
||||
self.results.put({'name': name, 'success': False, 'output': str(e)})
|
||||
print(f"\nException for {name}: {exception}")
|
||||
self.results.put({'name': name, 'success': False, 'output': str(exception)})
|
||||
return
|
||||
|
||||
result = result_container.get('result')
|
||||
success = True
|
||||
note = ''
|
||||
output = ''
|
||||
|
||||
if isinstance(result, tuple):
|
||||
if len(result) == 3:
|
||||
success, note, output = result
|
||||
elif len(result) == 2:
|
||||
success, output = result
|
||||
elif isinstance(result, dict):
|
||||
success = bool(result.get('success', True))
|
||||
note = result.get('note', '')
|
||||
output = result.get('output', '')
|
||||
elif isinstance(result, bool):
|
||||
success = result
|
||||
elif result is not None:
|
||||
output = str(result)
|
||||
|
||||
if success:
|
||||
final_text = f"{Colors.GREEN}[COMPLETED]{Colors.RESET} {name}"
|
||||
if note:
|
||||
final_text += f" {note}"
|
||||
self._update_task_line(task_id, final_text)
|
||||
self.results.put({'name': name, 'success': True, 'output': output})
|
||||
else:
|
||||
self._update_task_line(task_id, f"{Colors.RED}[FAILED]{Colors.RESET} {name}")
|
||||
if output:
|
||||
with self.lock:
|
||||
print(f"\nError output for {name}:\n{output}")
|
||||
self.results.put({'name': name, 'success': False, 'output': output})
|
||||
|
||||
def run_all(self):
|
||||
"""Execute all tasks in parallel and return overall success status."""
|
||||
|
|
@ -143,7 +220,7 @@ def install_brew_packages():
|
|||
# Development tools
|
||||
"node", "go", "gcc", "gdb", "automake", "cmake", "jsdoc3", "oven-sh/bun/bun",
|
||||
# Text processing and search
|
||||
"ripgrep", "the_silver_searcher", "fd", "fzf", "bat", "ccat", "tree", "jq",
|
||||
"ripgrep", "the_silver_searcher", "fd", "fzf", "bat", "ccat", "tree", "jq", "yq",
|
||||
# File management
|
||||
"yazi", "sevenzip",
|
||||
# Git and version control
|
||||
|
|
@ -184,21 +261,133 @@ def install_brew_packages():
|
|||
print("✅ All packages already installed")
|
||||
return True
|
||||
|
||||
def get_npm_updates(packages):
|
||||
packages = list(dict.fromkeys(packages))
|
||||
|
||||
list_result = subprocess.run(
|
||||
['npm', 'list', '-g', '--json', '--depth=0'],
|
||||
capture_output=True,
|
||||
text=True
|
||||
)
|
||||
|
||||
installed_versions = {}
|
||||
if list_result.stdout.strip():
|
||||
try:
|
||||
data = json.loads(list_result.stdout)
|
||||
dependencies = data.get('dependencies', {})
|
||||
if isinstance(dependencies, dict):
|
||||
for name, info in dependencies.items():
|
||||
if isinstance(info, dict):
|
||||
version = info.get('version')
|
||||
if version:
|
||||
installed_versions[name] = version
|
||||
except json.JSONDecodeError:
|
||||
pass
|
||||
|
||||
outdated_cmd = ['npm', 'outdated', '-g', '--json'] + packages
|
||||
outdated_result = subprocess.run(
|
||||
outdated_cmd,
|
||||
capture_output=True,
|
||||
text=True
|
||||
)
|
||||
|
||||
if outdated_result.returncode not in (0, 1):
|
||||
raise subprocess.CalledProcessError(
|
||||
outdated_result.returncode,
|
||||
outdated_cmd,
|
||||
output=outdated_result.stdout,
|
||||
stderr=outdated_result.stderr
|
||||
)
|
||||
|
||||
outdated_info = {}
|
||||
stdout = outdated_result.stdout.strip()
|
||||
if stdout and stdout != 'null':
|
||||
try:
|
||||
data = json.loads(stdout)
|
||||
if isinstance(data, dict):
|
||||
outdated_info = data
|
||||
except json.JSONDecodeError:
|
||||
pass
|
||||
|
||||
to_install = []
|
||||
updates = []
|
||||
installs = []
|
||||
|
||||
for package in packages:
|
||||
if package in outdated_info:
|
||||
to_install.append(package)
|
||||
updates.append(package)
|
||||
continue
|
||||
if package not in installed_versions:
|
||||
to_install.append(package)
|
||||
installs.append(package)
|
||||
|
||||
return to_install, updates, installs
|
||||
|
||||
def format_package_for_install(package):
|
||||
return package if '@' in package[1:] else f"{package}@latest"
|
||||
|
||||
def create_npm_update_action(packages):
|
||||
packages = list(packages)
|
||||
|
||||
def action():
|
||||
to_install, updates, installs = get_npm_updates(packages)
|
||||
if not to_install:
|
||||
return True, "(up to date)", ""
|
||||
|
||||
install_args = ['npm', 'install', '-g', '-f'] + [format_package_for_install(pkg) for pkg in to_install]
|
||||
|
||||
try:
|
||||
result = subprocess.run(
|
||||
install_args,
|
||||
capture_output=True,
|
||||
text=True,
|
||||
check=True
|
||||
)
|
||||
note_parts = []
|
||||
if updates:
|
||||
note_parts.append(f"updated: {', '.join(updates)}")
|
||||
if installs:
|
||||
note_parts.append(f"installed: {', '.join(installs)}")
|
||||
note = f"({'; '.join(note_parts)})" if note_parts else ''
|
||||
output_parts = [part for part in [result.stdout, result.stderr] if part]
|
||||
return True, note, "".join(output_parts)
|
||||
except subprocess.CalledProcessError as err:
|
||||
output_parts = [part for part in [err.stdout, err.stderr] if part]
|
||||
output = "".join(output_parts) or str(err)
|
||||
return False, "", output
|
||||
|
||||
return action
|
||||
|
||||
def schedule_npm_updates(runner):
|
||||
if shutil.which('npm') is None:
|
||||
print(f"{Colors.RED}❌ npm not found; skipping npm package checks{Colors.RESET}")
|
||||
return
|
||||
|
||||
packages = [
|
||||
"@anthropic-ai/claude-code",
|
||||
"ccusage",
|
||||
"ccstatusline",
|
||||
"@openai/codex",
|
||||
"instant-markdown-d",
|
||||
]
|
||||
|
||||
runner.add_task("Node Apps Update", action=create_npm_update_action(packages))
|
||||
|
||||
def main():
|
||||
runner = ParallelTaskRunner()
|
||||
|
||||
|
||||
# Install missing brew packages first
|
||||
if not install_brew_packages():
|
||||
print("❌ Brew package installation failed")
|
||||
sys.exit(1)
|
||||
|
||||
runner.add_task("Claude Code Update", "npm install -g -f @anthropic-ai/claude-code ccusage ccstatusline @openai/codex@latest")
|
||||
|
||||
schedule_npm_updates(runner)
|
||||
runner.add_task("Homebrew Update", "brew update && brew upgrade --greedy")
|
||||
runner.add_task("Config Git Pull", "cd ~/.config && git pull")
|
||||
runner.add_task("Neovim Config Git Pull", "cd ~/.config/nvim && git pull")
|
||||
runner.add_task("SConfig Git Pull", "cd ~/.sconfig && git pull")
|
||||
runner.add_task("Node Apps Update", "npm install -g -f instant-markdown-d")
|
||||
|
||||
|
||||
success = runner.run_all()
|
||||
sys.exit(0 if success else 1)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue