mirror of
https://github.com/ArchiveBox/ArchiveBox.git
synced 2026-01-03 01:15:57 +10:00
88 lines
3.0 KiB
Python
88 lines
3.0 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Install a binary using apt package manager.
|
|
|
|
Usage: on_Dependency__install_using_apt_provider.py --dependency-id=<uuid> --bin-name=<name> [--custom-cmd=<cmd>]
|
|
Output: InstalledBinary JSONL record to stdout after installation
|
|
|
|
Environment variables:
|
|
MACHINE_ID: Machine UUID (set by orchestrator)
|
|
"""
|
|
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
import rich_click as click
|
|
from abx_pkg import Binary, AptProvider, BinProviderOverrides
|
|
|
|
# Fix pydantic forward reference issue
|
|
AptProvider.model_rebuild()
|
|
|
|
|
|
@click.command()
|
|
@click.option('--dependency-id', required=True, help="Dependency UUID")
|
|
@click.option('--bin-name', required=True, help="Binary name to install")
|
|
@click.option('--bin-providers', default='*', help="Allowed providers (comma-separated)")
|
|
@click.option('--custom-cmd', default=None, help="Custom install command (overrides default)")
|
|
@click.option('--overrides', default=None, help="JSON-encoded overrides dict")
|
|
def main(dependency_id: str, bin_name: str, bin_providers: str, custom_cmd: str | None, overrides: str | None):
|
|
"""Install binary using apt package manager."""
|
|
|
|
# Check if apt provider is allowed
|
|
if bin_providers != '*' and 'apt' not in bin_providers.split(','):
|
|
click.echo(f"apt provider not allowed for {bin_name}", err=True)
|
|
sys.exit(0) # Not an error, just skip
|
|
|
|
# Use abx-pkg AptProvider to install binary
|
|
provider = AptProvider()
|
|
if not provider.INSTALLER_BIN:
|
|
click.echo("apt not available on this system", err=True)
|
|
sys.exit(1)
|
|
|
|
click.echo(f"Installing {bin_name} via apt...", err=True)
|
|
|
|
try:
|
|
# Parse overrides if provided
|
|
overrides_dict = None
|
|
if overrides:
|
|
try:
|
|
overrides_dict = json.loads(overrides)
|
|
click.echo(f"Using custom install overrides: {overrides_dict}", err=True)
|
|
except json.JSONDecodeError:
|
|
click.echo(f"Warning: Failed to parse overrides JSON: {overrides}", err=True)
|
|
|
|
binary = Binary(name=bin_name, binproviders=[provider], overrides=overrides_dict or {}).install()
|
|
except Exception as e:
|
|
click.echo(f"apt install failed: {e}", err=True)
|
|
sys.exit(1)
|
|
|
|
if not binary.abspath:
|
|
click.echo(f"{bin_name} not found after apt install", err=True)
|
|
sys.exit(1)
|
|
|
|
machine_id = os.environ.get('MACHINE_ID', '')
|
|
|
|
# Output InstalledBinary JSONL record to stdout
|
|
record = {
|
|
'type': 'InstalledBinary',
|
|
'name': bin_name,
|
|
'abspath': str(binary.abspath),
|
|
'version': str(binary.version) if binary.version else '',
|
|
'sha256': binary.sha256 or '',
|
|
'binprovider': 'apt',
|
|
'machine_id': machine_id,
|
|
'dependency_id': dependency_id,
|
|
}
|
|
print(json.dumps(record))
|
|
|
|
# Log human-readable info to stderr
|
|
click.echo(f"Installed {bin_name} at {binary.abspath}", err=True)
|
|
click.echo(f" version: {binary.version}", err=True)
|
|
|
|
sys.exit(0)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|