Changes Initial commit

ota_agent/registry_login.py

@@ -0,0 +1,70 @@
+from __future__ import annotations
+
+import asyncio
+import logging
+
+from .compose_manager import ComposeManager
+from .models import CommandResult
+
+logger = logging.getLogger(__name__)
+
+
+class RegistryLoginManager:
+    def __init__(
+        self,
+        *,
+        enabled: bool,
+        server: str,
+        username: str,
+        password: str,
+        timeout_seconds: int,
+        compose_manager: ComposeManager,
+    ) -> None:
+        self.enabled = enabled
+        self.server = server
+        self.username = username
+        self.password = password
+        self.timeout_seconds = timeout_seconds
+        self.compose_manager = compose_manager
+
+    async def login_if_needed(self) -> CommandResult:
+        if not self.enabled:
+            return CommandResult(success=True, stdout="registry login skipped", returncode=0)
+
+        command = [
+            "docker",
+            "login",
+            self.server,
+            "-u",
+            self.username,
+            "-p",
+            self.password,
+        ]
+        logger.info("开始执行私有仓库登录: %s", self.server)
+        result = await self._run_command(command)
+        if result.success:
+            logger.info("私有仓库登录成功: %s", self.server)
+        else:
+            logger.error("私有仓库登录失败: %s", result.stderr or result.stdout)
+        return result
+
+    async def _run_command(self, command: list[str]) -> CommandResult:
+        process = await asyncio.create_subprocess_exec(
+            *command,
+            cwd=str(self.compose_manager.working_dir),
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE,
+        )
+        try:
+            stdout, stderr = await asyncio.wait_for(process.communicate(), timeout=self.timeout_seconds)
+        except asyncio.TimeoutError:
+            process.kill()
+            await process.wait()
+            return CommandResult(success=False, stderr=f"registry login timeout after {self.timeout_seconds}s", returncode=-1)
+
+        return CommandResult(
+            success=process.returncode == 0,
+            stdout=stdout.decode("utf-8", errors="ignore"),
+            stderr=stderr.decode("utf-8", errors="ignore"),
+            returncode=process.returncode or 0,
+        )