Добавить заголовки прокси в указанное местоположение с помощью регулярного выражения вложенного местоположения

Я пытаюсь настроить конечную точку WebSocket на своем Rails API с помощью Nginx и Puma.

Что у меня есть (работает, но некрасиво)

Следующая конфигурация Nginx работает отлично, однако я чувствую, что могу использовать что-то более умное, чтобы избежать дублирования на обоих @puma и @puma_ws названные места:

upstream puma {
    server unix:///path/to/socket.sock;
}

server {
    listen 80;
    server_name example.com;

    root /var/www/public;

    location / {
        try_files $uri/index.html $uri @puma;
    }

    location ~ ^/api/websocket {
        try_files $uri/index.html $uri @puma_ws;
    }

    location @puma {
       proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
       proxy_set_header Host $http_host;

       proxy_redirect off;

       proxy_pass http://puma;
   }

   location @puma_ws {
       # These two lines are the only difference compared to @puma
       proxy_set_header Upgrade $http_upgrade;
       proxy_set_header Connection "upgrade";

       proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
       proxy_set_header Host $http_host;

       proxy_redirect off;

       proxy_pass http://puma;
   }
}

Примечание: я использую регулярное выражение для расположения WS (~ ^/api/websocket), потому что в моем реальном варианте использования мне нужно иметь более одной конечной точки WS. Я упростил его для простоты.

Первоначальная идея

Моя первая идея заключалась в том, чтобы иметь только одно названное место @puma, который будет иметь вложенное местоположение с регулярным выражением, которое добавит только два необходимых proxy_set_header.

Таким образом у меня будет только один try_files с единственным @puma именованное местоположение, которое само добавляло бы заголовки, используя вложенный location

Однако afaik не может быть вложенным location блок в названном месте.

Есть ли у вас лучшая идея добавить эти заголовки на основе теста на фактическом URI?

Спасибо!