From 0b1e3055d5a2a0a4b2c1e99e95ccacc05c06a322 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20B=C3=B6hm?= <philipp-boehm@live.de>
Date: Tue, 15 Jul 2014 19:20:11 +0200
Subject: [PATCH] Added settings through environment vars

---
 ddns.go              | 38 +++++++++++++++++++++++++++++++++-----
 templates/index.html |  2 +-
 2 files changed, 34 insertions(+), 6 deletions(-)

diff --git a/ddns.go b/ddns.go
index def7bd8..028a823 100644
--- a/ddns.go
+++ b/ddns.go
@@ -13,6 +13,8 @@ import (
 	"strings"
 )
 
+var DdnsDomain string
+
 func HandleErr(err error) {
 	if err != nil {
 		log.Fatal(err)
@@ -74,9 +76,17 @@ func RunWebService() {
 	r.HTMLTemplates = html
 
 	r.GET("/", func(g *gin.Context) {
-		g.HTML(200, "index.html", nil)
+        g.HTML(200, "index.html", gin.H{ "domain": DdnsDomain })
 	})
 
+	r.GET("/available/:hostname", func(c *gin.Context) {
+		hostname := c.Params.ByName("hostname")
+
+        c.JSON(200, gin.H{
+            "available": ! conn.HostExist(hostname),
+        })
+    })
+
 	r.GET("/new/:hostname", func(c *gin.Context) {
 		hostname := c.Params.ByName("hostname")
 
@@ -90,9 +100,11 @@ func RunWebService() {
 
 		conn.SaveHost(host)
 
-		c.String(200, fmt.Sprintf(
-			"Go to /update/%s/%s for updating your IP address",
-			host.Hostname, host.Token))
+        c.JSON(200, gin.H{
+            "hostname": host.Hostname,
+            "token": host.Token,
+            "update_link": fmt.Sprintf("/update/%s/%s", host.Hostname, host.Token),
+        })
 	})
 
 	r.GET("/update/:hostname/:token", func(c *gin.Context) {
@@ -115,7 +127,7 @@ func RunWebService() {
 
 		ip, err := GetRemoteAddr(c.Req)
 		if err != nil {
-			c.String(500, "You sender IP address is not in the right format")
+			c.String(500, "Your sender IP address is not in the right format")
 		}
 
 		host.Ip = ip
@@ -141,12 +153,28 @@ func GetRemoteAddr(req *http.Request) (string, error) {
 	}
 }
 
+func ExtractConfigVariables() {
+
+    // get the domain in the right format
+    DdnsDomain = os.Getenv("DDNS_DOMAIN")
+    if DdnsDomain == "" {
+        log.Fatal(
+            "You have to set your Subdomain through the DDNS_DOMAIN env variable")
+    }
+    if ! strings.HasPrefix(DdnsDomain, ".") {
+        DdnsDomain = "." + DdnsDomain
+    }
+
+}
+
 func main() {
 
 	if len(os.Args) < 2 {
 		usage()
 	}
 
+	ExtractConfigVariables()
+
 	cmd := os.Args[1]
 
 	switch cmd {
diff --git a/templates/index.html b/templates/index.html
index 9d9e2a0..080c9f8 100644
--- a/templates/index.html
+++ b/templates/index.html
@@ -124,7 +124,7 @@
                     <div class="form-group">
                         <div class="input-group">
                             <input class="form-control input-lg" type="text" placeholder="my-own-hostname">
-                            <div class="input-group-addon input-lg">.d.pboehm.de</div>
+                            <div class="input-group-addon input-lg">{{.domain}}</div>
                         </div>
                     </div>
                 </form>