Harbor adapter

import asyncio
import tempfile
from pathlib import Path

from harbor.models.task.config import EnvironmentConfig
from harbor.models.trial.paths import TrialPaths
from harbor_box_environment import BoxEnvironment


async def main() -> None:
    with tempfile.TemporaryDirectory() as raw:
        tmp = Path(raw)
        # Harbor needs an environment directory. Boat reads only the WORKDIR of the Dockerfile.
        env_dir = tmp / "environment"
        env_dir.mkdir()
        (env_dir / "Dockerfile").write_text("FROM ubuntu:24.04\nWORKDIR /workspace\n")
        trial_paths = TrialPaths(tmp / "trial")
        trial_paths.mkdir()

        env = BoxEnvironment(
            environment_dir=env_dir,
            environment_name="harbor-hello",
            session_id="hello-session",
            trial_paths=trial_paths,
            task_env_config=EnvironmentConfig(workdir="/workspace"),
            ttl_seconds=600,  # auto-stop 10 minutes after creation
        )

        await env.start(force_build=False)  # creates a sandbox and waits until it is ready
        try:
            await env.upload_file(__file__, "/workspace/hello_sandbox.py")
            result = await env.exec("echo 'hello from Boat' > notes.txt && cat notes.txt && uname -a")
            print("exit:", result.return_code)
            print(result.stdout)
        finally:
            await env.stop(delete=False)  # stop; delete=True fails in 0.1.2


if __name__ == "__main__":
    asyncio.run(main())