migrate.go

/*
Copyright 2023 The Kubernetes Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
	http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package rescaffold

import (
	"errors"
	"fmt"
	"os"
	"os/exec"
	"strings"

	log "github.com/sirupsen/logrus"

	"github.com/spf13/afero"
	"sigs.k8s.io/kubebuilder/v3/pkg/config"
	"sigs.k8s.io/kubebuilder/v3/pkg/config/store"
	"sigs.k8s.io/kubebuilder/v3/pkg/config/store/yaml"
	"sigs.k8s.io/kubebuilder/v3/pkg/machinery"
	"sigs.k8s.io/kubebuilder/v3/pkg/model/resource"
	"sigs.k8s.io/kubebuilder/v3/pkg/plugin/util"
	"sigs.k8s.io/kubebuilder/v3/pkg/plugins/golang/deploy-image/v1alpha1"
)

type MigrateOptions struct {
	InputDir  string
	OutputDir string
}

const DefaultOutputDir = "output-dir"
const grafanaPluginKey = "grafana.kubebuilder.io/v1-alpha"

func (opts *MigrateOptions) Rescaffold() error {
	config := yaml.New(machinery.Filesystem{FS: afero.NewOsFs()})
	if err := config.LoadFrom(fmt.Sprintf("%s/%s", opts.InputDir, yaml.DefaultPath)); err != nil {
		log.Fatalf("Failed to load PROJECT file %v", err)
	}
	// create output directory
	// nolint: gosec
	if err := os.MkdirAll(opts.OutputDir, 0755); err != nil {
		log.Fatalf("Failed to create output directory %v", err)
	}
	// use the new directory to set up the new project
	if err := os.Chdir(opts.OutputDir); err != nil {
		log.Fatalf("Failed to change the current working directory %v", err)
	}
	// init project with plugins
	if err := kubebuilderInit(config); err != nil {
		log.Fatalf("Failed to run init subcommand %v", err)
	}
	// call edit subcommands to enable or disable multigroup layout
	if err := kubebuilderEdit(config); err != nil {
		log.Fatalf("Failed to run edit subcommand %v", err)
	}
	// create APIs and Webhooks
	if err := kubebuilderCreate(config); err != nil {
		log.Fatalf("Failed to run create API subcommand %v", err)
	}
	// plugin specific migration
	if err := migrateGrafanaPlugin(config, opts.InputDir, opts.OutputDir); err != nil {
		log.Fatalf("Failed to run grafana plugin migration %v", err)
	}
	if err := migrateDeployImagePlugin(config); err != nil {
		log.Fatalf("Failed to run deploy-image plugin migration %v", err)
	}
	return nil
}

func (opts *MigrateOptions) Validate() error {
	cwd, err := os.Getwd()
	if err != nil {
		log.Fatal(err)
	}
	// get PROJECT path from command args
	inputPath, err := getInputPath(cwd, opts.InputDir)
	if err != nil {
		log.Fatal(err)
	}
	opts.InputDir = inputPath
	// get output path from command args
	opts.OutputDir, err = getOutputPath(cwd, opts.OutputDir)
	if err != nil {
		log.Fatal(err)
	}
	// check whether the kubebuilder binary is accessible
	_, err = exec.LookPath("kubebuilder")
	return err
}

func getInputPath(currentWorkingDirectory string, inputPath string) (string, error) {
	if inputPath == "" {
		inputPath = currentWorkingDirectory
	}
	projectPath := fmt.Sprintf("%s/%s", inputPath, yaml.DefaultPath)
	if _, err := os.Stat(projectPath); os.IsNotExist(err) {
		return "", fmt.Errorf("PROJECT path: %s does not exist. %v", projectPath, err)
	}
	return inputPath, nil
}

func getOutputPath(currentWorkingDirectory, outputPath string) (string, error) {
	if outputPath == "" {
		outputPath = fmt.Sprintf("%s/%s", currentWorkingDirectory, DefaultOutputDir)
	}
	_, err := os.Stat(outputPath)
	if err == nil {
		return "", fmt.Errorf("Output path: %s already exists. %v", outputPath, err)
	}
	if os.IsNotExist(err) {
		return outputPath, nil
	}
	return "", err
}

func kubebuilderInit(store store.Store) error {
	var args []string
	args = append(args, "init")
	args = append(args, getInitArgs(store)...)
	return util.RunCmd("kubebuilder init", "kubebuilder", args...)
}

func kubebuilderEdit(store store.Store) error {
	if store.Config().IsMultiGroup() {
		args := []string{"edit", "--multigroup"}
		return util.RunCmd("kubebuilder edit", "kubebuilder", args...)
	}
	return nil
}

func kubebuilderCreate(store store.Store) error {
	resources, err := store.Config().GetResources()
	if err != nil {
		return err
	}

	for _, r := range resources {
		if err = createAPI(r); err != nil {
			return err
		}
		if err = createWebhook(r); err != nil {
			return err
		}
	}

	return nil
}

func migrateGrafanaPlugin(store store.Store, src, des string) error {
	var grafanaPlugin struct{}
	err := store.Config().DecodePluginConfig(grafanaPluginKey, grafanaPlugin)
	// If the grafana plugin is not found, we don't need to migrate
	if err != nil {
		if errors.As(err, &config.PluginKeyNotFoundError{}) {
			log.Info("Grafana plugin is not found, skip the migration")
			return nil
		}
		return fmt.Errorf("failed to decode grafana plugin config %v", err)
	}
	err = kubebuilderGrafanaEdit()
	if err != nil {
		return err
	}
	err = grafanaConfigMigrate(src, des)
	if err != nil {
		return err
	}
	return kubebuilderGrafanaEdit()
}

func migrateDeployImagePlugin(store store.Store) error {
	deployImagePlugin := v1alpha1.PluginConfig{}
	err := store.Config().DecodePluginConfig("deploy-image.go.kubebuilder.io/v1-alpha", &deployImagePlugin)
	// If the deploy-image plugin is not found, we don't need to migrate
	if err != nil {
		if errors.As(err, &config.PluginKeyNotFoundError{}) {
			log.Printf("deploy-image plugin is not found, skip the migration")
			return nil
		}
		return fmt.Errorf("failed to decode deploy-image plugin config %v", err)
	}

	for _, r := range deployImagePlugin.Resources {
		if err = createAPIWithDeployImage(r); err != nil {
			return err
		}
	}
	return nil
}

func createAPIWithDeployImage(resource v1alpha1.ResourceData) error {
	var args []string
	args = append(args, "create")
	args = append(args, "api")
	args = append(args, getGVKFlagsFromDeployImage(resource)...)
	args = append(args, getDeployImageOptions(resource)...)
	return util.RunCmd("kubebuilder create api", "kubebuilder", args...)
}

func getInitArgs(store store.Store) []string {
	var args []string
	plugins := store.Config().GetPluginChain()
	if len(plugins) > 0 {
		args = append(args, "--plugins", strings.Join(plugins, ","))
	}
	domain := store.Config().GetDomain()
	if domain != "" {
		args = append(args, "--domain", domain)
	}
	return args
}

func getGVKFlags(resource resource.Resource) []string {
	var args []string

	if len(resource.Plural) > 0 {
		args = append(args, "--plural", resource.Plural)
	}
	if len(resource.Group) > 0 {
		args = append(args, "--group", resource.Group)
	}
	if len(resource.Version) > 0 {
		args = append(args, "--version", resource.Version)
	}
	if len(resource.Kind) > 0 {
		args = append(args, "--kind", resource.Kind)
	}
	return args
}

func getGVKFlagsFromDeployImage(resource v1alpha1.ResourceData) []string {
	var args []string
	if len(resource.Group) > 0 {
		args = append(args, "--group", resource.Group)
	}
	if len(resource.Version) > 0 {
		args = append(args, "--version", resource.Version)
	}
	if len(resource.Kind) > 0 {
		args = append(args, "--kind", resource.Kind)
	}
	return args
}

func getDeployImageOptions(resource v1alpha1.ResourceData) []string {
	var args []string
	if len(resource.Options.Image) > 0 {
		args = append(args, fmt.Sprintf("--image=%s", resource.Options.Image))
	}
	if len(resource.Options.ContainerCommand) > 0 {
		args = append(args, fmt.Sprintf("--image-container-command=%s", resource.Options.ContainerCommand))
	}
	if len(resource.Options.ContainerPort) > 0 {
		args = append(args, fmt.Sprintf("--image-container-port=%s", resource.Options.ContainerPort))
	}
	if len(resource.Options.RunAsUser) > 0 {
		args = append(args, fmt.Sprintf("--run-as-user=%s", resource.Options.RunAsUser))
	}
	args = append(args, fmt.Sprintf("--plugins=\"%s\"", "deploy-image/v1-alpha"))
	return args
}

func createAPI(resource resource.Resource) error {
	var args []string
	args = append(args, "create")
	args = append(args, "api")
	args = append(args, getGVKFlags(resource)...)
	args = append(args, getAPIResourceFlags(resource)...)
	return util.RunCmd("kubebuilder create api", "kubebuilder", args...)
}

func getAPIResourceFlags(resource resource.Resource) []string {
	var args []string
	if resource.API == nil || resource.API.IsEmpty() {
		// create API without creating resource
		args = append(args, "--resource=false")
	} else {
		args = append(args, "--resource")
		if resource.API.Namespaced {
			args = append(args, "--namespaced")
		}
	}

	if resource.Controller {
		args = append(args, "--controller")
	} else {
		args = append(args, "--controller=false")
	}
	return args
}

func createWebhook(resource resource.Resource) error {
	if resource.Webhooks == nil || resource.Webhooks.IsEmpty() {
		return nil
	}
	var args []string
	args = append(args, "create")
	args = append(args, "webhook")
	args = append(args, getGVKFlags(resource)...)
	args = append(args, getWebhookResourceFlags(resource)...)
	return util.RunCmd("kubebuilder create webhook", "kubebuilder", args...)
}

func getWebhookResourceFlags(resource resource.Resource) []string {
	var args []string
	if resource.HasConversionWebhook() {
		args = append(args, "--conversion")
	}
	if resource.HasValidationWebhook() {
		args = append(args, "--programmatic-validation")
	}
	if resource.HasDefaultingWebhook() {
		args = append(args, "--defaulting")
	}
	return args
}

func copyFile(src, des string) error {
	// nolint:gosec
	bytesRead, err := os.ReadFile(src)
	if err != nil {
		return fmt.Errorf("Source file path: %s does not exist. %v", src, err)
	}
	//Copy all the contents to the desitination file
	// nolint:gosec
	return os.WriteFile(des, bytesRead, 0755)
}

func grafanaConfigMigrate(src, des string) error {
	grafanaConfig := fmt.Sprintf("%s/%s", src, "grafana/custom-metrics/config.yaml")
	if _, err := os.Stat(grafanaConfig); os.IsNotExist(err) {
		return fmt.Errorf("Grafana Config path: %s does not exist. %v", grafanaConfig, err)
	}
	return copyFile(grafanaConfig, fmt.Sprintf("%s/%s", des, "grafana/custom-metrics/config.yaml"))
}

func kubebuilderGrafanaEdit() error {
	args := []string{"edit", "--plugins", grafanaPluginKey}
	err := util.RunCmd("kubebuilder edit", "kubebuilder", args...)
	if err != nil {
		return fmt.Errorf("Failed to run edit subcommand for Grafana Plugin %v", err)
	}
	return nil
}