porter/api/server/handlers/api_contract/update.go

package api_contract

import (
	"database/sql"
	"encoding/base64"
	"errors"
	"fmt"
	"net/http"

	"github.com/bufbuild/connect-go"
	"github.com/google/uuid"
	helpers "github.com/porter-dev/api-contracts/generated/go/helpers"
	porterv1 "github.com/porter-dev/api-contracts/generated/go/porter/v1"
	"github.com/porter-dev/porter/api/server/handlers"
	"github.com/porter-dev/porter/api/server/shared"
	"github.com/porter-dev/porter/api/server/shared/apierrors"
	"github.com/porter-dev/porter/api/server/shared/config"
	"github.com/porter-dev/porter/api/types"
	"github.com/porter-dev/porter/internal/models"
)

type APIContractUpdateHandler struct {
	handlers.PorterHandlerReadWriter
}

func NewAPIContractUpdateHandler(
	config *config.Config,
	decoderValidator shared.RequestDecoderValidator,
	writer shared.ResultWriter,
) *APIContractUpdateHandler {
	return &APIContractUpdateHandler{
		PorterHandlerReadWriter: handlers.NewDefaultPorterHandler(config, decoderValidator, writer),
	}
}

// ServeHTTP parses the Porter API contract for validity, and forwards the requests for handling on to another service
// For now, this handling cluster creation only, by inserting a row into the cluster table in order to create an ID for this cluster, as well as stores the raw request JSON for updating later
func (c *APIContractUpdateHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	project, _ := ctx.Value(types.ProjectScope).(*models.Project)
	user, _ := ctx.Value(types.UserScope).(*models.User)

	var apiContract porterv1.Contract

	err := helpers.UnmarshalContractObjectFromReader(r.Body, &apiContract)
	if err != nil {
		e := fmt.Errorf("error parsing api contract: %w", err)
		c.HandleAPIError(w, r, apierrors.NewErrInternal(e))
		return
	}

	existingClusters, err := c.Config().Repo.Cluster().ListClustersByProjectID(uint(apiContract.Cluster.ProjectId))
	if err != nil {
		if !errors.Is(err, sql.ErrNoRows) {
			e := fmt.Errorf("error listing clusters for given project ID: %w", err)
			c.HandleAPIError(w, r, apierrors.NewErrInternal(e))
			return
		}
	}
	for _, cluster := range existingClusters {
		if cluster.Name == apiContract.Cluster.GetEksKind().ClusterName {
			e := fmt.Errorf("cluster already exists in project %d called %s", cluster.ProjectID, cluster.Name)
			c.HandleAPIError(w, r, apierrors.NewErrInternal(e))
			return
		}
	}

	if !project.CapiProvisionerEnabled && !c.Config().EnableCAPIProvisioner {
		// return dummy data if capi provisioner disabled in project settings, and as env var
		// TODO: remove this stub when we can spin up all services locally, easily
		clusterID := apiContract.Cluster.ClusterId
		if apiContract.Cluster.ClusterId == 0 {
			dbcli := models.Cluster{
				ProjectID:                         uint(apiContract.Cluster.ProjectId),
				Status:                            "UPDATING_UNAVAILABLE",
				ProvisionedBy:                     "CAPI",
				CloudProvider:                     "AWS",
				CloudProviderCredentialIdentifier: apiContract.Cluster.CloudProviderCredentialsId,
				Name:                              apiContract.Cluster.GetEksKind().ClusterName,
				VanityName:                        apiContract.Cluster.GetEksKind().ClusterName,
			}
			dbcl, err := c.Config().Repo.Cluster().CreateCluster(&dbcli)
			if err != nil {
				e := fmt.Errorf("error updating mock contract: %w", err)
				c.HandleAPIError(w, r, apierrors.NewErrInternal(e))
				return
			}
			clusterID = int32(dbcl.ID)
		}

		by, err := helpers.MarshalContractObject(ctx, &apiContract)
		if err != nil {
			e := fmt.Errorf("error marshalling mock api contract: %w", err)
			c.HandleAPIError(w, r, apierrors.NewErrInternal(e))
			return
		}
		b64Contract := base64.StdEncoding.EncodeToString([]byte(by))

		revisionInput := models.APIContractRevision{
			ID:             uuid.New(),
			ClusterID:      int(clusterID),
			ProjectID:      int(apiContract.Cluster.ProjectId),
			Base64Contract: b64Contract,
		}
		revision, err := c.Config().Repo.APIContractRevisioner().Insert(ctx, revisionInput)
		if err != nil {
			e := fmt.Errorf("error updating mock contract: %w", err)
			c.HandleAPIError(w, r, apierrors.NewErrInternal(e))
			return
		}
		resp := &porterv1.ContractRevision{
			ClusterId:  int32(clusterID),
			ProjectId:  apiContract.Cluster.ProjectId,
			RevisionId: revision.ID.String(),
		}
		w.WriteHeader(http.StatusCreated)
		c.WriteResult(w, r, resp)
		return
	}

	apiContract.User = &porterv1.User{
		Id: int32(user.ID),
	}
	updateRequest := connect.NewRequest(&porterv1.UpdateContractRequest{
		Contract: &apiContract,
	})
	revision, err := c.Config().ClusterControlPlaneClient.UpdateContract(ctx, updateRequest)
	if err != nil {
		e := fmt.Errorf("error sending contract for update: %w", err)
		c.HandleAPIError(w, r, apierrors.NewErrInternal(e))
		return
	}

	w.WriteHeader(http.StatusCreated)
	c.WriteResult(w, r, revision.Msg)
}