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

9
unifi/dns_record.generated.go generated
View File

@@ -28,7 +28,7 @@ type DNSRecord struct {
Enabled bool `json:"enabled"`
Key string `json:"key,omitempty"` // .{1,128}
Port int `json:"port,omitempty"`
Priority string `json:"priority,omitempty"` // .{1,128}
Priority int `json:"priority,omitempty"` // .{1,128}
RecordType string `json:"record_type,omitempty"` // A|AAAA|CNAME|MX|NS|PTR|SOA|SRV|TXT
Ttl int `json:"ttl,omitempty"`
Value string `json:"value,omitempty"` // .{1,256}
@@ -59,17 +59,14 @@ func (dst *DNSRecord) UnmarshalJSON(b []byte) error {
}
func (c *Client) listDNSRecord(ctx context.Context, site string) ([]DNSRecord, error) {
var respBody struct {
Meta meta `json:"meta"`
Data []DNSRecord `json:"data"`
}
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.Data, nil
return respBody, nil
}
func (c *Client) getDNSRecord(ctx context.Context, site, id string) (*DNSRecord, error) {

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
}