package discover import ( "context" "fmt" "edge-infra.dev/pkg/edge/api/graph/model" "edge-infra.dev/pkg/edge/edgecli" "edge-infra.dev/pkg/edge/edgecli/constructors" "edge-infra.dev/pkg/edge/edgecli/flagutil" "edge-infra.dev/pkg/lib/cli/command" "edge-infra.dev/pkg/lib/cli/rags" ) func NewCmd(cfg *edgecli.Config) *command.Command { var cmd *command.Command flagsets := flagutil.GetConnectionFlags(cfg) cmd = &command.Command{ ShortUsage: "edge terminal disk discover", ShortHelp: "Edit the disk discovery", Flags: append( flagsets, &rags.Rag{ Name: flagutil.StoreFlag, Value: &rags.String{}, Usage: "Name of the Store - Must be unique", Required: true, }, &rags.Rag{ Name: flagutil.BannerFlag, Value: &rags.String{}, Usage: "Banner name", Required: true, }, &rags.Rag{ Name: flagutil.TerminalID, Value: &rags.String{}, Usage: "ID of Terminal", Required: true, }, &rags.Rag{ Name: flagutil.DiscoverDisksFlag, Value: &rags.String{}, Usage: "Disk discovery option", Required: true, }, ), Exec: func(_ context.Context, _ []string) error { if err := flagutil.ValidateRequiredFlags(cmd.Rags); err != nil { return err } registrar, err := constructors.BuildRegistrar(cmd.Rags) if err != nil { return err } terminalID := flagutil.GetStringFlag(cmd.Rags, flagutil.TerminalID) discoverDisk := flagutil.GetStringFlag(cmd.Rags, flagutil.DiscoverDisksFlag) reqCtx, cancelReq := context.WithTimeout(context.Background(), flagutil.BffReqTimeout) defer cancelReq() updatedTerminal := model.TerminalUpdateInput{ DiscoverDisks: (*model.TerminalDiscoverDisksType)(&discoverDisk), } _, err = registrar.ModifyTerminal(reqCtx, terminalID, &updatedTerminal) if err != nil { fmt.Println("an error occurred whilst modifying the terminal discover disks option") return err } return nil }, } return cmd }