From 75339ffba160432e705e9f6a063e10c0c6bb9d3e Mon Sep 17 00:00:00 2001
From: Amir Raminfar <findamir@gmail.com>
Date: Sun, 25 Nov 2018 12:10:46 -0800
Subject: [PATCH] Docker hostname added

---
 assets/index.html |   2 +-
 main.go           | 221 +++++++++++++++++++++++-----------------------
 2 files changed, 113 insertions(+), 110 deletions(-)
diff --git a/assets/index.html b/assets/index.html
index fab3b6bd..171f3895 100644
--- a/assets/index.html
+++ b/assets/index.html
@@ -3,7 +3,7 @@
   <head>
     <meta charset="utf-8" />
     <meta name="viewport" content="width=device-width, initial-scale=1" />
-    <title>Dozzle</title>
+    <title>{{ .Hostname }} - Dozzle</title>
     <link href="https://fonts.googleapis.com/css?family=Roboto|Roboto+Mono|Gafata" rel="stylesheet" />
     <link href="styles.scss" rel="stylesheet" />
     <script>
diff --git a/main.go b/main.go
index f769a417..ad966e9e 100644
--- a/main.go
+++ b/main.go
@@ -1,143 +1,146 @@
 package main
 
 import (
-	"context"
-	"encoding/binary"
-	"encoding/json"
-	"fmt"
-	"html/template"
-	"log"
-	"net/http"
-	"strings"
+    "context"
+    "encoding/binary"
+    "encoding/json"
+    "fmt"
+    "html/template"
+    "log"
+    "net/http"
+    "os"
+    "strings"
 
-	"github.com/docker/docker/api/types"
-	"github.com/docker/docker/client"
-	"github.com/gobuffalo/packr"
-	"github.com/gorilla/mux"
-	"github.com/gorilla/websocket"
-	flag "github.com/spf13/pflag"
+    "github.com/docker/docker/api/types"
+    "github.com/docker/docker/client"
+    "github.com/gobuffalo/packr"
+    "github.com/gorilla/mux"
+    "github.com/gorilla/websocket"
+    flag "github.com/spf13/pflag"
 )
 
 var (
-	cli      *client.Client
-	addr     = ""
-	ssl      = false
-	base     = "/"
-	upgrader = websocket.Upgrader{}
-	version  = "dev"
-	commit   = "none"
-	date     = "unknown"
+    cli      *client.Client
+    addr     = ""
+    ssl      = false
+    base     = "/"
+    upgrader = websocket.Upgrader{}
+    version  = "dev"
+    commit   = "none"
+    date     = "unknown"
 )
 
 func init() {
-	flag.StringVar(&addr, "addr", ":8080", "http service address")
-	flag.StringVar(&base, "base", "/", "base address of the application to mount")
-	flag.BoolVarP(&ssl, "ssl", "s", false, "Uses websockets over ssl if enabled")
+    flag.StringVar(&addr, "addr", ":8080", "http service address")
+    flag.StringVar(&base, "base", "/", "base address of the application to mount")
+    flag.BoolVarP(&ssl, "ssl", "s", false, "Uses websockets over ssl if enabled")
 
-	var err error
-	cli, err = client.NewClientWithOpts(client.FromEnv)
-	if err != nil {
-		log.Fatal(err)
-	}
-	flag.Parse()
+    var err error
+    cli, err = client.NewClientWithOpts(client.FromEnv)
+    if err != nil {
+        log.Fatal(err)
+    }
+    flag.Parse()
 }
 
 func main() {
-	r := mux.NewRouter()
+    r := mux.NewRouter()
 
-	if base != "/" {
-		r.HandleFunc(base, http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
-			http.Redirect(w, req, base+"/", http.StatusMovedPermanently)
-		}))
-	}
+    if base != "/" {
+        r.HandleFunc(base, http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
+            http.Redirect(w, req, base+"/", http.StatusMovedPermanently)
+        }))
+    }
 
-	s := r.PathPrefix(base).Subrouter()
-	box := packr.NewBox("./static")
+    s := r.PathPrefix(base).Subrouter()
+    box := packr.NewBox("./static")
 
-	s.HandleFunc("/api/containers.json", listContainers)
-	s.HandleFunc("/api/logs", logs)
-	s.HandleFunc("/version", versionHandler)
-	s.PathPrefix("/").Handler(http.StripPrefix(base, http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
-		fileServer := http.FileServer(box)
-		if box.Has(req.URL.Path) && req.URL.Path != "" && req.URL.Path != "/" {
-			fileServer.ServeHTTP(w, req)
-		} else {
-			handleIndex(box, w)
-		}
-	})))
+    s.HandleFunc("/api/containers.json", listContainers)
+    s.HandleFunc("/api/logs", logs)
+    s.HandleFunc("/version", versionHandler)
+    s.PathPrefix("/").Handler(http.StripPrefix(base, http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
+        fileServer := http.FileServer(box)
+        if box.Has(req.URL.Path) && req.URL.Path != "" && req.URL.Path != "/" {
+            fileServer.ServeHTTP(w, req)
+        } else {
+            handleIndex(box, w)
+        }
+    })))
 
-	log.Fatal(http.ListenAndServe(addr, r))
+    log.Fatal(http.ListenAndServe(addr, r))
 }
 
 func versionHandler(w http.ResponseWriter, r *http.Request) {
-	fmt.Fprintln(w, version)
-	fmt.Fprintln(w, commit)
-	fmt.Fprintln(w, date)
+    fmt.Fprintln(w, version)
+    fmt.Fprintln(w, commit)
+    fmt.Fprintln(w, date)
 }
 
 func listContainers(w http.ResponseWriter, r *http.Request) {
-	containers, err := cli.ContainerList(context.Background(), types.ContainerListOptions{})
-	if err != nil {
-		log.Fatal(err)
-	}
-	json.NewEncoder(w).Encode(containers)
+    containers, err := cli.ContainerList(context.Background(), types.ContainerListOptions{})
+    if err != nil {
+        log.Fatal(err)
+    }
+    json.NewEncoder(w).Encode(containers)
 }
 
 func handleIndex(box packr.Box, w http.ResponseWriter) {
-	text, _ := box.FindString("index.html")
-	text = strings.Replace(text, "__BASE__", "{{ .Base }}", -1)
-	tmpl, err := template.New("index.html").Parse(text)
-	if err != nil {
-		panic(err)
-	}
+    text, _ := box.FindString("index.html")
+    text = strings.Replace(text, "__BASE__", "{{ .Base }}", -1)
+    tmpl, err := template.New("index.html").Parse(text)
+    if err != nil {
+        panic(err)
+    }
 
-	path := ""
-	if base != "/" {
-		path = base
-	}
-	data := struct {
-		Base string
-		SSL  bool
-	}{path, ssl}
-	err = tmpl.Execute(w, data)
-	if err != nil {
-		panic(err)
-	}
+    path := ""
+    if base != "/" {
+        path = base
+    }
+    hostname, _ := os.Hostname()
+    data := struct {
+        Base     string
+        SSL      bool
+        Hostname string
+    }{path, ssl, hostname}
+    err = tmpl.Execute(w, data)
+    if err != nil {
+        panic(err)
+    }
 }
 
 func logs(w http.ResponseWriter, r *http.Request) {
-	id := r.URL.Query().Get("id")
-	c, err := upgrader.Upgrade(w, r, nil)
-	if err != nil {
-		log.Fatal(err)
-		return
-	}
-	defer c.Close()
+    id := r.URL.Query().Get("id")
+    c, err := upgrader.Upgrade(w, r, nil)
+    if err != nil {
+        log.Fatal(err)
+        return
+    }
+    defer c.Close()
 
-	options := types.ContainerLogsOptions{ShowStdout: true, ShowStderr: true, Follow: true, Tail: "300", Timestamps: true}
-	reader, err := cli.ContainerLogs(context.Background(), id, options)
-	defer reader.Close()
-	if err != nil {
-		log.Fatal(err)
-	}
+    options := types.ContainerLogsOptions{ShowStdout: true, ShowStderr: true, Follow: true, Tail: "300", Timestamps: true}
+    reader, err := cli.ContainerLogs(context.Background(), id, options)
+    defer reader.Close()
+    if err != nil {
+        log.Fatal(err)
+    }
 
-	hdr := make([]byte, 8)
-	content := make([]byte, 1024, 1024*1024)
-	for {
-		_, err := reader.Read(hdr)
-		if err != nil {
-			log.Panicln(err)
-		}
-		count := binary.BigEndian.Uint32(hdr[4:])
-		n, err := reader.Read(content[:count])
-		if err != nil {
-			log.Println(err)
-			break
-		}
-		err = c.WriteMessage(websocket.TextMessage, content[:n])
-		if err != nil {
-			log.Println(err)
-			break
-		}
-	}
+    hdr := make([]byte, 8)
+    content := make([]byte, 1024, 1024*1024)
+    for {
+        _, err := reader.Read(hdr)
+        if err != nil {
+            log.Panicln(err)
+        }
+        count := binary.BigEndian.Uint32(hdr[4:])
+        n, err := reader.Read(content[:count])
+        if err != nil {
+            log.Println(err)
+            break
+        }
+        err = c.WriteMessage(websocket.TextMessage, content[:n])
+        if err != nil {
+            log.Println(err)
+            break
+        }
+    }
 }
