69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
from pathlib import Path
|
|
from typing import List
|
|
|
|
import httpx
|
|
import typer
|
|
from plumbum import FG, ProcessExecutionError, local
|
|
from rich.console import Console
|
|
from rich.progress import Progress
|
|
|
|
TOOLS = Path("tools")
|
|
ORIGINAL = Path("original")
|
|
MODIFIED = Path("modified")
|
|
DECOMPILED = Path("decompiled")
|
|
PATCHES = Path("patches")
|
|
CONFIGS = Path("configs")
|
|
|
|
|
|
def ensure_dirs():
|
|
for d in [TOOLS, ORIGINAL, MODIFIED, DECOMPILED, PATCHES, CONFIGS]:
|
|
d.mkdir(exist_ok=True)
|
|
|
|
|
|
def run(console: Console, cmd: List[str], hide_output=True):
|
|
prog = local[cmd[0]][cmd[1:]]
|
|
try:
|
|
prog() if hide_output else prog & FG
|
|
except ProcessExecutionError as e:
|
|
console.print(f"[red]Ошибка при выполнении команды: {' '.join(cmd)}")
|
|
console.print(e.stderr)
|
|
raise typer.Exit(1)
|
|
|
|
|
|
def download(console: Console, url: str, dest: Path):
|
|
"""Скачивание файла по URL"""
|
|
console.print(f"[cyan]Скачивание {url} → {dest.name}")
|
|
|
|
with httpx.Client(follow_redirects=True, timeout=60.0) as client:
|
|
with client.stream("GET", url) as response:
|
|
response.raise_for_status()
|
|
total = int(response.headers.get("Content-Length", 0))
|
|
|
|
dest.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
with open(dest, "wb") as f, Progress(console=console) as progress:
|
|
task = progress.add_task("Загрузка", total=total if total else None)
|
|
for chunk in response.iter_bytes(chunk_size=8192):
|
|
f.write(chunk)
|
|
progress.update(task, advance=len(chunk))
|
|
|
|
|
|
def select_apk(console) -> Path:
|
|
"""Выбор APK файла из папки original"""
|
|
apks = list(ORIGINAL.glob("*.apk"))
|
|
|
|
if not apks:
|
|
raise BuildError("Нет apk-файлов в папке original")
|
|
|
|
if len(apks) == 1:
|
|
console.print(f"[green]Выбран {apks[0].name}")
|
|
return apks[0]
|
|
|
|
console.print("[cyan]Доступные APK файлы:")
|
|
options = {str(i): apk for i, apk in enumerate(apks, 1)}
|
|
for k, v in options.items():
|
|
console.print(f" {k}. {v.name}")
|
|
|
|
choice = Prompt.ask("Выберите номер", choices=list(options.keys()))
|
|
return options[choice]
|