From 5831f368bc090dfa92e161fa93084bd2b01d1b4d Mon Sep 17 00:00:00 2001
From: Shinnosuke Sawada-Dazai <shin@warashi.dev>
Date: Wed, 4 Dec 2024 14:40:44 +0900
Subject: [PATCH] Implement preparing manifest applier in kubernetes plugin
 (#5389)

* Implement FindDeployTarget method to retrieve deploy target by name

Signed-off-by: Shinnosuke Sawada-Dazai <shin@warashi.dev>

* Implement ensureK8sSyncStage

Signed-off-by: Shinnosuke Sawada-Dazai <shin@warashi.dev>

---------

Signed-off-by: Shinnosuke Sawada-Dazai <shin@warashi.dev>
---
 .../plugin/kubernetes/deployment/server.go    | 41 +++++++++++++++++--
 pkg/configv1/piped.go                         | 10 +++++
 2 files changed, 48 insertions(+), 3 deletions(-)

diff --git a/pkg/app/pipedv1/plugin/kubernetes/deployment/server.go b/pkg/app/pipedv1/plugin/kubernetes/deployment/server.go
index 3893b6cc5e..b5a6984037 100644
--- a/pkg/app/pipedv1/plugin/kubernetes/deployment/server.go
+++ b/pkg/app/pipedv1/plugin/kubernetes/deployment/server.go
@@ -15,7 +15,9 @@
 package deployment
 
 import (
+	"cmp"
 	"context"
+	"encoding/json"
 	"errors"
 	"time"
 
@@ -33,10 +35,20 @@ import (
 	"google.golang.org/grpc/status"
 )
 
+const (
+	defaultKubectlVersion = "1.18.2"
+)
+
 type toolClient interface {
 	InstallTool(ctx context.Context, name, version, script string) (string, error)
 }
 
+type toolRegistry interface {
+	Kubectl(ctx context.Context, version string) (string, error)
+	Kustomize(ctx context.Context, version string) (string, error)
+	Helm(ctx context.Context, version string) (string, error)
+}
+
 type loader interface {
 	// LoadManifests renders and loads all manifests for application.
 	LoadManifests(ctx context.Context, input provider.LoaderInput) ([]provider.Manifest, error)
@@ -60,7 +72,12 @@ type logPersister interface {
 type DeploymentService struct {
 	deployment.UnimplementedDeploymentServiceServer
 
+	// this field is set with the plugin configuration
+	// the plugin configuration is sent from piped while initializing the plugin
+	pluginConfig *config.PipedPlugin
+
 	logger       *zap.Logger
+	toolRegistry toolRegistry
 	loader       loader
 	logPersister logPersister
 }
@@ -71,9 +88,12 @@ func NewDeploymentService(
 	toolClient toolClient,
 	logPersister logPersister,
 ) *DeploymentService {
+	toolRegistry := toolregistry.NewRegistry(toolClient)
+
 	return &DeploymentService{
 		logger:       logger.Named("planner"),
-		loader:       provider.NewLoader(toolregistry.NewRegistry(toolClient)),
+		toolRegistry: toolRegistry,
+		loader:       provider.NewLoader(toolRegistry),
 		logPersister: logPersister,
 	}
 }
@@ -244,8 +264,23 @@ func (a *DeploymentService) executeK8sSyncStage(ctx context.Context, input *depl
 
 	// TODO: implement annotateConfigHash to ensure restart of workloads when config changes
 
-	// Get the applier for the target cluster.
-	var applier applier // TODO: build applier from the plugin config
+	// Get the deploy target config.
+	var deployTargetConfig kubeconfig.KubernetesDeployTargetConfig
+	deployTarget := a.pluginConfig.FindDeployTarget(input.GetDeployment().GetDeployTargets()[0]) // TODO: check if there is a deploy target
+	if err := json.Unmarshal(deployTarget.Config, &deployTargetConfig); err != nil {             // TODO: do not unmarshal the config here, but in the initialization of the plugin
+		lp.Errorf("Failed while unmarshalling deploy target config (%v)", err)
+		return nil, status.Error(codes.Internal, err.Error())
+	}
+
+	// Get the kubectl tool path.
+	kubectlPath, err := a.toolRegistry.Kubectl(ctx, cmp.Or(cfg.Spec.Input.KubectlVersion, deployTargetConfig.KubectlVersion, defaultKubectlVersion))
+	if err != nil {
+		lp.Errorf("Failed while getting kubectl tool (%v)", err)
+		return nil, status.Error(codes.Internal, err.Error())
+	}
+
+	// Create the applier for the target cluster.
+	applier := provider.NewApplier(provider.NewKubectl(kubectlPath), cfg.Spec.Input, deployTargetConfig, a.logger)
 
 	// Start applying all manifests to add or update running resources.
 	if err := applyManifests(ctx, applier, manifests, cfg.Spec.Input.Namespace, lp); err != nil {
diff --git a/pkg/configv1/piped.go b/pkg/configv1/piped.go
index 15e8084783..dce2fe68fa 100644
--- a/pkg/configv1/piped.go
+++ b/pkg/configv1/piped.go
@@ -1310,3 +1310,13 @@ type PipedDeployTarget struct {
 	// The configuration of the deploy target.
 	Config json.RawMessage `json:"config"`
 }
+
+// FindDeployTarget finds the deploy target by the given name.
+func (p *PipedPlugin) FindDeployTarget(name string) *PipedDeployTarget {
+	for _, dt := range p.DeployTargets {
+		if dt.Name == name {
+			return &dt
+		}
+	}
+	return nil
+}