control which languages get patched

boozook/archive.py

@@ -3,7 +3,7 @@
 import itertools
 import os
 from pathlib import Path
-from typing import Iterable, Iterator, MutableMapping, Optional, Sequence
+from typing import Iterable, Iterator, MutableMapping, Optional, Sequence, Set
 
 from boozook.codex import stk
 from boozook.codex.stk_compress import recompress_archive
@@ -39,6 +39,8 @@ def game_search(base_dir, patterns=('*',), patches=()):
 class GameBase:
     base_dir: str
     patches: Optional[Sequence[str]]
+    allowed_patches: Optional[Set[str]] = None
+    restricted_patches: Optional[Set[str]] = None
 
     _patched: dict[tuple[str, str], bytes] = field(default_factory=dict)
 
@@ -48,6 +50,11 @@ def search(self, patterns):
     def patch(self, fname: str, data: bytes, alias: str | None = None):
         if not alias:
             alias = fname
+        if self.allowed_patches:
+            if not any(Path(alias).match(pattern) for pattern in self.allowed_patches):
+                return
+        if any(Path(alias).match(pattern) for pattern in self.restricted_patches or ()):
+            return
         self._patched[(fname, alias)] = data
 
     def rebuild(self, target='patch'):
@@ -76,8 +83,16 @@ def rebuild(self, target='patch'):
                 raise ValueError(f'archive {arc} was not found')
 
 
-def open_game(base_dir, patches=()):
-    return GameBase(base_dir, patches=patches)
+def open_game(
+    base_dir,
+    patches=(),
+    allowed_patches=(),
+):
+    return GameBase(
+        base_dir,
+        patches=patches,
+        allowed_patches=set(allowed_patches),
+    )
 
 
 class DirectoryBackedArchive(MutableMapping[str, bytes]):

boozook/text.py

@@ -126,14 +126,20 @@ def encode(game, patterns, texts_dir):
         action='store_true',
         help='create modifed game resource with the changes',
     )
+    parser.add_argument(
+        '--allowed',
+        '-i',
+        action='append',
+        help='allow only specific patterns to be modified',
+    )
     args = parser.parse_args()
 
     patterns = TEXT_PATTERNS
 
     texts_dir = Path('texts')
     os.makedirs(texts_dir, exist_ok=True)
 
-    game = archive.open_game(args.directory)
+    game = archive.open_game(args.directory, allowed_patches=args.allowed or ())
     if not args.rebuild:
         decode(game, patterns, texts_dir)
     else: