automana/client/workspace.go

package client

import "fmt"
import "net/url"

type Workspace struct {
	GID  string `json:"gid"`
	Name string `json:"name"`
}

type workspacesResponse struct {
	Data     []*Workspace `json:"data"`
	NextPage *nextPage    `json:"next_page"`
}

func (c *Client) InWorkspace(name string) (*WorkspaceClient, error) {
	wrk, err := c.GetWorkspaceByName(name)
	if err != nil {
		return nil, err
	}

	return &WorkspaceClient{
		client:    c,
		workspace: wrk,
	}, nil
}

func (c *Client) GetWorkspaces() ([]*Workspace, error) {
	ret := []*Workspace{}

	values := &url.Values{}

	for {
		resp := &workspacesResponse{}
		err := c.get("workspaces", values, resp)
		if err != nil {
			return nil, err
		}

		ret = append(ret, resp.Data...)

		if resp.NextPage == nil {
			break
		}

		values.Set("offset", resp.NextPage.Offset)
	}

	return ret, nil
}

func (c *Client) GetWorkspaceByName(name string) (*Workspace, error) {
	wrks, err := c.GetWorkspaces()
	if err != nil {
		return nil, err
	}

	for _, wrk := range wrks {
		if wrk.Name == name {
			return wrk, nil
		}
	}

	return nil, fmt.Errorf("Workspace `%s` not found", name)
}

func (wrk *Workspace) String() string {
	return fmt.Sprintf("%s (%s)", wrk.GID, wrk.Name)
}