reCoreD-UI/models/domain.go

package models

import (
	"fmt"
	"strings"
)

// Domain domain data structure
type Domain struct {
	ID         uint   `gorm:"primaryKey" json:"id"`
	DomainName string `gorm:"unique,not null,size:255" json:"domain_name"`

	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:"type:uint;not null;default:86400" json:"refresh_interval"`
	RetryInterval   uint32 `gorm:"type:uint;not null;default:7200" json:"retry_interval"`
	ExpiryPeriod    uint32 `gorm:"type:uint;not null;default:3600000" json:"expiry_period"`
	NegativeTtl     uint32 `gorm:"type:uint;not null;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
}

func (d *Domain) GetValue() Domain {
	return *d
}

type IDomain interface {
	EmailSOAForamt() string
	WithDotEnd() string
	GenerateSOA() SOARecord
	GetValue() Domain
}