Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(193)

Unified Diff: milo/appengine/common/middleware.go

Issue 2748073006: Milo Refactor: Remove theme support (Closed)
Patch Set: Fix builder.html pointer Created 3 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: milo/appengine/common/middleware.go
diff --git a/milo/appengine/common/middleware.go b/milo/appengine/common/middleware.go
new file mode 100644
index 0000000000000000000000000000000000000000..daef78d87690eb9cab334f9cff9f673c2a3beeb2
--- /dev/null
+++ b/milo/appengine/common/middleware.go
@@ -0,0 +1,92 @@
+// Copyright 2015 The LUCI Authors. All rights reserved.
+// Use of this source code is governed under the Apache License, Version 2.0
+// that can be found in the LICENSE file.
+
+package common
+
+import (
+ "net/http"
+ "strings"
+
+ "golang.org/x/net/context"
+
+ "github.com/luci/gae/service/info"
+ "github.com/luci/luci-go/appengine/gaeauth/server"
+ "github.com/luci/luci-go/appengine/gaemiddleware"
+ "github.com/luci/luci-go/common/clock"
+ "github.com/luci/luci-go/server/analytics"
+ "github.com/luci/luci-go/server/auth"
+ "github.com/luci/luci-go/server/auth/identity"
+ "github.com/luci/luci-go/server/router"
+ "github.com/luci/luci-go/server/templates"
+)
+
+// GetTemplateBundles is used to render HTML templates. It provides base args
+// passed to all templates.
+func GetTemplateBundle() *templates.Bundle {
+ return &templates.Bundle{
+ Loader: templates.FileSystemLoader("templates"),
+ DebugMode: info.IsDevAppServer,
+ DefaultTemplate: "base",
+ DefaultArgs: func(c context.Context) (templates.Args, error) {
+ r := getRequest(c)
+ path := r.URL.Path
+ loginURL, err := auth.LoginURL(c, path)
+ if err != nil {
+ return nil, err
+ }
+ logoutURL, err := auth.LogoutURL(c, path)
+ if err != nil {
+ return nil, err
+ }
+ return templates.Args{
+ "AppVersion": strings.Split(info.VersionID(c), ".")[0],
+ "IsAnonymous": auth.CurrentIdentity(c) == identity.AnonymousIdentity,
+ "User": auth.CurrentUser(c),
+ "LoginURL": loginURL,
+ "LogoutURL": logoutURL,
+ "CurrentTime": clock.Now(c),
+ "Analytics": analytics.Snippet(c),
+ "RequestID": info.RequestID(c),
+ }, nil
+ },
+ FuncMap: funcMap,
+ }
+}
+
+// Base returns the basic LUCI appengine middlewares.
+func Base() router.MiddlewareChain {
+ methods := auth.Authenticator{
+ &server.OAuth2Method{Scopes: []string{server.EmailScope}},
+ server.CookieAuth,
+ &server.InboundAppIDAuthMethod{},
+ }
+ m := gaemiddleware.BaseProd().Extend(auth.Use(methods), auth.Authenticate)
+ m = m.Extend(withRequestMiddleware)
+ m = m.Extend(templates.WithTemplates(GetTemplateBundle()))
+ return m
+}
+
+// The context key, so that we can embed the http.Request object into
+// the context.
+var requestKey = "http.request"
+
+// WithRequest returns a context with the http.Request object
+// in it.
+func WithRequest(c context.Context, r *http.Request) context.Context {
+ return context.WithValue(c, &requestKey, r)
+}
+
+// withRequestMiddleware is a middleware that installs a request into the context.
+// This is used for various things in the default template.
+func withRequestMiddleware(c *router.Context, next router.Handler) {
+ c.Context = WithRequest(c.Context, c.Request)
+ next(c)
+}
+
+func getRequest(c context.Context) *http.Request {
+ if req, ok := c.Value(&requestKey).(*http.Request); ok {
+ return req
+ }
+ panic("No http.request found in context")
+}

Powered by Google App Engine
This is Rietveld 408576698