From a63e75a7de4a983b724f81bac4befe935c34b16d Mon Sep 17 00:00:00 2001
From: Sebastian Florek <sebastian@plural.sh>
Date: Wed, 18 Dec 2024 14:38:21 +0100
Subject: [PATCH] allow configuring infrastructure stack cluster

---
 catalogs/data/airbyte/airbyte-stack.yaml | 2 +-
 catalogs/data/dagster/dagster.yaml       | 2 +-
 catalogs/data/mlflow/mlflow-stack.yaml   | 2 +-
 setup/catalogs/data/airbyte.yaml         | 4 ++++
 setup/catalogs/data/dagster.yaml         | 4 ++++
 setup/catalogs/data/mlflow.yaml          | 4 ++++
 6 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/catalogs/data/airbyte/airbyte-stack.yaml b/catalogs/data/airbyte/airbyte-stack.yaml
index 5ce96af6..a2abd4e7 100644
--- a/catalogs/data/airbyte/airbyte-stack.yaml
+++ b/catalogs/data/airbyte/airbyte-stack.yaml
@@ -18,5 +18,5 @@ spec:
   configuration:
     version: '1.8'
   clusterRef:
-    name: {{ context.cluster }}
+    name: {{ context.stackCluster }}
     namespace: infra
diff --git a/catalogs/data/dagster/dagster.yaml b/catalogs/data/dagster/dagster.yaml
index 2359a58b..0dcde1c3 100644
--- a/catalogs/data/dagster/dagster.yaml
+++ b/catalogs/data/dagster/dagster.yaml
@@ -18,7 +18,7 @@ spec:
   configuration:
     version: '1.8'
   clusterRef:
-    name: {{ context.cluster }}
+    name: {{ context.stackCluster }}
     namespace: infra
 ---
 apiVersion: deployments.plural.sh/v1alpha1
diff --git a/catalogs/data/mlflow/mlflow-stack.yaml b/catalogs/data/mlflow/mlflow-stack.yaml
index ae6a2e4e..fcc25737 100644
--- a/catalogs/data/mlflow/mlflow-stack.yaml
+++ b/catalogs/data/mlflow/mlflow-stack.yaml
@@ -18,5 +18,5 @@ spec:
   configuration:
     version: '1.8'
   clusterRef:
-    name: {{ context.cluster }}
+    name: {{ context.stackCluster }}
     namespace: infra
diff --git a/setup/catalogs/data/airbyte.yaml b/setup/catalogs/data/airbyte.yaml
index 360d211c..fa4478f3 100644
--- a/setup/catalogs/data/airbyte.yaml
+++ b/setup/catalogs/data/airbyte.yaml
@@ -51,6 +51,10 @@ spec:
   - name: cluster
     type: STRING
     documentation: Handle of the cluster you want to deploy airbyte to.
+  - name: stackCluster
+    type: STRING
+    documentation: Handle of the cluster used to run Infrastructure Stacks for provisioning the infrastructure. Defaults to the management cluster.
+    default: mgmt
   - name: cloud
     type: ENUM
     documentation: Cloud provider you want to deploy airbyte to.
diff --git a/setup/catalogs/data/dagster.yaml b/setup/catalogs/data/dagster.yaml
index 31783e47..a56d9f3d 100644
--- a/setup/catalogs/data/dagster.yaml
+++ b/setup/catalogs/data/dagster.yaml
@@ -41,6 +41,10 @@ spec:
     - name: cluster
       type: STRING
       documentation: The cluster you want to deploy to.
+    - name: stackCluster
+      type: STRING
+      documentation: Handle of the cluster used to run Infrastructure Stacks for provisioning the infrastructure. Defaults to the management cluster.
+      default: mgmt
     - name: cloud
       type: ENUM
       documentation: The cloud you want to deploy to.
diff --git a/setup/catalogs/data/mlflow.yaml b/setup/catalogs/data/mlflow.yaml
index 871906e7..a8971d8c 100644
--- a/setup/catalogs/data/mlflow.yaml
+++ b/setup/catalogs/data/mlflow.yaml
@@ -54,6 +54,10 @@ spec:
     - name: cluster
       type: STRING
       documentation: Handle of the cluster you want to deploy mlflow to.
+    - name: stackCluster
+      type: STRING
+      documentation: Handle of the cluster used to run Infrastructure Stacks for provisioning the infrastructure. Defaults to the management cluster.
+      default: mgmt
     - name: cloud
       type: ENUM
       documentation: Cloud provider you want to deploy mlflow to.