aboutsummaryrefslogtreecommitdiffstats
path: root/packages/meshbay-node/src/meshbay_node/transport/webrtc/disk.py
diff options
context:
space:
mode:
Diffstat (limited to 'packages/meshbay-node/src/meshbay_node/transport/webrtc/disk.py')
-rw-r--r--packages/meshbay-node/src/meshbay_node/transport/webrtc/disk.py45
1 files changed, 34 insertions, 11 deletions
diff --git a/packages/meshbay-node/src/meshbay_node/transport/webrtc/disk.py b/packages/meshbay-node/src/meshbay_node/transport/webrtc/disk.py
index e298942..386b81f 100644
--- a/packages/meshbay-node/src/meshbay_node/transport/webrtc/disk.py
+++ b/packages/meshbay-node/src/meshbay_node/transport/webrtc/disk.py
@@ -1,8 +1,15 @@
-"""Blocking disk work the session hands to `off_disk`."""
+"""
+Blocking disk work the session hands to `off_disk`.
+Every call takes the path's long form (`long_path`): a file past Windows'
+MAX_PATH is indexed, so it is served, deleted and written beside like any other.
+"""
+
+import os
import shutil
from pathlib import Path
+from meshbay_common.paths import long_path
from meshbay_common.protocol import file_chunk_wire
from meshbay_node.roots import ROOT_NOT_SERVED, RootSet, entry_abs_path
@@ -22,7 +29,7 @@ def _locate(roots: RootSet, entry) -> tuple[Path | None, str | None]:
path = entry_abs_path(roots, entry)
if path is None:
return None, ROOT_NOT_SERVED
- if not path.exists():
+ if not os.path.exists(long_path(path)):
return None, "File not on disk"
return path, None
@@ -37,15 +44,16 @@ def _mkdir_if_absent(target: Path) -> str | None:
`mkdir` would raise where a refusal was meant. Blocking; called through
`off_disk`.
"""
- if target.exists():
+ if os.path.exists(long_path(target)):
return "Already exists"
- target.mkdir(parents=False)
+ os.mkdir(long_path(target))
return None
def _is_empty_dir(target: Path) -> bool:
"""Blocking; called through `off_disk`."""
- return not any(target.iterdir())
+ with os.scandir(long_path(target)) as listing:
+ return not any(listing)
def _rmdir_if_empty(target: Path) -> bool:
@@ -57,9 +65,10 @@ def _rmdir_if_empty(target: Path) -> bool:
operator's browser, and a file can land in between. Two awaits here would
reopen the same window one size smaller. Blocking; called through `off_disk`.
"""
- if any(target.iterdir()):
- return False
- target.rmdir()
+ with os.scandir(long_path(target)) as listing:
+ if any(listing):
+ return False
+ os.rmdir(long_path(target))
return True
@@ -71,7 +80,7 @@ def _read_and_encrypt(
file_id: str = "",
) -> dict:
"""Read one chunk off disk and encrypt it. Blocking; called through `off_disk`."""
- with open(file_path, "rb") as f:
+ with open(long_path(file_path), "rb") as f:
f.seek(chunk_index * CHUNK_SIZE)
plaintext = f.read(CHUNK_SIZE)
return file_chunk_wire(gek, plaintext, chunk_index, file_hash, file_id)
@@ -80,10 +89,24 @@ def _read_and_encrypt(
def _free_bytes(directory: Path) -> int:
"""Space left on the filesystem holding `directory`. Blocking; called
through `off_disk`, since a `statvfs` wakes a sleeping disk like any stat."""
- return shutil.disk_usage(directory).free
+ return shutil.disk_usage(long_path(directory)).free
def _append_chunk(tmp_path: Path, chunk_bytes: bytes, first: bool) -> None:
"""Add one chunk to a partial upload. Blocking; called through `off_disk`."""
- with open(tmp_path, "wb" if first else "ab") as f:
+ with open(long_path(tmp_path), "wb" if first else "ab") as f:
f.write(chunk_bytes)
+
+
+def _unlink(path: Path, missing_ok: bool = False) -> None:
+ """Remove one file. Blocking; called through `off_disk`."""
+ try:
+ os.unlink(long_path(path))
+ except FileNotFoundError:
+ if not missing_ok:
+ raise
+
+
+def _exists(path: Path) -> bool:
+ """Blocking; called through `off_disk`."""
+ return os.path.exists(long_path(path))