Finish migration to ResourceUri and fix lint_files_dispatch handler (handle partial results correctly)

Author: Aksem

docs/wm-protocol.md

@@ -9,7 +9,7 @@ connect to it.
 - TCP on `127.0.0.1`, random free port
 - Content-Length framing (same as LSP): `Content-Length: N\r\n\r\n{json_body}`
 - Discovery: port written to `.venvs/dev_workspace/cache/finecode/wm_port`
-- Auto-stops when the last client disconnects (after a 5s grace period) or if no client connects 30 seconds after start of WM Server
+- Auto-stops when the last client disconnects (after a 30s grace period by default, configurable via `--disconnect-timeout`) or if no client connects within 30 seconds after WM Server startup
 
 ## JSON-RPC 2.0
 
@@ -715,11 +715,13 @@ Clients can log or display this path so the user can open the file directly when
 
 #### `server/shutdown`
 
-Explicitly shut down the WM Server.
+Explicitly shut down the WM Server. Clients can use this when they intentionally
+want the WM to stop or restart, rather than waiting for disconnect-timeout
+auto-shutdown.
 
 - **Type:** request
 - **Clients:** any
-- **Status:** stub
+- **Status:** implemented
 
 **Params:** `{}`
 

extensions/fine_python_package_info/fine_python_package_info/is_artifact_published_to_registry_py_handler.py

@@ -3,6 +3,7 @@
 from finecode_extension_api import code_action
 from finecode_extension_api.actions.artifact import get_src_artifact_registries_action
 from finecode_extension_api.actions.publishing import is_artifact_published_to_registry_action
+from finecode_extension_api.resource_uri import resource_uri_to_path
 from finecode_extension_api.interfaces import (
     iactionrunner,
     ihttpclient,
@@ -132,7 +133,7 @@ async def run(
             except KeyError as exception:
                 raise code_action.ActionFailedException("File object has no 'filename' key") from exception
 
-            is_published_by_dist_path = {dist_path: dist_path.name in published_file_names for dist_path in dist_artifact_paths}
+            is_published_by_dist_path = {dist_path: resource_uri_to_path(dist_path).name in published_file_names for dist_path in dist_artifact_paths}
         else:
             is_published_by_dist_path = {dist_path: False for dist_path in dist_artifact_paths}
 

finecode_builtin_handlers/src/finecode_builtin_handlers/create_envs_discover_envs.py

@@ -7,6 +7,7 @@
     ilogger,
     iprojectinfoprovider,
 )
+from finecode_extension_api.resource_uri import path_to_resource_uri
 
 
 @dataclasses.dataclass
@@ -52,10 +53,10 @@ async def run(
             envs = [
                 create_envs_action.EnvInfo(
                     name=env_name,
-                    venv_dir_path=self.runner_info_provider.get_venv_dir_path_of_env(
-                        env_name
+                    venv_dir_path=path_to_resource_uri(
+                        self.runner_info_provider.get_venv_dir_path_of_env(env_name)
                     ),
-                    project_def_path=project_def_path,
+                    project_def_path=path_to_resource_uri(project_def_path),
                 )
                 for env_name in deps_groups
             ]

finecode_builtin_handlers/src/finecode_builtin_handlers/install_env_install_deps.py

@@ -9,6 +9,7 @@
     InstallEnvsRunResult,
 )
 from finecode_extension_api.interfaces import iactionrunner, ilogger
+from finecode_extension_api.resource_uri import path_to_resource_uri, resource_uri_to_path
 from finecode_builtin_handlers.dependency_config_utils import process_raw_deps
 
 
@@ -53,19 +54,20 @@ async def run(
             .get(env.name, {})
             .get("dependencies", {})
         )
+        project_def_path = resource_uri_to_path(env.project_def_path)
         dependencies: list[dict] = []
         process_raw_deps(
             env_raw_deps,
             env_deps_config,
             dependencies,
             deps_groups,
-            project_def_path=env.project_def_path,
+            project_def_path=project_def_path,
         )
 
         install_deps_payload = install_deps_in_env_action.InstallDepsInEnvRunPayload(
             env_name=env.name,
             venv_dir_path=env.venv_dir_path,
-            project_dir_path=env.project_def_path.parent,
+            project_dir_path=path_to_resource_uri(project_def_path.parent),
             dependencies=[
                 install_deps_in_env_action.Dependency(
                     name=dep["name"],

finecode_builtin_handlers/src/finecode_builtin_handlers/install_env_read_config.py

@@ -6,6 +6,7 @@
     InstallEnvsRunResult,
 )
 from finecode_extension_api.interfaces import ilogger, iprojectinfoprovider
+from finecode_extension_api.resource_uri import resource_uri_to_path
 from finecode_builtin_handlers import dependency_config_utils
 
 
@@ -32,11 +33,12 @@ async def run(
         payload: install_env_action.InstallEnvRunPayload,
         run_context: install_env_action.InstallEnvRunContext,
     ) -> InstallEnvsRunResult:
+        project_def_path = resource_uri_to_path(payload.env.project_def_path)
         project_raw_config = await self.project_info_provider.get_project_raw_config(
-            payload.env.project_def_path
+            project_def_path
         )
         dependency_config_utils.make_project_config_pip_compatible(
-            project_raw_config, payload.env.project_def_path
+            project_raw_config, project_def_path
         )
         run_context.project_def = project_raw_config
         return InstallEnvsRunResult(errors=[])

finecode_builtin_handlers/src/finecode_builtin_handlers/install_envs_discover_envs.py

@@ -8,6 +8,7 @@
     ilogger,
     iprojectinfoprovider,
 )
+from finecode_extension_api.resource_uri import path_to_resource_uri
 
 
 @dataclasses.dataclass
@@ -55,10 +56,10 @@ async def run(
             envs = [
                 EnvInfo(
                     name=env_name,
-                    venv_dir_path=self.runner_info_provider.get_venv_dir_path_of_env(
-                        env_name
+                    venv_dir_path=path_to_resource_uri(
+                        self.runner_info_provider.get_venv_dir_path_of_env(env_name)
                     ),
-                    project_def_path=project_def_path,
+                    project_def_path=path_to_resource_uri(project_def_path),
                 )
                 for env_name in deps_groups
             ]

finecode_builtin_handlers/src/finecode_builtin_handlers/lint_files_dispatch.py

@@ -1,10 +1,10 @@
-import asyncio
 import dataclasses
 
 from finecode_extension_api import code_action
 from finecode_extension_api.actions.artifact import group_src_artifact_files_by_lang_action
 from finecode_extension_api.actions.code_quality import lint_files_action
 from finecode_extension_api.interfaces import iactionrunner, ilogger
+from finecode_extension_api.resource_uri import ResourceUri
 
 
 @dataclasses.dataclass
@@ -17,7 +17,7 @@ class LintFilesDispatchHandler(
         LintFilesDispatchHandlerConfig,
     ]
 ):
-    """Group files by language once and dispatch to lint_{lang}_files subactions concurrently.
+    """Group files by language once and dispatch to lint_{lang}_files subactions.
 
     Subaction names follow the convention: language "python" maps to "lint_python_files",
     "javascript" maps to "lint_javascript_files", etc. Each subaction must be registered
@@ -32,18 +32,30 @@ def __init__(
         self.action_runner = action_runner
         self.logger = logger
 
+    async def _lint_file(
+        self,
+        subaction: iactionrunner.ActionDeclaration[lint_files_action.LintFilesAction],
+        file_uri: ResourceUri,
+        meta: code_action.RunActionMeta,
+    ) -> lint_files_action.LintFilesRunResult:
+        return await self.action_runner.run_action(
+            action=subaction,
+            payload=lint_files_action.LintFilesRunPayload(file_paths=[file_uri]),
+            meta=meta,
+        )
+
     async def run(
         self,
         payload: lint_files_action.LintFilesRunPayload,
         run_context: lint_files_action.LintFilesRunContext,
-    ) -> lint_files_action.LintFilesRunResult:
+    ) -> None:
         subactions_by_lang = self.action_runner.get_actions_for_parent(
             lint_files_action.LintFilesAction
         )
 
         if not subactions_by_lang:
             self.logger.debug("LintFilesDispatchHandler: no language subactions registered")
-            return lint_files_action.LintFilesRunResult(messages={})
+            return
 
         # Group files by language — single pass, O(files).
         group_action = self.action_runner.get_action_by_source(
@@ -59,27 +71,17 @@ async def run(
         )
         files_by_lang = files_by_lang_result.files_by_lang
 
-        # Dispatch concurrently — each subaction receives only its language's files.
-        lint_tasks: list[asyncio.Task[lint_files_action.LintFilesRunResult]] = []
-        try:
-            async with asyncio.TaskGroup() as tg:
-                for lang, files in files_by_lang.items():
-                    if not files:
-                        continue
-                    lint_tasks.append(
-                        tg.create_task(
-                            self.action_runner.run_action(
-                                action=subactions_by_lang[lang],
-                                payload=lint_files_action.LintFilesRunPayload(file_paths=files),
-                                meta=run_context.meta,
-                            )
-                        )
-                    )
-        except ExceptionGroup as eg:
-            error_str = ". ".join([str(e) for e in eg.exceptions])
-            raise code_action.ActionFailedException(error_str) from eg
+        # Build reverse mapping: file → language subaction.
+        file_to_subaction: dict[ResourceUri, iactionrunner.ActionDeclaration[lint_files_action.LintFilesAction]] = {}
+        for lang, files in files_by_lang.items():
+            for file_uri in files:
+                file_to_subaction[file_uri] = subactions_by_lang[lang]
 
-        result = lint_files_action.LintFilesRunResult(messages={})
-        for task in lint_tasks:
-            result.update(task.result())
-        return result
+        # Schedule per-file coroutines via partial_result_scheduler so that
+        # run_action can execute them concurrently and send partial results.
+        for file_uri in payload.file_paths:
+            if file_uri in file_to_subaction:
+                run_context.partial_result_scheduler.schedule(
+                    file_uri,
+                    self._lint_file(file_to_subaction[file_uri], file_uri, run_context.meta),
+                )

finecode_dev_extensions/src/finecode_dev_extensions/publish_and_verify_artifact_action.py

@@ -1,13 +1,13 @@
 import dataclasses
-import pathlib
 
 from finecode_extension_api import code_action, textstyler
+from finecode_extension_api.resource_uri import ResourceUri
 
 
 @dataclasses.dataclass
 class PublishAndVerifyArtifactRunPayload(code_action.RunActionPayload):
-    src_artifact_def_path: pathlib.Path
-    dist_artifact_paths: list[pathlib.Path]
+    src_artifact_def_path: ResourceUri
+    dist_artifact_paths: list[ResourceUri]
     force: bool = False
 
 

finecode_dev_extensions/src/finecode_dev_extensions/publish_and_verify_artifact_handler.py

@@ -1,5 +1,4 @@
 import dataclasses
-import pathlib
 
 from finecode_extension_api import code_action
 from finecode_extension_api.actions.publishing import (
@@ -45,8 +44,8 @@ async def run(
     ) -> PublishAndVerifyArtifactRunResult:
         run_meta = run_context.meta
 
-        src_artifact_def_path: pathlib.Path = payload.src_artifact_def_path
-        dist_artifact_paths: list[pathlib.Path] = payload.dist_artifact_paths
+        src_artifact_def_path = payload.src_artifact_def_path
+        dist_artifact_paths = payload.dist_artifact_paths
 
         # Publish the artifact
         publish_action = self.action_runner.get_action_by_source(