Little refactoring (#85)

* refactoring code

Rework registering of methods if any and peoper lookup of supported templates in SimpleView

Author: nicola-spb

router/routes.go

@@ -320,23 +320,18 @@ func (r *Router) add(activeRoute *route, ctrlfn func() controller.Controller, mi
 			}
 		}
 	}
-	// register methods if any
-	if len(activeRoute.methods) > 0 {
-		r.HandleFunc(activeRoute.pattern, func(w http.ResponseWriter, req *http.Request) {
-			ctx := base.NewContext(w, req)
-			r.prepareContext(ctx)
-			chain := chainMiddleware(ctx, m...)
-			chain.ThenFunc(r.wrapController(ctx, activeRoute.fn, ctrlfn())).ServeHTTP(w, req)
-		}).Methods(activeRoute.methods...)
-		return nil
-	}
-	r.HandleFunc(activeRoute.pattern, func(w http.ResponseWriter, req *http.Request) {
+	route := r.HandleFunc(activeRoute.pattern, func(w http.ResponseWriter, req *http.Request) {
 		ctx := base.NewContext(w, req)
 		r.prepareContext(ctx)
 		chain := chainMiddleware(ctx, m...)
 		chain.ThenFunc(r.wrapController(ctx, activeRoute.fn, ctrlfn())).ServeHTTP(w, req)
 	})
 
+	// register methods if any
+	if len(activeRoute.methods) > 0 {
+		route.Methods(activeRoute.methods...)
+
+	}
 	return nil
 }
 

view/view.go

@@ -45,7 +45,7 @@ func NewSimpleView(viewDir string) (View, error) {
 func (s *SimpleView) load(dir string) (View, error) {
 
 	// supported is the list of file extensions that will be parsed as templates
-	supported := []string{".tpl", ".html", ".tmpl"}
+	supported := map[string]bool{".tpl": true, ".html": true, ".tmpl": true}
 
 	werr := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
 		if err != nil {
@@ -56,14 +56,7 @@ func (s *SimpleView) load(dir string) (View, error) {
 		}
 
 		extension := filepath.Ext(path)
-		found := false
-		for _, ext := range supported {
-			if ext == extension {
-				found = true
-				break
-			}
-		}
-		if !found {
+		if _, ok := supported[extension]; !ok {
 			return nil
 		}
 
@@ -87,8 +80,8 @@ func (s *SimpleView) load(dir string) (View, error) {
 		name = strings.TrimSuffix(name, extension) // remove extension
 
 		t := s.tmpl.New(name)
-		_, err = t.Parse(string(data))
-		if err != nil {
+		
+		if _, err = t.Parse(string(data)); err != nil {
 			return err
 		}
 		return nil