init commit v0.7master

1 files changed, 249 insertions, 0 deletions

diff --git a/package.py b/package.py
new file mode 100644
index 0000000..2667c7d
--- /dev/null
+++ b/package.py
@@ -0,0 +1,249 @@
+import os
+import sys
+import tarfile
+import shutil
+import tempfile
+import subprocess
+import urllib.request
+from urllib.parse import urlsplit, urlunsplit
+
+from logs import info, warn, error
+from paths import TMP_DIR, CACHE_DIR, REPOS_DIR
+from pathlib import Path
+from uzbekdb import database
+from database import (
+    download_repos,
+    get_repos_urls,
+    install_package,
+    remove_package,
+    is_installed,
+)
+
+
+_NAME = os.path.basename(sys.argv[0])
+
+
+def _random_tmp_dir(tmp_dir=TMP_DIR):
+    return tempfile.mkdtemp(prefix="qulay_", dir=tmp_dir)
+
+
+def _download(url, dest):
+    try:
+        urllib.request.urlretrieve(url, dest)
+    except Exception as e:
+        return e
+    else:
+        return 1
+    return 0
+
+
+def _extract(archive, dest):
+    with tarfile.open(archive) as tar:
+        tar.extractall(dest)
+
+
+def _read_depends(path):
+    depends_file = os.path.join(path, "depends.uz")
+    if not os.path.exists(depends_file):
+        return []
+    with open(depends_file) as f:
+        return [line.strip() for line in f if line.strip()]
+
+
+def _check_pkg_structure(path):
+    required = ["install.sh", "remove.sh"]
+    for file in required:
+        if not os.path.exists(os.path.join(path, file)):
+            error("!! package is corrupted", True)
+
+
+def _get_packages(dest_dir="/"):
+    found = []
+    packages_path = os.path.join(dest_dir, REPOS_DIR.lstrip(os.sep))
+    for repo_name in os.listdir(packages_path):
+        repo_path = os.path.join(packages_path, repo_name)
+        if os.path.isdir(repo_path):
+            pkgs_file = os.path.join(repo_path, "pkgs.uz")
+            if os.path.exists(pkgs_file):
+                try:
+                    with open(pkgs_file, "r", encoding="utf-8") as f:
+                        pkgs_data = database.loads(f.read())
+                        for name, data in pkgs_data.items():
+                            found.append({
+                                "name": name,
+                                "repo": repo_name,
+                                "data": data
+                            })
+                except Exception:
+                    continue
+    return found
+
+    
+def _find_package(name, dest_dir="/"):
+    found = []
+    packages_path = os.path.join(dest_dir, REPOS_DIR.lstrip(os.sep))
+    
+    if not os.path.exists(packages_path):
+        return "-n"
+
+    for repo_name in os.listdir(packages_path):
+        repo_path = os.path.join(packages_path, repo_name)
+        if os.path.isdir(repo_path):
+            pkgs_file = os.path.join(repo_path, "pkgs.uz")
+            if os.path.exists(pkgs_file):
+                try:
+                    with open(pkgs_file, "r", encoding="utf-8") as f:
+                        pkgs_data = database.loads(f.read())
+                        
+                        if isinstance(pkgs_data, dict) and name in pkgs_data:
+                            found.append({
+                                "name": name,
+                                "repo": repo_name,
+                                "data": pkgs_data[name]
+                            })
+                except Exception:
+                    continue
+
+    if not found: 
+        return None
+    # if len(found) > 1: 
+    #     return "-r"
+    
+    return found
+
+
+def _find_repo_package(result, repo, name):
+    for pkg in result:
+        if pkg["repo"] == repo:
+            return pkg
+    return None
+    
+
+def install(name, args=[], dest_dir="/"):
+    # verbose=False, ask=False, reinstall=False, disable_logs=False, disable_deps=False
+    verbose = ":v" in args
+    ask = ":a" in args
+    reinstall = ":r" in args
+    disable_logs = ":nl" in args
+    disable_deps = ":nd" in args
+
+    if "/" in name:
+        parts = name.split("/", 1)
+        repo_name = parts[0]
+        name = parts[1]
+        
+        pkg = _find_repo_package(_find_package(name), repo_name, name)
+
+        if not pkg:
+            error(f"!! package {name} not found in {repo_name}", True, disable_logs)
+            return 1
+
+        pkg_data = pkg["data"]
+        version = pkg_data[1]
+    else:     
+        result = _find_package(name)
+        if not result:
+            error(f"!! package {name} not found", True, disable_logs)
+            return 1 
+
+        if len(result) > 1: 
+            error(f"!! {len(result)} same pkgs found in different repos:", True, disable_logs)
+            for pkg in result:
+                print(f"   {pkg['repo']}/{pkg['name']} - {pkg['data'][0]} - {pkg['data'][1]}")
+            return 1
+
+        repo_name = result[0]["repo"]
+        pkg_data = result[0]["data"]
+
+        version = pkg_data[1]
+
+    pkg_path = os.path.join(dest_dir, REPOS_DIR.lstrip(os.sep), f"{repo_name}/packages/{name}")
+
+    if ask:
+        download_answer = input(
+            f"-> confirm installing '{name}' with version {version} [Y/n]: "
+        ).strip().lower()
+
+        if download_answer not in ("y", "yes", ""):
+            sys.exit(0)
+
+    # -- install depends --
+    depends = _read_depends(pkg_path)
+    if not disable_deps:
+        for dep in depends:
+            if not is_installed(dep) or reinstall:
+                # -- copy args to dep install --
+                install(dep, args, dest_dir=dest_dir)
+    
+    # -- run install.sh --
+    try:
+        subprocess.run(
+            ["sh", "install.sh"],
+            cwd=pkg_path,
+            check=True,
+            env=os.environ
+        )
+    except Exception as e:
+        error(f"!! {e}", True, disable_logs)
+        return 0
+
+    if not disable_logs:
+        install_package(f"{repo_name}/{name}", version, dest_dir=dest_dir)
+
+    info(f":: {name} installed successfully", True, disable_logs)
+
+
+def remove(name, args=[], dest_dir="/"):
+    verbose = ":v" in args
+    ask = ":a" in args
+    disable_logs = ":nl" in args
+    # disable_deps = "/nd" in args
+
+    if "/" in name:
+        parts = name.split("/", 1)
+        repo_name = parts[0]
+        name = parts[1]
+        
+        pkg = _find_repo_package(_find_package(name), repo_name, name)
+
+        if not pkg:
+            error(f"!! package {name} not found in {repo_name}", True, disable_logs)
+            return 1
+
+        pkg_data = pkg["data"]
+        version = pkg_data[1]
+    else:     
+        result = _find_package(name)
+        if not result:
+            error(f"!! package {name} not found", True, disable_logs)
+            return 1 
+
+        if len(result) > 1: 
+            error(f"!! {len(result)} same pkgs found in different repos:", True, disable_logs)
+            for pkg in result:
+                print(f"   {pkg['repo']}/{pkg['name']} - {pkg['data'][0]} - {pkg['data'][1]}")
+            return 1
+
+        repo_name = result[0]["repo"]
+        pkg_data = result[0]["data"]
+
+        version = pkg_data[1]
+
+    # -- super uzbek package remover da --
+    pkg_path = os.path.join(dest_dir, REPOS_DIR.lstrip(os.sep), f"{repo_name}/packages/{name}")
+
+    try:
+        subprocess.run(
+            ["sh", "remove.sh"],
+            cwd=pkg_path,
+            check=True,
+            env=os.environ
+        )
+    except Exception as e:
+        error(f"!! {e}", True, disable_logs)
+        return 0
+
+    if not disable_logs:
+        remove_package(f"{repo_name}/{name}", dest_dir=dest_dir)
+    
+    info(f":: {name} removed successfully", True, disable_logs)