this is the write policy for my relay
This commit is contained in:
parent
8f81637c86
commit
43387d611d
1 changed files with 107 additions and 0 deletions
107
write-policy.py
Normal file
107
write-policy.py
Normal file
|
@ -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()
|
Loading…
Reference in a new issue