d3v1l/matrixion
1
0
Fork 0
mirror of https://github.com/Nheko-Reborn/nheko.git synced 2024-11-22 03:00:46 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Try to fix the flat-manager script for flat-manager behind reverse proxies

Browse source
This commit is contained in:
Nicolas Werner 2023-04-15 01:40:45 +02:00
parent 24230bacef
commit b757140bec
No known key found for this signature in database
GPG key ID: C8D75E610773F2D9
1 changed files with 21 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
scripts/flat-manager-client
View file

@ -26,6 +26,7 @@ import os
import sys import sys
import time import time
import traceback import traceback
from yarl import URL
from argparse import ArgumentParser from argparse import ArgumentParser
from functools import reduce from functools import reduce
from urllib.parse import urljoin, urlparse, urlsplit, urlunparse, urlunsplit from urllib.parse import urljoin, urlparse, urlsplit, urlunparse, urlunsplit
@ -375,7 +376,7 @@ async def wait_for_job(session, job_url, token):
if start_after and start_after > now: if start_after and start_after > now:
print("Waiting %d seconds before starting job" % (int(start_after - now))) print("Waiting %d seconds before starting job" % (int(start_after - now)))
if job_status > 0 and old_job_status == 0: if job_status > 0 and old_job_status == 0:
print("/ Job was started"); print("/ Job was started")
old_job_status = job_status old_job_status = job_status
log = job['log'] log = job['log']
if len(log) > 0: if len(log) > 0:
@ -441,14 +442,14 @@ async def commit_build(session, build_url, eol, eol_rebase, token_type, wait, to
raise ApiError(resp, await resp.text()) raise ApiError(resp, await resp.text())
job = await resp.json() job = await resp.json()
job_url = resp.headers['location']; job_url = keep_host(resp.headers['location'], build_url)
if wait: if wait:
print("Waiting for commit job") print("Waiting for commit job")
job = await wait_for_job(session, job_url, token); job = await wait_for_job(session, job_url, token)
reparse_job_results(job) reparse_job_results(job)
job["location"] = job_url job["location"] = keep_host(job_url, build_url)
return job return job
@ -478,14 +479,14 @@ async def publish_build(session, build_url, wait, token):
raise ApiError(resp, await resp.text()) raise ApiError(resp, await resp.text())
job = await resp.json() job = await resp.json()
job_url = resp.headers['location']; job_url = keep_host(resp.headers['location'], build_url)
if wait: if wait:
print("Waiting for publish job") print("Waiting for publish job")
job = await wait_for_job(session, job_url, token); job = await wait_for_job(session, job_url, token)
reparse_job_results(job) reparse_job_results(job)
job["location"] = job_url job["location"] = keep_host(job_url, build_url)
return job return job
@ -547,7 +548,7 @@ async def create_command(session, args):
if resp.status != 200: if resp.status != 200:
raise ApiError(resp, await resp.text()) raise ApiError(resp, await resp.text())
data = await resp.json() data = await resp.json()
data["location"] = resp.headers['location'] data["location"] = keep_host(resp.headers['location'], build_url)
if not args.print_output: if not args.print_output:
print(resp.headers['location']) print(resp.headers['location'])
return data return data
@ -565,6 +566,14 @@ def should_skip_delta(id, globs):
return True return True
return False return False
# work around for url_for returning http urls when flat-manager is behind a reverse proxy and aiohttp not keeping the Authorization header across redirects in version < 4
def keep_host(location, original):
loc_url = URL(location)
org_url = URL(original)
loc_url.scheme = org_url.scheme
loc_url.authority = org_url.authority
return str(loc_url)
def build_url_to_api(build_url): def build_url_to_api(build_url):
parts = urlparse(build_url) parts = urlparse(build_url)
path = os.path.dirname(os.path.dirname(parts.path)) path = os.path.dirname(os.path.dirname(parts.path))
@ -652,11 +661,11 @@ async def push_command(session, args):
update_job_url = build_url_to_api(args.build_url) + "/job/" + str(update_job_id) update_job_url = build_url_to_api(args.build_url) + "/job/" + str(update_job_id)
if args.wait_update: if args.wait_update:
print("Waiting for repo update job") print("Waiting for repo update job")
update_job = await wait_for_job (session, update_job_url, token); update_job = await wait_for_job (session, update_job_url, token)
else: else:
update_job = await get_job(session, update_job_url, token) update_job = await get_job(session, update_job_url, token)
reparse_job_results(update_job) reparse_job_results(update_job)
update_job["location"] = update_job_url update_job["location"] = keep_host(update_job_url, update_job_url)
data = await get_build(session, args.build_url, args.token) data = await get_build(session, args.build_url, args.token)
if commit_job: if commit_job:
@ -679,11 +688,11 @@ async def publish_command(session, args):
update_job_url = build_url_to_api(args.build_url) + "/job/" + str(update_job_id) update_job_url = build_url_to_api(args.build_url) + "/job/" + str(update_job_id)
if args.wait_update: if args.wait_update:
print("Waiting for repo update job") print("Waiting for repo update job")
update_job = await wait_for_job(session, update_job_url, args.token); update_job = await wait_for_job(session, update_job_url, args.token)
else: else:
update_job = await get_job(session, update_job_url, args.token) update_job = await get_job(session, update_job_url, args.token)
reparse_job_results(update_job) reparse_job_results(update_job)
update_job["location"] = update_job_url update_job["location"] = keep_host(update_job_url, args.build_url)
return job return job
async def purge_command(session, args): async def purge_command(session, args):