klauvsteinen/go-unifi
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add method to update site

Browse Source
This commit is contained in:
Kurt McAlpine
2020-10-10 14:57:40 +13:00
committed by Paul Tyng
parent d6dded588f
commit cb425568b4
1 changed files with 30 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
unifi/sites.go
View File

@@ -1,6 +1,9 @@
package unifi package unifi
import "context" import (
"context"
"fmt"
)
type Site struct { type Site struct {
ID string `json:"_id,omitempty"` ID string `json:"_id,omitempty"`
@@ -30,13 +33,13 @@ func (c *Client) ListSites(ctx context.Context) ([]Site, error) {
return respBody.Data, nil return respBody.Data, nil
} }
func (c *Client) CreateSite(ctx context.Context, Description string) ([]Site, error) { func (c *Client) CreateSite(ctx context.Context, description string) ([]Site, error) {
reqBody := struct { reqBody := struct {
Cmd string `json:"cmd"` Cmd string `json:"cmd"`
Desc string `json:"desc"` Desc string `json:"desc"`
}{ }{
Cmd: "add-site", Cmd: "add-site",
Desc: Description, Desc: description,
} }
var respBody struct { var respBody struct {
@@ -52,13 +55,13 @@ func (c *Client) CreateSite(ctx context.Context, Description string) ([]Site, er
return respBody.Data, nil return respBody.Data, nil
} }
func (c *Client) DeleteSite(ctx context.Context, ID string) ([]Site, error) { func (c *Client) DeleteSite(ctx context.Context, id string) ([]Site, error) {
reqBody := struct { reqBody := struct {
Cmd string `json:"cmd"` Cmd string `json:"cmd"`
Site string `json:"site"` Site string `json:"site"`
}{ }{
Cmd: "delete-site", Cmd: "delete-site",
Site: ID, Site: id,
} }
var respBody struct { var respBody struct {
@@ -73,3 +76,25 @@ func (c *Client) DeleteSite(ctx context.Context, ID string) ([]Site, error) {
return respBody.Data, nil return respBody.Data, nil
} }
func (c *Client) UpdateSite(ctx context.Context, name, description string) ([]Site, error) {
reqBody := struct {
Cmd string `json:"cmd"`
Desc string `json:"desc"`
}{
Cmd: "update-site",
Desc: description,
}
var respBody struct {
Meta meta `json:"meta"`
Data []Site `json:"data"`
}
err := c.do(ctx, "POST", fmt.Sprintf("s/%s/cmd/sitemgr", name), reqBody, &respBody)
if err != nil {
return nil, err
}
return respBody.Data, nil
}