feat: add zones (#46)

api/zone.go

@@ -0,0 +1,12 @@
+package api
+
+import (
+	"github.com/maas/gomaasclient/entity"
+)
+
+// Zone is an interface defining API behaviour for zones
+type Zone interface {
+	Get(name string) (*entity.Zone, error)
+	Update(name string, params *entity.ZoneParams) (*entity.Zone, error)
+	Delete(name string) error
+}

api/zones.go

@@ -0,0 +1,11 @@
+package api
+
+import (
+	"github.com/maas/gomaasclient/entity"
+)
+
+// Zones is an interface for listing and creating Zone objects
+type Zones interface {
+	Get() ([]entity.Zone, error)
+	Create(params *entity.ZoneParams) (*entity.Zone, error)
+}

client/client.go

@@ -78,6 +78,8 @@ func constructClient(apiClient *APIClient) *Client {
 		BootSourceSelections:  &BootSourceSelections{APIClient: *apiClient},
 		BootResource:          &BootResource{APIClient: *apiClient},
 		BootResources:         &BootResources{APIClient: *apiClient},
+		Zone:                  &Zone{APIClient: *apiClient},
+		Zones:                 &Zones{APIClient: *apiClient},
 	}
 
 	return &client
@@ -132,6 +134,8 @@ type Client struct {
 	BootSourceSelections  api.BootSourceSelections
 	BootResource          api.BootResource
 	BootResources         api.BootResources
+	Zone                  api.Zone
+	Zones                 api.Zones
 }
 
 // GetAPIClient returns a MAAS API client.

client/zone.go

@@ -0,0 +1,48 @@
+package client
+
+import (
+	"encoding/json"
+	"net/url"
+
+	"github.com/google/go-querystring/query"
+	"github.com/maas/gomaasclient/entity"
+)
+
+// Zone Contains functionality for manipulating the Zone entity.
+type Zone struct {
+	APIClient APIClient
+}
+
+func (z *Zone) client(name string) APIClient {
+	return z.APIClient.GetSubObject("zones").GetSubObject(name)
+}
+
+// Get Zone details.
+func (z *Zone) Get(name string) (*entity.Zone, error) {
+	zone := new(entity.Zone)
+	err := z.client(name).Get("", url.Values{}, func(data []byte) error {
+		return json.Unmarshal(data, zone)
+	})
+
+	return zone, err
+}
+
+// Update Zone.
+func (z *Zone) Update(name string, params *entity.ZoneParams) (*entity.Zone, error) {
+	qsp, err := query.Values(params)
+	if err != nil {
+		return nil, err
+	}
+
+	zone := new(entity.Zone)
+	err = z.client(name).Put(qsp, func(data []byte) error {
+		return json.Unmarshal(data, zone)
+	})
+
+	return zone, err
+}
+
+// Delete Zone.
+func (z *Zone) Delete(name string) error {
+	return z.client(name).Delete()
+}

client/zones.go

@@ -0,0 +1,44 @@
+//nolint:dupl // disable dupl check on client for now
+package client
+
+import (
+	"encoding/json"
+	"net/url"
+
+	"github.com/google/go-querystring/query"
+	"github.com/maas/gomaasclient/entity"
+)
+
+// Zones implements api.Zones
+type Zones struct {
+	APIClient APIClient
+}
+
+func (z *Zones) client() APIClient {
+	return z.APIClient.GetSubObject("zones")
+}
+
+// Get fetches a list of Zone objects
+func (z *Zones) Get() ([]entity.Zone, error) {
+	zones := make([]entity.Zone, 0)
+	err := z.client().Get("", url.Values{}, func(data []byte) error {
+		return json.Unmarshal(data, &zones)
+	})
+
+	return zones, err
+}
+
+// Create creates a new Zone
+func (z *Zones) Create(params *entity.ZoneParams) (*entity.Zone, error) {
+	qsp, err := query.Values(params)
+	if err != nil {
+		return nil, err
+	}
+
+	zone := new(entity.Zone)
+	err = z.client().Post("", qsp, func(data []byte) error {
+		return json.Unmarshal(data, zone)
+	})
+
+	return zone, err
+}

entity/zone.go

@@ -7,3 +7,8 @@ type Zone struct {
 	ResourceURI string `json:"resource_uri,omitempty"`
 	ID          int    `json:"id,omitempty"`
 }
+
+type ZoneParams struct {
+	Name        string `url:"name,omitempty"`
+	Description string `url:"description,omitempty"`
+}