From 0f917ab1c2a82597228f4aa4770e1205a47fdcbe Mon Sep 17 00:00:00 2001
From: Alexandre Archambault <alexarchambault@users.noreply.github.com>
Date: Thu, 15 Jun 2023 18:14:20 +0200
Subject: [PATCH] Take spark.ui.reverseProxyUrl into account (#323)

---
 .../NotebookSparkSessionBuilder.scala                | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/modules/almond-spark/src/main/scala/org/apache/spark/sql/almondinternals/NotebookSparkSessionBuilder.scala b/modules/almond-spark/src/main/scala/org/apache/spark/sql/almondinternals/NotebookSparkSessionBuilder.scala
index bc93ebd..15b8d56 100644
--- a/modules/almond-spark/src/main/scala/org/apache/spark/sql/almondinternals/NotebookSparkSessionBuilder.scala
+++ b/modules/almond-spark/src/main/scala/org/apache/spark/sql/almondinternals/NotebookSparkSessionBuilder.scala
@@ -83,7 +83,17 @@ class NotebookSparkSessionBuilder(implicit
 
       val session = super.getOrCreate()
 
-      for (url <- session.sparkContext.uiWebUrl)
+      val reverseProxyUrlOpt = session.sparkContext.getConf.getOption("spark.ui.reverseProxyUrl")
+        .filter { _ =>
+          session.sparkContext.getConf.getOption("spark.ui.reverseProxy").contains("true")
+        }
+      val uiUrlOpt = reverseProxyUrlOpt
+        .map { reverseProxyUrl =>
+          val appId = session.sparkContext.applicationId
+          s"$reverseProxyUrl/proxy/$appId"
+        }
+        .orElse(session.sparkContext.uiWebUrl)
+      for (url <- uiUrlOpt)
         html(s"""<a target="_blank" href="$url">Spark UI</a>""")
 
       session.sparkContext.addSparkListener(