// Copyright 2017 PingCAP, Inc.
//
// 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,
// See the License for the specific language governing permissions and
// limitations under the License.package spec

package cmd

import (
	"encoding/json"

	"github.com/pingcap/tidb-cloud-manager/managerctl/cmd/util"
	"github.com/spf13/cobra"
)

//NewCmdUpgradeCluster implements the upgrade subcommand of cluster
func NewCmdUpgradeCluster(cli util.Client) *cobra.Command {
	var options ClusterOptions

	cmd := &cobra.Command{
		Use:   "upgrade -f FILENAME -n NAMESPACE",
		Short: "Upgrade a tidb cluster from a file and namespace.",
		Long: `Upgrade a tidb cluster from a file and namespace.

JSON formats are accepted, Only support for modifying tikv、pd、tidb's version in this interface.`,
		Run: func(cmd *cobra.Command, args []string) {
			util.ValidCmdFlags(cmd)
			util.CheckErr(util.ValidManagerAddr(cmdFlags.URL))
			util.CheckErr(RunUpgradeCluster(cli, &options))
		},
	}

	cmd.Flags().StringVarP(&options.ClusterCfg, "clustercfg", "f", "", "Cluster config for upgrade tidb cluster.")
	cmd.Flags().StringVarP(&options.Namespace, "namespace", "n", "", "Namespace of tidb cluster.")
	return cmd
}

func RunUpgradeCluster(cli util.Client, options *ClusterOptions) error {
	tc, err := util.LoadClusterConfig(options.ClusterCfg)
	if err != nil {
		return err
	}
	tcBytes, err := json.Marshal(tc)
	if err != nil {
		return err
	}
	fullURL := util.GetClusterURL(cmdFlags.URL, options.Namespace)
	resp, err := cli.DoPut(fullURL, util.ContentType, tcBytes)
	if err != nil {
		return err
	}
	return util.OutputResp(resp)
}