<?php
class TM_AskIt_Controller_Router extends Mage_Core_Controller_Varien_Router_Abstract
{
    public function initControllerRouters($observer)
    {
        $front = $observer->getEvent()->getFront();
        $front->addRouter($this->getPrefixPart(), $this);
    }

    /**
     *
     * @return string
     */
    protected function getPrefixPart()
    {
        return Mage::helper('askit')->getRouteUrlPrefix();
    }

    protected function redirect($url, $code = 301)
    {
        Mage::app()->getFrontController()->getResponse()
            ->setRedirect($url)
            ->setHttpResponseCode($code)
            ->sendResponse();
        // exit;
    }

    /**
     *
     * @param Zend_Controller_Request_Http $request
     * @return boolean
     */
    public function match(Zend_Controller_Request_Http $request)
    {
        if (!Mage::isInstalled()) {
            return false;
        }

        $identifier = trim($request->getPathInfo(), '/');
        $parts = explode('/', $identifier);

        $prefix = $this->getPrefixPart();
        $oldPrefix = 'productsquestions'; // old static prefix redirect for seo
        if ($oldPrefix == $parts[0]) {
            $uri = $request->getRequestUri();
            $uri = str_replace($oldPrefix, $prefix, $uri);
            $this->redirect($uri);
            $request->setDispatched(true);
            return true;
        }

        if ($prefix != $parts[0]) {
            return false;
        }
        if (empty($parts[1])) {
            $request->setModuleName('askit')
                ->setControllerName('index')
                ->setActionName('index')
                ->setAlias(
                    Mage_Core_Model_Url_Rewrite::REWRITE_REQUEST_PATH_ALIAS,
                    $identifier
                );
            return true;
        }

        return $this->matchProduct($request)
            || $this->matchCategory($request)
            || $this->matchPage($request);
    }

    /**
     *
     * @param Zend_Controller_Request_Http $request
     * @return boolean
     */
    protected function matchProduct(Zend_Controller_Request_Http $request)
    {
        $identifier = trim($request->getPathInfo(), '/');

        $_identifier = trim(
            $this->strReplaceOnce('/' . $this->getPrefixPart(), '', $request->getPathInfo()),
            '/'
        );

        $rewriteModel = Mage::getModel('core/url_rewrite')
            ->setStoreId($this->getStoreId())
            ->loadByRequestPath($_identifier);

        $productId = $rewriteModel->getProductId();
        $product = null;
        if ($productId) {
            $product = Mage::getModel('catalog/product')->load($productId);
        }

        if (!$product) {
            $parts = explode('/', $_identifier);
            $productCollection = Mage::getModel('catalog/product')
                ->getCollection()
                ->addAttributeToFilter('url_path', $parts[0]);

            if (!$productCollection->count()) {
                return false;
            }
            $product = $productCollection->getFirstItem();
            $product->load($product->getId());

            if (count($parts) > 1) {
                $url = Mage::getUrl() . $this->getPrefixPart() . '/' . $parts[0];
                $this->redirect($url);
                $request->setDispatched(true);
                return true;
            }
        } else {
            if ($request->getRequestUri() == $request->getPathInfo()
                && rtrim($request->getPathInfo(), '/') != $request->getPathInfo()
            ) {
                $url = Mage::getUrl() . $identifier;
                $this->redirect($url);
                $request->setDispatched(true);
                return true;
            }
        }

        if (!$product) {
            return false;
        }

        $request->setModuleName('askit')
            ->setControllerName('index')
            ->setActionName('product')
            ->setParam('product_id', $product->getId())
            ->setAlias(
                Mage_Core_Model_Url_Rewrite::REWRITE_REQUEST_PATH_ALIAS,
                $identifier
            );
        Mage::register('current_product', $product);
        Mage::register('product', $product);
        Mage::register('productId', $product->getId());
        return true;
    }

    /**
     *
     * @param Zend_Controller_Request_Http $request
     * @return boolean
     */
    protected function matchCategory(Zend_Controller_Request_Http $request)
    {
        $identifier = trim($request->getPathInfo(), '/');

        $_identifier = trim(
            $this->strReplaceOnce('/' . $this->getPrefixPart(), '', $request->getPathInfo()),
            '/'
        );

        $rewriteModel = Mage::getModel('core/url_rewrite');
        $rewriteModel->setStoreId($this->getStoreId());
        $rewriteModel->loadByRequestPath($_identifier);

        if (!$rewriteModel->getId() || !$rewriteModel->getCategoryId()) {
            return false;
        }
        $categoryId = $rewriteModel->getCategoryId();

        $catogory = Mage::getModel('catalog/category')->load($categoryId);

        if (!$catogory) {
            return false;
        }
        if ($request->getRequestUri() == $request->getPathInfo()
            && rtrim($request->getPathInfo(), '/') != $request->getPathInfo()
        ) {
            $url = Mage::getUrl() . $identifier;
            $this->redirect($url);
            $request->setDispatched(true);
            return true;
        }
        $request->setModuleName('askit')
            ->setControllerName('index')
            ->setActionName('category')
            ->setParam('category_id', $categoryId)
            ->setAlias(
                Mage_Core_Model_Url_Rewrite::REWRITE_REQUEST_PATH_ALIAS,
                $identifier
            );
        Mage::register('current_category', $catogory);
        return true;
    }

    /**
     *
     * @param Zend_Controller_Request_Http $request
     * @return boolean
     */
    protected function matchPage(Zend_Controller_Request_Http $request)
    {
        $identifier = trim($request->getPathInfo(), '/');

        $_identifier = trim(
            $this->strReplaceOnce('/' . $this->getPrefixPart(), '', $request->getPathInfo()),
            '/'
        );

        $parts = explode('/', $_identifier);
        $page   = Mage::getModel('cms/page');
        $pageId = $page->checkIdentifier($parts[0], $this->getStoreId());

        if (!$pageId) {
            return false;
        }
        if (count($parts) > 1) {
            $url = Mage::getUrl() . $this->getPrefixPart() . '/' . $parts[0];
            $this->redirect($url);
            $request->setDispatched(true);
            return true;
        }
        if ($request->getRequestUri() == $request->getPathInfo()
            && rtrim($request->getPathInfo(), '/') != $request->getPathInfo()
        ) {
            $url = Mage::getUrl() . $identifier;
            $this->redirect($url);
            $request->setDispatched(true);
            return true;
        }
        $request->setModuleName('askit')
            ->setControllerName('index')
            ->setActionName('page')
            ->setParam('page_id', $pageId)
            ->setAlias(
                Mage_Core_Model_Url_Rewrite::REWRITE_REQUEST_PATH_ALIAS,
                $identifier
            );
        return true;
    }

    protected function strReplaceOnce($search, $replace, $subject)
    {
        $newstring = $subject;
        $pos = strpos($subject, $search);
        if ($pos !== false) {
            $newstring = substr_replace($subject, $replace, $pos, strlen($search));
        }
        return $newstring;
    }

    protected function getStoreId()
    {
        return Mage::app()->getStore()->getId();
    }
}