dns: refactor record formatting and sort show results #28
@@ -6,6 +6,7 @@ import (
|
|||||||
"io"
|
"io"
|
||||||
"net/netip"
|
"net/netip"
|
||||||
"sort"
|
"sort"
|
||||||
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"darvaza.org/core"
|
"darvaza.org/core"
|
||||||
@@ -38,6 +39,53 @@ func SortAddrRecords(s []AddrRecord) []AddrRecord {
|
|||||||
return s
|
return s
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// SortRecords sorts a slice of [libdns.Record], by Name, Type and Value
|
||||||
|
func SortRecords(s []libdns.Record) []libdns.Record {
|
||||||
|
sort.Slice(s, func(i, j int) bool {
|
||||||
|
return lessRecord(s[i], s[j])
|
||||||
|
})
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
func lessRecord(a, b libdns.Record) bool {
|
||||||
|
aName := strings.ToLower(a.Name)
|
||||||
|
bName := strings.ToLower(b.Name)
|
||||||
|
|
||||||
|
switch {
|
||||||
|
case aName < bName:
|
||||||
|
karasz marked this conversation as resolved
Outdated
|
|||||||
|
return true
|
||||||
|
case aName > bName:
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
aType := strings.ToUpper(a.Type)
|
||||||
|
bType := strings.ToUpper(b.Type)
|
||||||
|
|
||||||
|
switch {
|
||||||
|
case aType < bType:
|
||||||
|
return true
|
||||||
|
case aType > bType:
|
||||||
|
return false
|
||||||
|
case aType == "A", aType == "AAAA":
|
||||||
|
// IP Addresses
|
||||||
|
var aa, ba netip.Addr
|
||||||
|
|
||||||
|
switch {
|
||||||
|
case aa.UnmarshalText([]byte(a.Value)) != nil:
|
||||||
|
// bad address on a
|
||||||
|
return true
|
||||||
|
case ba.UnmarshalText([]byte(b.Value)) != nil:
|
||||||
|
// bad address on b
|
||||||
|
return false
|
||||||
|
default:
|
||||||
|
return aa.Less(ba)
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
// text
|
||||||
|
return a.Value < b.Value
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// SortRegions sorts regions. first by length those 3-character
|
// SortRegions sorts regions. first by length those 3-character
|
||||||
// or shorter, and then by length. It's mostly aimed at
|
// or shorter, and then by length. It's mostly aimed at
|
||||||
// supporting ISO-3166 order
|
// supporting ISO-3166 order
|
||||||
|
|||||||
+37
-8
@@ -1,11 +1,15 @@
|
|||||||
package dns
|
package dns
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"darvaza.org/core"
|
"darvaza.org/core"
|
||||||
|
"github.com/libdns/libdns"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Show shows current DNS entries
|
// Show shows current DNS entries
|
||||||
@@ -15,15 +19,40 @@ func (mgr *Manager) Show(ctx context.Context, names ...string) error {
|
|||||||
return core.Wrap(err, "GetRecords")
|
return core.Wrap(err, "GetRecords")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
SortRecords(recs)
|
||||||
|
return writeRecords(recs, os.Stdout)
|
||||||
|
}
|
||||||
|
|
||||||
|
func writeRecords(recs []libdns.Record, w io.Writer) error {
|
||||||
|
var buf bytes.Buffer
|
||||||
|
|
||||||
for _, rr := range recs {
|
for _, rr := range recs {
|
||||||
_, _ = fmt.Printf("%s\t%v\tIN\t%s\t%s\t; %s\n",
|
_ = fmtRecord(&buf, rr)
|
||||||
rr.Name,
|
_, _ = buf.WriteRune('\n')
|
||||||
int(rr.TTL/time.Second),
|
}
|
||||||
rr.Type,
|
_, _ = fmt.Fprintf(&buf, "; %v records\n", len(recs))
|
||||||
rr.Value,
|
|
||||||
rr.ID)
|
_, err := buf.WriteTo(w)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func fmtRecord(w io.Writer, rr libdns.Record) error {
|
||||||
|
ttl := int(rr.TTL / time.Second)
|
||||||
|
if ttl < 1 {
|
||||||
|
ttl = 1
|
||||||
}
|
}
|
||||||
|
|
||||||
_, _ = fmt.Printf("; %v records\n", len(recs))
|
_, err := fmt.Fprintf(w, "%s\t%v\tIN\t%s\t%s",
|
||||||
return nil
|
rr.Name,
|
||||||
|
ttl,
|
||||||
|
rr.Type,
|
||||||
|
rr.Value)
|
||||||
|
|
||||||
|
if err == nil {
|
||||||
|
if rr.ID != "" {
|
||||||
|
_, err = fmt.Fprintf(w, "\t; %s", rr.ID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user
I think here you might want to optimize and compare
strings.ToLower(a.Name)tostrings.ToLower(b.Name)