cmd/hello: add VR support at /vrXe/hello-vr

Signed-off-by: Christine Dodrill <xe@tailscale.com>

1 files changed, 73 insertions, 8 deletions

diff --git a/cmd/hello/hello.go b/cmd/hello/hello.go
index c31ccb703..6d31c2400 100644
--- a/cmd/hello/hello.go
+++ b/cmd/hello/hello.go
@@ -43,10 +43,12 @@ func main() {
 		return
 	}
 	if !devMode() {
-		tmpl = template.Must(template.New("home").Parse(slurpHTML()))
+		tmpl = template.Must(template.New("home").Parse(slurpHTML("hello.tmpl.html")))
+		vrTmpl = template.Must(template.New("vr").Parse(slurpHTML("hellovr.tmpl.html")))
 	}
 
 	http.HandleFunc("/", root)
+	http.HandleFunc("/vr", vr)
 	log.Printf("Starting hello server.")
 
 	errc := make(chan error, 1)
@@ -69,8 +71,8 @@ func main() {
 	log.Fatal(<-errc)
 }
 
-func slurpHTML() string {
-	slurp, err := ioutil.ReadFile("hello.tmpl.html")
+func slurpHTML(name string) string {
+	slurp, err := ioutil.ReadFile(name)
 	if err != nil {
 		log.Fatal(err)
 	}
@@ -79,14 +81,20 @@ func slurpHTML() string {
 
 func devMode() bool { return *httpsAddr == "" && *httpAddr != "" }
 
-func getTmpl() (*template.Template, error) {
+func getTmpl(name string) (*template.Template, error) {
 	if devMode() {
-		return template.New("home").Parse(slurpHTML())
+		return template.New(name).Parse(slurpHTML(name))
+	}
+	switch name {
+	case "hellovr.tmpl.html":
+		return vrTmpl, nil
+	default:
+		return tmpl, nil
 	}
-	return tmpl, nil
 }
 
 var tmpl *template.Template // not used in dev mode, initialized by main after flag parse
+var vrTmpl *template.Template
 
 type tmplData struct {
 	DisplayName   string // "Foo Barberson"
@@ -106,7 +114,64 @@ func root(w http.ResponseWriter, r *http.Request) {
 		http.Redirect(w, r, "https://"+host, http.StatusFound)
 		return
 	}
-	if r.RequestURI != "/" {
+	if r.RequestURI != "/" && r.RequestURI != "/vr" {
+		http.Redirect(w, r, "/", http.StatusFound)
+		return
+	}
+	ip, _, err := net.SplitHostPort(r.RemoteAddr)
+	if err != nil {
+		http.Error(w, "no remote addr", 500)
+		return
+	}
+	tmpl, err := getTmpl("hello.tmpl.html")
+	if err != nil {
+		w.Header().Set("Content-Type", "text/plain")
+		http.Error(w, "template error: "+err.Error(), 500)
+		return
+	}
+
+	who, err := whoIs(ip)
+	var data tmplData
+	if err != nil {
+		if devMode() {
+			log.Printf("warning: using fake data in dev mode due to whois lookup error: %v", err)
+			data = tmplData{
+				DisplayName:   "Taily Scalerson",
+				LoginName:     "taily@scaler.son",
+				ProfilePicURL: "https://placekitten.com/200/200",
+				MachineName:   "scaled",
+				MachineOS:     "Linux",
+				IP:            "100.1.2.3",
+			}
+		} else {
+			log.Printf("whois(%q) error: %v", ip, err)
+			http.Error(w, "Your Tailscale works, but we failed to look you up.", 500)
+			return
+		}
+	} else {
+		data = tmplData{
+			DisplayName:   who.UserProfile.DisplayName,
+			LoginName:     who.UserProfile.LoginName,
+			ProfilePicURL: who.UserProfile.ProfilePicURL,
+			MachineName:   firstLabel(who.Node.ComputedName),
+			MachineOS:     who.Node.Hostinfo.OS,
+			IP:            ip,
+		}
+	}
+	w.Header().Set("Content-Type", "text/html; charset=utf-8")
+	tmpl.Execute(w, data)
+}
+
+func vr(w http.ResponseWriter, r *http.Request) {
+	if r.TLS == nil && *httpsAddr != "" {
+		host := r.Host
+		if strings.Contains(r.Host, "100.101.102.103") {
+			host = "hello.ipn.dev"
+		}
+		http.Redirect(w, r, "https://"+host, http.StatusFound)
+		return
+	}
+	if r.RequestURI != "/" && r.RequestURI != "/vr" {
 		http.Redirect(w, r, "/", http.StatusFound)
 		return
 	}
@@ -115,7 +180,7 @@ func root(w http.ResponseWriter, r *http.Request) {
 		http.Error(w, "no remote addr", 500)
 		return
 	}
-	tmpl, err := getTmpl()
+	tmpl, err := getTmpl("hellovr.tmpl.html")
 	if err != nil {
 		w.Header().Set("Content-Type", "text/plain")
 		http.Error(w, "template error: "+err.Error(), 500)