package securitycompliance import ( "context" "fmt" "time" "edge-infra.dev/pkg/edge/api/graph/model" "edge-infra.dev/pkg/edge/edgeadmin/registrar" "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" ) // NewCmd command to update banner security compliance func NewCmd(cfg *edgecli.Config) *command.Command { flagsets := append(flagutil.GetCommonFlags(), flagutil.GetConnectionFlags(cfg)...) var cmd *command.Command cmd = &command.Command{ ShortUsage: "edge banner security-compliance", ShortHelp: "Update the banner security compliance", Flags: append(flagsets, &rags.Rag{ Name: flagutil.BannerFlag, Value: &rags.String{}, Usage: "Banner name", Required: true, }, &rags.Rag{ Name: flagutil.OptInEdgeSecurityCompliance, Value: &rags.String{}, Usage: "Opt in edge security compliance", Required: true, }, ), Exec: func(_ context.Context, _ []string) error { //validates required field value, value cannot be empty or space if err := flagutil.ValidateRequiredFlags(cmd.Rags); err != nil { return err } registrar, err := constructors.BuildRegistrar(cmd.Rags) if err != nil { return err } updated, err := updateEdgeSecurityCompliance(cmd, registrar) if err != nil { fmt.Println("failed to update edge security compliance") return err } if updated.StatusCode == 200 { fmt.Println("edge security compliance updated") } else { fmt.Printf("edge security compliance not updated") } return nil }, } return cmd } func updateEdgeSecurityCompliance(cmd *command.Command, registrar *registrar.Registrar) (*model.EdgeResponsePayload, error) { bannerName := flagutil.GetStringFlag(cmd.Rags, flagutil.BannerFlag) optIn := flagutil.GetStringFlag(cmd.Rags, flagutil.OptInEdgeSecurityCompliance) reqCtx, cancelReq := context.WithTimeout(context.Background(), time.Duration(30)*time.Second) defer cancelReq() updated, err := registrar.UpdateEdgeSecurityCompliance(reqCtx, bannerName, optIn) if err != nil { return &model.EdgeResponsePayload{}, err } return updated, nil }