61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
package models
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
type Domain struct {
|
|
ID int `gorm:"primaryKey" json:"id"`
|
|
DomainName string `gorm:"unique,not null,size:255" json:"domain_name"`
|
|
|
|
//SOA Info
|
|
MainDNS string `gorm:"not null;size:255" json:"main_dns"`
|
|
AdminEmail string `gorm:"not null;size:255" json:"admin_email"`
|
|
SerialNumber int64 `gorm:"not null;default:1" json:"serial_number"`
|
|
RefreshInterval uint32 `gorm:"not null;size:255,default:\"86400\"" json:"refresh_interval"`
|
|
RetryInterval uint32 `gorm:"not null;size:255,default:\"7200\"" json:"retry_interval"`
|
|
ExpiryPeriod uint32 `gorm:"not null;size:255,default:\"3600000\"" json:"expiry_period"`
|
|
NegativeTtl uint32 `gorm:"not null;size:255,default:\"86400\"" json:"negative_ttl"`
|
|
}
|
|
|
|
func (d *Domain) EmailSOAForamt() string {
|
|
s := strings.Split(d.AdminEmail, "@")
|
|
s[0] = strings.Replace(s[0], ".", "\\", -1)
|
|
if !strings.HasSuffix(s[1], ".") {
|
|
s[1] = fmt.Sprintf("%s.", s[1])
|
|
}
|
|
return strings.Join(s, ".")
|
|
}
|
|
|
|
func (d *Domain) WithDotEnd() string {
|
|
if strings.HasSuffix(d.DomainName, ".") {
|
|
return d.DomainName
|
|
} else {
|
|
return fmt.Sprintf("%s.", d.DomainName)
|
|
}
|
|
}
|
|
|
|
func (d *Domain) GenerateSOA() SOARecord {
|
|
var ns string
|
|
if !strings.HasSuffix(d.MainDNS, ".") {
|
|
ns = fmt.Sprintf("%s.", d.MainDNS)
|
|
} else {
|
|
ns = d.MainDNS
|
|
}
|
|
r := SOARecord{}
|
|
r.Ns = ns
|
|
r.MBox = d.EmailSOAForamt()
|
|
r.Refresh = d.RefreshInterval
|
|
r.Retry = d.RetryInterval
|
|
r.Expire = d.ExpiryPeriod
|
|
r.MinTtl = d.NegativeTtl
|
|
return r
|
|
}
|
|
|
|
type IDomain interface {
|
|
EmailSOAForamt() string
|
|
WithDotEnd() string
|
|
GenerateSOA() SOARecord
|
|
}
|