From ba920a033a1f3f59cf9632d50158829e423aa8c1 Mon Sep 17 00:00:00 2001
From: Puneet Singh <101257603+puneet-bdp@users.noreply.github.com>
Date: Tue, 6 Aug 2024 19:17:52 +0400
Subject: [PATCH] [DNA-0000]: Add Trino original user header for impersonated
 users (#29)

* [DNA-00000]: Fix dependencies version

* [DNA-00000]: Add Trino original user header
---
 redash/query_runner/trino.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/redash/query_runner/trino.py b/redash/query_runner/trino.py
index faba4819de..1a62d79bd8 100644
--- a/redash/query_runner/trino.py
+++ b/redash/query_runner/trino.py
@@ -112,7 +112,10 @@ def run_query(self, query, user):
             host=self.configuration.get("host", ""),
             port=self.configuration.get("port", 8080),
             catalog=self.configuration.get("catalog", "hive"),
-            http_headers={trino.constants.HEADER_CLIENT_INFO: str(user)},
+            http_headers={
+                trino.constants.HEADER_CLIENT_INFO: str(user),
+                "X-Trino-Original-User": str(user)
+            },
             source="Redash",
             schema=self.configuration.get("schema", "default"),
             user=self.configuration.get("username"),