Client update

unifi/dns_record.go

@@ -6,8 +6,51 @@ package unifi
 import (
 	"context"
 	"fmt"
+	"encoding/json"
 )
 
+type DNSRecord struct {
+	ID     string `json:"_id,omitempty"`
+	SiteID string `json:"site_id,omitempty"`
+
+	Hidden   bool   `json:"attr_hidden,omitempty"`
+	HiddenID string `json:"attr_hidden_id,omitempty"`
+	NoDelete bool   `json:"attr_no_delete,omitempty"`
+	NoEdit   bool   `json:"attr_no_edit,omitempty"`
+
+	Enabled    bool   `json:"enabled"`
+	Key        string `json:"key,omitempty"` // .{1,128}
+	Port       int    `json:"port,omitempty"`
+	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}
+	Weight     int    `json:"weight,omitempty"`
+}
+
+func (dst *DNSRecord) UnmarshalJSON(b []byte) error {
+	type Alias DNSRecord
+	aux := &struct {
+		Port   emptyStringInt `json:"port"`
+		Ttl    emptyStringInt `json:"ttl"`
+		Weight emptyStringInt `json:"weight"`
+
+		*Alias
+	}{
+		Alias: (*Alias)(dst),
+	}
+
+	err := json.Unmarshal(b, &aux)
+	if err != nil {
+		return fmt.Errorf("unable to unmarshal alias: %w", err)
+	}
+	dst.Port = int(aux.Port)
+	dst.Ttl = int(aux.Ttl)
+	dst.Weight = int(aux.Weight)
+
+	return nil
+}
+
 func (c *Client) ListDNSRecord(ctx context.Context, site string) ([]DNSRecord, error) {
 	var respBody []DNSRecord