summaryrefslogtreecommitdiff
path: root/www/wiki/extensions/SemanticMediaWiki/src/MediaWiki/DeepRedirectTargetResolver.php
blob: b5b7e3e43123817f96ebaf856a46ae86881361a0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
<?php

namespace SMW\MediaWiki;

use RuntimeException;
use Title;

/**
 * @license GNU GPL v2+
 * @since 2.1
 *
 * @author mwjames
 */
class DeepRedirectTargetResolver {

	/**
	 * @var PageCreator
	 */
	private $pageCreator = null;

	/**
	 * Track titles to prevent circular references caused by double redirects
	 * on the same title
	 *
	 * @var array
	 */
	private $recursiveResolverTracker = [];

	/**
	 * @since 2.1
	 *
	 * @param PageCreator $pageCreator
	 */
	public function __construct( PageCreator $pageCreator ) {
		$this->pageCreator = $pageCreator;
	}

	/**
	 * @since  2.1
	 *
	 * @param Title $title
	 *
	 * @return Title|null
	 * @throws RuntimeException
	 */
	public function findRedirectTargetFor( Title $title ) {
		return $this->doResolveRedirectTarget( $title );
	}

	protected function isValidRedirectTarget( $title ) {
		return $title instanceof Title && $title->isValidRedirectTarget();
	}

	protected function isRedirect( $title ) {
		return $title instanceof Title && $title->isRedirect();
	}

	private function doResolveRedirectTarget( Title $title ) {

		$this->addToResolverTracker( $title );

		if ( $this->isCircularByKnownRedirectTarget( $title ) ) {
			throw new RuntimeException( "Circular redirect for {$title->getPrefixedDBkey()} detected." );
		}

		if ( $this->isRedirect( $title ) ) {
			$title = $this->pageCreator->createPage( $title )->getRedirectTarget();

			if ( $title instanceof Title ) {
				$title = $this->doResolveRedirectTarget( $title );
			}
		}

		if ( $this->isValidRedirectTarget( $title ) ) {
			return $title;
		}

		throw new RuntimeException( "Redirect target is unresolvable" );
	}

	private function addToResolverTracker( $title ) {

		if ( !isset( $this->recursiveResolverTracker[$title->getPrefixedDBkey()] ) ) {
			$this->recursiveResolverTracker[$title->getPrefixedDBkey()] = 0;
		}

		return $this->recursiveResolverTracker[$title->getPrefixedDBkey()]++;
	}

	private function isCircularByKnownRedirectTarget( $title ) {
		return isset( $this->recursiveResolverTracker[$title->getPrefixedDBkey()] ) && $this->recursiveResolverTracker[$title->getPrefixedDBkey()] > 1;
	}

}