resforge 0.3.0__tar.gz → 0.3.1__tar.gz

{resforge-0.3.0 → resforge-0.3.1}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: resforge
-Version: 0.3.0
+Version: 0.3.1
 Summary: A type-safe Python DSL for cross-platform mobile resource generation
 Project-URL: Homepage, https://kipila.dev
-Project-URL: Repository, https://github.com/ok100/resforge
+Project-URL: Repository, https://github.com/kipila-dev/resforge
 Author-email: Ondrej Kipila <ondrej@kipila.dev>
 License: MIT
 License-File: LICENSE
@@ -159,7 +159,6 @@ with AssetCatalog("App", "Assets") as ac:
 
 ## Roadmap
 
-- SwiftUI color extension generation (`SwiftUIWriter`)
 - Asset Catalog image support (ImageSet, IconSet)
 - Android `res/drawable` vector asset support
 

{resforge-0.3.0 → resforge-0.3.1}/README.md

@@ -136,7 +136,6 @@ with AssetCatalog("App", "Assets") as ac:
 
 ## Roadmap
 
-- SwiftUI color extension generation (`SwiftUIWriter`)
 - Asset Catalog image support (ImageSet, IconSet)
 - Android `res/drawable` vector asset support
 

{resforge-0.3.0 → resforge-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "resforge"
-version = "0.3.0"
+version = "0.3.1"
 description = "A type-safe Python DSL for cross-platform mobile resource generation"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -36,7 +36,7 @@ dev = ["pytest", "defusedxml"]
 
 [project.urls]
 Homepage = "https://kipila.dev"
-Repository = "https://github.com/ok100/resforge"
+Repository = "https://github.com/kipila-dev/resforge"
 
 [tool.ruff]
 target-version = "py312"

resforge-0.3.1/resforge/_utils.py

@@ -0,0 +1,62 @@
+import os
+from collections.abc import Callable, Generator
+from contextlib import contextmanager, suppress
+from functools import wraps
+from pathlib import Path
+from tempfile import NamedTemporaryFile
+from typing import IO, Concatenate
+
+
+def require_context[T, **P, R](
+    func: Callable[Concatenate[T, P], R],
+) -> Callable[Concatenate[T, P], R]:
+    """Ensures a method is only called within an active context.
+
+    Raises:
+        RuntimeError: If the method is called while the instance's
+            `_active` attribute is False or missing.
+
+    """
+
+    @wraps(func)
+    def wrapper(self: T, *args: P.args, **kwargs: P.kwargs) -> R:
+        if not getattr(self, "_active", False):
+            msg = f"'{func.__name__}' requires an active 'with' context."
+            raise RuntimeError(msg)
+        return func(self, *args, **kwargs)
+
+    return wrapper
+
+
+@contextmanager
+def atomic_write(target_path: Path) -> Generator[IO[bytes], None, None]:
+    """Yields a temporary file, then atomically replaces target_path on success."""
+    target_path = Path(target_path)
+    target_path.parent.mkdir(parents=True, exist_ok=True)
+
+    temp_path = None
+    try:
+        with NamedTemporaryFile(
+            dir=target_path.parent, delete=False, suffix=".tmp"
+        ) as tf:
+            temp_path = Path(tf.name)
+            yield tf
+
+            tf.flush()
+            os.fsync(tf.fileno())
+
+        temp_path.replace(target_path)
+
+        if os.name == "posix":
+            flags = os.O_RDONLY
+            if hasattr(os, "O_DIRECTORY"):
+                flags |= os.O_DIRECTORY
+            dir_fd = os.open(target_path.parent, flags)
+            try:
+                os.fsync(dir_fd)
+            finally:
+                os.close(dir_fd)
+    finally:
+        if temp_path and temp_path.exists():
+            with suppress(OSError):
+                temp_path.unlink()

{resforge-0.3.0 → resforge-0.3.1}/resforge/android/compose.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from pathlib import Path
 from typing import TYPE_CHECKING, Self
 
-from resforge._utils import require_context
+from resforge._utils import atomic_write, require_context
 from resforge.codegen.kotlin import KotlinFile, KotlinObject
 from resforge.types import Color
 
@@ -119,7 +119,8 @@ class ComposeWriter(_BaseComposeScope):
     def __exit__(self, exc_type, *_) -> None:
         try:
             if exc_type is None:
-                self._kotlin_file.write(self._path)
+                with atomic_write(self._path) as tf:
+                    self._kotlin_file.write(tf)
         finally:
             self._active = False
 

{resforge-0.3.0 → resforge-0.3.1}/resforge/android/values.py

@@ -4,7 +4,7 @@ from pathlib import Path
 from re import Pattern
 from typing import Self
 
-from resforge._utils import require_context
+from resforge._utils import atomic_write, require_context
 from resforge.types import Color
 
 from .types import Dimension, PluralValues
@@ -49,15 +49,15 @@ class ValuesWriter:
     def __exit__(self, exc_type, *_) -> None:
         try:
             if exc_type is None:
-                self._path.parent.mkdir(parents=True, exist_ok=True)
                 ET.indent(self._root, space=" " * 4, level=0)
                 tree = ET.ElementTree(self._root)
-                tree.write(
-                    self._path,
-                    encoding="utf-8",
-                    xml_declaration=True,
-                    short_empty_elements=True,
-                )
+                with atomic_write(self._path) as tf:
+                    tree.write(
+                        tf,
+                        encoding="utf-8",
+                        xml_declaration=True,
+                        short_empty_elements=True,
+                    )
         finally:
             self._active = False
 

{resforge-0.3.0 → resforge-0.3.1}/resforge/apple/_base.py

@@ -4,11 +4,13 @@ from abc import ABC, abstractmethod
 from pathlib import Path
 from typing import Any, Self
 
+from resforge._utils import atomic_write
+
 
 def write_contents(path: str | Path, contents: dict[str, Any]) -> None:
     path = Path(path) / "Contents.json"
-    with path.open("w") as f:
-        json.dump(contents, f, indent=2)
+    with atomic_write(path) as tf:
+        tf.write(json.dumps(contents, indent=2).encode())
 
 
 class AssetNode(ABC):

{resforge-0.3.0 → resforge-0.3.1}/resforge/codegen/kotlin.py

@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 from pathlib import Path
-from typing import Literal, Self
+from typing import IO, Literal, Self
 
 PropertyMutability = Literal["val", "var"]
 
@@ -90,8 +90,11 @@ class KotlinFile:
 
         return "\n\n".join(sections) + "\n"
 
-    def write(self, path: str | Path) -> None:
-        path = Path(path)
-        path.parent.mkdir(parents=True, exist_ok=True)
-        with path.open("w") as f:
-            f.write(self.render())
+    def write(self, dest: str | Path | IO[bytes]) -> None:
+        if isinstance(dest, (str, Path)):
+            path = Path(dest)
+            path.parent.mkdir(parents=True, exist_ok=True)
+            with path.open("wb") as f:
+                f.write(self.render().encode())
+        else:
+            dest.write(self.render().encode())

resforge-0.3.1/tests/test_atomic_write.py

@@ -0,0 +1,35 @@
+import pytest
+
+from resforge._utils import atomic_write
+
+
+def test_atomic_replace_existing(tmp_path):
+    target = tmp_path / "output.bin"
+    target.write_bytes(b"old")
+    with atomic_write(target) as f:
+        f.write(b"new")
+    assert target.read_bytes() == b"new"
+    assert not any(tmp_path.glob("*.tmp"))
+
+
+def test_creates_parent_dirs(tmp_path):
+    target = tmp_path / "a" / "b" / "c" / "output.bin"
+    with atomic_write(target) as f:
+        f.write(b"hello")
+    assert target.exists()
+
+
+def test_no_corruption_on_exception(tmp_path):
+    target = tmp_path / "output.bin"
+    target.write_bytes(b"original")
+
+    def _broken_write() -> None:
+        with atomic_write(target) as f:
+            f.write(b"new corrupted data")
+            raise RuntimeError("boom")
+
+    with pytest.raises(RuntimeError):
+        _broken_write()
+
+    assert target.read_bytes() == b"original"
+    assert not any(tmp_path.glob("*.tmp"))

resforge-0.3.0/resforge/_utils.py

@@ -1,24 +0,0 @@
-from collections.abc import Callable
-from functools import wraps
-from typing import Concatenate
-
-
-def require_context[T, **P, R](
-    func: Callable[Concatenate[T, P], R],
-) -> Callable[Concatenate[T, P], R]:
-    """Ensures a method is only called within an active context.
-
-    Raises:
-        RuntimeError: If the method is called while the instance's
-            `_active` attribute is False or missing.
-
-    """
-
-    @wraps(func)
-    def wrapper(self: T, *args: P.args, **kwargs: P.kwargs) -> R:
-        if not getattr(self, "_active", False):
-            msg = f"'{func.__name__}' requires an active 'with' context."
-            raise RuntimeError(msg)
-        return func(self, *args, **kwargs)
-
-    return wrapper