mirror of
https://github.com/StackExchange/dnscontrol.git
synced 2024-09-20 14:56:20 +08:00
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package rwth
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"strings"
|
|
|
|
"github.com/StackExchange/dnscontrol/v4/models"
|
|
"github.com/StackExchange/dnscontrol/v4/pkg/prettyzone"
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
// Print the generateZoneFileHelper
|
|
func (api *rwthProvider) printRecConfig(rr models.RecordConfig) string {
|
|
// Similar to prettyzone
|
|
// Fake types are commented out.
|
|
prefix := ""
|
|
_, ok := dns.StringToType[rr.Type]
|
|
if !ok {
|
|
prefix = ";"
|
|
}
|
|
|
|
// ttl
|
|
ttl := ""
|
|
if rr.TTL != 172800 && rr.TTL != 0 {
|
|
ttl = fmt.Sprint(rr.TTL)
|
|
}
|
|
|
|
// type
|
|
typeStr := rr.Type
|
|
|
|
// the remaining line
|
|
target := rr.GetTargetCombined()
|
|
|
|
// comment
|
|
comment := ";"
|
|
|
|
return fmt.Sprintf("%s%s%s\n",
|
|
prefix, prettyzone.FormatLine([]int{10, 5, 2, 5, 0}, []string{rr.NameFQDN, ttl, "IN", typeStr, target}), comment)
|
|
}
|
|
|
|
// NewRR returns custom dns.NewRR with RWTH default TTL
|
|
func NewRR(s string) (dns.RR, error) {
|
|
if len(s) > 0 && s[len(s)-1] != '\n' { // We need a closing newline
|
|
return ReadRR(strings.NewReader(s + "\n"))
|
|
}
|
|
return ReadRR(strings.NewReader(s))
|
|
}
|
|
|
|
// ReadRR reads an RR from r.
|
|
func ReadRR(r io.Reader) (dns.RR, error) {
|
|
zp := dns.NewZoneParser(r, ".", "")
|
|
zp.SetDefaultTTL(172800)
|
|
zp.SetIncludeAllowed(true)
|
|
rr, _ := zp.Next()
|
|
return rr, zp.Err()
|
|
}
|