chore: import upstream snapshot with attribution

python/mirage/resource/databricks_volume/__init__.py

@@ -0,0 +1,25 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+from mirage.resource.databricks_volume.config import DatabricksVolumeConfig
+
+__all__ = ["DatabricksVolumeConfig", "DatabricksVolumeResource"]
+
+
+def __getattr__(name: str):
+    if name == "DatabricksVolumeResource":
+        from mirage.resource.databricks_volume.databricks_volume import \
+            DatabricksVolumeResource
+        return DatabricksVolumeResource
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

python/mirage/resource/databricks_volume/config.py

@@ -0,0 +1,52 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+import posixpath
+
+from pydantic import BaseModel, ConfigDict, Field, field_validator
+
+
+class DatabricksVolumeConfig(BaseModel):
+    model_config = ConfigDict(frozen=True,
+                              populate_by_name=True,
+                              serialize_by_alias=True)
+
+    catalog: str
+    # Public config key is still "schema"
+    # internal name avoids warning overwriting BaseModel.schema.
+    schema_name: str = Field(alias="schema")
+    volume: str
+    root_path: str = "/"
+    host: str | None = None
+    token: str | None = None
+    profile: str | None = None
+    timeout: int = 30
+
+    @field_validator("catalog", "schema_name", "volume")
+    @classmethod
+    def validate_volume_part(cls, value: str) -> str:
+        if not value or "/" in value:
+            raise ValueError("must be a non-empty path segment")
+        return value
+
+    @field_validator("root_path")
+    @classmethod
+    def normalize_root_path(cls, value: str) -> str:
+        stripped = value.strip("/")
+        if not stripped:
+            return "/"
+        if any(part == ".." for part in stripped.split("/")):
+            raise ValueError("root_path must not contain '..' segments")
+        normalized = posixpath.normpath("/" + stripped)
+        return "/" if normalized == "/." else normalized

python/mirage/resource/databricks_volume/databricks_volume.py

@@ -0,0 +1,103 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+import dataclasses
+from typing import Any
+
+from mirage.accessor.databricks_volume import DatabricksVolumeAccessor
+from mirage.commands.builtin.databricks_volume import \
+    COMMANDS as DATABRICKS_VOLUME_COMMANDS
+from mirage.core.databricks_volume.copy import copy
+from mirage.core.databricks_volume.create import create
+from mirage.core.databricks_volume.exists import exists
+from mirage.core.databricks_volume.glob import resolve_glob as _resolve_glob
+from mirage.core.databricks_volume.mkdir import mkdir
+from mirage.core.databricks_volume.read import read_bytes
+from mirage.core.databricks_volume.readdir import readdir
+from mirage.core.databricks_volume.rename import rename
+from mirage.core.databricks_volume.rm import rm_recursive
+from mirage.core.databricks_volume.rmdir import rmdir
+from mirage.core.databricks_volume.stat import stat as databricks_stat
+from mirage.core.databricks_volume.stream import range_read, read_stream
+from mirage.core.databricks_volume.unlink import unlink
+from mirage.core.databricks_volume.write import write_bytes
+from mirage.ops.databricks_volume import OPS as DATABRICKS_VOLUME_OPS
+from mirage.resource.base import BaseResource
+from mirage.resource.databricks_volume.config import DatabricksVolumeConfig
+from mirage.resource.databricks_volume.prompt import PROMPT
+from mirage.types import PathSpec, ResourceName
+from mirage.utils.key_prefix import mount_key
+
+_DATABRICKS_VOLUME_OPS = {
+    "read_bytes": read_bytes,
+    "readdir": readdir,
+    "stat": databricks_stat,
+    "read_stream": read_stream,
+    "range_read": range_read,
+    "exists": exists,
+    "write": write_bytes,
+    "create": create,
+    "unlink": unlink,
+    "mkdir": mkdir,
+    "rmdir": rmdir,
+    "copy": copy,
+    "rename": rename,
+    "rm_recursive": rm_recursive,
+}
+
+
+class DatabricksVolumeResource(BaseResource):
+    name: str = ResourceName.DATABRICKS_VOLUME
+    caches_reads: bool = True
+    _ops: dict[str, Any] = _DATABRICKS_VOLUME_OPS
+    PROMPT: str = PROMPT
+
+    def __init__(
+        self,
+        config: DatabricksVolumeConfig,
+        client: Any | None = None,
+    ) -> None:
+        super().__init__()
+        self.config = config
+        self.accessor = DatabricksVolumeAccessor(self.config, client)
+
+        for fn in DATABRICKS_VOLUME_COMMANDS:
+            self.register(fn)
+        for fn in DATABRICKS_VOLUME_OPS:
+            self.register_op(fn)
+
+    async def resolve_glob(self, paths, prefix: str = ""):
+        if prefix:
+            paths = [
+                dataclasses.replace(p,
+                                    resource_path=mount_key(p.virtual, prefix))
+                if isinstance(p, PathSpec) else p for p in paths
+            ]
+        return await _resolve_glob(self.accessor, paths, self._index)
+
+    def get_state(self) -> dict:
+        redacted = ["token"]
+        cfg = self.config.model_dump()
+        for field in redacted:
+            if cfg.get(field) is not None:
+                cfg[field] = "<REDACTED>"
+        return {
+            "type": self.name,
+            "needs_override": True,
+            "redacted_fields": redacted,
+            "config": cfg,
+        }
+
+    def load_state(self, state: dict) -> None:
+        pass

python/mirage/resource/databricks_volume/prompt.py

@@ -0,0 +1,26 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+PROMPT = """\
+{prefix}
+  Remote Databricks Unity Catalog volume filesystem.
+  IMPORTANT: This is a remote mount - every file/dir access is an API round-trip.
+  Prefer targeted reads (ls, head, grep, rg, find with a path/glob) over full
+  scans; recursive ops (rm -r, cp -r, tree, grep -r) over a large tree are slow.
+  Create parent dirs before writing: `mkdir -p /path && cmd > /path/out`.
+  mv/cp are non-atomic full copies (no server-side rename) - avoid on large files.
+  sed -i is not supported; transform and redirect to a new file instead.
+  Read/analyze: ls, cat, head, tail, grep, rg, find, tree, stat, wc, sort, uniq,
+    cut, nl, tr, sed, awk, jq, diff.
+  Write: touch, rm, rm -r, mkdir, mkdir -p, cp, mv, and >/>> redirects."""