From 04e5d5b71fe39039baf72b27411e7629edbcc6a2 Mon Sep 17 00:00:00 2001
From: Unknown <joe2010xtmf@163.com>
Date: Wed, 19 Mar 2014 05:31:38 -0400
Subject: [PATCH] Use custom logger middleware

---
 modules/middleware/logger.go | 44 ++++++++++++++++++++++++++++++++++++
 web.go                       | 13 ++++++++++-
 2 files changed, 56 insertions(+), 1 deletion(-)
 create mode 100644 modules/middleware/logger.go

diff --git a/modules/middleware/logger.go b/modules/middleware/logger.go
new file mode 100644
index 00000000..9bbedb29
--- /dev/null
+++ b/modules/middleware/logger.go
@@ -0,0 +1,44 @@
+// Copyright 2014 The Gogs Authors. All rights reserved.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+
+package middleware
+
+import (
+	"fmt"
+	"log"
+	"net/http"
+	"runtime"
+	"time"
+
+	"github.com/codegangsta/martini"
+)
+
+var isWindows bool
+
+func init() {
+	isWindows = runtime.GOOS == "windows"
+}
+
+func Logger() martini.Handler {
+	return func(res http.ResponseWriter, req *http.Request, ctx martini.Context, log *log.Logger) {
+		start := time.Now()
+		log.Printf("Started %s %s", req.Method, req.URL.Path)
+
+		rw := res.(martini.ResponseWriter)
+		ctx.Next()
+
+		content := fmt.Sprintf("Completed %v %s in %v", rw.Status(), http.StatusText(rw.Status()), time.Since(start))
+		if !isWindows {
+			switch rw.Status() {
+			case 200:
+				content = fmt.Sprintf("\033[1;32%s\033[0m", content)
+			case 304:
+				content = fmt.Sprintf("\033[1;33%s\033[0m", content)
+			case 404:
+				content = fmt.Sprintf("\033[1;31%s\033[0m", content)
+			}
+		}
+		log.Println(content)
+	}
+}
diff --git a/web.go b/web.go
index 018d8ffd..97cfdcc4 100644
--- a/web.go
+++ b/web.go
@@ -46,11 +46,22 @@ func checkRunMode() {
 	log.Info("Run Mode: %s", strings.Title(martini.Env))
 }
 
+func newMartini() *martini.ClassicMartini {
+	r := martini.NewRouter()
+	m := martini.New()
+	m.Use(middleware.Logger())
+	m.Use(martini.Recovery())
+	m.Use(martini.Static("public"))
+	m.MapTo(r, (*martini.Routes)(nil))
+	m.Action(r.Handle)
+	return &martini.ClassicMartini{m, r}
+}
+
 func runWeb(*cli.Context) {
 	checkRunMode()
 	log.Info("%s %s", base.AppName, base.AppVer)
 
-	m := martini.Classic()
+	m := newMartini()
 
 	// Middlewares.
 	m.Use(render.Renderer(render.Options{Funcs: []template.FuncMap{base.TemplateFuncs}}))