onepassword-operator/pkg/onepassword/connect_setup.go

package onepassword

import (
	"context"
	"os"

	appsv1 "k8s.io/api/apps/v1"
	corev1 "k8s.io/api/core/v1"
	"k8s.io/apimachinery/pkg/api/errors"
	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/types"
	"k8s.io/apimachinery/pkg/util/yaml"
	"sigs.k8s.io/controller-runtime/pkg/client"
	logf "sigs.k8s.io/controller-runtime/pkg/log"
)

var logConnectSetup = logf.Log.WithName("ConnectSetup")
var deploymentPath = "../config/connect/deployment.yaml"
var servicePath = "../config/connect/service.yaml"

func SetupConnect(ctx context.Context, kubeClient client.Client, deploymentNamespace string) error {
	err := setupService(ctx, kubeClient, servicePath, deploymentNamespace)
	if err != nil {
		return err
	}

	err = setupDeployment(ctx, kubeClient, deploymentPath, deploymentNamespace)
	if err != nil {
		return err
	}

	return nil
}

func setupDeployment(
	ctx context.Context,
	kubeClient client.Client,
	deploymentPath string,
	deploymentNamespace string,
) error {
	existingDeployment := &appsv1.Deployment{}

	// check if deployment has already been created
	err := kubeClient.Get(ctx, types.NamespacedName{
		Name:      "onepassword-connect",
		Namespace: deploymentNamespace,
	}, existingDeployment)
	if err != nil {
		if errors.IsNotFound(err) {
			logConnectSetup.Info("No existing Connect deployment found. Creating Deployment")
			return createDeployment(ctx, kubeClient, deploymentPath, deploymentNamespace)
		}
	}
	return err
}

func createDeployment(
	ctx context.Context,
	kubeClient client.Client,
	deploymentPath string,
	deploymentNamespace string,
) error {
	deployment, err := getDeploymentToCreate(deploymentPath, deploymentNamespace)
	if err != nil {
		return err
	}

	err = kubeClient.Create(ctx, deployment)
	if err != nil {
		return err
	}

	return nil
}

func getDeploymentToCreate(deploymentPath string, deploymentNamespace string) (*appsv1.Deployment, error) {
	f, err := os.Open(deploymentPath)
	if err != nil {
		return nil, err
	}
	deployment := &appsv1.Deployment{
		ObjectMeta: v1.ObjectMeta{
			Namespace: deploymentNamespace,
		},
	}

	err = yaml.NewYAMLOrJSONDecoder(f, 4096).Decode(deployment)
	if err != nil {
		return nil, err
	}
	return deployment, nil
}

func setupService(ctx context.Context, kubeClient client.Client, servicePath string, deploymentNamespace string) error {
	existingService := &corev1.Service{}

	// check if service has already been created
	err := kubeClient.Get(ctx, types.NamespacedName{
		Name:      "onepassword-connect",
		Namespace: deploymentNamespace,
	}, existingService)
	if err != nil {
		if errors.IsNotFound(err) {
			logConnectSetup.Info("No existing Connect service found. Creating Service")
			return createService(ctx, kubeClient, servicePath, deploymentNamespace)
		}
	}
	return err
}

func createService(
	ctx context.Context,
	kubeClient client.Client,
	servicePath string,
	deploymentNamespace string,
) error {
	f, err := os.Open(servicePath)
	if err != nil {
		return err
	}
	service := &corev1.Service{
		ObjectMeta: v1.ObjectMeta{
			Namespace: deploymentNamespace,
		},
	}

	err = yaml.NewYAMLOrJSONDecoder(f, 4096).Decode(service)
	if err != nil {
		return err
	}

	err = kubeClient.Create(ctx, service)
	if err != nil {
		return err
	}

	return nil
}