Merge pull request #31 from serverscom/move-kbm-under-hosts-resource

move-kbm-under-hosts-resource

Author: abozhiy

cmd/base/const.go

@@ -1,5 +1,6 @@
 package base
 
 const (
-	ENDPOINT = "https://api.servers.com/v1"
+	ENDPOINT      = "https://api.servers.com/v1"
+	ErrNoContexts = "no contexts found, log in first: 'srvctl login <context-name>' or, if you don't want to create the context, you may pass SC_TOKEN every time you use srvctl"
 )

cmd/base/hooks.go

@@ -1,6 +1,7 @@
 package base
 
 import (
+	"errors"
 	"fmt"
 	"html/template"
 	"os"
@@ -131,7 +132,7 @@ func CheckEmptyContexts(cmdContext *CmdContext) func(cmd *cobra.Command, args []
 		manager := cmdContext.GetManager()
 
 		if len(manager.GetContexts()) == 0 {
-			return fmt.Errorf("no contexts found, log in first: 'srvctl login <context-name>'")
+			return errors.New(ErrNoContexts)
 		}
 		return nil
 	}

cmd/entities/hosts/drive-slots.go

@@ -23,4 +23,20 @@ func newListDSDriveSlotsCmd(cmdContext *base.CmdContext) *cobra.Command {
 	return base.NewListCmd("list-drive-slots <id>", "Dedicated server drive slots", factory, cmdContext, opts)
 }
 
-// TODO add list drive slots for KBM after adding KBMDriveSlots in go client
+func newListKBMDriveSlotsCmd(cmdContext *base.CmdContext) *cobra.Command {
+	factory := func(verbose bool, args ...string) serverscom.Collection[serverscom.HostDriveSlot] {
+		scClient := cmdContext.GetClient().SetVerbose(verbose).GetScClient()
+		if len(args) == 0 {
+			log.Fatal("Missing KBM node ID")
+		}
+		id := args[0]
+		return scClient.Hosts.KubernetesBaremetalNodeDriveSlots(id)
+	}
+
+	opts := base.NewListOptions(
+		&base.BaseListOptions[serverscom.HostDriveSlot]{},
+		&base.SearchPatternOption[serverscom.HostDriveSlot]{},
+	)
+
+	return base.NewListCmd("list-drive-slots <id>", "KBM node drive slots", factory, cmdContext, opts...)
+}

cmd/entities/hosts/hosts.go

@@ -71,6 +71,8 @@ func NewCmd(cmdContext *base.CmdContext) *cobra.Command {
 			extraCmds: []func(*base.CmdContext) *cobra.Command{
 				newUpdateKBMCmd,
 				newListKBMCmd,
+				newListKBMNetworksCmd,
+				newListKBMDriveSlotsCmd,
 			},
 		},
 		{