chore: import upstream snapshot with attribution

python/mirage/ops/ssh/__init__.py

@@ -0,0 +1,44 @@
+# ========= 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 logging
+
+from mirage.commands.optional import try_load_command
+from mirage.ops.ssh.create import create
+from mirage.ops.ssh.mkdir import mkdir
+from mirage.ops.ssh.read.read import read
+from mirage.ops.ssh.readdir import readdir
+from mirage.ops.ssh.rename import rename
+from mirage.ops.ssh.rmdir import rmdir
+from mirage.ops.ssh.stat import stat
+from mirage.ops.ssh.truncate import truncate
+from mirage.ops.ssh.unlink import unlink
+from mirage.ops.ssh.write import write as write_bytes
+
+_logger = logging.getLogger(__name__)
+
+read_feather = try_load_command("mirage.ops.ssh.read.read_feather",
+                                "read_feather", "parquet")
+read_hdf5 = try_load_command("mirage.ops.ssh.read.read_hdf5", "read_hdf5",
+                             "hdf5")
+read_orc = try_load_command("mirage.ops.ssh.read.read_orc", "read_orc",
+                            "parquet")
+read_parquet = try_load_command("mirage.ops.ssh.read.read_parquet",
+                                "read_parquet", "parquet")
+
+OPS = [
+    c for c in (create, mkdir, read, read_feather, read_hdf5, read_orc,
+                read_parquet, readdir, rename, rmdir, stat, truncate, unlink,
+                write_bytes) if c is not None
+]

python/mirage/ops/ssh/create.py

@@ -0,0 +1,23 @@
+# ========= 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.accessor.ssh import SSHAccessor
+from mirage.core.ssh.write import write_bytes
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("create", resource="ssh", write=True)
+async def create(accessor: SSHAccessor, path: PathSpec, **kwargs) -> None:
+    await write_bytes(accessor, path, b"")

python/mirage/ops/ssh/mkdir.py

@@ -0,0 +1,23 @@
+# ========= 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.accessor.ssh import SSHAccessor
+from mirage.core.ssh.mkdir import mkdir as mkdir_impl
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("mkdir", resource="ssh", write=True)
+async def mkdir(accessor: SSHAccessor, path: PathSpec, **kwargs) -> None:
+    await mkdir_impl(accessor, path)

python/mirage/ops/ssh/read/__init__.py

@@ -0,0 +1,13 @@
+# ========= 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. =========

python/mirage/ops/ssh/read/read.py

@@ -0,0 +1,24 @@
+# ========= 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.accessor.ssh import SSHAccessor
+from mirage.core.ssh.read import read_bytes
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("read", resource="ssh")
+async def read(accessor: SSHAccessor, path: PathSpec, *, index,
+               **kwargs) -> bytes:
+    return await read_bytes(accessor, path, index)

python/mirage/ops/ssh/read/read_feather.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. =========
+
+from mirage.accessor.ssh import SSHAccessor
+from mirage.core.filetype.feather import cat as feather_cat
+from mirage.core.ssh.read import read_bytes
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("read", resource="ssh", filetype=".feather")
+async def read_feather(accessor: SSHAccessor, path: PathSpec, *, index,
+                       **kwargs) -> bytes:
+    raw = await read_bytes(accessor, path, index)
+    return feather_cat(raw)

python/mirage/ops/ssh/read/read_hdf5.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. =========
+
+from mirage.accessor.ssh import SSHAccessor
+from mirage.core.filetype.hdf5 import cat as hdf5_cat
+from mirage.core.ssh.read import read_bytes
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("read", resource="ssh", filetype=".hdf5")
+async def read_hdf5(accessor: SSHAccessor, path: PathSpec, *, index,
+                    **kwargs) -> bytes:
+    raw = await read_bytes(accessor, path, index)
+    return hdf5_cat(raw)

python/mirage/ops/ssh/read/read_orc.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. =========
+
+from mirage.accessor.ssh import SSHAccessor
+from mirage.core.filetype.orc import cat as orc_cat
+from mirage.core.ssh.read import read_bytes
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("read", resource="ssh", filetype=".orc")
+async def read_orc(accessor: SSHAccessor, path: PathSpec, *, index,
+                   **kwargs) -> bytes:
+    raw = await read_bytes(accessor, path, index)
+    return orc_cat(raw)

python/mirage/ops/ssh/read/read_parquet.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. =========
+
+from mirage.accessor.ssh import SSHAccessor
+from mirage.core.filetype.parquet import cat as parquet_cat
+from mirage.core.ssh.read import read_bytes
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("read", resource="ssh", filetype=".parquet")
+async def read_parquet(accessor: SSHAccessor, path: PathSpec, *, index,
+                       **kwargs) -> bytes:
+    raw = await read_bytes(accessor, path, index)
+    return parquet_cat(raw)

python/mirage/ops/ssh/readdir.py

@@ -0,0 +1,24 @@
+# ========= 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.accessor.ssh import SSHAccessor
+from mirage.core.ssh.readdir import readdir as readdir_impl
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("readdir", resource="ssh")
+async def readdir(accessor: SSHAccessor, path: PathSpec, *, index,
+                  **kwargs) -> list[str]:
+    return await readdir_impl(accessor, path, index)

python/mirage/ops/ssh/rename.py

@@ -0,0 +1,24 @@
+# ========= 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.accessor.ssh import SSHAccessor
+from mirage.core.ssh.rename import rename as rename_impl
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("rename", resource="ssh", write=True)
+async def rename(accessor: SSHAccessor, src: PathSpec, dst: PathSpec,
+                 **kwargs) -> None:
+    await rename_impl(accessor, src, dst)

python/mirage/ops/ssh/rmdir.py

@@ -0,0 +1,23 @@
+# ========= 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.accessor.ssh import SSHAccessor
+from mirage.core.ssh.rmdir import rmdir as rmdir_impl
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("rmdir", resource="ssh", write=True)
+async def rmdir(accessor: SSHAccessor, path: PathSpec, **kwargs) -> None:
+    await rmdir_impl(accessor, path)

python/mirage/ops/ssh/stat.py

@@ -0,0 +1,24 @@
+# ========= 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.accessor.ssh import SSHAccessor
+from mirage.core.ssh.stat import stat as stat_impl
+from mirage.ops.registry import op
+from mirage.types import FileStat, PathSpec
+
+
+@op("stat", resource="ssh")
+async def stat(accessor: SSHAccessor, path: PathSpec, *, index,
+               **kwargs) -> FileStat:
+    return await stat_impl(accessor, path, index)

python/mirage/ops/ssh/truncate.py

@@ -0,0 +1,29 @@
+# ========= 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.accessor.ssh import SSHAccessor
+from mirage.core.ssh.read import read_bytes
+from mirage.core.ssh.write import write_bytes
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("truncate", resource="ssh", write=True)
+async def truncate(accessor: SSHAccessor, path: PathSpec, length: int,
+                   **kwargs) -> None:
+    try:
+        data = await read_bytes(accessor, path)
+    except FileNotFoundError:
+        data = b""
+    await write_bytes(accessor, path, data[:length].ljust(length, b"\0"))

python/mirage/ops/ssh/unlink.py

@@ -0,0 +1,23 @@
+# ========= 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.accessor.ssh import SSHAccessor
+from mirage.core.ssh.unlink import unlink as unlink_impl
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("unlink", resource="ssh", write=True)
+async def unlink(accessor: SSHAccessor, path: PathSpec, **kwargs) -> None:
+    await unlink_impl(accessor, path)

python/mirage/ops/ssh/write.py

@@ -0,0 +1,24 @@
+# ========= 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.accessor.ssh import SSHAccessor
+from mirage.core.ssh.write import write_bytes
+from mirage.ops.registry import op
+from mirage.types import PathSpec
+
+
+@op("write", resource="ssh", write=True)
+async def write(accessor: SSHAccessor, path: PathSpec, data: bytes,
+                **kwargs) -> None:
+    await write_bytes(accessor, path, data)