mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 05:53:43 +00:00
e1aa8b1cb0
This works around tus/tusd#322, which in turn is caused by aws/aws-sdk-go-v2#1816. This requires separate authentication via service account key. Fixes: #34186.
85 lines
3.4 KiB
Python
85 lines
3.4 KiB
Python
import os
|
|
from typing import Any
|
|
from urllib.parse import SplitResult
|
|
|
|
from django.conf import settings
|
|
from django.core.management.base import BaseCommand, CommandError, CommandParser
|
|
from typing_extensions import override
|
|
|
|
from scripts.lib.zulip_tools import get_config, get_config_file
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = """Starts the tusd server"""
|
|
|
|
@override
|
|
def add_arguments(self, parser: CommandParser) -> None:
|
|
parser.add_argument(
|
|
"listen", help="[Port, or address:port, to bind HTTP server to]", type=str
|
|
)
|
|
local_port = 80
|
|
config_file = get_config_file()
|
|
if get_config(config_file, "application_server", "http_only", False):
|
|
local_port = int(
|
|
get_config(config_file, "application_server", "nginx_listen_port", "80")
|
|
)
|
|
parser.add_argument(
|
|
"hooks_http",
|
|
help="[An HTTP endpoint to which hook events will be sent to]",
|
|
default=f"http://127.0.0.1:{local_port}/api/internal/tusd",
|
|
nargs="?",
|
|
)
|
|
|
|
@override
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
listen = options["listen"]
|
|
if listen.isdigit():
|
|
addr, port = "127.0.0.1", int(listen)
|
|
else:
|
|
r = SplitResult("", listen, "", "", "")
|
|
if r.port is None:
|
|
raise CommandError(f"{listen!r} does not have a valid port number.")
|
|
addr, port = r.hostname or "127.0.0.1", r.port
|
|
|
|
hooks_http = options["hooks_http"]
|
|
|
|
# https://tus.github.io/tusd/getting-started/configuration/
|
|
# We do not set a maximum upload size, as the pre-create hooks
|
|
# will set the max size that they want, based on the intended
|
|
# use of the uploaded file.
|
|
tusd_args = [
|
|
"tusd",
|
|
"-base-path=/api/v1/tus/",
|
|
f"-port={port}",
|
|
f"-host={addr}",
|
|
"-behind-proxy",
|
|
f"-hooks-http={hooks_http}",
|
|
"-hooks-http-forward-headers=Cookie,Authorization",
|
|
"--hooks-enabled-events=pre-create,pre-finish,pre-terminate",
|
|
"-disable-download",
|
|
"--show-startup-logs=false",
|
|
]
|
|
env_vars = os.environ.copy()
|
|
if settings.LOCAL_UPLOADS_DIR is not None:
|
|
assert settings.LOCAL_FILES_DIR is not None
|
|
tusd_args.append(f"-upload-dir={settings.LOCAL_FILES_DIR}")
|
|
elif settings.S3_ENDPOINT_URL in (
|
|
"https://storage.googleapis.com",
|
|
"https://storage.googleapis.com/",
|
|
):
|
|
tusd_args.append(f"-gcs-bucket={settings.S3_AUTH_UPLOADS_BUCKET}")
|
|
env_vars["GCS_SERVICE_ACCOUNT_FILE"] = "/etc/zulip/gcp_key.json"
|
|
else:
|
|
tusd_args.append(f"-s3-bucket={settings.S3_AUTH_UPLOADS_BUCKET}")
|
|
if settings.S3_ENDPOINT_URL is not None:
|
|
tusd_args.append(f"-s3-endpoint={settings.S3_ENDPOINT_URL}")
|
|
if settings.S3_KEY is not None:
|
|
env_vars["AWS_ACCESS_KEY_ID"] = settings.S3_KEY
|
|
if settings.S3_SECRET_KEY is not None:
|
|
env_vars["AWS_SECRET_ACCESS_KEY"] = settings.S3_SECRET_KEY
|
|
if settings.S3_REGION is not None:
|
|
env_vars["AWS_REGION"] = settings.S3_REGION
|
|
if settings.S3_SKIP_CHECKSUM:
|
|
env_vars["AWS_REQUEST_CHECKSUM_CALCULATION"] = "when_required"
|
|
os.execvpe("tusd", tusd_args, env_vars)
|