Refactor dns part of app to it's own package
dns package changes dns settings through a client, a client is able to: • GetRecords to get a list of records of a type and host. For example, GetRecords can get all "AAAA" records for host "domain.example.com" • CreateRecord will add a dns entry • DeleteRecord will delete a record on dnsimple Update main package to use dns package • main package will now create a record and remove all existing records when updating • Add special case to do nothing if a dns name has only one value Create utils package • Move splitHost into a utils packagemaster
Buddy Sandidge
8 years ago
parent
59b259e55f
commit
f05b4b4278
@ -0,0 +1,115 @@
|
||||
package dns
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
|
||||
"git.xbudex.com/buddy/update-dns/utils"
|
||||
"github.com/dnsimple/dnsimple-go/dnsimple"
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
// Client holds metadata for the dns api client
|
||||
type Client struct {
|
||||
client *dnsimple.Client
|
||||
accountID string
|
||||
}
|
||||
|
||||
// Record is the type used for a Zone record
|
||||
type Record dnsimple.ZoneRecord
|
||||
|
||||
// New returns an implementation of a DNS interface
|
||||
func New(oauthToken string) *Client {
|
||||
credentials := dnsimple.NewOauthTokenCredentials(oauthToken)
|
||||
client := dnsimple.NewClient(credentials)
|
||||
return &Client{
|
||||
client: client,
|
||||
}
|
||||
}
|
||||
|
||||
// GetRecords returns all record for a host
|
||||
func (d *Client) GetRecords(host, kind string) ([]Record, error) {
|
||||
accountID, err := d.getAccountID()
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "unable to get account id")
|
||||
}
|
||||
|
||||
subdomain, domain, err := utils.SplitHost(host)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "unable to parse host")
|
||||
}
|
||||
|
||||
records, err := d.client.Zones.ListRecords(accountID, domain, &dnsimple.ZoneRecordListOptions{})
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "could not get list of records")
|
||||
}
|
||||
|
||||
ret := []Record{}
|
||||
for _, record := range records.Data {
|
||||
if record.ZoneID == domain && record.Name == subdomain && record.Type == kind {
|
||||
ret = append(ret, Record(record))
|
||||
}
|
||||
}
|
||||
return ret, nil
|
||||
}
|
||||
|
||||
// CreateRecord adds a record to DNS
|
||||
func (d *Client) CreateRecord(host string, record Record) error {
|
||||
accountID, err := d.getAccountID()
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "unable to get account id")
|
||||
}
|
||||
|
||||
subdomain, domain, err := utils.SplitHost(host)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "unable to parse host")
|
||||
}
|
||||
|
||||
newRecord := dnsimple.ZoneRecord(record)
|
||||
newRecord.Name = subdomain
|
||||
if _, err := d.client.Zones.CreateRecord(accountID, domain, newRecord); err != nil {
|
||||
return errors.Wrap(err, "could not create record")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// UpdateRecord returns all record for a host
|
||||
func (d *Client) UpdateRecord(record Record) error {
|
||||
accountID, err := d.getAccountID()
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "unable to get account id")
|
||||
}
|
||||
|
||||
if _, err := d.client.Zones.UpdateRecord(accountID, record.ZoneID, record.ID, dnsimple.ZoneRecord(record)); err != nil {
|
||||
return errors.Wrap(err, "could not update record")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// DeleteRecord returns all record for a host
|
||||
func (d *Client) DeleteRecord(record Record) error {
|
||||
accountID, err := d.getAccountID()
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "unable to get account id")
|
||||
}
|
||||
|
||||
if _, err := d.client.Zones.DeleteRecord(accountID, record.ZoneID, record.ID); err != nil {
|
||||
return errors.Wrap(err, "could not delete record")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *Client) getAccountID() (string, error) {
|
||||
if d.accountID != "" {
|
||||
return d.accountID, nil
|
||||
}
|
||||
|
||||
whoamiResponse, err := d.client.Identity.Whoami()
|
||||
if err != nil {
|
||||
return "", errors.Wrap(err, "could not get account information from dnsimple")
|
||||
}
|
||||
if whoamiResponse.Data.Account == nil {
|
||||
return "", errors.New("could not get account information")
|
||||
}
|
||||
d.accountID = strconv.Itoa(whoamiResponse.Data.Account.ID)
|
||||
return d.accountID, nil
|
||||
}
|
||||
@ -0,0 +1,25 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
// SplitHost returns hostname into domain and subdomain strings.
|
||||
// For example:
|
||||
// "some.subdomain.example.com" → ("some.subdomain", "example.com", nil)
|
||||
// "example.com" → ("", "example.com", nil)
|
||||
func SplitHost(host string) (subdomain string, domain string, err error) {
|
||||
subdomains := strings.Split(host, ".")
|
||||
|
||||
if len(subdomains) >= 3 {
|
||||
domain = strings.Join(subdomains[len(subdomains)-2:], ".")
|
||||
subdomain = strings.Join(subdomains[:len(subdomains)-2], ".")
|
||||
} else if len(subdomains) == 2 {
|
||||
domain = strings.Join(subdomains, ".")
|
||||
} else {
|
||||
err = errors.New("invalid domain")
|
||||
}
|
||||
return subdomain, domain, err
|
||||
}
|
||||
Loading…
Reference in New Issue