try to use referer when the path is not forced

igrigorik · Feb 27, 2016 · df8ebf9 · df8ebf9
1 parent bfec0d6
commit df8ebf9
Showing 1 changed file with 16 additions and 9 deletions.
diff --git a/ga-beacon/ga-beacon.go b/ga-beacon/ga-beacon.go
@@ -103,16 +103,23 @@ func handler(w http.ResponseWriter, r *http.Request) {
 
 	// /account -> account template
 	if len(params) == 1 {
-		templateParams := struct {
-			Account string
-		}{
-			Account: params[0],
-		}
-		if err := pageTemplate.ExecuteTemplate(w, "page.html", templateParams); err != nil {
-			http.Error(w, "could not show account page", 500)
-			c.Errorf("Cannot execute template: %v", err)
+		referer := strings.Replace(strings.Replace(r.Header.Get("Referer"), "http://", "", 1), "https://", "", 1);
+		if len(referer) != 0 {
+			params = strings.SplitN(strings.Trim(r.URL.Path, "/") + "/" + referer, "/", 2)
+		} else {
+			templateParams := struct {
+				Account string
+				Referer string
+			}{
+				Account: params[0],
+				Referer: r.Header.Get("Referer"),
+			}
+			if err := pageTemplate.ExecuteTemplate(w, "page.html", templateParams); err != nil {
+				http.Error(w, "could not show account page", 500)
+				c.Errorf("Cannot execute template: %v", err)
+			}
+			return
 		}
-		return
 	}
 
 	// /account/page -> GIF + log pageview to GA collector