|
| 1 | +package rtype |
| 2 | + |
| 3 | +import ( |
| 4 | + "fmt" |
| 5 | + |
| 6 | + "github.com/StackExchange/dnscontrol/v4/models" |
| 7 | + "github.com/StackExchange/dnscontrol/v4/pkg/domaintags" |
| 8 | + "github.com/StackExchange/dnscontrol/v4/pkg/rtypecontrol" |
| 9 | + "github.com/miekg/dns" |
| 10 | +) |
| 11 | + |
| 12 | +func init() { |
| 13 | + rtypecontrol.Register(&DS{}) |
| 14 | +} |
| 15 | + |
| 16 | +// DS RR. |
| 17 | +type DS struct { |
| 18 | + dns.DS |
| 19 | +} |
| 20 | + |
| 21 | +// Name returns the DNS record type as a string. |
| 22 | +func (handle *DS) Name() string { |
| 23 | + return "DS" |
| 24 | +} |
| 25 | + |
| 26 | +// FromArgs fills in the RecordConfig from []any, which is typically from a parsed config file. |
| 27 | +func (handle *DS) FromArgs(dcn *domaintags.DomainNameVarieties, rec *models.RecordConfig, args []any) error { |
| 28 | + if err := rtypecontrol.PaveArgs(args[1:], "wbbs"); err != nil { |
| 29 | + return fmt.Errorf("ERROR: (%s) [DS(%q, %v)]: %w", |
| 30 | + rec.FilePos, |
| 31 | + rec.Name, rtypecontrol.StringifyQuoted(args[1:]), |
| 32 | + err) |
| 33 | + } |
| 34 | + fields := &dns.DS{ |
| 35 | + KeyTag: args[1].(uint16), |
| 36 | + Algorithm: args[2].(uint8), |
| 37 | + DigestType: args[3].(uint8), |
| 38 | + Digest: args[4].(string), |
| 39 | + } |
| 40 | + |
| 41 | + return handle.FromStruct(dcn, rec, args[0].(string), fields) |
| 42 | +} |
| 43 | + |
| 44 | +// FromStruct fills in the RecordConfig from a struct, typically from an API response. |
| 45 | +func (handle *DS) FromStruct(dcn *domaintags.DomainNameVarieties, rec *models.RecordConfig, name string, fields any) error { |
| 46 | + // Fields is of type "any" thus we must validate the type. It should be the "inner" type of .F, not the outer type, rtype.DS{}. |
| 47 | + ds, ok := fields.(*dns.DS) |
| 48 | + if !ok { |
| 49 | + return fmt.Errorf("fields is not *dns.DS, got %T", fields) |
| 50 | + } |
| 51 | + rec.F = &DS{*ds} |
| 52 | + |
| 53 | + rec.ZonefilePartial = rec.GetTargetRFC1035Quoted() |
| 54 | + rec.Comparable = rec.ZonefilePartial |
| 55 | + |
| 56 | + handle.CopyToLegacyFields(rec) |
| 57 | + return nil |
| 58 | +} |
| 59 | + |
| 60 | +// CopyToLegacyFields populates the legacy fields of the RecordConfig using the fields in .F. |
| 61 | +func (handle *DS) CopyToLegacyFields(rec *models.RecordConfig) { |
| 62 | + ds := rec.F.(*DS) |
| 63 | + _ = rec.SetTargetDS(ds.KeyTag, ds.Algorithm, ds.DigestType, ds.Digest) |
| 64 | +} |
| 65 | + |
| 66 | +// CopyFromLegacyFields uses the the legacy fields to populate .F |
| 67 | +func (handle *DS) CopyFromLegacyFields(rec *models.RecordConfig) { |
| 68 | + // Copy fields: |
| 69 | + rec.F = &DS{ |
| 70 | + dns.DS{ |
| 71 | + KeyTag: rec.DsKeyTag, |
| 72 | + Algorithm: rec.DsAlgorithm, |
| 73 | + DigestType: rec.DsDigestType, |
| 74 | + Digest: rec.DsDigest, |
| 75 | + }, |
| 76 | + } |
| 77 | + |
| 78 | + // Fix up ZonefilePartial and Comparable: |
| 79 | + rec.ZonefilePartial = rec.GetTargetRFC1035Quoted() |
| 80 | + rec.Comparable = rec.ZonefilePartial |
| 81 | +} |
0 commit comments