add check upgrade in ua

Signed-off-by: Patrick Zhao <zhaoyu@koderover.com>

Author: PetrusZ

pkg/cli/upgradeassistant/cmd/check.go

@@ -0,0 +1,85 @@
+/*
+Copyright 2021 The KodeRover 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 cmd
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+
+	"github.com/koderover/zadig/v2/pkg/shared/client/plutusenterprise"
+	e "github.com/koderover/zadig/v2/pkg/tool/errors"
+	"github.com/koderover/zadig/v2/pkg/tool/log"
+)
+
+func init() {
+	rootCmd.AddCommand(checkUpgradeCmd)
+
+	checkUpgradeCmd.PersistentFlags().StringP("from-version", "f", oldestVersion, "current version to migrate from")
+	checkUpgradeCmd.PersistentFlags().StringP("to-version", "t", "", "target version to migrate to")
+	_ = viper.BindPFlag("fromVersion", checkUpgradeCmd.PersistentFlags().Lookup("from-version"))
+	_ = viper.BindPFlag("toVersion", checkUpgradeCmd.PersistentFlags().Lookup("to-version"))
+}
+
+var checkUpgradeCmd = &cobra.Command{
+	Use:   "check",
+	Short: "check if the upgrade is possible",
+	Long:  `check if the upgrade is possible.`,
+	PreRunE: func(cmd *cobra.Command, args []string) error {
+		return preRun()
+	},
+	Run: func(cmd *cobra.Command, args []string) {
+		if err := runCheckUpgrade(); err != nil {
+			log.Fatal(err)
+		}
+	},
+	PostRun: func(cmd *cobra.Command, args []string) {
+		if err := postRun(); err != nil {
+			fmt.Println(err)
+		}
+	},
+}
+
+func runCheckUpgrade() error {
+	from := viper.GetString("fromVersion")
+	to := viper.GetString("toVersion")
+
+	if len(from) == 0 {
+		from = oldestVersion
+	}
+	if len(to) == 0 {
+		return fmt.Errorf("target version not assigned")
+	}
+
+	log.Infof("Checking upgrade from %s to %s", from, to)
+
+	resp, err := plutusenterprise.New().CheckUpgrade(from, to)
+	if err != nil {
+		nerr := e.ErrUpgradeNotAllowed.AddErr(err)
+		log.Error(nerr)
+		return nerr
+	}
+	if !resp.AllowUpgrade {
+		nerr := e.ErrUpgradeNotAllowed
+		log.Error(nerr)
+		return nerr
+	}
+
+	log.Info("Upgrade check passed")
+	return nil
+}

pkg/cli/upgradeassistant/cmd/migrate.go

@@ -29,6 +29,8 @@ import (
 	_ "github.com/koderover/zadig/v2/pkg/cli/upgradeassistant/cmd/migrate"
 	"github.com/koderover/zadig/v2/pkg/cli/upgradeassistant/internal/upgradepath"
 	"github.com/koderover/zadig/v2/pkg/setting"
+	"github.com/koderover/zadig/v2/pkg/shared/client/plutusenterprise"
+	e "github.com/koderover/zadig/v2/pkg/tool/errors"
 	"github.com/koderover/zadig/v2/pkg/tool/log"
 	mongotool "github.com/koderover/zadig/v2/pkg/tool/mongo"
 )
@@ -159,7 +161,19 @@ func run() error {
 		upgradepath.AddHandler(upgradepath.VersionDatas.VersionIndex(rh.FromVersion), upgradepath.VersionDatas.VersionIndex(rh.ToVersion), rh.Fn)
 	}
 
-	err := upgradepath.UpgradeWithBestPath(from, to)
+	resp, err := plutusenterprise.New().CheckUpgrade(from, to)
+	if err != nil {
+		nerr := e.ErrUpgradeNotAllowed.AddErr(err)
+		log.Error(nerr)
+		return nerr
+	}
+	if !resp.AllowUpgrade {
+		nerr := e.ErrUpgradeNotAllowed
+		log.Error(nerr)
+		return nerr
+	}
+
+	err = upgradepath.UpgradeWithBestPath(from, to)
 	if err == nil {
 		log.Info("Migration finished")
 	}

pkg/shared/client/plutusenterprise/plutusenterprise.go

@@ -72,3 +72,26 @@ func (c *Client) Health() error {
 	_, err := c.Get(url)
 	return err
 }
+
+type UpgradeCheckResponse struct {
+	AllowUpgrade        bool   `json:"allow_upgrade"`
+	IsUpgrade           bool   `json:"is_upgrade"`
+	FromVersion         string `json:"from_version"`
+	ToVersion           string `json:"to_version"`
+	MaintenanceExpireAt int64  `json:"maintenance_expire_at"`
+}
+
+type checkUpgradePermissionReq struct {
+	FromVersion string `json:"from_version"`
+	ToVersion   string `json:"to_version"`
+}
+
+func (c *Client) CheckUpgrade(fromVersion, toVersion string) (*UpgradeCheckResponse, error) {
+	url := "/license/upgrade/check"
+	res := &UpgradeCheckResponse{}
+	_, err := c.Post(url, httpclient.SetBody(&checkUpgradePermissionReq{
+		FromVersion: fromVersion,
+		ToVersion:   toVersion,
+	}), httpclient.SetResult(res))
+	return res, err
+}

pkg/tool/errors/http_errors.go

@@ -932,7 +932,8 @@ var (
 	//-----------------------------------------------------------------------------------------------
 	// License APIs Range: 7050 - 7059
 	//-----------------------------------------------------------------------------------------------
-	ErrLicenseInvalid = NewHTTPError(7050, "用户许可证不可用，请检查许可证后重试")
+	ErrLicenseInvalid    = NewHTTPError(7050, "用户许可证不可用，请检查许可证后重试")
+	ErrUpgradeNotAllowed = NewHTTPError(6695, "当前许可证维保期已过，不允许升级")
 
 	//-----------------------------------------------------------------------------------------------
 	// Istio Grayscale APIs Range: 7060 - 7069