Barmaja-PyArduino 2.0.0__py3-none-any.whl

Barmaja_PyArduino/installer.py

@@ -0,0 +1,151 @@
+from __future__ import annotations
+
+import os
+import shutil
+import sys
+import zipfile
+from dataclasses import dataclass
+from pathlib import Path
+
+from .environment import ensure_runtime_dirs, pictoblox_module_dirs, run_text
+
+@dataclass(frozen=True)
+class InstallTarget:
+    kind: str
+    path: Path
+    label: str
+
+@dataclass(frozen=True)
+class InstallResult:
+    target: InstallTarget
+    ok: bool
+    message: str
+
+
+def discover_install_targets() -> tuple[InstallTarget, ...]:
+    targets = [InstallTarget("python", Path(sys.executable), f"Python {sys.version_info.major}.{sys.version_info.minor}")]
+    for p in pictoblox_module_dirs():
+        targets.append(InstallTarget("pictoblox", p, "PictoBlox"))
+    return tuple(targets)
+
+
+def _extract_pure_python_wheel(wheel: Path, target_dir: Path) -> None:
+    with zipfile.ZipFile(wheel) as z:
+        names = z.namelist()
+        if any(".data/" in n or n.endswith((".pyd", ".so", ".dll", ".dylib")) for n in names):
+            raise ValueError("PictoBlox offline installation only accepts the Barmaja pure-Python wheel.")
+        target_dir.mkdir(parents=True, exist_ok=True)
+        # Replace, never merge: stale modules from an older version must not survive.
+        existing = target_dir / "Barmaja_PyArduino"
+        if existing.exists():
+            shutil.rmtree(existing)
+        for member in names:
+            if member.startswith("Barmaja_PyArduino/"):
+                z.extract(member, target_dir)
+    if not (target_dir / "Barmaja_PyArduino" / "__init__.py").exists():
+        raise ValueError("Wheel does not contain Barmaja_PyArduino package files.")
+
+
+def install_offline_wheel(wheel: str | Path, target: InstallTarget) -> InstallResult:
+    wheel = Path(wheel).resolve()
+    if not wheel.exists():
+        return InstallResult(target, False, f"wheel not found: {wheel}")
+    try:
+        if target.kind == "python":
+            cp = run_text([str(target.path), "-m", "pip", "install", "--no-index", "--no-deps", str(wheel)], timeout=120)
+            if cp.returncode != 0:
+                return InstallResult(target, False, (cp.stderr or cp.stdout).strip() or "pip install failed")
+            return InstallResult(target, True, "installed")
+        if target.kind == "pictoblox":
+            _extract_pure_python_wheel(wheel, target.path)
+            return InstallResult(target, True, "installed into PictoBlox python_modules")
+        return InstallResult(target, False, "unsupported install target")
+    except Exception as e:
+        return InstallResult(target, False, str(e))
+
+
+PINNED_LIBRARY_FOLDERS = {
+    "Servo": "1.3.0",
+    "Ultrasonic": "3.0.0",
+    "DHT sensor library": "1.4.7",
+    "Adafruit Unified Sensor": "1.1.15",
+    "LiquidCrystal I2C": "1.1.4",
+}
+
+
+def install_private_arduino_libraries(source_dir: str | Path) -> tuple[str, ...]:
+    """Install bundled pinned Arduino libraries into Barmaja's private library dir.
+
+    This avoids modifying or overwriting the user's Arduino IDE libraries.
+    """
+    source = Path(source_dir)
+    dest_root = ensure_runtime_dirs()["root"] / "arduino-libraries"
+    dest_root.mkdir(parents=True, exist_ok=True)
+    installed=[]
+    for child in sorted(source.iterdir()) if source.exists() else []:
+        if not child.is_dir():
+            continue
+        props = child / "library.properties"
+        if not props.exists():
+            continue
+        text = props.read_text(encoding="utf-8", errors="replace")
+        if "name=" not in text:
+            continue
+        dest = dest_root / child.name
+        if dest.exists(): shutil.rmtree(dest)
+        shutil.copytree(child, dest)
+        installed.append(child.name)
+    return tuple(installed)
+
+
+def install_bundled_runtime(source_dir: str | Path) -> Path:
+    source = Path(source_dir)
+    root = ensure_runtime_dirs()["root"]
+    dest = root / "runtime"
+    if dest.exists():
+        shutil.rmtree(dest)
+    shutil.copytree(source, dest)
+
+    # The Full Pack is relocatable at install time. Rebuild the Arduino CLI
+    # config with the actual destination instead of shipping machine-specific
+    # absolute paths.
+    data = dest / "arduino-data"
+    user = dest / "arduino-user"
+    downloads = dest / "downloads"
+    for folder in (data, user, downloads):
+        folder.mkdir(parents=True, exist_ok=True)
+    build_cache = root / "build-cache"
+    build_cache.mkdir(parents=True, exist_ok=True)
+    def q(path: Path) -> str:
+        return '"' + str(path).replace("\\", "/").replace('"', '\\"') + '"'
+    config = (
+        "board_manager:\n"
+        "  additional_urls: []\n"
+        "build_cache:\n"
+        f"  path: {q(build_cache)}\n"
+        "directories:\n"
+        f"  data: {q(data)}\n"
+        f"  downloads: {q(downloads)}\n"
+        f"  user: {q(user)}\n"
+        "locale: en\n"
+        "logging:\n"
+        "  format: text\n"
+        "  level: warn\n"
+        "metrics:\n"
+        "  enabled: false\n"
+        "output:\n"
+        "  no_color: true\n"
+        "updater:\n"
+        "  enable_notification: false\n"
+    )
+    (dest / "arduino-cli.yaml").write_text(config, encoding="utf-8")
+    return dest
+
+
+def install_offline_package(wheel: str | Path, *, include_pictoblox: bool = True) -> tuple[InstallResult, ...]:
+    results=[]
+    for target in discover_install_targets():
+        if target.kind == "pictoblox" and not include_pictoblox:
+            continue
+        results.append(install_offline_wheel(wheel, target))
+    return tuple(results)