11package hosts
22
33import (
4- "log"
5-
64 serverscom "github.com/serverscom/serverscom-go-client/pkg"
75 "github.com/serverscom/srvctl/cmd/base"
86 "github.com/spf13/cobra"
@@ -11,11 +9,7 @@ import (
119func newListDSNetworksCmd (cmdContext * base.CmdContext ) * cobra.Command {
1210 factory := func (verbose bool , args ... string ) serverscom.Collection [serverscom.Network ] {
1311 scClient := cmdContext .GetClient ().SetVerbose (verbose ).GetScClient ()
14- if len (args ) == 0 {
15- log .Fatal ("Missing dedicated server ID" )
16- }
17- id := args [0 ]
18- return scClient .Hosts .DedicatedServerNetworks (id )
12+ return scClient .Hosts .DedicatedServerNetworks (args [0 ])
1913 }
2014
2115 opts := base .NewListOptions (
@@ -27,7 +21,11 @@ func newListDSNetworksCmd(cmdContext *base.CmdContext) *cobra.Command {
2721 & base.AdditionalOption [serverscom.Network ]{},
2822 )
2923
30- return base .NewListCmd ("list-networks <id>" , "Dedicated server networks" , factory , cmdContext , opts ... )
24+ cmd := base .NewListCmd ("list-networks" , "Dedicated server networks" , factory , cmdContext , opts ... )
25+ cmd .Use = "list-networks <id>"
26+ cmd .Args = cobra .ExactArgs (1 )
27+
28+ return cmd
3129}
3230
3331func newGetDSNetworkCmd (cmdContext * base.CmdContext ) * cobra.Command {
@@ -160,11 +158,7 @@ func newDeleteDSNetworkCmd(cmdContext *base.CmdContext) *cobra.Command {
160158func newListKBMNetworksCmd (cmdContext * base.CmdContext ) * cobra.Command {
161159 factory := func (verbose bool , args ... string ) serverscom.Collection [serverscom.Network ] {
162160 scClient := cmdContext .GetClient ().SetVerbose (verbose ).GetScClient ()
163- if len (args ) == 0 {
164- log .Fatal ("Missing KBM node ID" )
165- }
166- id := args [0 ]
167- return scClient .Hosts .KubernetesBaremetalNodeNetworks (id )
161+ return scClient .Hosts .KubernetesBaremetalNodeNetworks (args [0 ])
168162 }
169163
170164 opts := base .NewListOptions (
@@ -176,5 +170,9 @@ func newListKBMNetworksCmd(cmdContext *base.CmdContext) *cobra.Command {
176170 & base.AdditionalOption [serverscom.Network ]{},
177171 )
178172
179- return base .NewListCmd ("list-networks <id>" , "KBM node networks" , factory , cmdContext , opts ... )
173+ cmd := base .NewListCmd ("list-networks" , "KBM node networks" , factory , cmdContext , opts ... )
174+ cmd .Use = "list-networks <id>"
175+ cmd .Args = cobra .ExactArgs (1 )
176+
177+ return cmd
180178}
0 commit comments