hopgo 0.1.0__tar.gz

hopgo-0.1.0/.gitignore

@@ -0,0 +1,10 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv

hopgo-0.1.0/.python-version

@@ -0,0 +1 @@
+3.14

hopgo-0.1.0/PKG-INFO

@@ -0,0 +1,7 @@
+Metadata-Version: 2.4
+Name: hopgo
+Version: 0.1.0
+Summary: A Pythonic multi-hop SSH/ADB/Docker automation framework
+Requires-Python: >=3.12
+Requires-Dist: fabric>=3.2
+Requires-Dist: rich>=13.0

hopgo-0.1.0/hopgo/__init__.py

@@ -0,0 +1,40 @@
+from .session import connect
+from .shell import Shell, Result
+from .hops import Hop, SSHHop, ADBHop, DockerHop
+from .files import FileInfo
+from .exceptions import (
+    HopError,
+    HopConnectionError,
+    HopSudoError,
+    HopSecurityError,
+    HopFileNotFoundError,
+    HopFileExistsError,
+    HopIsDirectoryError,
+    HopPermissionError,
+    HopTimeoutError,
+    HopFileSizeError,
+)
+
+MAX_READ_SIZE = 10 * 1024 * 1024
+
+__all__ = [
+    "connect",
+    "Shell",
+    "Result",
+    "Hop",
+    "SSHHop",
+    "ADBHop",
+    "DockerHop",
+    "FileInfo",
+    "HopError",
+    "HopConnectionError",
+    "HopSudoError",
+    "HopSecurityError",
+    "HopFileNotFoundError",
+    "HopFileExistsError",
+    "HopIsDirectoryError",
+    "HopPermissionError",
+    "HopTimeoutError",
+    "HopFileSizeError",
+    "MAX_READ_SIZE",
+]

hopgo-0.1.0/hopgo/cli.py

@@ -0,0 +1,11 @@
+#!/usr/bin/env python3
+"""HopGo CLI - Multi-hop SSH/ADB/Docker automation framework."""
+
+
+def main() -> None:
+    print("HopGo - A Pythonic multi-hop automation framework")
+    print("Use 'from hopgo import connect' in Python to get started.")
+
+
+if __name__ == "__main__":
+    main()

hopgo-0.1.0/hopgo/exceptions.py

@@ -0,0 +1,38 @@
+class HopError(Exception):
+    pass
+
+
+class HopConnectionError(HopError):
+    pass
+
+
+class HopSudoError(HopError):
+    pass
+
+
+class HopSecurityError(HopError):
+    pass
+
+
+class HopFileNotFoundError(HopError):
+    pass
+
+
+class HopFileExistsError(HopError):
+    pass
+
+
+class HopIsDirectoryError(HopError):
+    pass
+
+
+class HopPermissionError(HopError):
+    pass
+
+
+class HopTimeoutError(HopError):
+    pass
+
+
+class HopFileSizeError(HopError):
+    pass

hopgo-0.1.0/hopgo/files.py

@@ -0,0 +1,16 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from datetime import datetime
+from typing import Optional
+
+
+@dataclass
+class FileInfo:
+    name: str
+    path: str
+    size: int
+    is_dir: bool
+    is_symlink: bool
+    modified: Optional[datetime] = None
+    permissions: str = ""

hopgo-0.1.0/hopgo/hops.py

@@ -0,0 +1,91 @@
+from __future__ import annotations
+
+import shlex
+from abc import ABC, abstractmethod
+from typing import Optional
+
+
+class Hop(ABC):
+    target_os: str = "linux"
+
+    def __init__(self, sudo_password: Optional[str] = None) -> None:
+        self.sudo_password = sudo_password
+
+    @abstractmethod
+    def wrap(self, cmd: str) -> str:
+        ...
+
+
+class SSHHop(Hop):
+    def __init__(
+        self,
+        host: str,
+        user: Optional[str] = None,
+        password: Optional[str] = None,
+        key_filename: Optional[str] = None,
+        port: int = 22,
+        sudo_password: Optional[str] = None,
+    ) -> None:
+        super().__init__(sudo_password=sudo_password)
+        self.host = host
+        self.user = user
+        self.password = password
+        self.key_filename = key_filename
+        self.port = port
+        self.target_os = "linux"
+
+    def wrap(self, cmd: str) -> str:
+        parts = ["ssh", "-o", "StrictHostKeyChecking=no"]
+        if self.port != 22:
+            parts += ["-p", str(self.port)]
+        if self.key_filename:
+            parts += ["-i", shlex.quote(self.key_filename)]
+        user_host = f"{self.user}@{self.host}" if self.user else self.host
+        parts.append(user_host)
+        parts.append(shlex.quote(cmd))
+        return " ".join(parts)
+
+
+class ADBHop(Hop):
+    def __init__(
+        self,
+        serial: Optional[str] = None,
+        user: Optional[str] = None,
+        sudo_password: Optional[str] = None,
+    ) -> None:
+        super().__init__(sudo_password=sudo_password)
+        self.serial = serial
+        self.user = user
+        self.target_os = "linux"
+
+    def wrap(self, cmd: str) -> str:
+        parts = ["adb"]
+        if self.serial:
+            parts += ["-s", shlex.quote(self.serial)]
+        parts.append("shell")
+        if self.user:
+            parts += ["run-as", shlex.quote(self.user)]
+        parts.append(shlex.quote(cmd))
+        return " ".join(parts)
+
+
+class DockerHop(Hop):
+    def __init__(
+        self,
+        container: str,
+        user: Optional[str] = None,
+        host_os: str = "linux",
+        sudo_password: Optional[str] = None,
+    ) -> None:
+        super().__init__(sudo_password=sudo_password)
+        self.container = container
+        self.user = user
+        self.target_os = host_os
+
+    def wrap(self, cmd: str) -> str:
+        parts = ["docker", "exec", "-i"]
+        if self.user:
+            parts += ["-u", shlex.quote(self.user)]
+        parts.append(shlex.quote(self.container))
+        parts += ["sh", "-c", shlex.quote(cmd)]
+        return " ".join(parts)

hopgo-0.1.0/hopgo/session.py

@@ -0,0 +1,36 @@
+from __future__ import annotations
+
+import sys
+from typing import Optional, List
+
+from .shell import Shell
+
+
+class Session:
+    def __init__(self, _ssh_conn=None) -> None:
+        self._ssh_conn = _ssh_conn
+
+
+def connect(
+    host: Optional[str] = None,
+    hops: Optional[List] = None,
+    **ssh_kwargs,
+) -> Shell:
+    ssh_conn = None
+    if host is not None:
+        from fabric import Connection
+        ssh_conn = Connection(host, **ssh_kwargs)
+        try:
+            ssh_conn.open()
+        except Exception as e:
+            from .exceptions import HopConnectionError
+            raise HopConnectionError(f"SSH connection to {host} failed: {e}")
+
+    session = Session(_ssh_conn=ssh_conn)
+    target_os = "windows" if sys.platform == "win32" else "linux"
+
+    return Shell(
+        _session=session,
+        _cwd="/",
+        _target_os=target_os,
+    )