From 8f97788f8937a72a461cb7a81e5c2b8d795316a2 Mon Sep 17 00:00:00 2001
From: Henrique Dias <hacdias@gmail.com>
Date: Sat, 12 Sep 2015 18:58:10 +0100
Subject: [PATCH] updates

---
 hugo.go | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/hugo.go b/hugo.go
index 2427313e..3c3be8e7 100644
--- a/hugo.go
+++ b/hugo.go
@@ -2,6 +2,7 @@ package hugo
 
 import (
 	"net/http"
+	"strings"
 
 	"github.com/spf13/hugo/commands"
 
@@ -14,26 +15,29 @@ func Setup(c *setup.Controller) (middleware.Middleware, error) {
 	commands.Execute()
 
 	return func(next middleware.Handler) middleware.Handler {
-		return &handler{
-			Next: next,
-		}
+		return &handler{Next: next}
 	}, nil
 }
 
 type handler struct{ Next middleware.Handler }
-type adminHandler struct{}
 
 func (h handler) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error) {
-	// do path matching
 	if middleware.Path(r.URL.Path).Matches("/admin") {
-		a := new(adminHandler)
-		return a.ServeHTTP(w, r)
+
+		if middleware.Path(r.URL.Path).Matches("/admin/new") {
+			w.Write([]byte("New"))
+		} else if middleware.Path(r.URL.Path).Matches("/admin/edit") {
+			var fileName string
+
+			fileName = strings.Replace(r.URL.Path, "/admin/edit", "", 1)
+
+			w.Write([]byte("Edit " + fileName))
+		} else {
+			w.Write([]byte("Admin"))
+		}
+
+		return 200, nil
 	}
 
 	return h.Next.ServeHTTP(w, r)
 }
-
-func (a adminHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error) {
-	w.Write([]byte("Admin area"))
-	return 200, nil
-}