feat: Add more replace options
All checks were successful
Build Docker / Build Docker (push) Successful in 29s
All checks were successful
Build Docker / Build Docker (push) Successful in 29s
This commit is contained in:
parent
b065530cad
commit
37379e4255
@ -1,3 +1,4 @@
|
||||
PROXY=http://nathanwoodburn:5000/
|
||||
TLD=woodburn
|
||||
RESTRICTED=["admin"]
|
||||
RESTRICTED=["admin"]
|
||||
REPLACE=["https://nathan.woodburn.au/"]
|
@ -4,4 +4,4 @@ Environment variables:
|
||||
- `PROXY` - The proxy to use for restricted TLDs. Example: `http://localhost:3128`
|
||||
- `TLD` - The TLD to restrict. Example: `g`
|
||||
- `RESTRICTED` - List of restricted paths. Example: `["path1", "path2"]` will require the user to have a .g domain to access `path1/*`, `path2/*`
|
||||
|
||||
- `REPLACE` - List of urls to replace with the host to fix redirects and links
|
4
main.py
4
main.py
@ -14,6 +14,8 @@ RESTRICTED = os.getenv('RESTRICTED')
|
||||
RESTRICTED = json.loads(RESTRICTED)
|
||||
RESTRICTED = [f'{i.lower()}/' for i in RESTRICTED]
|
||||
TLD = os.getenv('TLD')
|
||||
REPLACEMENT = os.getenv('REPLACE')
|
||||
REPLACEMENT = json.loads(REPLACEMENT)
|
||||
|
||||
# Load cookies
|
||||
cookies = []
|
||||
@ -118,6 +120,8 @@ def catch_all(path):
|
||||
if 'text/html' in res.headers['Content-Type']:
|
||||
content = res.content.decode('utf-8')
|
||||
content = content.replace(URL, request.host_url)
|
||||
for i in REPLACEMENT:
|
||||
content = content.replace(i, request.host_url)
|
||||
response = make_response(content, res.status_code, headers)
|
||||
return response
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user