/*
 * @Author: jffan
 * @Date: 2024-07-31 14:18:43
 * @LastEditTime: 2024-08-15 09:36:55
 * @LastEditors: jffan
 * @FilePath: \gitee-tcas\cmd\policy\set.go
 * @Description: set policy
 */
package policy

import (
	"fmt"
	"gitee.com/nanhu-lab/tcas-cli/cmd/initconfig"
	consts "gitee.com/nanhu-lab/tcas-cli/constants"
	"gitee.com/nanhu-lab/tcas-cli/manager"
	"gitee.com/nanhu-lab/tcas-cli/tees"
	"gitee.com/nanhu-lab/tcas-cli/utils/file"
	"gitee.com/nanhu-lab/tcas-cli/utils/tools"

	"github.com/sirupsen/logrus"
	"github.com/spf13/cobra"
)

var policySetCmd = &cobra.Command{
	Use:   "set",
	Short: "set policy",
	Long:  `set policy`,
	Run: func(cmd *cobra.Command, args []string) {
		filepath, _ := cmd.Flags().GetString("rego-file")
		if filepath == "" {
			logrus.Errorf(consts.ColorRed + "Error: Please set the rego policy file path first! use `--rego-file /path/to/file`" + consts.OutReset)
			return
		}
		var fileBase64 string
		var err error
		if file.IsExists(filepath) {
			fileBase64, err = file.FileToBase64(filepath)
			if err != nil {
				logrus.Errorf("Error: transfer base64 failed：", err)
				return
			}
		} else {
			logrus.Errorf(consts.ColorRed + "Error: The file path you set is not exist! Please check it!" + consts.OutReset)
			return
		}

		name, _ := cmd.Flags().GetString("name")
		logrus.Debugf(name)
		if name == "" {
			name = tools.GenerateName("policy")
			logrus.Debugf("There is no name set So We generate a policy name (" + consts.ColorYellow + name + consts.OutReset + ") for you! ")
		}
		url, _ := cmd.Flags().GetString("url")
		logrus.Debugf("policy set url:" + consts.ColorYellow + url + consts.OutReset)
		attestationType, _ := cmd.Flags().GetString("type")
		logrus.Debugf("policy set type:" + consts.ColorYellow + attestationType + consts.OutReset)

		tcasConfig, err := initconfig.ReadTCASInfoFromConfig(url)
		if err != nil {
			logrus.Errorf("get tcasConfig failed, please run <tcascli init> first")
			return
		}
		m, err := manager.New(tcasConfig, tees.GetCollectors())
		if err != nil {
			logrus.Errorf("create attest manager failed, error: %s", err)
			return
		}
		if err != nil {
			logrus.Errorf("create attest manager failed, error: %s", err)
			return
		}
		res, err := m.SetPolicy(name, fileBase64, attestationType)
		if err != nil {
			logrus.Errorf("Request failed: %v", err)
			return
		}
		if res.Code == 200 {
			fmt.Println(consts.ColorGreen + "set policy successful, policy id: " + res.PolicyID + consts.OutReset)
		} else {
			logrus.Errorf(consts.ColorRed + "set policy failed:" + res.Message + consts.OutReset)
		}
	},
}

func init() {
	Cmd.AddCommand(policySetCmd)
	//set parameter for policy set
	policySetCmd.Flags().StringP("url", "u", "https://api.trustcluster.cc", "optional, tcas's api url")
	policySetCmd.Flags().StringP("name", "n", "", "must, policy name")
	policySetCmd.MarkFlagRequired("name")

	policySetCmd.Flags().StringP("type", "t", "trust_node", "optional, the attestation-type of policy, support `trust_node` or `trust_cluster`")
	policySetCmd.Flags().StringP("rego-file", "f", "", "must, the path of policy file in rego format")
	policySetCmd.MarkFlagRequired("rego-file")
}