From 6ac39ef687238def1a7175e52bba67d388a70564 Mon Sep 17 00:00:00 2001
From: nickago <ngonella@calpoly.edu>
Date: Tue, 25 Aug 2015 11:06:11 -0700
Subject: [PATCH] Added a link to error page that returns users to the root
 page

---
 api/context.go           | 21 +++++++++++++++++----
 api/templates/error.html |  1 +
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/api/context.go b/api/context.go
index 8babf85f2f..aaf304e2ca 100644
--- a/api/context.go
+++ b/api/context.go
@@ -4,14 +4,15 @@
 package api
 
 import (
-	l4g "code.google.com/p/log4go"
-	"github.com/mattermost/platform/model"
-	"github.com/mattermost/platform/store"
-	"github.com/mattermost/platform/utils"
 	"net"
 	"net/http"
 	"net/url"
 	"strings"
+
+	l4g "code.google.com/p/log4go"
+	"github.com/mattermost/platform/model"
+	"github.com/mattermost/platform/store"
+	"github.com/mattermost/platform/utils"
 )
 
 var sessionCache *utils.Cache = utils.NewLru(model.SESSION_CACHE_SIZE)
@@ -431,10 +432,22 @@ func IsPrivateIpAddress(ipAddress string) bool {
 }
 
 func RenderWebError(err *model.AppError, w http.ResponseWriter, r *http.Request) {
+
+	protocol := "http"
+	if utils.Cfg.ServiceSettings.UseSSL {
+		forwardProto := r.Header.Get(model.HEADER_FORWARDED_PROTO)
+		if forwardProto != "http" {
+			protocol = "https"
+		}
+	}
+
+	SiteURL := protocol + "://" + r.Host
+
 	m := make(map[string]string)
 	m["Message"] = err.Message
 	m["Details"] = err.DetailedError
 	m["SiteName"] = utils.Cfg.ServiceSettings.SiteName
+	m["SiteURL"] = SiteURL
 
 	w.WriteHeader(err.StatusCode)
 	ServerTemplates.ExecuteTemplate(w, "error.html", m)
diff --git a/api/templates/error.html b/api/templates/error.html
index f38bb81a1e..3474c9e1ea 100644
--- a/api/templates/error.html
+++ b/api/templates/error.html
@@ -14,6 +14,7 @@
 			<div class="error__icon"><i class="fa fa-exclamation-triangle"></i></div>
 			<h2>{{ .SiteName }} needs your help:</h2>
 			<p>{{.Message}}</p>
+			<a href="{{.SiteURL}}">Go back to team site</a>
 		</div>
 	</div>
 </body>