From 7efdf5717fc55bf150facf47e284536ada771f5b Mon Sep 17 00:00:00 2001
From: Mohamed Mehdi BEN AISSA
 <diode-ben-aissa.consultant@dgfip.finances.gouv.fr>
Date: Fri, 27 Oct 2023 15:06:32 +0200
Subject: [PATCH] fix(knox): fix broken yarn links for livy

---
 .../services/livy/0.4.3/rewrite.xml.j2        | 58 +++++++++++++++++++
 .../services/livy/0.4.3/service.xml.j2        | 35 +++++++++++
 roles/knox/gateway/tasks/config.yml           | 25 ++++++++
 3 files changed, 118 insertions(+)
 create mode 100644 roles/knox/common/templates/services/livy/0.4.3/rewrite.xml.j2
 create mode 100644 roles/knox/common/templates/services/livy/0.4.3/service.xml.j2

diff --git a/roles/knox/common/templates/services/livy/0.4.3/rewrite.xml.j2 b/roles/knox/common/templates/services/livy/0.4.3/rewrite.xml.j2
new file mode 100644
index 00000000..c8a7f3c0
--- /dev/null
+++ b/roles/knox/common/templates/services/livy/0.4.3/rewrite.xml.j2
@@ -0,0 +1,58 @@
+<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
+<!--
+   Licensed to the Apache Software Foundation (ASF) under one or more
+   contributor license agreements.  See the NOTICE file distributed with
+   this work for additional information regarding copyright ownership.
+   The ASF licenses this file to You under the Apache License, Version 2.0
+   (the "License"); you may not use this file except in compliance with
+   the License.  You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+-->
+<rules>
+  <!-- These should be removed at some point since Livy
+       doesn't have /v1 in the REST API specification -->
+  <rule dir="IN" name="LIVYSERVER/livy/root/v1/inbound/" pattern="*://*:*/**/livy/v1">
+    <rewrite template="{$serviceUrl[LIVYSERVER]}"/>
+  </rule>
+  <rule dir="IN" name="LIVYSERVER/livy/path/v1/inbound" pattern="*://*:*/**/livy/v1/{path=**}?{**}">
+    <rewrite template="{$serviceUrl[LIVYSERVER]}/{path=**}?{**}"/>
+  </rule>
+
+  <!-- Prefer these rules without v1 in the url -->
+  <rule dir="IN" name="LIVYSERVER/livy/root/inbound" pattern="*://*:*/**/livy">
+    <rewrite template="{$serviceUrl[LIVYSERVER]}"/>
+  </rule>
+  <rule dir="IN" name="LIVYSERVER/livy/path/inbound" pattern="*://*:*/**/livy/{path=**}?{**}">
+    <rewrite template="{$serviceUrl[LIVYSERVER]}/{path=**}?{**}"/>
+  </rule>
+
+  <rule dir="OUT" name="LIVYSERVER/livy/outbound/sparkurl" pattern="*://*:*/proxy/{**}">
+    <rewrite template="{$frontend[url]}/yarn/proxy/{**}"/>
+  </rule>
+
+  <rule dir="OUT" name="LIVYSERVER/livy/outbound/logs" pattern="{scheme}://{host}:{port}/node/containerlogs/{**}?{**}">
+    <rewrite template="{$frontend[url]}/yarn/node/containerlogs/{**}?{**}?{scheme}?host={$hostmap(host)}?{port}"/>
+  </rule>
+
+  <rule dir="OUT" name="LIVYSERVER/livy/outbound/logs2" pattern="{scheme}://{host}:{port}/node/containerlogs/{**}">
+    <rewrite template="{$frontend[url]}/yarn/node/containerlogs/{**}?{scheme}?host={$hostmap(host)}?{port}"/>
+  </rule>
+
+  <rule dir="OUT" name="LIVYSERVER/livy/outbound/headers/ui" pattern="{scheme}://{host}:{port}/ui/">
+    <rewrite template="{$frontend[url]}/livy/ui/"/>
+  </rule>
+
+  <filter name="LIVYSERVER/livy/outbound/headers">
+    <content type="application/x-http-headers">
+      <apply path="Location" rule="LIVYSERVER/livy/outbound/headers/ui"/>
+    </content>
+  </filter>
+
+</rules>
diff --git a/roles/knox/common/templates/services/livy/0.4.3/service.xml.j2 b/roles/knox/common/templates/services/livy/0.4.3/service.xml.j2
new file mode 100644
index 00000000..f7434565
--- /dev/null
+++ b/roles/knox/common/templates/services/livy/0.4.3/service.xml.j2
@@ -0,0 +1,35 @@
+<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
+<!--
+   Licensed to the Apache Software Foundation (ASF) under one or more
+   contributor license agreements.  See the NOTICE file distributed with
+   this work for additional information regarding copyright ownership.
+   The ASF licenses this file to You under the Apache License, Version 2.0
+   (the "License"); you may not use this file except in compliance with
+   the License.  You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+-->
+<service role="LIVYSERVER" name="livy" version="0.4.3">
+  <metadata>
+    <type>API_AND_UI</type>
+    <context>/livy/</context>
+    <shortDesc>Livy Server</shortDesc>
+    <description>Apache Livy is a service that enables easy interaction with a Spark cluster over a REST interface.</description>
+  </metadata>
+  <routes>
+    <route path="/livy/**?**"/>
+    <route path="/livy">
+      <rewrite apply="LIVYSERVER/livy/outbound/headers" to="response.headers"/>
+    </route>
+    <route path="/livy/">
+      <rewrite apply="LIVYSERVER/livy/outbound/headers" to="response.headers"/>
+    </route>
+  </routes>
+  <dispatch classname="org.apache.knox.gateway.livy.LivyDispatch"/>
+</service>
diff --git a/roles/knox/gateway/tasks/config.yml b/roles/knox/gateway/tasks/config.yml
index 0214af2f..88601910 100644
--- a/roles/knox/gateway/tasks/config.yml
+++ b/roles/knox/gateway/tasks/config.yml
@@ -72,6 +72,31 @@
     group: "{{ knox_group }}"
     mode: "644"
 
+# Livy Spark 2 service definition
+- name: Create Livy Spark 2 service dir
+  ansible.builtin.file:
+    path: "{{ knox_data_dir }}/data/services/livy/0.4.3"
+    state: directory
+    owner: "{{ knox_user }}"
+    group: "{{ knox_group }}"
+    mode: "755"
+
+- name: Template Livy Spark 2 service.xml
+  ansible.builtin.template:
+    src: services/livy/0.4.3/service.xml.j2
+    dest: "{{ knox_data_dir }}/data/services/livy/0.4.3/service.xml"
+    owner: "{{ knox_user }}"
+    group: "{{ knox_group }}"
+    mode: "644"
+
+- name: Template Livy Spark 2 rewrite.xml
+  ansible.builtin.template:
+    src: services/livy/0.4.3/rewrite.xml.j2
+    dest: "{{ knox_data_dir }}/data/services/livy/0.4.3/rewrite.xml"
+    owner: "{{ knox_user }}"
+    group: "{{ knox_group }}"
+    mode: "644"
+
 - name: Template Knox gateway-site.xml
   ansible.builtin.template:
     src: gateway-site.xml.j2