package cmd

import (
	"fmt"

	"os"

	"path/filepath"

	"github.com/rancher/kontainer-engine/store"
	"github.com/rancher/kontainer-engine/utils"
	"github.com/sirupsen/logrus"
	"github.com/urfave/cli"
)

// RmCommand defines the remove command
func RmCommand() cli.Command {
	return cli.Command{
		Name:      "remove",
		ShortName: "rm",
		Usage:     "Remove kubernetes clusters",
		Action:    rmCluster,
		Flags: []cli.Flag{
			cli.BoolFlag{
				Name:  "force,f",
				Usage: "force to remove a cluster",
			},
		},
	}
}

func rmCluster(ctx *cli.Context) error {
	for _, name := range ctx.Args() {
		if name == "" || name == "--help" {
			return cli.ShowCommandHelp(ctx, "remove")
		}
		clusters, err := store.GetAllClusterFromStore()
		if err != nil {
			return err
		}
		cluster, ok := clusters[name]
		if !ok {
			return fmt.Errorf("cluster %v can't be found", name)
		}
		rpcClient, _, err := runRPCDriver(cluster.DriverName)
		if err != nil {
			return err
		}
		configGetter := cliConfigGetter{
			name: name,
			ctx:  ctx,
		}
		cluster.ConfigGetter = configGetter
		cluster.PersistStore = cliPersistStore{}
		cluster.Driver = rpcClient
		if err := cluster.Remove(); err != nil {
			if !ctx.Bool("force") {
				return err
			}
		}
		clusterFilePath := filepath.Join(utils.HomeDir(), "clusters", cluster.Name)
		logrus.Debugf("Deleting cluster storage path %v", clusterFilePath)
		if err := os.RemoveAll(clusterFilePath); err != nil && !os.IsNotExist(err) {
			return err
		}

		config, err := getConfigFromFile()
		if err != nil {
			return err
		}
		deleteConfigByName(&config, name)
		if err := setConfigToFile(config); err != nil {
			return err
		}
		fmt.Println(cluster.Name)
	}
	return nil
}