CI框架 -- 核心文件 之 Hooks.php

1 $hook[‘pre_controller‘] = array(
2                                 ‘class‘    => ‘TestHook‘, //调用的类名
3                                 ‘function‘ => ‘test‘, //调用的函数名
4                                 ‘filename‘ => ‘TestHook.php‘, //文件名
5                                 ‘filepath‘ => ‘hooks‘, //文件或脚本路径，以application为基准
6                                 ‘params‘   => ‘array()‘//传递给脚本参数，可选
7                                 );

  1 /**
  2  * 钩子嘛，就是在不修改系统核心文件的基础上来改变或增加系统的核心运行功能
  3  */
  4 class CI_Hooks {
  5  
  6     /**
  7      * 检测hook是否开启
  8      */
  9     var $enabled        = FALSE;
 10     /**
 11      * config/hooks.php中的hooks配置信息
 12      */
 13     var $hooks            = array();
 14     //防止死循环，因为钩子程序里面可能还还有钩子
 15     var $in_progress    = FALSE;
 16  
 17     //构造函数
 18     function __construct()
 19     {
 20         $this->_initialize();
 21         log_message(‘debug‘, "Hooks Class Initialized");
 22     }
 23  
 24     /**
 25      * 初始化，获取hooks配合
 26      */
 27     function _initialize()
 28     {
 29         $CFG =& load_class(‘Config‘, ‘core‘);
 30  
 31         // 检测配置是否开启钩子
 32         if ($CFG->item(‘enable_hooks‘) == FALSE)
 33         {
 34             return;
 35         }
 36  
 37         // 检测是否配置钩子
 38         if (defined(‘ENVIRONMENT‘) AND is_file(APPPATH.‘config/‘.ENVIRONMENT.‘/hooks.php‘))
 39         {
 40             include(APPPATH.‘config/‘.ENVIRONMENT.‘/hooks.php‘);
 41         }
 42         elseif (is_file(APPPATH.‘config/hooks.php‘))
 43         {
 44             include(APPPATH.‘config/hooks.php‘);
 45         }
 46  
 47  
 48         if ( ! isset($hook) OR ! is_array($hook))
 49         {
 50             return;
 51         }
 52  
 53         $this->hooks =& $hook;
 54         $this->enabled = TRUE;
 55     }
 56  
 57     // --------------------------------------------------------------------
 58  
 59     /**
 60      * 运行钩子程序，外部就是这样调用：
 61          *     $EXT =& load_class(‘Hooks‘, ‘core‘);
 62      *      $EXT->_call_hook(‘pre_system‘);
 63      */
 64     function _call_hook($which = ‘‘)
 65     {
 66         if ( ! $this->enabled OR ! isset($this->hooks[$which]))
 67         {
 68             return FALSE;
 69         }
 70                 //CI支持多次钩子，那么就是二维数组
 71         if (isset($this->hooks[$which][0]) AND is_array($this->hooks[$which][0]))
 72         {
 73             foreach ($this->hooks[$which] as $val)
 74             {
 75                 $this->_run_hook($val);
 76             }
 77         }
 78         else
 79         {
 80                         //一个钩子直接运行钩子
 81             $this->_run_hook($this->hooks[$which]);
 82         }
 83  
 84         return TRUE;
 85     }
 86  
 87     // --------------------------------------------------------------------
 88  
 89     /**
 90      * Run Hook
 91      *
 92      * Runs a particular hook
 93      *
 94      * @access    private
 95      * @param    array    the hook details
 96      * @return    bool
 97      */
 98     function _run_hook($data)
 99     {
100         if ( ! is_array($data))
101         {
102             return FALSE;
103         }
104         
105                 //防止死循环，因为钩子程序里面可能还还有钩子
106         if ($this->in_progress == TRUE)
107         {
108             return;
109         }
110  
111         //设置路径
112                 //下面可以filepathfilepath就以那个文件夹(application)为基准，application/hooks下, 你可以把hooks 作为你的filepath
113                 
114         if ( ! isset($data[‘filepath‘]) OR ! isset($data[‘filename‘]))
115         {
116             return FALSE;
117         }
118  
119         $filepath = APPPATH.$data[‘filepath‘].‘/‘.$data[‘filename‘];
120  
121         if ( ! file_exists($filepath))
122         {
123             return FALSE;
124         }
125  
126         // -----------------------------------
127         // Set class/function name
128         // -----------------------------------
129  
130         $class        = FALSE;
131         $function    = FALSE;
132         $params        = ‘‘;
133  
134         if (isset($data[‘class‘]) AND $data[‘class‘] != ‘‘)
135         {
136             $class = $data[‘class‘];
137         }
138  
139         if (isset($data[‘function‘]))
140         {
141             $function = $data[‘function‘];
142         }
143  
144         if (isset($data[‘params‘]))
145         {
146             $params = $data[‘params‘];
147         }
148  
149         if ($class === FALSE AND $function === FALSE)
150         {
151             return FALSE;
152         }
153  
154         //不用多说了吧
155         $this->in_progress = TRUE;
156  
157         //获取钩子配置信息成功后，运行钩子程序
158         if ($class !== FALSE)
159         {
160             if ( ! class_exists($class))
161             {
162                 require($filepath);
163             }
164  
165             $HOOK = new $class;
166             $HOOK->$function($params);
167         }
168         else
169         {
170             if ( ! function_exists($function))
171             {
172                 require($filepath);
173             }
174  
175             $function($params);
176         }
177                 // //执行相应程序完毕后，重新把当前hook的状态改为非运行中，以让它可以再次被触发。
178         $this->in_progress = FALSE;
179         return TRUE;
180     }
181  
182 }