from __future__ import annotations

import tempfile
from pathlib import Path

from pyro_mcp import Pyro


def main() -> None:
    pyro = Pyro()
    with (
        tempfile.TemporaryDirectory(prefix="pyro-workspace-seed-") as seed_dir,
        tempfile.TemporaryDirectory(prefix="pyro-workspace-sync-") as sync_dir,
    ):
        Path(seed_dir, "note.txt").write_text("hello from seed\n", encoding="utf-8")
        Path(sync_dir, "note.txt").write_text("hello from sync\n", encoding="utf-8")
        created = pyro.create_workspace(environment="debian:12", seed_path=seed_dir)
        workspace_id = str(created["workspace_id"])
        try:
            pyro.push_workspace_sync(workspace_id, sync_dir)
            result = pyro.exec_workspace(workspace_id, command="cat note.txt")
            print(result["stdout"], end="")
            logs = pyro.logs_workspace(workspace_id)
            print(f"workspace_id={workspace_id} command_count={logs['count']}")
        finally:
            pyro.delete_workspace(workspace_id)


if __name__ == "__main__":
    main()