[proxy] make Haketilo popup functional

* Ad hoc payload creation was additionally fixed in this commit.
* Addition on newly created script blocking/allowing rules to pattern tree was additionally fixed in this commit. It is no longer necessary to restart Haketilo for new rules to come into effect.

1 files changed, 18 insertions, 0 deletions

diff --git a/src/hydrilla/proxy/state_impl/rules.py b/src/hydrilla/proxy/state_impl/rules.py
index 2fed2c1..1761b04 100644
--- a/src/hydrilla/proxy/state_impl/rules.py
+++ b/src/hydrilla/proxy/state_impl/rules.py
@@ -148,6 +148,8 @@ class ConcreteRuleStore(st.RuleStore):
 
             (rule_id,), = cursor.fetchall()
 
+            self.state.rebuild_structures(payloads=False)
+
             return ConcreteRuleRef(str(rule_id), self.state)
 
     def get_display_infos(self, allow: t.Optional[bool] = None) \
@@ -176,3 +178,19 @@ class ConcreteRuleStore(st.RuleStore):
             result.append(st.RuleDisplayInfo(ref, pattern, allow_scripts))
 
         return result
+
+    def get_by_pattern(self, pattern: str) -> st.RuleRef:
+        with self.state.cursor() as cursor:
+            cursor.execute(
+                'SELECT rule_id FROM rules WHERE pattern = ?;',
+                (url_patterns.normalize_pattern(pattern),)
+            )
+
+            rows = cursor.fetchall()
+
+        if rows == []:
+            raise st.MissingItemError()
+
+        (rule_id,), = rows
+
+        return ConcreteRuleRef(str(rule_id), self.state)