From e2ffefead8df885d8e9eea2815b5a415bb98491f Mon Sep 17 00:00:00 2001
From: Shubhra Deshpande <shubhrajayant+github.com>
Date: Fri, 4 Oct 2024 18:30:21 -0400
Subject: [PATCH] Added scale up component for postgres and clair resource

---
 .../scale-down-clair/clair-pg-scale-up.patch.yaml           | 6 ++++++
 .../scale-down-clair/clair.deployment-scale-up.patch.yaml   | 6 ++++++
 .../clairpgupgrade/scale-down-clair/kustomization.yaml      | 6 ++++--
 pkg/kustomize/kustomize.go                                  | 3 ++-
 4 files changed, 18 insertions(+), 3 deletions(-)
 create mode 100644 kustomize/components/clairpgupgrade/scale-down-clair/clair-pg-scale-up.patch.yaml
 create mode 100644 kustomize/components/clairpgupgrade/scale-down-clair/clair.deployment-scale-up.patch.yaml

diff --git a/kustomize/components/clairpgupgrade/scale-down-clair/clair-pg-scale-up.patch.yaml b/kustomize/components/clairpgupgrade/scale-down-clair/clair-pg-scale-up.patch.yaml
new file mode 100644
index 000000000..875b1827f
--- /dev/null
+++ b/kustomize/components/clairpgupgrade/scale-down-clair/clair-pg-scale-up.patch.yaml
@@ -0,0 +1,6 @@
+apiVersion: apps/v1
+kind: Deployment
+metadata:
+  name: clair-app
+spec:
+  replicas: 2
diff --git a/kustomize/components/clairpgupgrade/scale-down-clair/clair.deployment-scale-up.patch.yaml b/kustomize/components/clairpgupgrade/scale-down-clair/clair.deployment-scale-up.patch.yaml
new file mode 100644
index 000000000..25835358e
--- /dev/null
+++ b/kustomize/components/clairpgupgrade/scale-down-clair/clair.deployment-scale-up.patch.yaml
@@ -0,0 +1,6 @@
+apiVersion: apps/v1
+kind: Deployment
+metadata:
+  name: clair-postgres
+spec:
+  replicas: 2
\ No newline at end of file
diff --git a/kustomize/components/clairpgupgrade/scale-down-clair/kustomization.yaml b/kustomize/components/clairpgupgrade/scale-down-clair/kustomization.yaml
index c81ba473a..52e98776d 100644
--- a/kustomize/components/clairpgupgrade/scale-down-clair/kustomization.yaml
+++ b/kustomize/components/clairpgupgrade/scale-down-clair/kustomization.yaml
@@ -2,5 +2,7 @@ apiVersion: kustomize.config.k8s.io/v1alpha1
 kind: Component
 patchesStrategicMerge:
   - ./clair.deployment.patch.yaml
-resources:
-  - "../base"
+  - ./clair-pg-scale-up.patch.yaml
+  - ./clair.deployment-scale-up.patch.yaml
+components:
+  - ../base
diff --git a/pkg/kustomize/kustomize.go b/pkg/kustomize/kustomize.go
index b2d64c5f9..b72e63ced 100644
--- a/pkg/kustomize/kustomize.go
+++ b/pkg/kustomize/kustomize.go
@@ -439,8 +439,9 @@ func KustomizationFor(
 	if ctx.NeedsClairPgUpgrade {
 		if v1.ComponentIsManaged(quay.Spec.Components, v1.ComponentClair) {
 			componentPaths = append(componentPaths, "../components/clairpgupgrade/scale-down-clair")
+		} else {
+			componentPaths = append(componentPaths, "../components/clairpgupgrade/base")
 		}
-		componentPaths = append(componentPaths, "../components/clairpgupgrade/base")
 	}
 
 	images := []types.Image{}