128 lines
3.6 KiB
Go
128 lines
3.6 KiB
Go
package unifi
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
)
|
|
|
|
//go:generate go run golang.org/x/tools/cmd/stringer -trimprefix DeviceState -type DeviceState
|
|
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
|
|
)
|
|
|
|
type LastConnection struct {
|
|
Mac string `json:"mac,omitempty"`
|
|
LastSeen int `json:"last_seen,omitempty"`
|
|
}
|
|
type PortTable struct {
|
|
PortIdx int `json:"port_idx,omitempty"`
|
|
Media string `json:"media,omitempty"`
|
|
PortPoe bool `json:"port_poe,omitempty"`
|
|
PoeCaps int `json:"poe_caps,omitempty"`
|
|
SpeedCaps int `json:"speed_caps,omitempty"`
|
|
LastConnection LastConnection `json:"last_connection,omitempty"`
|
|
OpMode string `json:"op_mode,omitempty"`
|
|
Forward string `json:"forward,omitempty"`
|
|
PoeMode string `json:"poe_mode,omitempty"`
|
|
FullDuplex bool `json:"full_duplex,omitempty"`
|
|
IsUplink bool `json:"is_uplink,omitempty"`
|
|
MacTableCount int `json:"mac_table_count,omitempty"`
|
|
PoeClass string `json:"poe_class,omitempty"`
|
|
PoeCurrent string `json:"poe_current,omitempty"`
|
|
PoeEnable bool `json:"poe_enable,omitempty"`
|
|
PoeGood bool `json:"poe_good,omitempty"`
|
|
PoePower string `json:"poe_power,omitempty"`
|
|
PoeVoltage string `json:"poe_voltage,omitempty"`
|
|
}
|
|
|
|
func (c *Client) ListDevice(ctx context.Context, site string) ([]Device, error) {
|
|
return c.listDevice(ctx, site)
|
|
}
|
|
|
|
func (c *Client) GetDeviceByMAC(ctx context.Context, site, mac string) (*Device, error) {
|
|
return c.getDevice(ctx, site, mac)
|
|
}
|
|
|
|
func (c *Client) DeleteDevice(ctx context.Context, site, id string) error {
|
|
return c.deleteDevice(ctx, site, id)
|
|
}
|
|
|
|
func (c *Client) CreateDevice(ctx context.Context, site string, d *Device) (*Device, error) {
|
|
return c.createDevice(ctx, site, d)
|
|
}
|
|
|
|
func (c *Client) UpdateDevice(ctx context.Context, site string, d *Device) (*Device, error) {
|
|
return c.updateDevice(ctx, site, d)
|
|
}
|
|
|
|
func (c *Client) GetDevice(ctx context.Context, site, id string) (*Device, error) {
|
|
devices, err := c.ListDevice(ctx, site)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, d := range devices {
|
|
if d.ID == id {
|
|
return &d, nil
|
|
}
|
|
}
|
|
|
|
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
|
|
}
|