From 780ccccd65006c3801c8e502791df91763f431f8 Mon Sep 17 00:00:00 2001
From: haiqing chen <2538302280@qq.com>
Date: Thu, 19 Dec 2024 14:36:42 +0800
Subject: [PATCH] /autoUpdate-kes-minioClinet-certificate

---
 pkg/controller/kes.go | 66 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)

diff --git a/pkg/controller/kes.go b/pkg/controller/kes.go
index 85a37585bb0..ffe4d4eb33d 100644
--- a/pkg/controller/kes.go
+++ b/pkg/controller/kes.go
@@ -152,6 +152,31 @@ func kesStatefulSetMatchesSpec(tenant *miniov2.Tenant, existingStatefulSet *apps
 	return true, nil
 }
 
+func (c *Controller) recreateKesCertsOnTenant(ctx context.Context, tenant *miniov2.Tenant, nsName types.NamespacedName) error {
+	err := c.kubeClientSet.CoreV1().Secrets(tenant.Namespace).Delete(ctx, tenant.KESTLSSecretName(), metav1.DeleteOptions{})
+	if err != nil {
+		return err
+	}
+
+	// Then delete the CSR
+	err = c.deleteCSR(ctx, tenant.KESCSRName())
+	if err != nil {
+		return err
+	}
+
+	// Finally re-create the certs on the tenant
+	return c.checkAndCreateKESCSR(ctx, nsName, tenant)
+}
+
+// recreateMinIOClientCerts re-creates the client certificates for MinIO
+func (c *Controller) recreateMinIOClientCerts(ctx context.Context, tenant *miniov2.Tenant, nsName types.NamespacedName) error {
+	err := c.kubeClientSet.CoreV1().Secrets(tenant.Namespace).Delete(ctx, tenant.MinIOClientTLSSecretName(), metav1.DeleteOptions{})
+	if err != nil {
+		return err
+	}
+	return c.checkAndCreateMinIOClientCertificates(ctx, nsName, tenant)
+}
+
 func (c *Controller) checkKESCertificatesStatus(ctx context.Context, tenant *miniov2.Tenant, nsName types.NamespacedName) (err error) {
 	if !tenant.ExternalClientCert() {
 		if err = c.checkAndCreateMinIOClientCertificates(ctx, nsName, tenant); err != nil {
@@ -179,6 +204,47 @@ func (c *Controller) checkKESCertificatesStatus(ctx context.Context, tenant *min
 			}
 		}
 	}
+	if tenant.AutoCert() {
+		// renew the KES certificates
+		tlsSecret, err := c.getCertificateSecret(ctx, tenant.Namespace, tenant.KESTLSSecretName())
+		if err != nil {
+			if k8serrors.IsNotFound(err) {
+				if err := c.checkAndCreateKESCSR(ctx, nsName, tenant); err != nil {
+					return err
+				}
+				// TLS secret not found, delete CSR if exists and start certificate generation process again
+				if err := c.deleteCSR(ctx, tenant.KESCSRName()); err != nil {
+					return err
+				}
+			} else {
+				return err
+			}
+		}
+
+		needsRenewal, err := c.certNeedsRenewal(tlsSecret)
+		if err != nil {
+			return err
+		}
+
+		if needsRenewal {
+			return c.recreateKesCertsOnTenant(ctx, tenant, nsName)
+		}
+
+		// renew the client certificates
+		tlsSecret, err = c.getCertificateSecret(ctx, tenant.Namespace, tenant.MinIOClientTLSSecretName())
+		if err != nil {
+			return err
+		}
+
+		needsRenewal, err = c.certNeedsRenewal(tlsSecret)
+
+		if err != nil {
+			return err
+		}
+		if needsRenewal {
+			return c.recreateMinIOClientCerts(ctx, tenant, nsName)
+		}
+	}
 	return nil
 }