reCoreD-UI/models/domain.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
}