Add API Framework Revel Source Files

[iec.git] / src / foundation / api / revel / before_after_filter.go

diff --git a/src/foundation/api/revel/before_after_filter.go b/src/foundation/api/revel/before_after_filter.go
new file mode 100644 (file)
index 0000000..e9af1d3
--- /dev/null
+++ b/src/foundation/api/revel/before_after_filter.go
@@ -0,0 +1,60 @@
+package revel
+
+import (
+       "reflect"
+)
+
+// Autocalls any defined before and after methods on the target controller
+// If either calls returns a value then the result is returned
+func BeforeAfterFilter(c *Controller, fc []Filter) {
+       defer func() {
+               if resultValue := beforeAfterFilterInvoke(FINALLY, c); resultValue != nil && !resultValue.IsNil() {
+                       c.Result = resultValue.Interface().(Result)
+               }
+       }()
+       defer func() {
+               if err := recover(); err != nil {
+                       if resultValue := beforeAfterFilterInvoke(PANIC, c); resultValue != nil && !resultValue.IsNil() {
+                               c.Result = resultValue.Interface().(Result)
+                       }
+                       panic(err)
+               }
+       }()
+       if resultValue := beforeAfterFilterInvoke(BEFORE, c); resultValue != nil && !resultValue.IsNil() {
+               c.Result = resultValue.Interface().(Result)
+       }
+       fc[0](c, fc[1:])
+       if resultValue := beforeAfterFilterInvoke(AFTER, c); resultValue != nil && !resultValue.IsNil() {
+               c.Result = resultValue.Interface().(Result)
+       }
+}
+
+func beforeAfterFilterInvoke(method When, c *Controller) (r *reflect.Value) {
+
+       if c.Type == nil {
+               return
+       }
+       var index []*ControllerFieldPath
+       switch method {
+       case BEFORE:
+               index = c.Type.ControllerEvents.Before
+       case AFTER:
+               index = c.Type.ControllerEvents.After
+       case FINALLY:
+               index = c.Type.ControllerEvents.Finally
+       case PANIC:
+               index = c.Type.ControllerEvents.Panic
+       }
+
+       if len(index) == 0 {
+               return
+       }
+       for _, function := range index {
+               result := function.Invoke(reflect.ValueOf(c.AppController), nil)[0]
+               if !result.IsNil() {
+                       return &result
+               }
+       }
+
+       return
+}