fixa aumento no serial do soa
This commit is contained in:
@@ -13,6 +13,8 @@ import (
|
||||
"time"
|
||||
)
|
||||
|
||||
const soaEditAPIIncrease = "INCREASE"
|
||||
|
||||
type HTTPClient interface {
|
||||
Do(*http.Request) (*http.Response, error)
|
||||
}
|
||||
@@ -26,25 +28,20 @@ type Client struct {
|
||||
|
||||
type Server struct {
|
||||
ID string `json:"id"`
|
||||
Type string `json:"type"`
|
||||
DaemonType string `json:"daemon_type"`
|
||||
Version string `json:"version"`
|
||||
URL string `json:"url"`
|
||||
ConfigURL string `json:"config_url"`
|
||||
ZonesURL string `json:"zones_url"`
|
||||
}
|
||||
|
||||
type Zone struct {
|
||||
ID string `json:"id,omitempty"`
|
||||
Name string `json:"name"`
|
||||
Type string `json:"type,omitempty"`
|
||||
Kind string `json:"kind,omitempty"`
|
||||
Serial uint64 `json:"serial,omitempty"`
|
||||
EditedSerial uint64 `json:"edited_serial,omitempty"`
|
||||
SOAEditAPI string `json:"soa_edit_api,omitempty"`
|
||||
Nameservers []string `json:"nameservers,omitempty"`
|
||||
Masters []string `json:"masters,omitempty"`
|
||||
RRSets []RRSet `json:"rrsets,omitempty"`
|
||||
ID string `json:"id,omitempty"`
|
||||
Name string `json:"name,omitempty"`
|
||||
Type string `json:"type,omitempty"`
|
||||
Kind string `json:"kind,omitempty"`
|
||||
Serial uint64 `json:"serial,omitempty"`
|
||||
SOAEditAPI string `json:"soa_edit_api,omitempty"`
|
||||
Nameservers []string `json:"nameservers,omitempty"`
|
||||
Masters []string `json:"masters,omitempty"`
|
||||
RRSets []RRSet `json:"rrsets,omitempty"`
|
||||
}
|
||||
|
||||
type RRSet struct {
|
||||
@@ -100,9 +97,50 @@ func (c *Client) ListZones(ctx context.Context) ([]Zone, error) {
|
||||
}
|
||||
|
||||
func (c *Client) CreateZone(ctx context.Context, zone Zone) (Zone, error) {
|
||||
zone.SOAEditAPI = soaEditAPIIncrease
|
||||
var created Zone
|
||||
err := c.do(ctx, http.MethodPost, c.path("/zones"), zone, &created)
|
||||
return created, err
|
||||
if err := c.do(ctx, http.MethodPost, c.path("/zones"), zone, &created); err != nil {
|
||||
return Zone{}, err
|
||||
}
|
||||
|
||||
zoneID := created.ID
|
||||
if zoneID == "" {
|
||||
zoneID = created.Name
|
||||
}
|
||||
if zoneID == "" {
|
||||
zoneID = zone.Name
|
||||
}
|
||||
if err := c.SetZoneSOAEditAPI(ctx, zoneID); err != nil {
|
||||
return Zone{}, fmt.Errorf("set SOA-EDIT-API for new zone %s: %w", zoneID, err)
|
||||
}
|
||||
|
||||
return created, nil
|
||||
}
|
||||
|
||||
func (c *Client) EnsureAllZonesSOAEditAPI(ctx context.Context) error {
|
||||
zones, err := c.ListZones(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, zone := range zones {
|
||||
zoneID := zone.ID
|
||||
if zoneID == "" {
|
||||
zoneID = zone.Name
|
||||
}
|
||||
if zoneID == "" {
|
||||
return fmt.Errorf("zone without id or name cannot be updated")
|
||||
}
|
||||
if err := c.SetZoneSOAEditAPI(ctx, zoneID); err != nil {
|
||||
return fmt.Errorf("set SOA-EDIT-API for %s: %w", zoneID, err)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Client) SetZoneSOAEditAPI(ctx context.Context, zoneID string) error {
|
||||
body := Zone{SOAEditAPI: soaEditAPIIncrease}
|
||||
return c.do(ctx, http.MethodPut, c.path("/zones/"+url.PathEscape(zoneID)), body, nil)
|
||||
}
|
||||
|
||||
func (c *Client) DeleteZone(ctx context.Context, zoneID string) error {
|
||||
@@ -116,19 +154,17 @@ func (c *Client) GetZone(ctx context.Context, zoneID string) (Zone, error) {
|
||||
}
|
||||
|
||||
func (c *Client) CreateRRSet(ctx context.Context, zoneID string, rrset RRSet) error {
|
||||
var before *Zone
|
||||
if allowsMultipleRecords(rrset.Type) {
|
||||
zone, err := c.GetZone(ctx, zoneID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("read zone before merging records: %w", err)
|
||||
}
|
||||
before = &zone
|
||||
if existing, ok := findRRSet(zone, rrset.Name, rrset.Type); ok {
|
||||
rrset.Records = mergeRecords(existing.Records, rrset.Records)
|
||||
}
|
||||
}
|
||||
|
||||
return c.patchZoneWithSerialBump(ctx, zoneID, before, []changeRRSet{{
|
||||
return c.patchZone(ctx, zoneID, []changeRRSet{{
|
||||
Name: rrset.Name,
|
||||
Type: rrset.Type,
|
||||
TTL: rrset.TTL,
|
||||
@@ -175,7 +211,7 @@ func recordKey(record Record) string {
|
||||
}
|
||||
|
||||
func (c *Client) DeleteRRSet(ctx context.Context, zoneID, name, recordType string) error {
|
||||
return c.patchZoneWithSerialBump(ctx, zoneID, nil, []changeRRSet{{
|
||||
return c.patchZone(ctx, zoneID, []changeRRSet{{
|
||||
Name: name,
|
||||
Type: recordType,
|
||||
ChangeType: "DELETE",
|
||||
@@ -189,40 +225,6 @@ func (c *Client) patchZone(ctx context.Context, zoneID string, rrsets []changeRR
|
||||
return c.do(ctx, http.MethodPatch, c.path("/zones/"+url.PathEscape(zoneID)), body, nil)
|
||||
}
|
||||
|
||||
func (c *Client) patchZoneWithSerialBump(ctx context.Context, zoneID string, before *Zone, rrsets []changeRRSet) error {
|
||||
if before == nil {
|
||||
zone, err := c.GetZone(ctx, zoneID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("read zone before change: %w", err)
|
||||
}
|
||||
before = &zone
|
||||
}
|
||||
|
||||
if err := c.patchZone(ctx, zoneID, rrsets); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
after, err := c.GetZone(ctx, zoneID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("read zone after change: %w", err)
|
||||
}
|
||||
if serialIncreased(*before, after) {
|
||||
return nil
|
||||
}
|
||||
|
||||
soa, err := bumpedSOA(after, before.Serial+1)
|
||||
if err != nil {
|
||||
return fmt.Errorf("bump SOA serial: %w", err)
|
||||
}
|
||||
return c.patchZone(ctx, zoneID, []changeRRSet{{
|
||||
Name: soa.Name,
|
||||
Type: soa.Type,
|
||||
TTL: soa.TTL,
|
||||
ChangeType: "REPLACE",
|
||||
Records: soa.Records,
|
||||
}})
|
||||
}
|
||||
|
||||
func (c *Client) path(suffix string) string {
|
||||
return "/api/v1/servers/" + url.PathEscape(c.serverID) + suffix
|
||||
}
|
||||
@@ -275,53 +277,3 @@ func (c *Client) do(ctx context.Context, method, apiPath string, in any, out any
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func serialIncreased(before, after Zone) bool {
|
||||
if after.Serial > before.Serial {
|
||||
return true
|
||||
}
|
||||
if (before.EditedSerial != 0 || after.EditedSerial != 0) && after.EditedSerial > before.EditedSerial {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func bumpedSOA(zone Zone, minimum uint64) (RRSet, error) {
|
||||
for _, rrset := range zone.RRSets {
|
||||
if !strings.EqualFold(rrset.Type, "SOA") {
|
||||
continue
|
||||
}
|
||||
if len(rrset.Records) != 1 {
|
||||
return RRSet{}, fmt.Errorf("expected exactly one SOA record, got %d", len(rrset.Records))
|
||||
}
|
||||
|
||||
record := rrset.Records[0]
|
||||
fields := strings.Fields(record.Content)
|
||||
if len(fields) != 7 {
|
||||
return RRSet{}, fmt.Errorf("SOA record must have 7 fields")
|
||||
}
|
||||
|
||||
current, err := strconv.ParseUint(fields[2], 10, 32)
|
||||
if err != nil {
|
||||
return RRSet{}, fmt.Errorf("parse SOA serial: %w", err)
|
||||
}
|
||||
if current == 1<<32-1 {
|
||||
return RRSet{}, fmt.Errorf("SOA serial is already at maximum uint32 value")
|
||||
}
|
||||
|
||||
next := current + 1
|
||||
if next < minimum {
|
||||
next = minimum
|
||||
}
|
||||
if next > 1<<32-1 {
|
||||
return RRSet{}, fmt.Errorf("next SOA serial exceeds maximum uint32 value")
|
||||
}
|
||||
|
||||
fields[2] = strconv.FormatUint(next, 10)
|
||||
record.Content = strings.Join(fields, " ")
|
||||
rrset.Records[0] = record
|
||||
return rrset, nil
|
||||
}
|
||||
|
||||
return RRSet{}, fmt.Errorf("SOA record not found")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user