Locations cmd

Makefile

@@ -15,6 +15,7 @@ generate: deps
 	mockgen --destination ./internal/mocks/racks_service.go --package=mocks --source ./vendor/github.com/serverscom/serverscom-go-client/pkg/racks.go
 	mockgen --destination ./internal/mocks/invoices_service.go --package=mocks --source ./vendor/github.com/serverscom/serverscom-go-client/pkg/invoices.go
 	mockgen --destination ./internal/mocks/account_service.go --package=mocks --source ./vendor/github.com/serverscom/serverscom-go-client/pkg/accounts.go
+	mockgen --destination ./internal/mocks/locations_service.go --package=mocks --source ./vendor/github.com/serverscom/serverscom-go-client/pkg/locations.go
 	sed -i '' 's|github.com/serverscom/srvctl/vendor/github.com/serverscom/serverscom-go-client/pkg|github.com/serverscom/serverscom-go-client/pkg|g' \
 	./internal/mocks/ssh_service.go \
 	./internal/mocks/hosts_service.go \
@@ -23,5 +24,6 @@ generate: deps
 	./internal/mocks/racks_service.go \
 	./internal/mocks/invoices_service.go \
 	./internal/mocks/account_service.go \
-	./internal/mocks/collection.go
+	./internal/mocks/collection.go \
+	./internal/mocks/locations_service.go
 

cmd/entities/hosts/hosts_test.go

@@ -26,22 +26,28 @@ var (
 	}
 	testDS = serverscom.DedicatedServer{
 		ID:      testId,
+		RackID:  testId,
 		Type:    "dedicated_server",
 		Title:   "example.aa",
 		Status:  "active",
 		Created: fixedTime,
 		Updated: fixedTime,
 	}
 	testKBM = serverscom.KubernetesBaremetalNode{
-		ID:      testId,
-		Type:    "kubernetes_baremetal_node",
-		Title:   "example.aa",
-		Status:  "active",
-		Created: fixedTime,
-		Updated: fixedTime,
+		ID:                          testId,
+		RackID:                      testId,
+		KubernetesClusterID:         testId,
+		KubernetesClusterNodeID:     testId,
+		KubernetesClusterNodeNumber: 1,
+		Type:                        "kubernetes_baremetal_node",
+		Title:                       "example.aa",
+		Status:                      "active",
+		Created:                     fixedTime,
+		Updated:                     fixedTime,
 	}
 	testSBM = serverscom.SBMServer{
 		ID:      testId,
+		RackID:  testId,
 		Type:    "sbm_server",
 		Title:   "example.aa",
 		Status:  "active",

cmd/entities/locations/get.go

@@ -0,0 +1,45 @@
+package locations
+
+import (
+	"fmt"
+	"strconv"
+
+	"github.com/serverscom/srvctl/cmd/base"
+	"github.com/spf13/cobra"
+)
+
+func newGetCmd(cmdContext *base.CmdContext) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "get <id>",
+		Short: "Get a location",
+		Long:  "Get a location by id",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			manager := cmdContext.GetManager()
+
+			ctx, cancel := base.SetupContext(cmd, manager)
+			defer cancel()
+
+			base.SetupProxy(cmd, manager)
+
+			scClient := cmdContext.GetClient().SetVerbose(manager.GetVerbose(cmd)).GetScClient()
+
+			id, err := strconv.Atoi(args[0])
+			if err != nil {
+				return fmt.Errorf("location id should be integer")
+			}
+			location, err := scClient.Locations.GetLocation(ctx, int64(id))
+			if err != nil {
+				return err
+			}
+
+			if location != nil {
+				formatter := cmdContext.GetOrCreateFormatter(cmd)
+				return formatter.Format(location)
+			}
+			return nil
+		},
+	}
+
+	return cmd
+}

cmd/entities/locations/list.go

@@ -0,0 +1,21 @@
+package locations
+
+import (
+	serverscom "github.com/serverscom/serverscom-go-client/pkg"
+	"github.com/serverscom/srvctl/cmd/base"
+	"github.com/spf13/cobra"
+)
+
+func newListCmd(cmdContext *base.CmdContext) *cobra.Command {
+	factory := func(verbose bool, args ...string) serverscom.Collection[serverscom.Location] {
+		scClient := cmdContext.GetClient().SetVerbose(verbose).GetScClient()
+		return scClient.Locations.Collection()
+	}
+
+	opts := base.NewListOptions(
+		&base.BaseListOptions[serverscom.Location]{},
+		&base.SearchPatternOption[serverscom.Location]{},
+	)
+
+	return base.NewListCmd("list", "locations", factory, cmdContext, opts...)
+}

cmd/entities/locations/locations.go

@@ -0,0 +1,41 @@
+package locations
+
+import (
+	"log"
+
+	serverscom "github.com/serverscom/serverscom-go-client/pkg"
+	"github.com/serverscom/srvctl/cmd/base"
+	"github.com/serverscom/srvctl/internal/output/entities"
+	"github.com/spf13/cobra"
+)
+
+func NewCmd(cmdContext *base.CmdContext) *cobra.Command {
+	locationEntity, err := entities.Registry.GetEntityFromValue(serverscom.Location{})
+	if err != nil {
+		log.Fatal(err)
+	}
+	entitiesMap := make(map[string]entities.EntityInterface)
+	entitiesMap["locations"] = locationEntity
+	cmd := &cobra.Command{
+		Use:   "locations",
+		Short: "Manage locations",
+		PersistentPreRunE: base.CombinePreRunE(
+			base.CheckFormatterFlags(cmdContext, entitiesMap),
+			base.CheckEmptyContexts(cmdContext),
+		),
+		Args: base.NoArgs,
+		Run:  base.UsageRun,
+	}
+
+	cmd.AddCommand(
+		newListCmd(cmdContext),
+		// newAddCmd(cmdContext),
+		newGetCmd(cmdContext),
+	// newUpdateCmd(cmdContext),
+	// newDeleteCmd(cmdContext),
+	)
+
+	base.AddFormatFlags(cmd)
+
+	return cmd
+}