From 69973a0c0cea573d444d46b5ae69bebc1567a9b8 Mon Sep 17 00:00:00 2001 From: Niall Sheridan <nsheridan@gmail.com> Date: Sat, 28 May 2016 23:28:32 +0100 Subject: [PATCH] Define the token tmpl as a string --- cmd/cashierd/main.go | 8 ++++---- exampleconfig.json | 3 +-- templates/{token.html => token.go} | 6 ++++-- 3 files changed, 9 insertions(+), 8 deletions(-) rename templates/{token.html => token.go} (95%) diff --git a/cmd/cashierd/main.go b/cmd/cashierd/main.go index 61461a74..2a8d2b80 100644 --- a/cmd/cashierd/main.go +++ b/cmd/cashierd/main.go @@ -13,7 +13,6 @@ import ( "log" "net/http" "os" - "path" "strings" "time" @@ -27,6 +26,7 @@ import ( "github.com/nsheridan/cashier/server/auth/google" "github.com/nsheridan/cashier/server/config" "github.com/nsheridan/cashier/server/signer" + "github.com/nsheridan/cashier/templates" ) var ( @@ -38,7 +38,6 @@ type appContext struct { cookiestore *sessions.CookieStore authprovider auth.Provider authsession *auth.Session - views *template.Template sshKeySigner *signer.KeySigner } @@ -154,7 +153,9 @@ func rootHandler(a *appContext, w http.ResponseWriter, r *http.Request) (int, er page := struct { Token string }{tok.AccessToken} - a.views.ExecuteTemplate(w, "token.html", page) + + tmpl := template.Must(template.New("token.html").Parse(templates.Token)) + tmpl.Execute(w, page) return http.StatusOK, nil } @@ -226,7 +227,6 @@ func main() { ctx := &appContext{ cookiestore: sessions.NewCookieStore([]byte(config.Server.CookieSecret)), authprovider: authprovider, - views: template.Must(template.ParseGlob(path.Join(config.Server.TemplateDir, "*"))), sshKeySigner: signer, } ctx.cookiestore.Options = &sessions.Options{ diff --git a/exampleconfig.json b/exampleconfig.json index cc4aca92..1ac84f27 100644 --- a/exampleconfig.json +++ b/exampleconfig.json @@ -4,8 +4,7 @@ "tls_key": "server.key", "tls_cert": "server.crt", "port": 443, - "cookie_secret": "supersecret", - "template_dir": "/go/src/github.com/nsheridan/cashier/templates" + "cookie_secret": "supersecret" }, "auth": { "provider": "google", diff --git a/templates/token.html b/templates/token.go similarity index 95% rename from templates/token.html rename to templates/token.go index 45131764..dbad371f 100644 --- a/templates/token.html +++ b/templates/token.go @@ -1,4 +1,6 @@ -<html> +package templates + +const Token = `<html> <head> <title>YOUR TOKEN!</title> <style> @@ -43,4 +45,4 @@ The token will expire in < 1 hour. </h2> </body> -</html> +</html>` -- GitLab