diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/zorglub/node.rb | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/lib/zorglub/node.rb b/lib/zorglub/node.rb index 6e0989f..e80c051 100644 --- a/lib/zorglub/node.rb +++ b/lib/zorglub/node.rb @@ -4,8 +4,15 @@ module Zorglub # class Node # + @hooks = { + :before_all => [], + :after_all => [], + } + # class << self # + attr_reader :hooks + # def engine engine=nil @engine = engine unless engine.nil? @engine ||= Config.engine @@ -42,6 +49,24 @@ module Zorglub node.content end # + def before_hooks obj + Node.hooks[:before_all].each do |blk| blk.call obj end + end + # + def before_all &blk + Node.hooks[:before_all]<< blk + Node.hooks[:before_all].uniq! + end + # + def after_hooks obj + Node.hooks[:after_all].each do |blk| blk.call obj end + end + # + def after_all &blk + Node.hooks[:after_all]<< blk + Node.hooks[:after_all].uniq! + end + # def error_404 node resp = node.response resp.status = 404 @@ -71,11 +96,13 @@ module Zorglub end # def feed! + Node.before_hooks self @content = self.send @action[:method], *@action[:args] e, v, l = Config.engine_proc(@action[:engine]), view, layout # TODO compile and cache @content = e.call v, self if e and File.exists? v @content = e.call l, self if e and File.exists? l + Node.after_hooks self @content end # |