add use feature

Author: joyqi

build/http.js

@@ -9,9 +9,11 @@ var interceptorOne = function (done, next) {
     console.log(this.request.agent);
 };
 
-http.on('/', interceptorOne, function (done) {
+http.on('/', function (done) {
     return done('json', this.text + this.request.method);
-});
+}).use(interceptorOne);
+
+http.use(interceptorOne);
 
 http.assets('/src', __dirname + '/../src');
 

build/router.js

@@ -1,6 +1,6 @@
 {
   "name": "node-tiny-http",
-  "version": "1.2.7",
+  "version": "1.2.8",
   "description": "A tiny node.js http framework",
   "main": "build/http.js",
   "scripts": {

examples/server.js

@@ -18,19 +18,23 @@ module.exports =
     # register result
     result: Result.register
 
+
+    # use default functions
+    use: Router.use
+
 
     # on method
-    on: (pattern, fn...) ->
-        Router.register null, pattern, fn
+    on: (pattern, fn, method = null) ->
+        Router.register method, pattern, fn
 
 
     # get method
-    get: (pattern, fn...) ->
+    get: (pattern, fn) ->
         Router.register 'GET', pattern, fn
 
 
     # post method
-    post: (pattern, fn...) ->
+    post: (pattern, fn) ->
         Router.register 'POST', pattern, fn
 
 

package.json

@@ -5,6 +5,9 @@ Result = require './result'
 # routes map
 routes = {}
 
+# default functions
+defaults = []
+
 
 # create match uri pattern
 match = (method, pattern) ->
@@ -35,14 +38,35 @@ match = (method, pattern) ->
 
 
 # register routes
-register = (method, pattern, actions) ->
+register = (method, pattern, fn) ->
     tester = match method, pattern
     functions = []
-
+    pushed = no
+
+    routes[pattern] =
+        get: ->
+            if not pushed
+                functions.push fn
+                pushed = yes
+
+            [tester, functions]
+        use: (actions...) ->
+            for action in actions
+                if action instanceof Array
+                    for item in action
+                        functions.push item
+                else
+                    functions.push action
+
+
+# register default functions
+use = (actions...) ->
     for action in actions
-        functions.push action if action not in functions
-
-    routes[pattern] = [tester, functions]
+        if action instanceof Array
+            for item in action
+                defaults.push item
+        else
+            defaults.push action
 
 
 # handler for http
@@ -63,7 +87,7 @@ handler = (result, options) ->
                 response.respond() if not response.responded
 
             for pattern, def of routes
-                [tester, functions] = def
+                [tester, functions] = def.get()
                 params = {}
                 index = -1
 
@@ -73,13 +97,13 @@ handler = (result, options) ->
 
                 do next = ->
                     index += 1
-                    fn = functions[index]
+                    fn = if index >= defaults.length then functions[index - defaults.length] else defaults[index]
                     fn.call context, done, next if fn?
 
                 return
 
             done 'notFound'
 
 
-module.exports = { register, handler }
+module.exports = { register, handler, use }