package control

import (
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/commonflags"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/key"
	object "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/modules/object"
	commonCmd "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/internal/common"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/control"
	rawclient "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/rpc/client"
	cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
	oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"

	"github.com/spf13/cobra"
)

var locateObjectCmd = &cobra.Command{
	Use:   "locate-object",
	Short: "List shards storing the object",
	Long:  "List shards storing the object",
	Run:   locateObject,
}

func initControlLocateObjectCmd() {
	initControlFlags(locateObjectCmd)

	flags := locateObjectCmd.Flags()

	flags.String(commonflags.CIDFlag, "", commonflags.CIDFlagUsage)
	_ = locateObjectCmd.MarkFlagRequired(commonflags.CIDFlag)

	flags.String(commonflags.OIDFlag, "", commonflags.OIDFlagUsage)
	_ = locateObjectCmd.MarkFlagRequired(commonflags.OIDFlag)

	flags.Bool(commonflags.JSON, false, "Print shard info as a JSON array")
}

func locateObject(cmd *cobra.Command, _ []string) {
	var cnr cid.ID
	var obj oid.ID

	_ = object.ReadObjectAddress(cmd, &cnr, &obj)

	pk := key.Get(cmd)

	body := new(control.ListShardsForObjectRequest_Body)
	body.SetContainerId(cnr.EncodeToString())
	body.SetObjectId(obj.EncodeToString())
	req := new(control.ListShardsForObjectRequest)
	req.SetBody(body)
	signRequest(cmd, pk, req)

	cli := getClient(cmd, pk)

	var err error
	var resp *control.ListShardsForObjectResponse
	err = cli.ExecRaw(func(client *rawclient.Client) error {
		resp, err = control.ListShardsForObject(client, req)
		return err
	})
	commonCmd.ExitOnErr(cmd, "rpc error: %w", err)

	verifyResponse(cmd, resp.GetSignature(), resp.GetBody())

	shards := resp.GetBody().GetShards()
	isJSON, _ := cmd.Flags().GetBool(commonflags.JSON)
	if isJSON {
		prettyPrintShardsJSON(cmd, shards)
	} else {
		prettyPrintShards(cmd, shards)
	}
}