mirror of https://github.com/fluxcd/flux2.git
Add install command with dry-run option
- build kustomization - apply output on the cluster - check rollout status of source and kustomize controllerspull/1/head
parent
42d2ed51cb
commit
94e0b3c9c3
@ -0,0 +1,97 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"fmt"
|
||||
"github.com/spf13/cobra"
|
||||
"io"
|
||||
"os"
|
||||
"os/exec"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
var installCmd = &cobra.Command{
|
||||
Use: "install",
|
||||
Short: "Install the toolkit components",
|
||||
Long: `
|
||||
The Install command deploys the toolkit components
|
||||
on the configured Kubernetes cluster in ~/.kube/config`,
|
||||
Example: ` install --manifests github.com/fluxcd/toolkit//manifests/install --dry-run`,
|
||||
RunE: installCmdRun,
|
||||
}
|
||||
|
||||
var (
|
||||
installDryRun bool
|
||||
installManifestsPath string
|
||||
installNamespace string
|
||||
)
|
||||
|
||||
func init() {
|
||||
installCmd.Flags().BoolVarP(&installDryRun, "dry-run", "", false,
|
||||
"only print the object that would be applied")
|
||||
installCmd.Flags().StringVarP(&installManifestsPath, "manifests", "", "",
|
||||
"path to the manifest directory")
|
||||
installCmd.Flags().StringVarP(&installNamespace, "namespace", "", "gitops-system",
|
||||
"the namespace scope for this installation")
|
||||
|
||||
rootCmd.AddCommand(installCmd)
|
||||
}
|
||||
|
||||
func installCmdRun(cmd *cobra.Command, args []string) error {
|
||||
if installManifestsPath == "" {
|
||||
return fmt.Errorf("no manifests specified")
|
||||
}
|
||||
|
||||
if !strings.HasPrefix(installManifestsPath, "github.com/") {
|
||||
if _, err := os.Stat(installManifestsPath); err != nil {
|
||||
return fmt.Errorf("manifests not found: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
timeout := time.Minute * 5
|
||||
ctx, cancel := context.WithTimeout(context.Background(), timeout)
|
||||
defer cancel()
|
||||
|
||||
dryRun := ""
|
||||
if installDryRun {
|
||||
dryRun = "--dry-run=client"
|
||||
}
|
||||
command := fmt.Sprintf("kustomize build %s | kubectl apply -f- %s",
|
||||
installManifestsPath, dryRun)
|
||||
c := exec.CommandContext(ctx, "/bin/sh", "-c", command)
|
||||
|
||||
var stdoutBuf, stderrBuf bytes.Buffer
|
||||
c.Stdout = io.MultiWriter(os.Stdout, &stdoutBuf)
|
||||
c.Stderr = io.MultiWriter(os.Stderr, &stderrBuf)
|
||||
|
||||
fmt.Println(`✚`, "installing...")
|
||||
err := c.Run()
|
||||
if err != nil {
|
||||
fmt.Println(`✗`, "install failed")
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
if installDryRun {
|
||||
fmt.Println(`✔`, "install dry-run finished")
|
||||
return nil
|
||||
}
|
||||
|
||||
fmt.Println(`✚`, "verifying installation...")
|
||||
for _, deployment := range []string{"source-controller", "kustomize-controller"} {
|
||||
command = fmt.Sprintf("kubectl -n %s rollout status deployment %s --timeout=2m",
|
||||
installNamespace, deployment)
|
||||
c = exec.CommandContext(ctx, "/bin/sh", "-c", command)
|
||||
c.Stdout = io.MultiWriter(os.Stdout, &stdoutBuf)
|
||||
c.Stderr = io.MultiWriter(os.Stderr, &stderrBuf)
|
||||
err := c.Run()
|
||||
if err != nil {
|
||||
fmt.Println(`✗`, "install failed")
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
fmt.Println(`✔`, "install finished")
|
||||
return nil
|
||||
}
|
Loading…
Reference in New Issue