mirror of
https://github.com/AsgardEternal/DiscordWhitelist.git
synced 2024-12-30 15:19:12 -06:00
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
import http.server
|
|
import os
|
|
import re
|
|
import socketserver
|
|
import traceback
|
|
import urllib.request
|
|
from urllib.parse import urlparse
|
|
from urllib.parse import parse_qs
|
|
from sys import stderr
|
|
|
|
import requests
|
|
|
|
PORT = 8000
|
|
|
|
|
|
class serveRA(http.server.SimpleHTTPRequestHandler):
|
|
def do_GET(self):
|
|
self.send_response(200)
|
|
self.send_header("Content-type", "text")
|
|
self.end_headers()
|
|
query = parse_qs(urlparse(self.path).query)
|
|
if 'grpName' in query:
|
|
grpName = query['grpName'][0]
|
|
grpfile = "./"
|
|
if os.path.exists(f"./wlgrps/{grpName}.cfg"):
|
|
grpfile += f"wlgrps/{grpName}.cfg"
|
|
elif os.path.exists(f"./extgrps/{grpName}.cfg"):
|
|
grpfile += f"extgrps/{grpName}.cfg"
|
|
else:
|
|
print("could not find admins file!", file=stderr)
|
|
return
|
|
try:
|
|
file = open(grpfile, 'rb')
|
|
firstline = file.readline().decode('utf-8')
|
|
file.seek(0)
|
|
if firstline.startswith('remotelist='):
|
|
remote = firstline.split('=')[1].strip()
|
|
response = requests.get(remote, headers={'Accept': 'text/html,*/*'})
|
|
responsetext = response.text
|
|
config = file.read().decode('utf-8')
|
|
confgrps = re.findall(r"^permissions/(.+)=(.+)", config, flags=re.M)
|
|
baseperm = re.match(r"^permissions=(.+)", config, flags=re.M)
|
|
responsetext = re.sub(r"^Group=(.+):(.+)", fr'Group=\1:{baseperm[0]}', responsetext, flags=re.M)
|
|
for congrp in confgrps:
|
|
responsetext = re.sub(rf"^Group=({congrp[0]}):(.+)", rf"Group=\1:{congrp[1]}", responsetext, flags=re.M)
|
|
self.wfile.write(responsetext.encode('utf-8'))
|
|
else:
|
|
self.copyfile(file, self.wfile)
|
|
file.close()
|
|
except:
|
|
print('failed to open file!', file=stderr)
|
|
print(traceback.format_exc())
|
|
return
|
|
|
|
|
|
def startServer():
|
|
handler = serveRA
|
|
|
|
with socketserver.TCPServer(("", PORT), handler) as httpd:
|
|
print('starting server!')
|
|
httpd.serve_forever()
|
|
|
|
|
|
startServer()
|