mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-12-27 00:19:36 -05:00
ec424b260d
* Move upgrade specific functions out of Terraform module * Always allow overwriting Terraform files * Ensure constellation-terraform dir does not exist on create --------- Signed-off-by: Daniel Weiße <dw@edgeless.systems>
82 lines
2.4 KiB
Go
82 lines
2.4 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package cloudcmd
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/edgelesssys/constellation/v2/cli/internal/terraform"
|
|
"github.com/edgelesssys/constellation/v2/internal/file"
|
|
)
|
|
|
|
// planUpgrade prepares a workspace and plans the possible Terraform migrations.
|
|
// In case of possible migrations, the diff is written to outWriter and this function returns true.
|
|
func planUpgrade(
|
|
ctx context.Context, tfClient tfUpgradePlanner, fileHandler file.Handler,
|
|
outWriter io.Writer, logLevel terraform.LogLevel, vars terraform.Variables,
|
|
templateDir, existingWorkspace, backupDir string,
|
|
) (bool, error) {
|
|
if err := ensureFileNotExist(fileHandler, backupDir); err != nil {
|
|
return false, fmt.Errorf("backup directory %s already exists: %w", backupDir, err)
|
|
}
|
|
|
|
// Backup old workspace
|
|
if err := fileHandler.CopyDir(existingWorkspace, backupDir); err != nil {
|
|
return false, fmt.Errorf("backing up old workspace: %w", err)
|
|
}
|
|
|
|
// Move the new embedded Terraform files into the workspace.
|
|
if err := tfClient.PrepareWorkspace(templateDir, vars); err != nil {
|
|
return false, fmt.Errorf("preparing terraform workspace: %w", err)
|
|
}
|
|
|
|
hasDiff, err := tfClient.Plan(ctx, logLevel)
|
|
if err != nil {
|
|
return false, fmt.Errorf("terraform plan: %w", err)
|
|
}
|
|
|
|
if hasDiff {
|
|
if err := tfClient.ShowPlan(ctx, logLevel, outWriter); err != nil {
|
|
return false, fmt.Errorf("terraform show plan: %w", err)
|
|
}
|
|
}
|
|
|
|
return hasDiff, nil
|
|
}
|
|
|
|
// restoreBackup replaces the existing Terraform workspace with the backup.
|
|
func restoreBackup(fileHandler file.Handler, workingDir, backupDir string) error {
|
|
if err := fileHandler.RemoveAll(workingDir); err != nil {
|
|
return fmt.Errorf("removing existing workspace: %w", err)
|
|
}
|
|
if err := fileHandler.CopyDir(
|
|
backupDir,
|
|
workingDir,
|
|
); err != nil {
|
|
return fmt.Errorf("replacing terraform workspace with backup: %w", err)
|
|
}
|
|
|
|
if err := fileHandler.RemoveAll(backupDir); err != nil {
|
|
return fmt.Errorf("removing backup directory: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// ensureFileNotExist checks if a single file or directory does not exist, returning an error if it does.
|
|
func ensureFileNotExist(fileHandler file.Handler, fileName string) error {
|
|
if _, err := fileHandler.Stat(fileName); err != nil {
|
|
if !os.IsNotExist(err) {
|
|
return fmt.Errorf("checking %q: %w", fileName, err)
|
|
}
|
|
return nil
|
|
}
|
|
return fmt.Errorf("%q already exists", fileName)
|
|
}
|