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

Add DNS Record

Browse Source
This commit is contained in:
appkins
2024-07-18 19:02:20 -05:00
parent 2736c307ea
commit 1477c472cc
4 changed files with 49 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
unifi/dns_record.go
View File

@@ -5,24 +5,64 @@ package unifi
import (
"context"
"fmt"
)
func (c *Client) ListDNSRecord(ctx context.Context, site string) ([]DNSRecord, error) {
return c.listDNSRecord(ctx, site)
var respBody []DNSRecord
err := c.do(ctx, "GET", fmt.Sprintf("%s/site/%s/static-dns", c.apiV2Path, site), nil, &respBody)
if err != nil {
return nil, err
}
return respBody, nil
}
func (c *Client) GetDNSRecord(ctx context.Context, site, id string) (*DNSRecord, error) {
return c.getDNSRecord(ctx, site, id)
var respBody DNSRecord
err := c.do(ctx, "GET", fmt.Sprintf("%s/site/%s/static-dns/%s", c.apiV2Path, site, id), nil, &respBody)
if err != nil {
return nil, err
}
// if len(respBody.Data) != 1 {
// return nil, &NotFoundError{}
// }
// d := respBody.Data[0]
return &respBody, nil
}
func (c *Client) DeleteDNSRecord(ctx context.Context, site, id string) error {
return c.deleteDNSRecord(ctx, site, id)
err := c.do(ctx, "DELETE", fmt.Sprintf("%s/site/%s/static-dns/%s", c.apiV2Path, site, id), struct{}{}, nil)
if err != nil {
return err
}
return nil
}
func (c *Client) CreateDNSRecord(ctx context.Context, site string, d *DNSRecord) (*DNSRecord, error) {
return c.createDNSRecord(ctx, site, d)
var respBody DNSRecord
err := c.do(ctx, "POST", fmt.Sprintf("%s/site/%s/static-dns", c.apiV2Path, site), d, &respBody)
if err != nil {
return nil, err
}
return &respBody, nil
}
func (c *Client) UpdateDNSRecord(ctx context.Context, site string, d *DNSRecord) (*DNSRecord, error) {
return c.updateDNSRecord(ctx, site, d)
var respBody DNSRecord
err := c.do(ctx, "PUT", fmt.Sprintf("%s/site/%s/static-dns/%s", c.apiV2Path, site, d.ID), d, &respBody)
if err != nil {
return nil, err
}
// if len(respBody) != nil {
// return nil, &NotFoundError{}
// }
return &respBody, nil
}