diff --git a/error.go b/error.go index 69e2023..fad75c7 100644 --- a/error.go +++ b/error.go @@ -3,22 +3,11 @@ package main import ( "bytes" "io" + "os" "text/template" "time" ) -var errPageRawTmpl = ` - - COW Proxy - -

{{.H1}}

- {{.Msg}} -
- Generated by COW ` + version + ` at {{.T}} - - -` - // Do not end with "\r\n" so we can add more header later var headRawTmpl = "HTTP/1.1 {{.CodeReason}}\r\n" + "Connection: keep-alive\r\n" + @@ -30,7 +19,24 @@ var headRawTmpl = "HTTP/1.1 {{.CodeReason}}\r\n" + var errPageTmpl, headTmpl *template.Template func init() { - var err error + hostName, err := os.Hostname() + if err != nil { + hostName = "unknown host" + } + + errPageRawTmpl := ` + + COW Proxy + +

{{.H1}}

+ {{.Msg}} +
+ Generated by COW ` + version + `
+ Host ` + hostName + `
+ {{.T}} + + +` if headTmpl, err = template.New("errorHead").Parse(headRawTmpl); err != nil { Fatal("Internal error on generating error head template") }