main.go

package main

import (
	"io/ioutil"
	"log"
	"net/http"
	"os"

	"github.com/gusga/ip-updater/domainer"
	"github.com/gusga/ip-updater/storage"
)

var IP_URL_SERVICE string
var DOMAIN string

func lookupRemoteIp() string {
	response, err := http.Get(IP_URL_SERVICE)
	if err != nil {
		log.Fatalf("Error ", err)
	}
	ip, err := ioutil.ReadAll(response.Body)
	defer response.Body.Close()
	if err != nil {
		log.Fatalf("Error parsing ip ", err)
	}
	return string(ip)
}
func init() {
	DOMAIN = os.Getenv("DOMAIN")
	IP_URL_SERVICE = os.Getenv("IP_URL_SERVICE")
	if DOMAIN == "" {
		log.Fatalln("DOMAIN env must be setted")
	}
	if IP_URL_SERVICE == "" {
		log.Fatalln("IP_URL_SERVICE env must be setted")
	}
	domainer.SetDomain(DOMAIN)
}

func main() {
	var domainData *domainer.DomainList
	ip := lookupRemoteIp()
	savedDomain, err := storage.GetDomainData(DOMAIN)
	defer storage.Close()
	if err != nil {
		log.Fatalln(err)
	}
	if savedDomain == "" {
		domainData, err = domainer.GetDomains()
		if err != nil {
			log.Fatalf("Could not retrieve domain records ", err)
		}
		
		domainData.SetIP(ip)

		err = domainData.Save()
			if err != nil {
			log.Fatalln(err)			
		}
		
	} else {
		domainData, err = domainer.DomainListFromDB(savedDomain)
		if err != nil {
			log.Fatalln(err)
		}

	}

	if domainData.CheckIP(ip) {
		log.Printf("The IP address has not changed %s\n", ip)
		return
	} else {
		err = domainData.UpdateDomainsIP(ip)
		if err != nil {
			log.Fatalln(err)
		}
	}
}