Fixes adding apps to flux-system namespace

cmd/clusters-service/pkg/server/automations.go

@@ -0,0 +1,263 @@
+package server
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"strings"
+	"time"
+
+	"github.com/fluxcd/go-git-providers/gitprovider"
+	helmv2 "github.com/fluxcd/helm-controller/api/v2beta1"
+	sourcev1 "github.com/fluxcd/source-controller/api/v1beta2"
+	"github.com/mkmik/multierror"
+	"github.com/spf13/viper"
+	"google.golang.org/grpc/codes"
+	grpcStatus "google.golang.org/grpc/status"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/types"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/yaml"
+
+	"github.com/weaveworks/weave-gitops-enterprise/cmd/clusters-service/pkg/git"
+	capiv1_proto "github.com/weaveworks/weave-gitops-enterprise/cmd/clusters-service/pkg/protos"
+	apiextensionsv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
+)
+
+// CreateAutomationsPullRequest receives a list of {kustomization, helmrelease, cluster}
+// generates a kustomization file and/or a helm release file for each provided cluster in the list
+// and creates a pull request for the generated files
+func (s *server) CreateAutomationsPullRequest(ctx context.Context, msg *capiv1_proto.CreateAutomationsPullRequestRequest) (*capiv1_proto.CreateAutomationsPullRequestResponse, error) {
+	gp, err := getGitProvider(ctx)
+	if err != nil {
+		return nil, grpcStatus.Errorf(codes.Unauthenticated, "error creating pull request: %s", err.Error())
+	}
+
+	if err := validateCreateAutomationsPR(msg); err != nil {
+		s.log.Error(err, "Failed to create pull request, message payload was invalid")
+		return nil, err
+	}
+
+	client, err := s.clientGetter.Client(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	repositoryURL := viper.GetString("capi-templates-repository-url")
+	if msg.RepositoryUrl != "" {
+		repositoryURL = msg.RepositoryUrl
+	}
+	baseBranch := viper.GetString("capi-templates-repository-base-branch")
+	if msg.BaseBranch != "" {
+		baseBranch = msg.BaseBranch
+	}
+
+	var clusters []string
+
+	var files []gitprovider.CommitFile
+
+	for _, c := range msg.ClusterAutomations {
+		cluster := createNamespacedName(c.Cluster.Name, c.Cluster.Namespace)
+
+		if c.Kustomization != nil {
+			kustomization, err := generateKustomizationFile(ctx, c.IsControlPlane, cluster, client, c.Kustomization, c.FilePath)
+
+			if err != nil {
+				return nil, err
+			}
+
+			files = append(files, kustomization)
+		}
+
+		if c.HelmRelease != nil {
+			helmRelease, err := generateHelmReleaseFile(ctx, c.IsControlPlane, cluster, client, c.HelmRelease, c.FilePath)
+
+			if err != nil {
+				return nil, err
+			}
+
+			files = append(files, helmRelease)
+		}
+
+		clusters = append(clusters, c.Cluster.Name)
+	}
+
+	if msg.HeadBranch == "" {
+		clusters := strings.Join(clusters, "")
+		msg.HeadBranch = getHash(msg.RepositoryUrl, clusters, msg.BaseBranch)
+	}
+	if msg.Title == "" {
+		msg.Title = "Gitops add cluster workloads"
+	}
+	if msg.Description == "" {
+		msg.Description = "Pull request to create cluster workloads"
+	}
+	if msg.CommitMessage == "" {
+		msg.CommitMessage = "Add Kustomization Manifests"
+	}
+	_, err = s.provider.GetRepository(ctx, *gp, repositoryURL)
+	if err != nil {
+		return nil, grpcStatus.Errorf(codes.Unauthenticated, "failed to access repo %s: %s", repositoryURL, err)
+	}
+
+	res, err := s.provider.WriteFilesToBranchAndCreatePullRequest(ctx, git.WriteFilesToBranchAndCreatePullRequestRequest{
+		GitProvider:       *gp,
+		RepositoryURL:     repositoryURL,
+		ReposistoryAPIURL: msg.RepositoryApiUrl,
+		HeadBranch:        msg.HeadBranch,
+		BaseBranch:        baseBranch,
+		Title:             msg.Title,
+		Description:       msg.Description,
+		CommitMessage:     msg.CommitMessage,
+		Files:             files,
+	})
+
+	if err != nil {
+		return nil, fmt.Errorf("unable to create pull request: %w", err)
+	}
+
+	return &capiv1_proto.CreateAutomationsPullRequestResponse{
+		WebUrl: res.WebURL,
+	}, nil
+}
+
+func generateHelmReleaseFile(
+	ctx context.Context,
+	isControlPlane bool,
+	cluster types.NamespacedName,
+	kubeClient client.Client,
+	helmRelease *capiv1_proto.HelmRelease,
+	filePath string) (gitprovider.CommitFile, error) {
+	kustomizationYAML := createHelmReleaseObject(helmRelease)
+
+	b, err := yaml.Marshal(kustomizationYAML)
+	if err != nil {
+		return gitprovider.CommitFile{}, fmt.Errorf("error marshalling %s helmrelease, %w", helmRelease.Metadata.Name, err)
+	}
+
+	hr := createNamespacedName(helmRelease.Metadata.Name, helmRelease.Metadata.Namespace)
+
+	helmReleasePath := getClusterResourcePath(isControlPlane, "helmrelease", cluster, hr)
+	if filePath != "" {
+		helmReleasePath = filePath
+	}
+
+	helmReleaseContent := string(b)
+
+	file := &gitprovider.CommitFile{
+		Path:    &helmReleasePath,
+		Content: &helmReleaseContent,
+	}
+
+	return *file, nil
+}
+
+func createHelmReleaseObject(hr *capiv1_proto.HelmRelease) *helmv2.HelmRelease {
+	generatedHelmRelease := helmv2.HelmRelease{
+		TypeMeta: metav1.TypeMeta{
+			APIVersion: helmv2.GroupVersion.Identifier(),
+			Kind:       helmv2.HelmReleaseKind,
+		},
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      hr.Metadata.Name,
+			Namespace: hr.Metadata.Namespace,
+		},
+		Spec: helmv2.HelmReleaseSpec{
+			Chart: helmv2.HelmChartTemplate{
+				Spec: helmv2.HelmChartTemplateSpec{
+					Chart: hr.Spec.Chart.Spec.Chart,
+					SourceRef: helmv2.CrossNamespaceObjectReference{
+						APIVersion: sourcev1.GroupVersion.Identifier(),
+						Kind:       sourcev1.HelmRepositoryKind,
+						Name:       hr.Spec.Chart.Spec.SourceRef.Name,
+						Namespace:  hr.Spec.Chart.Spec.SourceRef.Namespace,
+					},
+				},
+			},
+			Interval: metav1.Duration{Duration: time.Minute * 10},
+			Values:   &apiextensionsv1.JSON{Raw: []byte(hr.Spec.Values)},
+		},
+	}
+
+	return &generatedHelmRelease
+}
+
+func validateCreateAutomationsPR(msg *capiv1_proto.CreateAutomationsPullRequestRequest) error {
+	var err error
+
+	if len(msg.ClusterAutomations) == 0 {
+		err = multierror.Append(err, fmt.Errorf(createClusterAutomationsRequiredErr))
+	}
+
+	for _, c := range msg.ClusterAutomations {
+		if c.Cluster == nil {
+			err = multierror.Append(err, fmt.Errorf("cluster object must be specified"))
+		} else {
+			if c.Cluster.Name == "" {
+				err = multierror.Append(err, fmt.Errorf("cluster name must be specified"))
+			}
+
+			invalidNamespaceErr := validateNamespace(c.Cluster.Namespace)
+			if invalidNamespaceErr != nil {
+				err = multierror.Append(err, invalidNamespaceErr)
+			}
+		}
+
+		if c.Kustomization != nil {
+			err = multierror.Append(err, validateKustomization(c.Kustomization))
+		} else if c.HelmRelease != nil {
+			err = multierror.Append(err, validateHelmRelease(c.HelmRelease))
+		} else {
+			err = multierror.Append(err, fmt.Errorf("cluster automation must contain either kustomization or helm release"))
+		}
+	}
+
+	return err
+}
+
+func validateHelmRelease(helmRelease *capiv1_proto.HelmRelease) error {
+	var err error
+
+	if helmRelease.Metadata == nil {
+		err = multierror.Append(err, errors.New("helmrelease metadata must be specified"))
+	} else {
+		if helmRelease.Metadata.Name == "" {
+			err = multierror.Append(err, fmt.Errorf("helmrelease name must be specified"))
+		}
+
+		invalidNamespaceErr := validateNamespace(helmRelease.Metadata.Namespace)
+		if invalidNamespaceErr != nil {
+			err = multierror.Append(err, invalidNamespaceErr)
+		}
+	}
+
+	if helmRelease.Spec.Chart == nil {
+		err = multierror.Append(
+			err,
+			fmt.Errorf("chart must be specified in HelmRelease %s",
+				helmRelease.Metadata.Name))
+	} else {
+		if helmRelease.Spec.Chart.Spec.Chart == "" {
+			err = multierror.Append(
+				err,
+				fmt.Errorf("chart name must be specified in HelmRelease %s",
+					helmRelease.Metadata.Name))
+		}
+
+		if helmRelease.Spec.Chart.Spec.SourceRef != nil {
+			if helmRelease.Spec.Chart.Spec.SourceRef.Name == "" {
+				err = multierror.Append(
+					err,
+					fmt.Errorf("sourceRef name must be specified in chart %s in HelmRelease %s",
+						helmRelease.Spec.Chart.Spec.Chart, helmRelease.Metadata.Name))
+			}
+
+			invalidNamespaceErr := validateNamespace(helmRelease.Spec.Chart.Spec.SourceRef.Namespace)
+			if invalidNamespaceErr != nil {
+				err = multierror.Append(err, invalidNamespaceErr)
+			}
+		}
+	}
+
+	return err
+}