From d14cec1fad87cc7f70b91fb512d49b1be3c8558f Mon Sep 17 00:00:00 2001 From: Kurt McAlpine Date: Mon, 12 Oct 2020 15:30:44 +1300 Subject: [PATCH] Add method to get site --- unifi/sites.go | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/unifi/sites.go b/unifi/sites.go index 137f116..eec4b08 100644 --- a/unifi/sites.go +++ b/unifi/sites.go @@ -33,6 +33,22 @@ func (c *Client) ListSites(ctx context.Context) ([]Site, error) { return respBody.Data, nil } +func (c *Client) GetSite(ctx context.Context, id string) (*Site, error) { + sites, err := c.ListSites(ctx) + + if err != nil { + return nil, err + } + + for _, s := range sites { + if s.ID == id { + return &s, nil + } + } + + return nil, &NotFoundError{} +} + func (c *Client) CreateSite(ctx context.Context, description string) ([]Site, error) { reqBody := struct { Cmd string `json:"cmd"`