api: Add sites

Author: lazyfrosch

api/response.go

@@ -123,12 +123,15 @@ func (c *Client) GetJSONItems(request *http.Request) (items []json.RawMessage, e
 
 		err = json.Unmarshal(response.Data, &dataItems)
 		if err != nil {
-			return
-		}
-
-		// append items to overall list
-		for _, item := range dataItems {
-			items = append(items, item)
+			// Fall back to adding response.Data to the item list
+			// This is useful when data is not an array, but an object
+			err = nil
+			items = append(items, response.Data)
+		} else {
+			// append each item to overall list
+			for _, item := range dataItems {
+				items = append(items, item)
+			}
 		}
 
 		// set nextCursor or break iteration when done

api/sites.go

@@ -0,0 +1,45 @@
+package api
+
+import (
+	"encoding/json"
+	"net/url"
+)
+
+type Site struct {
+	ID       string `json:"id"`
+	Name     string `json:"name"`
+	SiteType string `json:"siteType"`
+}
+
+type SiteResult struct {
+	// Only license info
+	AllSites json.RawMessage `json:"allSites"`
+	Sites    []*Site
+}
+
+func (c *Client) GetSites(values url.Values) (data []*Site, err error) {
+	req, err := c.NewRequest("GET", "v2.1/sites?"+values.Encode(), nil)
+	if err != nil {
+		return
+	}
+
+	res, err := c.GetJSONItems(req)
+	if err != nil {
+		return
+	}
+
+	for _, page := range res {
+		p := &SiteResult{}
+
+		err = json.Unmarshal(page, p)
+		if err != nil {
+			return
+		}
+
+		for _, site := range p.Sites {
+			data = append(data, site)
+		}
+	}
+
+	return
+}

api/sites_test.go

@@ -0,0 +1,35 @@
+package api_test
+
+import (
+	"github.com/jarcoal/httpmock"
+	"github.com/stretchr/testify/assert"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"testing"
+)
+
+func TestClient_GetSites(t *testing.T) {
+	c, cleanup := testClient()
+	defer cleanup()
+
+	httpmock.RegisterResponder("GET", "https://euce1-test.sentinelone.net/web/api/v2.1/sites",
+		func(req *http.Request) (*http.Response, error) {
+			data, err := ioutil.ReadFile("testdata/sites.json")
+			assert.NoError(t, err)
+
+			return httpmock.NewBytesResponse(200, data), nil
+		})
+
+	sites, err := c.GetSites(url.Values{})
+	assert.NoError(t, err)
+	assert.Len(t, sites, 1)
+}
+
+func TestClient_GetSites_Integration(t *testing.T) {
+	c := envClient(t)
+
+	sites, err := c.GetSites(url.Values{})
+	assert.NoError(t, err)
+	assert.NotNil(t, sites)
+}

api/testdata/sites.json

@@ -0,0 +1,40 @@
+{
+  "data": {
+    "allSites": {
+      "totalLicenses": "integer",
+      "activeLicenses": "integer"
+    },
+    "sites": [
+      {
+        "isDefault": "boolean",
+        "name": "My Site",
+        "totalLicenses": "integer",
+        "suite": "Core",
+        "registrationToken": "eyJ1cmwiOiAiaHR0cHM6Ly9jb25zb2xlLnNlbnRpbmVsb25lLm5ldCIsICJzaXRlX2tleSI6ICIwNzhkYjliMWUyOTA1Y2NhIn0=",
+        "state": "active",
+        "unlimitedExpiration": "boolean",
+        "activeLicenses": 31,
+        "id": "182483948009279260",
+        "externalId": "string",
+        "siteType": "string",
+        "updatedAt": "2018-02-27T04:49:26.257525Z",
+        "creatorId": "225494730938493804",
+        "expiration": "2018-02-27T04:49:26.257525Z",
+        "healthStatus": "boolean",
+        "accountId": "225494730938493804",
+        "accountName": "SentinelOne",
+        "creator": "string",
+        "unlimitedLicenses": "boolean",
+        "createdAt": "2018-02-27T04:49:26.257525Z",
+        "sku": "Core"
+      }
+    ]
+  },
+  "pagination": {
+    "nextCursor": "",
+    "totalItems": 580
+  },
+  "errors": [
+    {}
+  ]
+}