orgs.go

package gapi

import (
	"bytes"
	"encoding/json"
	"errors"
	"fmt"
	"io/ioutil"
)

type Org struct {
	Id   int64  `json:"id"`
	Name string `json:"name"`
}

type OrgPreferences struct {
	Theme            string `json:"theme,omitempty"`
	HomeDashboardUid string `json:"homeDashboardUid,omitempty"`
	HomeDashboardId  int64  `json:"homeDashboardId,omitempty"`
	Timezone         string `json:"timezone,omitempty"`
}

func (c *Client) Orgs() ([]Org, error) {
	orgs := make([]Org, 0)

	req, err := c.newRequest("GET", "/api/orgs/", nil, nil)
	if err != nil {
		return orgs, err
	}
	resp, err := c.Do(req)
	if err != nil {
		return orgs, err
	}
	if resp.StatusCode != 200 {
		return orgs, errors.New(resp.Status)
	}
	data, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return orgs, err
	}
	err = json.Unmarshal(data, &orgs)
	return orgs, err
}

func (c *Client) OrgByName(name string) (Org, error) {
	org := Org{}
	req, err := c.newRequest("GET", fmt.Sprintf("/api/orgs/name/%s", name), nil, nil)
	if err != nil {
		return org, err
	}
	resp, err := c.Do(req)
	if err != nil {
		return org, err
	}
	if resp.StatusCode != 200 {
		return org, errors.New(resp.Status)
	}
	data, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return org, err
	}
	err = json.Unmarshal(data, &org)
	return org, err
}

func (c *Client) Org(id int64) (Org, error) {
	org := Org{}
	req, err := c.newRequest("GET", fmt.Sprintf("/api/orgs/%d", id), nil, nil)
	if err != nil {
		return org, err
	}
	resp, err := c.Do(req)
	if err != nil {
		return org, err
	}
	if resp.StatusCode != 200 {
		return org, errors.New(resp.Status)
	}
	data, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return org, err
	}
	err = json.Unmarshal(data, &org)
	return org, err
}

func (c *Client) NewOrg(name string) (int64, error) {
	dataMap := map[string]string{
		"name": name,
	}
	data, err := json.Marshal(dataMap)
	id := int64(0)
	req, err := c.newRequest("POST", "/api/orgs", nil, bytes.NewBuffer(data))
	if err != nil {
		return id, err
	}
	resp, err := c.Do(req)
	if err != nil {
		return id, err
	}
	if resp.StatusCode != 200 {
		return id, errors.New(resp.Status)
	}
	data, err = ioutil.ReadAll(resp.Body)
	if err != nil {
		return id, err
	}
	tmp := struct {
		Id int64 `json:"orgId"`
	}{}
	err = json.Unmarshal(data, &tmp)
	if err != nil {
		return id, err
	}
	id = tmp.Id
	return id, err
}

func (c *Client) UpdateOrg(id int64, name string) error {
	dataMap := map[string]string{
		"name": name,
	}
	data, err := json.Marshal(dataMap)
	req, err := c.newRequest("PUT", fmt.Sprintf("/api/orgs/%d", id), nil, bytes.NewBuffer(data))
	if err != nil {
		return err
	}
	resp, err := c.Do(req)
	if err != nil {
		return err
	}
	if resp.StatusCode != 200 {
		return errors.New(resp.Status)
	}
	return err
}

func (c *Client) GetOrgPreferences() (*OrgPreferences, error) {
	var prefs *OrgPreferences
	req, err := c.newRequest("GET", "/api/org/preferences", nil, nil)
	if err != nil {
		return prefs, err
	}
	resp, err := c.Do(req)
	if err != nil {
		return prefs, err
	}
	if resp.StatusCode != 200 {
		return prefs, errors.New(resp.Status)
	}
	data, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return prefs, err
	}
	err = json.Unmarshal(data, &prefs)
	if err != nil {
		return prefs, err
	}

	if prefs.HomeDashboardId != 0 {
		uid, err := c.GetDashboardUidById(prefs.HomeDashboardId)
		if err != nil {
			return prefs, err
		}
		prefs.HomeDashboardUid = uid
	}
	return prefs, err
}

func (c *Client) UpdateOrgPreferences(prefs *OrgPreferences) error {
	if prefs.HomeDashboardUid != "" {
		id, err := c.GetDashboardIdByUid(prefs.HomeDashboardUid)
		if err != nil {
			return err
		}
		prefs.HomeDashboardId = id
	}

	data, err := json.Marshal(prefs)
	req, err := c.newRequest("PUT", "/api/org/preferences", nil, bytes.NewBuffer(data))
	if err != nil {
		return err
	}
	resp, err := c.Do(req)
	if err != nil {
		return err
	}
	if resp.StatusCode != 200 {
		return errors.New(resp.Status)
	}
	return nil
}

func (c *Client) DeleteOrg(id int64) error {
	req, err := c.newRequest("DELETE", fmt.Sprintf("/api/orgs/%d", id), nil, nil)
	if err != nil {
		return err
	}
	resp, err := c.Do(req)
	if err != nil {
		return err
	}
	if resp.StatusCode != 200 {
		return errors.New(resp.Status)
	}
	return err
}