Files
automana/client/workspace.go
2021-09-26 04:25:06 +00:00

71 lines
1.2 KiB
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,
rateLimitSearch: NewRateLimitPerMinute(60, 60),
}, 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)
}