No public description

PiperOrigin-RevId: 928823128
Change-Id: Ieb09fc821092b4d8a626a2a9f54534a1edebaf4c

Author: kibergus

google/antigravity/connections/local/local_connection.py

@@ -230,6 +230,11 @@ def normalize_wire_path(path: str) -> str:
     # urlparse("file:///abs/path").path == "/abs/path"
     # url2pathname converts URL path to platform-native path
     return urllib.request.url2pathname(parsed.path)
+  if parsed.scheme == "cns":
+    # urlparse("cns://el-d/home/user/...").netloc == "el-d"
+    # urlparse("cns://el-d/home/user/...").path == "/home/user/..."
+    # Convert to the canonical /cns/<cell>/... absolute path format.
+    return "/cns/" + parsed.netloc + parsed.path
   return path
 
 

google/antigravity/connections/local/local_connection_test.py

@@ -1377,6 +1377,31 @@ def test_step_from_dict_normalizes_file_uri_arguments(self):
         "/dev/shm/workspace/foo.py",
     )
 
+  def test_step_from_dict_normalizes_cns_uri_arguments(self):
+    """Verifies that LocalConnectionStep.from_dict normalizes cns:// URIs.
+
+    Why: The CNS-backed filesystem uses cns:// URIs as path representations.
+    The workspace_only policy compares canonical_path against /cns/... paths
+    provided by the user, so cns:// must be translated to /cns/... for
+    policy matching to work correctly.
+    """
+    step = local_connection.LocalConnectionStep.from_dict({
+        "step_index": 1,
+        "trajectory_id": "traj_1",
+        "state": "STATE_WAITING_FOR_USER",
+        "create_file": {"path": "cns://el-d/home/user/workspace/kittens.md"},
+    })
+    self.assertEqual(len(step.tool_calls), 1)
+    self.assertEqual(
+        step.tool_calls[0].args.get("path"),
+        "/cns/el-d/home/user/workspace/kittens.md",
+    )
+    self.assertNotIn("canonical_path", step.tool_calls[0].args)
+    self.assertEqual(
+        step.tool_calls[0].canonical_path,
+        "/cns/el-d/home/user/workspace/kittens.md",
+    )
+
 
 class LocalConnectionToolCallNoRunnerTest(unittest.IsolatedAsyncioTestCase):
   """Tests for tool call handling when no ToolRunner is configured."""