| // Copyright 2020 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| package drac |
| |
| import ( |
| "fmt" |
| |
| "github.com/maruel/subcommands" |
| "go.chromium.org/luci/auth/client/authcli" |
| "go.chromium.org/luci/common/cli" |
| "go.chromium.org/luci/common/errors" |
| "go.chromium.org/luci/grpc/prpc" |
| |
| "infra/cmd/shivas/cmdhelp" |
| "infra/cmd/shivas/site" |
| "infra/cmd/shivas/utils" |
| "infra/cmdsupport/cmdlib" |
| ufspb "infra/unifiedfleet/api/v1/models" |
| ufsAPI "infra/unifiedfleet/api/v1/rpc" |
| ufsUtil "infra/unifiedfleet/app/util" |
| ) |
| |
| // UpdateDracCmd Update drac by given name. |
| var UpdateDracCmd = &subcommands.Command{ |
| UsageLine: "drac [Options...]", |
| ShortDesc: "Update a drac on a machine", |
| LongDesc: cmdhelp.UpdateDracLongDesc, |
| CommandRun: func() subcommands.CommandRun { |
| c := &updateDrac{} |
| c.authFlags.Register(&c.Flags, site.DefaultAuthOptions) |
| c.envFlags.Register(&c.Flags) |
| c.commonFlags.Register(&c.Flags) |
| |
| c.Flags.StringVar(&c.newSpecsFile, "f", "", cmdhelp.DracFileText) |
| c.Flags.BoolVar(&c.interactive, "i", false, "enable interactive mode for input") |
| |
| c.Flags.StringVar(&c.vlanName, "vlan", "", "the vlan to assign the drac to") |
| c.Flags.BoolVar(&c.deleteVlan, "delete-vlan", false, "if deleting the ip assignment for the drac") |
| c.Flags.StringVar(&c.ip, "ip", "", "the ip to assign the drac to") |
| |
| c.Flags.StringVar(&c.machineName, "machine", "", "name of the machine to update the association of the drac to a different machine") |
| c.Flags.StringVar(&c.dracName, "name", "", "name of the drac to update") |
| c.Flags.StringVar(&c.macAddress, "mac", "", "the mac address of the drac to add."+cmdhelp.ClearFieldHelpText) |
| c.Flags.StringVar(&c.switchName, "switch", "", "the name of the switch that this drac is connected to. "+cmdhelp.ClearFieldHelpText) |
| c.Flags.StringVar(&c.switchPort, "switch-port", "", "the port of the switch that this drac is connected to. "+cmdhelp.ClearFieldHelpText) |
| c.Flags.StringVar(&c.tags, "tags", "", "comma separated tags. You can only append/add new tags here. "+cmdhelp.ClearFieldHelpText) |
| return c |
| }, |
| } |
| |
| type updateDrac struct { |
| subcommands.CommandRunBase |
| authFlags authcli.Flags |
| envFlags site.EnvFlags |
| |
| newSpecsFile string |
| interactive bool |
| commonFlags site.CommonFlags |
| |
| machineName string |
| dracName string |
| macAddress string |
| switchName string |
| switchPort string |
| tags string |
| vlanName string |
| deleteVlan bool |
| ip string |
| } |
| |
| func (c *updateDrac) Run(a subcommands.Application, args []string, env subcommands.Env) int { |
| if err := c.innerRun(a, args, env); err != nil { |
| cmdlib.PrintError(a, err) |
| return 1 |
| } |
| return 0 |
| } |
| |
| func (c *updateDrac) innerRun(a subcommands.Application, args []string, env subcommands.Env) error { |
| if err := c.validateArgs(); err != nil { |
| return err |
| } |
| ctx := cli.GetContext(a, c, env) |
| ns, err := c.envFlags.Namespace() |
| if err != nil { |
| return err |
| } |
| ctx = utils.SetupContext(ctx, ns) |
| hc, err := cmdlib.NewHTTPClient(ctx, &c.authFlags) |
| if err != nil { |
| return err |
| } |
| e := c.envFlags.Env() |
| if c.commonFlags.Verbose() { |
| fmt.Printf("Using UFS service %s\n", e.UnifiedFleetService) |
| } |
| ic := ufsAPI.NewFleetPRPCClient(&prpc.Client{ |
| C: hc, |
| Host: e.UnifiedFleetService, |
| Options: site.DefaultPRPCOptions, |
| }) |
| var drac ufspb.Drac |
| if c.interactive { |
| utils.GetDracInteractiveInput(ctx, ic, &drac, true) |
| } else { |
| if c.newSpecsFile != "" { |
| if err = utils.ParseJSONFile(c.newSpecsFile, &drac); err != nil { |
| return err |
| } |
| if drac.GetMachine() == "" { |
| return errors.New(fmt.Sprintf("machine field is empty in json. It is a required parameter for json input.")) |
| } |
| } else { |
| c.parseArgs(&drac) |
| } |
| } |
| if err := utils.PrintExistingDrac(ctx, ic, drac.Name); err != nil { |
| return err |
| } |
| drac.Name = ufsUtil.AddPrefix(ufsUtil.DracCollection, drac.Name) |
| if !ufsUtil.ValidateTags(drac.Tags) { |
| return fmt.Errorf(ufsAPI.InvalidTags) |
| } |
| res, err := ic.UpdateDrac(ctx, &ufsAPI.UpdateDracRequest{ |
| Drac: &drac, |
| NetworkOption: &ufsAPI.NetworkOption{ |
| Vlan: c.vlanName, |
| Delete: c.deleteVlan, |
| Ip: c.ip, |
| }, |
| UpdateMask: utils.GetUpdateMask(&c.Flags, map[string]string{ |
| "machine": "machine", |
| "mac": "macAddress", |
| "switch": "switch", |
| "switch-port": "portName", |
| "tags": "tags", |
| }), |
| }) |
| if err != nil { |
| return err |
| } |
| res.Name = ufsUtil.RemovePrefix(res.Name) |
| fmt.Println("The drac after update:") |
| utils.PrintProtoJSON(res, !utils.NoEmitMode(false)) |
| if c.deleteVlan { |
| fmt.Printf("Successfully deleted vlan of drac %s\n", res.Name) |
| } |
| if c.vlanName != "" { |
| // Log the assigned IP |
| if dhcp, err := ic.GetDHCPConfig(ctx, &ufsAPI.GetDHCPConfigRequest{ |
| Hostname: res.Name, |
| }); err == nil { |
| utils.PrintProtoJSON(dhcp, false) |
| fmt.Println("Successfully added dhcp config to drac: ", res.Name) |
| } |
| } |
| return nil |
| } |
| |
| func (c *updateDrac) parseArgs(drac *ufspb.Drac) { |
| drac.Name = c.dracName |
| if c.macAddress == utils.ClearFieldValue { |
| drac.MacAddress = "" |
| } else { |
| drac.MacAddress = c.macAddress |
| } |
| drac.SwitchInterface = &ufspb.SwitchInterface{} |
| drac.Machine = c.machineName |
| if c.switchName == utils.ClearFieldValue { |
| drac.GetSwitchInterface().Switch = "" |
| } else { |
| drac.GetSwitchInterface().Switch = c.switchName |
| } |
| if c.switchPort == utils.ClearFieldValue { |
| drac.GetSwitchInterface().PortName = "" |
| } else { |
| drac.GetSwitchInterface().PortName = c.switchPort |
| } |
| if c.tags == utils.ClearFieldValue { |
| drac.Tags = nil |
| } else { |
| drac.Tags = utils.GetStringSlice(c.tags) |
| } |
| } |
| |
| func (c *updateDrac) validateArgs() error { |
| if c.newSpecsFile != "" && c.interactive { |
| return cmdlib.NewQuietUsageError(c.Flags, "Wrong usage!!\nThe interactive & JSON mode cannot be specified at the same time.") |
| } |
| if c.newSpecsFile != "" || c.interactive { |
| if c.dracName != "" { |
| return cmdlib.NewQuietUsageError(c.Flags, "Wrong usage!!\nThe interactive/JSON mode is specified. '-name' cannot be specified at the same time.") |
| } |
| if c.switchName != "" { |
| return cmdlib.NewQuietUsageError(c.Flags, "Wrong usage!!\nThe interactive/JSON mode is specified. '-switch' cannot be specified at the same time.") |
| } |
| if c.switchPort != "" { |
| return cmdlib.NewQuietUsageError(c.Flags, "Wrong usage!!\nThe interactive/JSON mode is specified. '-switch-port' cannot be specified at the same time.") |
| } |
| if c.macAddress != "" { |
| return cmdlib.NewQuietUsageError(c.Flags, "Wrong usage!!\nThe interactive/JSON mode is specified. '-mac' cannot be specified at the same time.") |
| } |
| if c.tags != "" { |
| return cmdlib.NewQuietUsageError(c.Flags, "Wrong usage!!\nThe interactive/JSON mode is specified. '-tags' cannot be specified at the same time.") |
| } |
| if c.machineName != "" { |
| return cmdlib.NewQuietUsageError(c.Flags, "Wrong usage!!\nThe interactive/JSON mode is specified. '-machine' cannot be specified at the same time.") |
| } |
| } |
| if c.newSpecsFile == "" && !c.interactive { |
| if c.dracName == "" { |
| return cmdlib.NewQuietUsageError(c.Flags, "Wrong usage!!\n'-name' is required, no mode ('-f' or '-i') is specified.") |
| } |
| if c.vlanName == "" && !c.deleteVlan && c.ip == "" && |
| c.machineName == "" && c.switchName == "" && c.switchPort == "" && |
| c.macAddress == "" && c.tags == "" { |
| return cmdlib.NewQuietUsageError(c.Flags, "Wrong usage!!\nNothing to update. Please provide any field to update") |
| } |
| } |
| return nil |
| } |
