package stackbuilders import ( "context" "fmt" "strconv" "time" portainer "github.com/portainer/portainer/api" ) type FileUploadMethodStackBuildProcess interface { // Set general stack information SetGeneralInfo(payload *StackPayload, endpoint *portainer.Endpoint) FileUploadMethodStackBuildProcess // Set unique stack information, e.g. swarm stack has swarmID, kubernetes stack has namespace SetUniqueInfo(payload *StackPayload) FileUploadMethodStackBuildProcess // Deploy stack based on the configuration Deploy(ctx context.Context, payload *StackPayload, endpoint *portainer.Endpoint) FileUploadMethodStackBuildProcess // Save the stack information to database SaveStack() (*portainer.Stack, error) // Get response from HTTP request. Use if it is needed GetResponse() string // Process the upload file SetUploadedFile(payload *StackPayload) FileUploadMethodStackBuildProcess Error() error } type FileUploadMethodStackBuilder struct { StackBuilder } func (b *FileUploadMethodStackBuilder) SetGeneralInfo(payload *StackPayload, endpoint *portainer.Endpoint) FileUploadMethodStackBuildProcess { stackID := b.dataStore.Stack().GetNextIdentifier() b.stack.ID = portainer.StackID(stackID) b.stack.EndpointID = endpoint.ID now := time.Now().Unix() b.stack.Status = portainer.StackStatusDeploying b.stack.CreationDate = now b.stack.DeploymentStatus = []portainer.StackDeploymentStatus{ {Status: portainer.StackStatusDeploying, Time: now}, } return b } func (b *FileUploadMethodStackBuilder) SetUniqueInfo(payload *StackPayload) FileUploadMethodStackBuildProcess { return b } func (b *FileUploadMethodStackBuilder) SetUploadedFile(payload *StackPayload) FileUploadMethodStackBuildProcess { if b.hasError() { return b } stackFolder := strconv.Itoa(int(b.stack.ID)) projectPath, err := b.fileService.StoreStackFileFromBytes(stackFolder, b.stack.EntryPoint, payload.StackFileContentBytes) if err != nil { b.err = fmt.Errorf("Unable to persist file on disk: %w", err) return b } b.stack.ProjectPath = projectPath return b } func (b *FileUploadMethodStackBuilder) Deploy(ctx context.Context, payload *StackPayload, endpoint *portainer.Endpoint) FileUploadMethodStackBuildProcess { if b.hasError() { return b } // Deploy the stack if err := b.deploymentConfiger.Deploy(ctx); err != nil { b.err = err return b } b.doCleanUp = false return b } func (b *FileUploadMethodStackBuilder) GetResponse() string { return "" }