package stackbuilders import ( "fmt" "strconv" "sync" httperror "github.com/portainer/libhttp/error" portainer "github.com/portainer/portainer/api" "github.com/portainer/portainer/api/dataservices" "github.com/portainer/portainer/api/filesystem" k "github.com/portainer/portainer/api/kubernetes" "github.com/portainer/portainer/api/stacks/deployments" "github.com/portainer/portainer/api/stacks/stackutils" ) type K8sStackFileContentBuilder struct { FileContentMethodStackBuilder stackCreateMut *sync.Mutex KuberneteDeployer portainer.KubernetesDeployer User *portainer.User } // CreateK8sStackFileContentBuilder creates a builder for the Kubernetes stack that will be deployed by file content method func CreateK8sStackFileContentBuilder(dataStore dataservices.DataStore, fileService portainer.FileService, stackDeployer deployments.StackDeployer, kuberneteDeployer portainer.KubernetesDeployer, user *portainer.User) *K8sStackFileContentBuilder { return &K8sStackFileContentBuilder{ FileContentMethodStackBuilder: FileContentMethodStackBuilder{ StackBuilder: CreateStackBuilder(dataStore, fileService, stackDeployer), }, stackCreateMut: &sync.Mutex{}, KuberneteDeployer: kuberneteDeployer, User: user, } } func (b *K8sStackFileContentBuilder) SetGeneralInfo(payload *StackPayload, endpoint *portainer.Endpoint) FileContentMethodStackBuildProcess { b.FileContentMethodStackBuilder.SetGeneralInfo(payload, endpoint) return b } func (b *K8sStackFileContentBuilder) SetUniqueInfo(payload *StackPayload) FileContentMethodStackBuildProcess { if b.hasError() { return b } b.stack.Name = payload.StackName b.stack.Type = portainer.KubernetesStack b.stack.EntryPoint = filesystem.ManifestFileDefaultName b.stack.Namespace = payload.Namespace b.stack.CreatedBy = b.User.Username b.stack.IsComposeFormat = payload.ComposeFormat return b } func (b *K8sStackFileContentBuilder) SetFileContent(payload *StackPayload) FileContentMethodStackBuildProcess { if b.hasError() { return b } stackFolder := strconv.Itoa(int(b.stack.ID)) projectPath, err := b.fileService.StoreStackFileFromBytes(stackFolder, b.stack.EntryPoint, []byte(payload.StackFileContent)) if err != nil { fileType := "Manifest" if b.stack.IsComposeFormat { fileType = "Compose" } errMsg := fmt.Sprintf("Unable to persist Kubernetes %s file on disk", fileType) b.err = httperror.InternalServerError(errMsg, err) return b } b.stack.ProjectPath = projectPath return b } func (b *K8sStackFileContentBuilder) Deploy(payload *StackPayload, endpoint *portainer.Endpoint) FileContentMethodStackBuildProcess { if b.hasError() { return b } b.stackCreateMut.Lock() defer b.stackCreateMut.Unlock() k8sAppLabel := k.KubeAppLabels{ StackID: int(b.stack.ID), StackName: b.stack.Name, Owner: stackutils.SanitizeLabel(b.stack.CreatedBy), Kind: "content", } k8sDeploymentConfig, err := deployments.CreateKubernetesStackDeploymentConfig(b.stack, b.KuberneteDeployer, k8sAppLabel, b.User, endpoint) if err != nil { b.err = httperror.InternalServerError("failed to create temp kub deployment files", err) return b } b.deploymentConfiger = k8sDeploymentConfig return b.FileContentMethodStackBuilder.Deploy(payload, endpoint) } func (b *K8sStackFileContentBuilder) GetResponse() string { return b.FileContentMethodStackBuilder.deploymentConfiger.GetResponse() }