Merge pull request #34 from serverscom/add-account-balance

Add account service; fix desc for invoices

Author: olegy89

pkg/accounts.go

@@ -0,0 +1,44 @@
+package serverscom
+
+import (
+	"context"
+	"encoding/json"
+)
+
+const (
+	accountBalancePath = "/account/balance"
+)
+
+// AccountService is an interface for interfacing with accoount endpoints
+// API documentation:
+// https://developers.servers.com/api-documentation/v1/#tag/Account
+type AccountService interface {
+
+	// Generic operations
+	GetBalance(ctx context.Context) (*AccountBalance, error)
+}
+
+// AccountHandler handles operations around account
+type AccountHandler struct {
+	client *Client
+}
+
+// GetBalance returns account balance information
+// Endpoint: https://developers.servers.com/api-documentation/v1/#tag/Account/operation/GetCurrentAccountBalance
+func (h *AccountHandler) GetBalance(ctx context.Context) (*AccountBalance, error) {
+	url := h.client.baseURL + accountBalancePath
+
+	body, err := h.client.buildAndExecRequest(ctx, "GET", url, nil)
+
+	if err != nil {
+		return nil, err
+	}
+
+	balance := new(AccountBalance)
+
+	if err := json.Unmarshal(body, &balance); err != nil {
+		return nil, err
+	}
+
+	return balance, nil
+}

pkg/accounts_test.go

@@ -0,0 +1,31 @@
+package serverscom
+
+import (
+	"context"
+	"testing"
+
+	. "github.com/onsi/gomega"
+)
+
+func TestGetAccountBalance(t *testing.T) {
+	g := NewGomegaWithT(t)
+
+	ts, client := newFakeServer().
+		WithRequestPath("/account/balance").
+		WithRequestMethod("GET").
+		WithResponseBodyStubFile("fixtures/account/get_balance_response.json").
+		WithResponseCode(200).
+		Build()
+
+	defer ts.Close()
+
+	ctx := context.TODO()
+
+	balance, err := client.Account.GetBalance(ctx)
+
+	g.Expect(err).To(BeNil())
+	g.Expect(balance).ToNot(BeNil())
+	g.Expect(balance.Currency).To(Equal("EUR"))
+	g.Expect(balance.CurrentBalance).To(Equal(float64(123.456)))
+	g.Expect(balance.NextInvoiceTotalDue).To(Equal(float64(0.0)))
+}

pkg/fixtures/account/get_balance_response.json

@@ -0,0 +1,5 @@
+{
+    "current_balance": 123.456,
+    "next_invoice_total_due": 0.0,
+    "currency": "EUR"
+}

pkg/invoices.go

@@ -22,7 +22,7 @@ type InvoiceService interface {
 	GetBillingInvoice(ctx context.Context, id string) (*Invoice, error)
 }
 
-// InvoiceHandler handles operations around hosts
+// InvoiceHandler handles operations around invoices
 type InvoiceHandler struct {
 	client *Client
 }

pkg/serverscom.go

@@ -42,6 +42,8 @@ type Client struct {
 
 	Invoices InvoiceService
 
+	Account AccountService
+
 	client *resty.Client
 }
 
@@ -101,6 +103,7 @@ func (cli *Client) configureResources() {
 	cli.CloudBlockStorageVolumes = &CloudBlockStorageVolumesHandler{cli}
 	cli.KubernetesClusters = &KubernetesClustersHandler{cli}
 	cli.Invoices = &InvoiceHandler{cli}
+	cli.Account = &AccountHandler{cli}
 }
 
 func (cli *Client) buildURL(path string, values ...interface{}) string {

pkg/types.go

@@ -1042,3 +1042,10 @@ type RealIPHeader struct {
 	Name     RealIPHeaderName `json:"name"`
 	Networks []string         `json:"networks"`
 }
+
+// AccountBalance represents account balance info
+type AccountBalance struct {
+	CurrentBalance      float64 `json:"current_balance"`
+	NextInvoiceTotalDue float64 `json:"next_invoice_total_due"`
+	Currency            string  `json:"currency"`
+}