dnsimple-updater.go

package main

import (
	"flag"
	"log"

	"io/ioutil"
	"net/http"

	"time"

	"github.com/weppos/dnsimple-go/dnsimple"
)

var (
	token    = flag.String("token", "", "Domain token")
	entryID  = flag.Int("entry", 0, "ID of the entry")
	domain   = flag.String("domain", "", "Domain name")
	interval = flag.Duration("interval", 60*time.Minute, "Interval of update")
)

func main() {
	flag.Parse()
	checkFlags()

	for {
		api := dnsimple.NewAuthenticatedClient(dnsimple.NewDomainTokenCredentials(*token))
		ip, err := getIP()
		if err != nil {
			log.Printf("Could not get public IP: %s", err)
			continue
		}
		log.Printf("Updating IP to %s", ip)

		_, _, err = api.Domains.UpdateRecord(*domain, *entryID, dnsimple.Record{
			Content: ip,
		})
		if err != nil {
			log.Printf("Could not update DNS record: %s", err)
			continue
		}

		time.Sleep(*interval)
	}
}

func checkFlags() {
	if *token == "" {
		log.Fatalf("Must provide -token")
	}
	if *entryID == 0 {
		log.Fatalf("Must provide -entry")
	}
	if *domain == "" {
		log.Fatalf("Must provide -domain")
	}
}

func getIP() (string, error) {
	resp, err := http.Get("https://icanhazip.com")
	if err != nil {
		return "", err
	}
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return "", err
	}
	return string(body), err
}