UrlRule.php 9.35 KB
Newer Older
Qiang Xue committed
1 2 3
<?php
/**
 * @link http://www.yiiframework.com/
Qiang Xue committed
4
 * @copyright Copyright (c) 2008 Yii Software LLC
Qiang Xue committed
5 6 7 8 9 10
 * @license http://www.yiiframework.com/license/
 */

namespace yii\web;

use yii\base\Object;
Qiang Xue committed
11
use yii\base\InvalidConfigException;
Qiang Xue committed
12 13

/**
14 15 16 17 18 19 20 21 22 23 24
 * UrlRule represents a rule used by [[UrlManager]] for parsing and generating URLs.
 *
 * To define your own URL parsing and creation logic you can extend from this class
 * and add it to [[UrlManager::rules]] like this:
 *
 * ~~~
 * 'rules' => [
 *     ['class' => 'MyUrlRule', 'pattern' => '...', 'route' => 'site/index', ...],
 *     // ...
 * ]
 * ~~~
Qiang Xue committed
25 26 27 28 29 30
 *
 * @author Qiang Xue <qiang.xue@gmail.com>
 * @since 2.0
 */
class UrlRule extends Object
{
Qiang Xue committed
31 32 33 34 35 36 37 38 39
	/**
	 * Set [[mode]] with this value to mark that this rule is for URL parsing only
	 */
	const PARSING_ONLY = 1;
	/**
	 * Set [[mode]] with this value to mark that this rule is for URL creation only
	 */
	const CREATION_ONLY = 2;

Qiang Xue committed
40 41 42 43
	/**
	 * @var string the name of this rule. If not set, it will use [[pattern]] as the name.
	 */
	public $name;
Qiang Xue committed
44
	/**
Qiang Xue committed
45 46
	 * @var string the pattern used to parse and create the path info part of a URL.
	 * @see host
Qiang Xue committed
47
	 */
Qiang Xue committed
48
	public $pattern;
Qiang Xue committed
49 50 51 52 53
	/**
	 * @var string the pattern used to parse and create the host info part of a URL.
	 * @see pattern
	 */
	public $host;
Qiang Xue committed
54 55 56 57
	/**
	 * @var string the route to the controller action
	 */
	public $route;
Qiang Xue committed
58
	/**
resurtm committed
59
	 * @var array the default GET parameters (name => value) that this rule provides.
Qiang Xue committed
60 61 62
	 * When this rule is used to parse the incoming request, the values declared in this property
	 * will be injected into $_GET.
	 */
Alexander Makarov committed
63
	public $defaults = [];
Qiang Xue committed
64 65 66 67 68 69 70 71 72 73 74 75 76
	/**
	 * @var string the URL suffix used for this rule.
	 * For example, ".html" can be used so that the URL looks like pointing to a static HTML page.
	 * If not, the value of [[UrlManager::suffix]] will be used.
	 */
	public $suffix;
	/**
	 * @var string|array the HTTP verb (e.g. GET, POST, DELETE) that this rule should match.
	 * Use array to represent multiple verbs that this rule may match.
	 * If this property is not set, the rule can match any verb.
	 * Note that this property is only used when parsing a request. It is ignored for URL creation.
	 */
	public $verb;
Qiang Xue committed
77
	/**
Qiang Xue committed
78
	 * @var integer a value indicating if this rule should be used for both request parsing and URL creation,
Qiang Xue committed
79
	 * parsing only, or creation only.
Qiang Xue committed
80 81
	 * If not set or 0, it means the rule is both request parsing and URL creation.
	 * If it is [[PARSING_ONLY]], the rule is for request parsing only.
Qiang Xue committed
82 83 84
	 * If it is [[CREATION_ONLY]], the rule is for URL creation only.
	 */
	public $mode;
Qiang Xue committed
85

Qiang Xue committed
86 87 88
	/**
	 * @var string the template for generating a new URL. This is derived from [[pattern]] and is used in generating URL.
	 */
Qiang Xue committed
89
	private $_template;
Qiang Xue committed
90 91 92
	/**
	 * @var string the regex for matching the route part. This is used in generating URL.
	 */
Qiang Xue committed
93
	private $_routeRule;
Qiang Xue committed
94 95 96
	/**
	 * @var array list of regex for matching parameters. This is used in generating URL.
	 */
Alexander Makarov committed
97
	private $_paramRules = [];
Qiang Xue committed
98 99 100
	/**
	 * @var array list of parameters used in the route.
	 */
Alexander Makarov committed
101
	private $_routeParams = [];
Qiang Xue committed
102

Qiang Xue committed
103 104 105
	/**
	 * Initializes this rule.
	 */
Qiang Xue committed
106
	public function init()
Qiang Xue committed
107
	{
Qiang Xue committed
108 109 110 111 112 113
		if ($this->pattern === null) {
			throw new InvalidConfigException('UrlRule::pattern must be set.');
		}
		if ($this->route === null) {
			throw new InvalidConfigException('UrlRule::route must be set.');
		}
Qiang Xue committed
114 115 116 117 118 119
		if ($this->verb !== null) {
			if (is_array($this->verb)) {
				foreach ($this->verb as $i => $verb) {
					$this->verb[$i] = strtoupper($verb);
				}
			} else {
Alexander Makarov committed
120
				$this->verb = [strtoupper($this->verb)];
Qiang Xue committed
121 122
			}
		}
Qiang Xue committed
123 124 125
		if ($this->name === null) {
			$this->name = $this->pattern;
		}
Qiang Xue committed
126

Qiang Xue committed
127
		$this->pattern = trim($this->pattern, '/');
128

Qiang Xue committed
129 130 131
		if ($this->host !== null) {
			$this->pattern = rtrim($this->host, '/') . rtrim('/' . $this->pattern, '/') . '/';
		} elseif ($this->pattern === '') {
Qiang Xue committed
132
			$this->_template = '';
Qiang Xue committed
133 134 135 136 137 138 139 140 141
			$this->pattern = '#^$#u';
			return;
		} else {
			$this->pattern = '/' . $this->pattern . '/';
		}

		$this->route = trim($this->route, '/');
		if (strpos($this->route, '<') !== false && preg_match_all('/<(\w+)>/', $this->route, $matches)) {
			foreach ($matches[1] as $name) {
Qiang Xue committed
142
				$this->_routeParams[$name] = "<$name>";
Qiang Xue committed
143 144 145
			}
		}

146 147 148 149 150 151 152 153 154 155
		$tr = [
			'.' => '\\.',
			'*' => '\\*',
			'$' => '\\$',
			'[' => '\\[',
			']' => '\\]',
			'(' => '\\(',
			')' => '\\)',
		];
		$tr2 = [];
Qiang Xue committed
156 157 158 159 160 161 162
		if (preg_match_all('/<(\w+):?([^>]+)?>/', $this->pattern, $matches, PREG_OFFSET_CAPTURE | PREG_SET_ORDER)) {
			foreach ($matches as $match) {
				$name = $match[1][0];
				$pattern = isset($match[2][0]) ? $match[2][0] : '[^\/]+';
				if (isset($this->defaults[$name])) {
					$length = strlen($match[0][0]);
					$offset = $match[0][1];
163
					if ($offset > 1 && $this->pattern[$offset - 1] === '/' && $this->pattern[$offset + $length] === '/') {
Qiang Xue committed
164
						$tr["/<$name>"] = "(/(?P<$name>$pattern))?";
Qiang Xue committed
165
					} else {
Qiang Xue committed
166
						$tr["<$name>"] = "(?P<$name>$pattern)?";
Qiang Xue committed
167 168 169 170
					}
				} else {
					$tr["<$name>"] = "(?P<$name>$pattern)";
				}
Qiang Xue committed
171
				if (isset($this->_routeParams[$name])) {
Qiang Xue committed
172 173
					$tr2["<$name>"] = "(?P<$name>$pattern)";
				} else {
Qiang Xue committed
174
					$this->_paramRules[$name] = $pattern === '[^\/]+' ? '' : "#^$pattern$#";
Qiang Xue committed
175 176 177 178
				}
			}
		}

Qiang Xue committed
179 180
		$this->_template = preg_replace('/<(\w+):?([^>]+)?>/', '<$1>', $this->pattern);
		$this->pattern = '#^' . trim(strtr($this->_template, $tr), '/') . '$#u';
Qiang Xue committed
181

182
		if (!empty($this->_routeParams)) {
Qiang Xue committed
183
			$this->_routeRule = '#^' . strtr($this->route, $tr2) . '$#u';
Qiang Xue committed
184 185 186
		}
	}

Qiang Xue committed
187
	/**
Qiang Xue committed
188
	 * Parses the given request and returns the corresponding route and parameters.
Qiang Xue committed
189
	 * @param UrlManager $manager the URL manager
Qiang Xue committed
190
	 * @param Request $request the request component
Qiang Xue committed
191 192 193
	 * @return array|boolean the parsing result. The route and the parameters are returned as an array.
	 * If false, it means this rule cannot be used to parse this path info.
	 */
Qiang Xue committed
194
	public function parseRequest($manager, $request)
Qiang Xue committed
195
	{
Qiang Xue committed
196 197 198 199
		if ($this->mode === self::CREATION_ONLY) {
			return false;
		}

200
		if ($this->verb !== null && !in_array($request->getMethod(), $this->verb, true)) {
Qiang Xue committed
201 202 203
			return false;
		}

204
		$pathInfo = $request->getPathInfo();
Qiang Xue committed
205 206 207 208 209 210 211 212 213
		$suffix = (string)($this->suffix === null ? $manager->suffix : $this->suffix);
		if ($suffix !== '' && $pathInfo !== '') {
			$n = strlen($suffix);
			if (substr($pathInfo, -$n) === $suffix) {
				$pathInfo = substr($pathInfo, 0, -$n);
				if ($pathInfo === '') {
					// suffix alone is not allowed
					return false;
				}
Qiang Xue committed
214
			} else {
Qiang Xue committed
215 216 217 218
				return false;
			}
		}

Qiang Xue committed
219
		if ($this->host !== null) {
220
			$pathInfo = strtolower($request->getHostInfo()) . ($pathInfo === '' ? '' : '/' . $pathInfo);
221 222
		}

Qiang Xue committed
223 224 225
		if (!preg_match($this->pattern, $pathInfo, $matches)) {
			return false;
		}
Qiang Xue committed
226 227 228 229 230
		foreach ($this->defaults as $name => $value) {
			if (!isset($matches[$name]) || $matches[$name] === '') {
				$matches[$name] = $value;
			}
		}
Qiang Xue committed
231
		$params = $this->defaults;
Alexander Makarov committed
232
		$tr = [];
Qiang Xue committed
233
		foreach ($matches as $name => $value) {
Qiang Xue committed
234 235
			if (isset($this->_routeParams[$name])) {
				$tr[$this->_routeParams[$name]] = $value;
Qiang Xue committed
236
				unset($params[$name]);
Qiang Xue committed
237
			} elseif (isset($this->_paramRules[$name])) {
Qiang Xue committed
238
				$params[$name] = $value;
Qiang Xue committed
239 240
			}
		}
Qiang Xue committed
241
		if ($this->_routeRule !== null) {
Qiang Xue committed
242 243 244 245
			$route = strtr($this->route, $tr);
		} else {
			$route = $this->route;
		}
Alexander Makarov committed
246
		return [$route, $params];
Qiang Xue committed
247 248
	}

Qiang Xue committed
249 250 251 252 253 254 255 256
	/**
	 * Creates a URL according to the given route and parameters.
	 * @param UrlManager $manager the URL manager
	 * @param string $route the route. It should not have slashes at the beginning or the end.
	 * @param array $params the parameters
	 * @return string|boolean the created URL, or false if this rule cannot be used for creating this URL.
	 */
	public function createUrl($manager, $route, $params)
Qiang Xue committed
257
	{
Qiang Xue committed
258
		if ($this->mode === self::PARSING_ONLY) {
Qiang Xue committed
259 260 261
			return false;
		}

Alexander Makarov committed
262
		$tr = [];
Qiang Xue committed
263 264 265

		// match the route part first
		if ($route !== $this->route) {
Qiang Xue committed
266 267
			if ($this->_routeRule !== null && preg_match($this->_routeRule, $route, $matches)) {
				foreach ($this->_routeParams as $name => $token) {
Qiang Xue committed
268 269 270 271 272
					if (isset($this->defaults[$name]) && strcmp($this->defaults[$name], $matches[$name]) === 0) {
						$tr[$token] = '';
					} else {
						$tr[$token] = $matches[$name];
					}
Qiang Xue committed
273 274 275 276 277 278 279 280 281
				}
			} else {
				return false;
			}
		}

		// match default params
		// if a default param is not in the route pattern, its value must also be matched
		foreach ($this->defaults as $name => $value) {
Qiang Xue committed
282
			if (isset($this->_routeParams[$name])) {
Qiang Xue committed
283 284
				continue;
			}
Qiang Xue committed
285 286 287 288
			if (!isset($params[$name])) {
				return false;
			} elseif (strcmp($params[$name], $value) === 0) { // strcmp will do string conversion automatically
				unset($params[$name]);
Qiang Xue committed
289
				if (isset($this->_paramRules[$name])) {
Qiang Xue committed
290 291
					$tr["<$name>"] = '';
				}
Qiang Xue committed
292
			} elseif (!isset($this->_paramRules[$name])) {
Qiang Xue committed
293 294 295 296 297
				return false;
			}
		}

		// match params in the pattern
Qiang Xue committed
298
		foreach ($this->_paramRules as $name => $rule) {
299
			if (isset($params[$name]) && !is_array($params[$name]) && ($rule === '' || preg_match($rule, $params[$name]))) {
Qiang Xue committed
300 301 302 303 304 305 306
				$tr["<$name>"] = urlencode($params[$name]);
				unset($params[$name]);
			} elseif (!isset($this->defaults[$name]) || isset($params[$name])) {
				return false;
			}
		}

Qiang Xue committed
307
		$url = trim(strtr($this->_template, $tr), '/');
Qiang Xue committed
308
		if ($this->host !== null) {
309 310 311 312 313
			$pos = strpos($url, '/', 8);
			if ($pos !== false) {
				$url = substr($url, 0, $pos) . preg_replace('#/+#', '/', substr($url, $pos));
			}
		} elseif (strpos($url, '//') !== false) {
Qiang Xue committed
314 315
			$url = preg_replace('#/+#', '/', $url);
		}
Qiang Xue committed
316 317 318 319 320

		if ($url !== '') {
			$url .= ($this->suffix === null ? $manager->suffix : $this->suffix);
		}

321
		if (!empty($params)) {
Qiang Xue committed
322 323 324 325
			$url .= '?' . http_build_query($params);
		}
		return $url;
	}
Qiang Xue committed
326
}