From 43387d611d4f669d66387b7088672f335791786c Mon Sep 17 00:00:00 2001
From: Miguel <contact@miguelalmodo.com>
Date: Sun, 30 Jun 2024 00:50:16 -0400
Subject: [PATCH] this is the write policy for my relay

---
 write-policy.py | 107 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 107 insertions(+)
 create mode 100644 write-policy.py

diff --git a/write-policy.py b/write-policy.py
new file mode 100644
index 0000000..b6b02d0
--- /dev/null
+++ b/write-policy.py
@@ -0,0 +1,107 @@
+#!/usr/bin/env python3
+
+import sys
+import json
+
+# ENTER YOUR HEX PUBKEY(S) BELOW:
+whitelist = {
+  "ec965405e11a6a6186b27fa451a2ffc1396ede7883d2ea11c32fbd2c63996966",
+  "11493e2efadce058f779ee4bd3b6bf9874fe571622704372b1b4d9051397ab32",
+  "60f1a5eb3310076ae334f432bc178f2ed4aac86abd704ee1626d4a283f1abdec",
+  "22f11e47397e83cdeb56933aad13a1dd2c4f70ed8c2f5d93dd70b1ea4cdfafa6",
+  "02f7fe308859d2d6178ada7caca89c2787e1ee2889378ee6e98d530e75cd77b8",
+  "d10a02474e3ff020d46753336e1c436a7373a84f6c22852749dabf63ea505515",
+  "c118d1b814a64266730e75f6c11c5ffa96d0681bfea594d564b43f3097813844",
+  "8f4761dcd2405aab2455bc66e7a54b8f8e5a88dc645e9717df665d1993b740fc",
+  "",
+  "",
+}
+
+# ENTER YOUR ALLOWED IP(S) BELOW:
+sources = {
+  "1.1.1.1",
+  "8.8.8.8"
+}
+
+# ENTER YOUR ALLOWED TYPE(S) BELOW:
+types = {
+  "Stream",
+  "Import",
+  "Sync"
+}
+
+def eprint(*args, **kwargs):
+  print(*args, **kwargs, file=sys.stderr, flush=True)
+
+def accept(request):
+  response = {
+    'id' : request['event']['id']
+  }
+
+  response['action'] = 'accept'
+  r = json.dumps(response,separators=(',', ':')) # output JSONL
+  print(r, end='\n', file=sys.stdout, flush=True)
+
+def reject(request):
+  response = {
+    'id' : request['event']['id']
+  }
+
+  response['action'] = 'reject'
+  response['msg'] = f"blocked: pubkey {request['event']['pubkey']} not in whitelist | SOURCE: {request['sourceInfo']}"
+  r = json.dumps(response,separators=(',', ':')) # output JSONL
+  print(r, end='\n', file=sys.stdout, flush=True)
+
+def main():
+  for line in sys.stdin:
+    request = json.loads(line)
+
+    try:
+      if request['type'] == 'lookback':
+        continue
+    except KeyError:
+      eprint("input without type in write policy plugin")
+      continue
+
+    if request['type'] != 'new':
+      eprint("unexpected request type in write policy plugin")
+      continue
+
+    try:
+      if not request['event']['id']:
+        eprint("input without event id in write policy plugin")
+        continue
+    except KeyError:
+      eprint("input without event id in write policy plugin")
+      continue
+
+    try:
+      if request['event']['pubkey'] in whitelist:
+        accept(request)
+        continue
+      elif int(request['event']['kind']) == 10002:
+        accept(request)
+        continue
+      elif request['sourceType'] in types:
+        accept(request)
+        continue
+      elif request['sourceInfo'] in sources:
+        accept(request)
+        continue
+      elif request.get("event", {}).get("tags"):
+        if p_tags:= [x for x in request['event']['tags'] if x[0] == 'p']:
+          pubkeys = [x[1] for x in p_tags]
+          if whitelist.intersection(pubkeys):
+            accept(request)
+            continue
+        reject(request)
+        continue
+      else:
+        reject(request)
+        continue
+    except KeyError:
+      eprint("poorly formed event input in write policy plugin")
+      continue
+
+if __name__=='__main__':
+  main()