map $http_upgrade $connection_upgrade {
    default upgrade;
    ''      close;
}

upstream rustfs_backend {
    server 8.153.151.51:9000;
    keepalive 64;
}

server {
    listen 80;
    server_name localhost;
    # 全局或针对存储服务设置最大上传限制 (100G)
    client_max_body_size 100G;
    # 指定 Docker 内置 DNS 解析器，并设置 30 秒缓存
    resolver 127.0.0.11 valid=30s ipv6=off;

    # =========================================================================
    # 1. RustFS 对象存储服务转发 (/storage/)
    # =========================================================================
    location /storage/ {
        # 核心：透传 Host，确保 RustFS 生成的 Presigned URL 包含公网地址
        proxy_set_header Host $http_host;
        proxy_set_header X-Real-IP $remote_addr;
        proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
        proxy_set_header X-Forwarded-Proto $scheme;

        # 转发至 RustFS
        # 注意：如果 RustFS 内部接口也是 /storage/...，请把末尾的 / 去掉
        proxy_pass http://rustfs_backend/;

        # HTTP/1.1 长连接支持
        proxy_http_version 1.1;
        proxy_set_header Connection "";

        # 性能优化：关闭双向 Buffer，实现大文件流式传输
        proxy_buffering off;
        proxy_request_buffering off;

        # 大文件传输超时设置
        proxy_connect_timeout 300s;
        proxy_read_timeout    3600s;
        proxy_send_timeout    3600s;
    }

    # =========================================================================
    # 2. Jupyter 服务配置
    # =========================================================================
    # 拒绝直接访问目录
    location ~ ^/jupyter/(?<workspace_id>[^/]+)/?$ {
        return 403 "Direct directory access is forbidden. Please specify a notebook path.";
    }

    # 精准匹配入口：仅允许访问特定 notebook 页面与配套资源
    location ~ ^/jupyter/(?<workspace_id>[^/]+)(?<rest_uri>/.*)$ {

        # 触发后端鉴权
        auth_request /internal-auth;

        auth_request_set $target_upstream $upstream_http_x_upstream_addr;
        auth_request_set $jupyter_token $upstream_http_x_jupyter_internal_token;

        # 拼接完整 URL 并输出到 Response Header 方便调试
        add_header X-Debug-Full-Url "$target_upstream/jupyter/$workspace_id$rest_uri$is_args$args" always;

        # 代理到具体的 Jupyter 子进程
        proxy_pass $target_upstream/jupyter/$workspace_id$rest_uri$is_args$args;
        proxy_set_header Authorization "token $jupyter_token";

        # 支持 WebSocket (Jupyter Kernel 必需)
        proxy_http_version 1.1;
        proxy_set_header Upgrade $http_upgrade;
        proxy_set_header Connection $connection_upgrade;

        proxy_set_header Host $host;
        proxy_set_header X-Real-IP $remote_addr;
        proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
    }

    # 2. 内部 Auth 子请求 location
    location = /internal-auth {
        internal;
        # 打到宿主机的 FastAPI 8000 端口
        proxy_pass http://backend:8000/api/v1/auth/jupyter;

        proxy_pass_request_body off;
        proxy_set_header Content-Length "";

        proxy_pass_header x-upstream-addr;
        proxy_pass_header x-jupyter-internal-token;

        proxy_set_header X-Original-Workspace-Id $workspace_id;
        proxy_set_header X-Original-URI $request_uri;

        proxy_set_header Cookie $http_cookie;
        proxy_set_header Authorization $http_authorization;
    }

    # 拒绝其余非法路径
    location /jupyter/ {
        return 403 "Access Denied";
    }
}