Finish refactoring the Exporter class and improve hooks handling

Author: kirkbrauer

packages/jumpstarter/jumpstarter/common/utils.py

@@ -5,20 +5,23 @@
 from datetime import timedelta
 from functools import partial
 from subprocess import Popen
+from typing import TYPE_CHECKING
 
 from anyio.from_thread import BlockingPortal, start_blocking_portal
 
 from jumpstarter.client import client_from_path
 from jumpstarter.config.env import JMP_DRIVERS_ALLOW, JUMPSTARTER_HOST
-from jumpstarter.driver import Driver
 from jumpstarter.exporter import Session
 from jumpstarter.utils.env import env
 
+if TYPE_CHECKING:
+    from jumpstarter.driver import Driver
+
 __all__ = ["env"]
 
 
 @asynccontextmanager
-async def serve_async(root_device: Driver, portal: BlockingPortal, stack: ExitStack):
+async def serve_async(root_device: "Driver", portal: BlockingPortal, stack: ExitStack):
     with Session(root_device=root_device) as session:
         async with session.serve_unix_async() as path:
             # SAFETY: the root_device instance is constructed locally thus considered trusted
@@ -31,7 +34,7 @@ async def serve_async(root_device: Driver, portal: BlockingPortal, stack: ExitSt
 
 
 @contextmanager
-def serve(root_device: Driver):
+def serve(root_device: "Driver"):
     with start_blocking_portal() as portal:
         with ExitStack() as stack:
             with portal.wrap_async_context_manager(serve_async(root_device, portal, stack)) as client:

packages/jumpstarter/jumpstarter/config/exporter.py

@@ -2,7 +2,7 @@
 
 from contextlib import asynccontextmanager, contextmanager, suppress
 from pathlib import Path
-from typing import Any, ClassVar, Literal, Optional, Self
+from typing import TYPE_CHECKING, Any, ClassVar, Literal, Optional, Self
 
 import grpc
 import yaml
@@ -15,7 +15,9 @@
 from jumpstarter.common.exceptions import ConfigurationError
 from jumpstarter.common.grpc import aio_secure_channel, ssl_channel_credentials
 from jumpstarter.common.importlib import import_class
-from jumpstarter.driver import Driver
+
+if TYPE_CHECKING:
+    from jumpstarter.driver import Driver
 
 
 class HookInstanceConfigV1Alpha1(BaseModel):
@@ -71,7 +73,7 @@ class ExporterConfigV1Alpha1DriverInstance(RootModel):
         | ExporterConfigV1Alpha1DriverInstanceProxy
     )
 
-    def instantiate(self) -> Driver:
+    def instantiate(self) -> "Driver":
         match self.root:
             case ExporterConfigV1Alpha1DriverInstanceBase():
                 driver_class = import_class(self.root.type, [], True)