From 339683fce3179deaa49efb1ba2db83375ae90acf Mon Sep 17 00:00:00 2001
From: Nick Hudson <nick.hudson@gmail.com>
Date: Mon, 19 Aug 2024 14:16:24 -0500
Subject: [PATCH] add ability to set DNS record as well

---
 charts/tembo-ai/templates/inference-gateway/service-lb.yaml | 5 +++++
 charts/tembo-ai/values.yaml                                 | 4 ++++
 2 files changed, 9 insertions(+)

diff --git a/charts/tembo-ai/templates/inference-gateway/service-lb.yaml b/charts/tembo-ai/templates/inference-gateway/service-lb.yaml
index 9cdd228af..4c74903bb 100644
--- a/charts/tembo-ai/templates/inference-gateway/service-lb.yaml
+++ b/charts/tembo-ai/templates/inference-gateway/service-lb.yaml
@@ -7,6 +7,11 @@ metadata:
     {{- include "tembo-ai.inferenceGateway.labels" . | nindent 4 }}
   annotations:
     {{- toYaml .Values.inferenceGateway.internalLoadBalancer.annotations | nindent 4 }}
+    {{- if .Values.inferenceGateway.internalLoadBalancer.route53.enabled }}
+    external-dns.alpha.kubernetes.io/hostname: {{ .Values.inferenceGateway.internalLoadBalancer.route53.recordName }}.{{ .Values.inferenceGateway.internalLoadBalancer.route53.zoneName }}
+    external-dns.alpha.kubernetes.io/ttl: {{ .Values.inferenceGateway.internalLoadBalancer.route53.ttl | quote }}
+    {{- end }}
+
 spec:
   type: LoadBalancer
   ports:
diff --git a/charts/tembo-ai/values.yaml b/charts/tembo-ai/values.yaml
index daf9a6292..9d715757e 100644
--- a/charts/tembo-ai/values.yaml
+++ b/charts/tembo-ai/values.yaml
@@ -60,6 +60,10 @@ inferenceGateway:
       service.beta.kubernetes.io/aws-load-balancer-type: nlb
       service.beta.kubernetes.io/aws-load-balancer-internal: "true"
       service.beta.kubernetes.io/aws-load-balancer-scheme: internal
+    route53:
+      enabled: false
+      zoneName: ""
+      recordName: "inference-gateway"
     port: 8080
   args: []
   command: