[proxy] Add support for script blocking/allowing rules

1 files changed, 45 insertions, 2 deletions

diff --git a/src/hydrilla/proxy/state.py b/src/hydrilla/proxy/state.py
index abea7a7..1ba87b4 100644
--- a/src/hydrilla/proxy/state.py
+++ b/src/hydrilla/proxy/state.py
@@ -133,6 +133,47 @@ class Store(ABC, t.Generic[RefType]):
         ...
 
 
+class RulePatternInvalid(HaketiloException):
+    pass
+
+@dc.dataclass(frozen=True)
+class RuleDisplayInfo:
+    ref:           'RuleRef'
+    pattern:       str
+    allow_scripts: bool
+
+# mypy needs to be corrected:
+# https://stackoverflow.com/questions/70999513/conflict-between-mix-ins-for-abstract-dataclasses/70999704#70999704
+@dc.dataclass(frozen=True, unsafe_hash=True) # type: ignore[misc]
+class RuleRef(Ref):
+    @abstractmethod
+    def remove(self) -> None:
+        ...
+
+    @abstractmethod
+    def update(
+            self,
+            *,
+            pattern: t.Optional[str] = None,
+            allow: t.Optional[bool] = None
+    ) -> None:
+        ...
+
+    @abstractmethod
+    def get_display_info(self) -> RuleDisplayInfo:
+        ...
+
+class RuleStore(Store[RuleRef]):
+    @abstractmethod
+    def get_display_infos(self, allow: t.Optional[bool] = None) \
+        -> t.Sequence[RuleDisplayInfo]:
+        ...
+
+    @abstractmethod
+    def add(self, pattern: str, allow: bool) -> RuleRef:
+        ...
+
+
 class RepoNameInvalid(HaketiloException):
     pass
 
@@ -161,8 +202,6 @@ class FileMissingError(FileInstallationError):
 class RepoApiVersionUnsupported(HaketiloException):
     pass
 
-# mypy needs to be corrected:
-# https://stackoverflow.com/questions/70999513/conflict-between-mix-ins-for-abstract-dataclasses/70999704#70999704
 @dc.dataclass(frozen=True, unsafe_hash=True) # type: ignore[misc]
 class RepoRef(Ref):
     """...."""
@@ -443,6 +482,10 @@ class HaketiloState(ABC):
         ...
 
     @abstractmethod
+    def rule_store(self) -> RuleStore:
+        ...
+
+    @abstractmethod
     def repo_store(self) -> RepoStore:
         """...."""
         ...