CheckActionRouteBehavior.class.php 8.32 KB
<?php
// +----------------------------------------------------------------------
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
// +----------------------------------------------------------------------
// | Copyright (c) 2006-2012 http://thinkphp.cn All rights reserved.
// +----------------------------------------------------------------------
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
// +----------------------------------------------------------------------
// | Author: liu21st <liu21st@gmail.com>
// +----------------------------------------------------------------------
namespace Behavior;

/**
 * 系统行为扩展:操作路由检测
 */
class CheckActionRouteBehavior
{

    // 行为扩展的执行入口必须是run
    public function run(&$config)
    {
        // 优先检测是否存在PATH_INFO
        $regx = trim($_SERVER['PATH_INFO'], '/');
        if (empty($regx)) {
            return;
        }
        // 路由定义文件优先于config中的配置定义
        // 路由处理
        $routes = $config['routes'];
        if (! empty($routes)) {
            $depr = C('URL_PATHINFO_DEPR');
            // 分隔符替换 确保路由定义使用统一的分隔符
            $regx = str_replace($depr, '/', $regx);
            $regx = substr_replace($regx, '', 0, strlen(__URL__));
            foreach ($routes as $rule => $route) {
                if (0 === strpos($rule, '/') && preg_match($rule, $regx, $matches)) { // 正则路由
                    return C('ACTION_NAME', $this->parseRegex($matches, $route, $regx));
                } else { // 规则路由
                    $len1 = substr_count($regx, '/');
                    $len2 = substr_count($rule, '/');
                    if ($len1 >= $len2) {
                        if ('$' == substr($rule, - 1, 1)) { // 完整匹配
                            if ($len1 != $len2) {
                                continue;
                            } else {
                                $rule = substr($rule, 0, - 1);
                            }
                        }
                        $match = $this->checkUrlMatch($regx, $rule);
                        if ($match) {
                            return C('ACTION_NAME', $this->parseRule($rule, $route, $regx));
                        }
                    }
                }
            }
        }
    }

    // 检测URL和规则路由是否匹配
    private function checkUrlMatch($regx, $rule)
    {
        $m1 = explode('/', $regx);
        $m2 = explode('/', $rule);
        $match = true; // 是否匹配
        foreach ($m2 as $key => $val) {
            if (':' == substr($val, 0, 1)) { // 动态变量
                if (strpos($val, '\\')) {
                    $type = substr($val, - 1);
                    if ('d' == $type && ! is_numeric($m1[$key])) {
                        $match = false;
                        break;
                    }
                } elseif (strpos($val, '^')) {
                    $array = explode('|', substr(strstr($val, '^'), 1));
                    if (in_array($m1[$key], $array)) {
                        $match = false;
                        break;
                    }
                }
            } elseif (0 !== strcasecmp($val, $m1[$key])) {
                $match = false;
                break;
            }
        }

        return $match;
    }

    // 解析规范的路由地址
    // 地址格式 操作?参数1=值1&参数2=值2...
    private function parseUrl($url)
    {
        $var = array();
        if (false !== strpos($url, '?')) { // 操作?参数1=值1&参数2=值2...
            $info = parse_url($url);
            $path = $info['path'];
            parse_str($info['query'], $var);
        } else { // 操作
            $path = $url;
        }
        $var[C('VAR_ACTION')] = $path;

        return $var;
    }

    // 解析规则路由
    // '路由规则'=>'操作?额外参数1=值1&额外参数2=值2...'
    // '路由规则'=>array('操作','额外参数1=值1&额外参数2=值2...')
    // '路由规则'=>'外部地址'
    // '路由规则'=>array('外部地址','重定向代码')
    // 路由规则中 :开头 表示动态变量
    // 外部地址中可以用动态变量 采用 :1 :2 的方式
    // 'news/:month/:day/:id'=>array('News/read?cate=1','status=1'),
    // 'new/:id'=>array('/new.php?id=:1',301), 重定向
    private function parseRule($rule, $route, $regx)
    {
        // 获取路由地址规则
        $url = is_array($route) ? $route[0] : $route;
        // 获取URL地址中的参数
        $paths = explode('/', $regx);
        // 解析路由规则
        $matches = array();
        $rule = explode('/', $rule);
        foreach ($rule as $item) {
            if (0 === strpos($item, ':')) {
                // 动态变量获取
                $pos = strpos($item, '^');
                if ($pos) {
                    $var = substr($item, 1, $pos - 1);
                } elseif (strpos($item, '\\')) {
                    $var = substr($item, 1, - 2);
                } else {
                    $var = substr($item, 1);
                }
                $matches[$var] = array_shift($paths);
            } else { // 过滤URL中的静态变量
                array_shift($paths);
            }
        }
        if (0 === strpos($url, '/') || 0 === strpos($url, 'http')) { // 路由重定向跳转
            if (strpos($url, ':')) { // 传递动态参数
                $values = array_values($matches);
                $url = preg_replace('/:(\d+)/e', '$values[\\1-1]', $url);
            }
            header("Location: $url", true, (is_array($route) && isset($route[1])) ? $route[1] : 301);
            exit();
        } else {
            // 解析路由地址
            $var = $this->parseUrl($url);
            // 解析路由地址里面的动态参数
            $values = array_values($matches);
            foreach ($var as $key => $val) {
                if (0 === strpos($val, ':')) {
                    $var[$key] = $values[substr($val, 1) - 1];
                }
            }
            $var = array_merge($matches, $var);
            // 解析剩余的URL参数
            if ($paths) {
                preg_replace('@(\w+)\/([^\/]+)@e', '$var[strtolower(\'\\1\')]=strip_tags(\'\\2\');', implode('/', $paths));
            }
            // 解析路由自动传入参数
            if (is_array($route) && isset($route[1])) {
                parse_str($route[1], $params);
                $var = array_merge($var, $params);
            }
            $action = $var[C('VAR_ACTION')];
            unset($var[C('VAR_ACTION')]);
            $_GET = array_merge($var, $_GET);

            return $action;
        }
    }

    // 解析正则路由
    // '路由正则'=>'[分组/模块/操作]?参数1=值1&参数2=值2...'
    // '路由正则'=>array('[分组/模块/操作]?参数1=值1&参数2=值2...','额外参数1=值1&额外参数2=值2...')
    // '路由正则'=>'外部地址'
    // '路由正则'=>array('外部地址','重定向代码')
    // 参数值和外部地址中可以用动态变量 采用 :1 :2 的方式
    // '/new\/(\d+)\/(\d+)/'=>array('News/read?id=:1&page=:2&cate=1','status=1'),
    // '/new\/(\d+)/'=>array('/new.php?id=:1&page=:2&status=1','301'), 重定向
    private function parseRegex($matches, $route, $regx)
    {
        // 获取路由地址规则
        $url = is_array($route) ? $route[0] : $route;
        $url = preg_replace('/:(\d+)/e', '$matches[\\1]', $url);
        if (0 === strpos($url, '/') || 0 === strpos($url, 'http')) { // 路由重定向跳转
            header("Location: $url", true, (is_array($route) && isset($route[1])) ? $route[1] : 301);
            exit();
        } else {
            // 解析路由地址
            $var = $this->parseUrl($url);
            // 解析剩余的URL参数
            $regx = substr_replace($regx, '', 0, strlen($matches[0]));
            if ($regx) {
                preg_replace('@(\w+)\/([^,\/]+)@e', '$var[strtolower(\'\\1\')]=strip_tags(\'\\2\');', $regx);
            }
            // 解析路由自动传入参数
            if (is_array($route) && isset($route[1])) {
                parse_str($route[1], $params);
                $var = array_merge($var, $params);
            }
            $action = $var[C('VAR_ACTION')];
            unset($var[C('VAR_ACTION')]);
            $_GET = array_merge($var, $_GET);
        }

        return $action;
    }
}