From cb425568b4365376f1bdd2f520b91ef11475985f Mon Sep 17 00:00:00 2001
From: Kurt McAlpine <kurt@linux.com>
Date: Sat, 10 Oct 2020 14:57:40 +1300
Subject: [PATCH] Add method to update site

---
 unifi/sites.go | 35 ++++++++++++++++++++++++++++++-----
 1 file changed, 30 insertions(+), 5 deletions(-)

diff --git a/unifi/sites.go b/unifi/sites.go
index 0534fd0..137f116 100644
--- a/unifi/sites.go
+++ b/unifi/sites.go
@@ -1,6 +1,9 @@
 package unifi
 
-import "context"
+import (
+	"context"
+	"fmt"
+)
 
 type Site struct {
 	ID string `json:"_id,omitempty"`
@@ -30,13 +33,13 @@ func (c *Client) ListSites(ctx context.Context) ([]Site, error) {
 	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 {
 		Cmd  string `json:"cmd"`
 		Desc string `json:"desc"`
 	}{
 		Cmd:  "add-site",
-		Desc: Description,
+		Desc: description,
 	}
 
 	var respBody struct {
@@ -52,13 +55,13 @@ func (c *Client) CreateSite(ctx context.Context, Description string) ([]Site, er
 	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 {
 		Cmd  string `json:"cmd"`
 		Site string `json:"site"`
 	}{
 		Cmd:  "delete-site",
-		Site: ID,
+		Site: id,
 	}
 
 	var respBody struct {
@@ -73,3 +76,25 @@ func (c *Client) DeleteSite(ctx context.Context, ID string) ([]Site, error) {
 
 	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
+}