141 lines
2.6 KiB
Go
141 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
// Standard
|
|
"fmt"
|
|
"maps"
|
|
"slices"
|
|
"strings"
|
|
)
|
|
|
|
type DNSRecord struct {
|
|
ID string `json:".id"`
|
|
Address string
|
|
Disabled string
|
|
Dynamic string
|
|
Name string
|
|
TTL string
|
|
Type string
|
|
}
|
|
|
|
type DomainPart struct {
|
|
Subparts map[string]*DomainPart `json:",omitempty"`
|
|
}
|
|
|
|
type RecordsTree struct {
|
|
Record DNSRecord
|
|
Children []RecordsTree
|
|
}
|
|
|
|
func SortDNSRecord(a, b DNSRecord) int {
|
|
aComponents := strings.Split(a.Name, ".")
|
|
bComponents := strings.Split(b.Name, ".")
|
|
|
|
slices.Reverse(aComponents)
|
|
slices.Reverse(bComponents)
|
|
|
|
for i, aComp := range aComponents {
|
|
if i >= len(bComponents) {
|
|
return -1
|
|
}
|
|
|
|
bComp := bComponents[i]
|
|
if aComp > bComp {
|
|
return 1
|
|
}
|
|
|
|
if aComp < bComp {
|
|
return -1
|
|
}
|
|
|
|
}
|
|
|
|
return 0
|
|
}
|
|
|
|
func (r DNSRecord) Parts() int {
|
|
return len(strings.Split(r.Name, "."))
|
|
}
|
|
|
|
func (r DNSRecord) Part(numParts int) string {
|
|
splitName := strings.Split(r.Name, ".")
|
|
slices.Reverse(splitName)
|
|
|
|
parts := []string{}
|
|
for i := range numParts {
|
|
if i >= len(splitName) {
|
|
break
|
|
}
|
|
parts = append(parts, splitName[i])
|
|
}
|
|
|
|
slices.Reverse(parts)
|
|
return strings.Join(parts, ".")
|
|
}
|
|
|
|
func (r DNSRecord) NameReversed() string {
|
|
parts := strings.Split(r.Name, ".")
|
|
slices.Reverse(parts)
|
|
return strings.Join(parts, ".")
|
|
}
|
|
|
|
func BuildRecordsTree(records []DNSRecord) *DomainPart {
|
|
topPart := new(DomainPart)
|
|
topPart.Subparts = make(map[string]*DomainPart)
|
|
|
|
for _, record := range records {
|
|
curPart := topPart
|
|
|
|
currentDomainNameSplit := strings.Split(record.Name, ".")
|
|
slices.Reverse(currentDomainNameSplit)
|
|
|
|
for _, part := range currentDomainNameSplit {
|
|
if nextDomainPart, found := curPart.Subparts[part]; !found {
|
|
newPart := new(DomainPart)
|
|
newPart.Subparts = make(map[string]*DomainPart)
|
|
curPart.Subparts[strings.ToLower(part)] = newPart
|
|
curPart = newPart
|
|
} else {
|
|
curPart = nextDomainPart
|
|
}
|
|
}
|
|
}
|
|
|
|
return topPart
|
|
}
|
|
|
|
func (dp *DomainPart) ToHTML(parts []string) string {
|
|
html := ""
|
|
|
|
sortedParts := slices.Sorted(maps.Keys(dp.Subparts))
|
|
|
|
for _, part := range sortedParts {
|
|
subpart := dp.Subparts[part]
|
|
newParts := append(parts, part)
|
|
|
|
reversedParts := make([]string, len(newParts))
|
|
copy(reversedParts, newParts)
|
|
slices.Reverse(reversedParts)
|
|
fqdn := strings.Join(reversedParts, ".")
|
|
|
|
var subHTML string
|
|
if len(subpart.Subparts) == 0 {
|
|
html += fmt.Sprintf(`<div class="record" data-record="%s">%s</div>`, fqdn, fqdn)
|
|
} else {
|
|
subHTML = subpart.ToHTML(newParts)
|
|
html += fmt.Sprintf(`
|
|
<div class="top" data-top="%s">
|
|
<div class="fqdn">%s</div>
|
|
<div class="records">%s</div>
|
|
</div>
|
|
`,
|
|
fqdn,
|
|
fqdn,
|
|
subHTML,
|
|
)
|
|
}
|
|
|
|
}
|
|
|
|
return html
|
|
}
|