From fc1b7c8a7495fad8d9a463029216e75a532915ce Mon Sep 17 00:00:00 2001
From: Elena Makarova <el-makarova@yandex-team.ru>
Date: Fri, 29 Aug 2025 13:25:49 +0300
Subject: [PATCH] fix(streaming): use CSRF-token if presented

---
 src/services/api/streaming.ts | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/services/api/streaming.ts b/src/services/api/streaming.ts
index 1b78247bf6..08dc2ba636 100644
--- a/src/services/api/streaming.ts
+++ b/src/services/api/streaming.ts
@@ -28,6 +28,12 @@ export interface StreamQueryOptions {
 }
 
 export class StreamingAPI extends BaseYdbAPI {
+    private csrfToken?: string;
+
+    setCSRFToken = (token: string) => {
+        this.csrfToken = token;
+    };
+
     async streamQuery<Action extends Actions>(
         params: StreamQueryParams<Action>,
         options: StreamQueryOptions,
@@ -48,6 +54,10 @@ export class StreamingAPI extends BaseYdbAPI {
             'Content-Type': 'application/json',
         });
 
+        if (this.csrfToken) {
+            headers.set('X-CSRF-Token', this.csrfToken);
+        }
+
         if (params.tracingLevel) {
             headers.set('X-Trace-Verbosity', String(params.tracingLevel));
         }