Add AdoptDevice and ForgetDevice methods (#45)

* Add `AdoptDevice` and `ForgetDevice` methods

* Address comments

Author: joshuaspence

fields/main.go

@@ -148,6 +148,8 @@ func NewResource(version string, structName string, resourcePath string) *Resour
 		baseType.Fields[" Key"] = NewFieldInfo("Key", "key", "string", "", false, false, "")
 	case resource.StructName == "Device":
 		baseType.Fields[" MAC"] = NewFieldInfo("MAC", "mac", "string", "", true, false, "")
+		baseType.Fields["Adopted"] = NewFieldInfo("Adopted", "adopted", "bool", "", false, false, "")
+		baseType.Fields["State"] = NewFieldInfo("State", "state", "DeviceState", "", false, false, "")
 	case resource.StructName == "User":
 		baseType.Fields[" IP"] = NewFieldInfo("IP", "ip", "string", "non-generated field", true, false, "")
 		baseType.Fields[" DevIdOverride"] = NewFieldInfo("DevIdOverride", "dev_id_override", "int", "non-generated field", true, false, "")

unifi/device.generated.go

@@ -2,6 +2,24 @@ package unifi
 
 import (
 	"context"
+	"fmt"
+)
+
+type DeviceState int
+
+const (
+	DeviceStateUnknown          DeviceState = 0
+	DeviceStateConnected        DeviceState = 1
+	DeviceStatePending          DeviceState = 2
+	DeviceStateFirmwareMismatch DeviceState = 3
+	DeviceStateUpgrading        DeviceState = 4
+	DeviceStateProvisioning     DeviceState = 5
+	DeviceStateHeartbeatMissed  DeviceState = 6
+	DeviceStateAdopting         DeviceState = 7
+	DeviceStateDeleting         DeviceState = 8
+	DeviceStateInformError      DeviceState = 9
+	DeviceStateAdoptFailed      DeviceState = 10
+	DeviceStateIsolated         DeviceState = 11
 )
 
 func (c *Client) ListDevice(ctx context.Context, site string) ([]Device, error) {
@@ -39,3 +57,46 @@ func (c *Client) GetDevice(ctx context.Context, site, id string) (*Device, error
 
 	return nil, &NotFoundError{}
 }
+
+func (c *Client) AdoptDevice(ctx context.Context, site, mac string) error {
+	reqBody := struct {
+		Cmd string `json:"cmd"`
+		MAC string `json:"mac"`
+	}{
+		Cmd: "adopt",
+		MAC: mac,
+	}
+
+	var respBody struct {
+		Meta meta `json:"meta"`
+	}
+
+	err := c.do(ctx, "POST", fmt.Sprintf("s/%s/cmd/devmgr", site), reqBody, &respBody)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (c *Client) ForgetDevice(ctx context.Context, site, mac string) error {
+	reqBody := struct {
+		Cmd  string   `json:"cmd"`
+		MACs []string `json:"macs"`
+	}{
+		Cmd:  "delete-device",
+		MACs: []string{mac},
+	}
+
+	var respBody struct {
+		Meta meta     `json:"meta"`
+		Data []Device `json:"data"`
+	}
+
+	err := c.do(ctx, "POST", fmt.Sprintf("s/%s/cmd/sitemgr", site), reqBody, &respBody)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}