<?php

namespace SilverStripe\CMS\Controllers;

use SilverStripe\ORM\Queries\SQLSelect;
use SilverStripe\CMS\Model\SiteTree;
use Extension;
use Director;
use SS_HTTPRequest;
use SS_HTTPResponse;
use Controller;
use SS_HTTPResponse_Exception;


class OldPageRedirector extends Extension {

	/**
	 * On every URL that generates a 404, we'll capture it here and see if we can
	 * find an old URL that it should be redirecting to.
	 *
	 * @param SS_HTTPRequest $request The request object
	 * @throws SS_HTTPResponse_Exception
	 */
	public function onBeforeHTTPError404($request) {
		// We need to get the URL ourselves because $request->allParams() only has a max of 4 params
		$params = preg_split('|/+|', $request->getURL());
		$cleanURL = trim(Director::makeRelative($request->getURL(false)), '/');

		$getvars = $request->getVars();
		unset($getvars['url']);

		$page = self::find_old_page($params);
		$cleanPage = trim(Director::makeRelative($page), '/');
		if (!$cleanPage) {
			$cleanPage = Director::makeRelative(RootURLController::get_homepage_link());
		}

		if ($page && $cleanPage != $cleanURL) {
			$res = new SS_HTTPResponse();
			$res->redirect(
				Controller::join_links(
					$page,
					($getvars) ? '?' . http_build_query($getvars) : null
				), 301);
			throw new SS_HTTPResponse_Exception($res);
		}
	}

	/**
	 * Attempt to find an old/renamed page from some given the URL as an array
	 *
	 * @param array $params The array of URL, e.g. /foo/bar as array('foo', 'bar')
	 * @param SiteTree $parent The current parent in the recursive flow
	 * @param boolean $redirect Whether we've found an old page worthy of a redirect
	 *
	 * @return string|boolean False, or the new URL
	 */
	static public function find_old_page($params, $parent = null, $redirect = false) {
		$parent = is_numeric($parent) ? SiteTree::get()->byID($parent) : $parent;
		$params = (array)$params;
		$URL = rawurlencode(array_shift($params));
		if (empty($URL)) {
			return false;
		}
		/** @var SiteTree $page */
		if ($parent) {
			$page = SiteTree::get()->filter(array('ParentID' => $parent->ID, 'URLSegment' => $URL))->first();
		} else {
			$page = SiteTree::get()->filter(array('URLSegment' => $URL))->first();
		}

		if (!$page) {
			// If we haven't found a candidate, lets resort to finding an old page with this URL segment
			$oldFilter = array(
				'"SiteTree_versions"."URLSegment"' => $URL,
				'"SiteTree_versions"."WasPublished"' => true
			);
			if($parent) {
				$oldFilter[] = array('"SiteTree_versions"."ParentID"' => $parent->ID);
			}
			$query = new SQLSelect(
				'"RecordID"',
				'"SiteTree_versions"',
				$oldFilter,
				'"LastEdited" DESC',
				null,
				null,
				1
			);
			$record = $query->execute()->first();
			if ($record) {
				$page = SiteTree::get()->byID($record['RecordID']);
				$redirect = true;
			}
		}

		if ($page && $page->canView()) {
			if (count($params)) {
				// We have to go deeper!
				$ret = self::find_old_page($params, $page, $redirect);
				if ($ret) {
					// A valid child page was found! We can return it
					return $ret;
				} else {
					// No valid page found.
					if ($redirect) {
						// If we had some redirect to be done, lets do it. imagine /foo/action -> /bar/action, we still want this redirect to happen if action isn't a page
						return $page->Link() . implode('/', $params);
					}
				}
			} else {
				// We've found the final, end all, page.
				return $page->Link();
			}
		}

		return false;
	}
}