<?php
/**
 * @package DJ-Catalog2
 * @copyright Copyright (C) DJ-Extensions.com, All rights reserved.
 * @license http://www.gnu.org/licenses GNU/GPL
 * @author url: http://dj-extensions.com
 * @author email contact@dj-extensions.com
 *
 */

defined('_JEXEC') or die;

require_once(JPATH_ROOT.'/components/com_djcatalog2/helpers/route.php');

function DJCatalog2BuildRoute(&$query)
{
	$segments = array();
	
	$app		= JFactory::getApplication();
	$menu	   = $app->getMenu('site');
	
	$params = JComponentHelper::getParams( 'com_djcatalog2' );
	$component_views = array('items' => 'items', 'item' => 'item',
		'producer' => 'producer', 'myitems' => 'myitems',
		'itemform' => 'itemform', 'producers' => 'producers',
		'cart'=>'cart', 'orders'=>'orders', 'order'=>'order',
		'checkout'=>'checkout', 'query' => 'query', 'map' => 'map',
		'archived' => 'archived', 'questions' => 'questions', 'question' => 'question',
		'compare' => 'compare', 'orderform' => 'orderform', 'questionform' => 'questionform',
		'addresses' => 'addresses', 'address' => 'address', 'configurable' => 'configurable' , 'wishes' => 'wishes');
	foreach($component_views as $view_name => $seotag) {
		$alias = $params->get('seo_'.$view_name.'_view', $view_name);
		$alias = JApplicationHelper::stringURLSafe($alias);
		if(trim(str_replace('-','',$alias)) != '') {
			$component_views[$view_name] = trim((string)$alias);
		}
	}
	
	$default_menu = $menu->getDefault();
	$menuItem = null;
	
	if (empty($query['Itemid'])) {
		//JLog::add(' empty Itemid '.print_r($query, true));
		unset($query['Itemid']);
		$menuItem = $menu->getActive();
	} else {
		//JLog::add(' NOT empty Itemid '.print_r($query, true));
		$menuItem = $menu->getItem($query['Itemid']);
	}
	
	//JLog::add(' menu Item '.@$menuItem->component.' '.print_r(@$menuItem->query,true).' -----------------------');
	
	//$option = (empty($menuItem->component)) ? null : $menuItem->component;
	
	$mOption  = (empty($menuItem->query['option'])) ? null : $menuItem->query['option'];
	$mView  = (empty($menuItem->query['view'])) ? null : $menuItem->query['view'];
	$mCatid = (empty($menuItem->query['cid'])) ? null : (int)$menuItem->query['cid'];
	$mProdCatid = (empty($menuItem->query['pcid'])) ? null : (int)$menuItem->query['pcid'];
	$mProdid   = (empty($menuItem->query['pid'])) ? null : (int)$menuItem->query['pid'];
	$mId	= (empty($menuItem->query['id'])) ? null : (int)$menuItem->query['id'];
	$mLayout = (empty($menuItem->query['layout'])) ? null : $menuItem->query['layout'];
	
	$view = !empty($query['view']) ? $query['view'] : null;
	$cid = !empty($query['cid']) ? $query['cid'] : null;
	$pcid = !empty($query['pcid']) ? $query['pcid'] : null;
	$pid = !empty($query['pid']) ? $query['pid'] : null;
	$id = !empty($query['id']) ? $query['id'] : null;
	$layout = !empty($query['layout']) ? $query['layout'] : null;
	
	$mECatid = (empty($menuItem->query['ecid'])) ? null : (int)$menuItem->query['ecid'];
	$mEId	= (empty($menuItem->query['eid'])) ? null : (int)$menuItem->query['eid'];
	
	$ecid = !empty($query['ecid']) ? $query['ecid'] : null;
	$eid = !empty($query['eid']) ? $query['eid'] : null;
	
	$qid = !empty($query['qid']) ? $query['qid'] : null;
	$oid = !empty($query['oid']) ? $query['oid'] : null;
	
	$task = !empty($query['task']) ? $query['task'] : null;
	
	// JoomSEF bug workaround
	if (isset($query['start']) && isset($query['limitstart'])) {
		if ((int)$query['limitstart'] != (int)$query['start'] && (int)$query['start'] > 0) {
			// let's make it clear - 'limitstart' has higher priority than 'start' parameter,
			// however ARTIO JoomSEF doesn't seem to respect that.
			$query['start'] = $query['limitstart'];
			unset($query['limitstart']);
		}
	}
	// JoomSEF workaround - end
	//if ($view && $option == 'com_djcatalog2') {
	if ($view) {
		if ($view != $mView || empty($query['Itemid'])) {
			if ($params->get('seo_skip_item_view', 0) == '0') {
				$segments[] = $view;
			} else {
				/*if (
				 !( ($view == 'item' || $view == 'items' || $view == 'archived') && ($mView == 'items' || $mView == 'archived') )
				 ||
				 !( $view == 'producer' && $mView == 'producers' )
				 ) {
				 $segments[] = $view;
				 }*/
				/*if ( (($mView == 'items' || $mView == 'archived') && ($view != 'item')) || ($mView == 'producers' && $view !='producer')  || $mOption != 'com_djcatalog2') {
				 $segments[] = $view;
				 }*/
				
				if ($mOption != 'com_djcatalog2'
					|| ($view != 'item' && $view != 'producer')
					|| ($view == 'item' && $mView != 'items' && $mView != 'archived')
					|| ($view == 'producer' && $mView != 'producers') ) {
						$segments[] = $view;
					}
					
				/*if ( ($view != 'item' && $view != 'producer') || $mOption != 'com_djcatalog2') {
				 $segments[] = $view;
				 }*/
			}
		}
		unset($query['view']);
		
		$mCountryId = (empty($menuItem->query['cntid'])) ? null : (int)$menuItem->query['cntid'];
		$mStateId   = (empty($menuItem->query['stid'])) ? null : (int)$menuItem->query['stid'];
		$countryid = !empty($query['cntid']) ? $query['cntid'] : null;
		$stateid = !empty($query['stid']) ? $query['stid'] : null;
		
		if ( $view != 'items' || (isset($query['cntid']) && $query['cntid'] === '') ||  (intval($mCountryId) == intval($countryid))) {
			unset($query['cntid']);
		}
		if ( $view != 'items' || (isset($query['stid']) && $query['stid'] === '') ||  (intval($mStateId) == intval($stateid))) {
			unset($query['stid']);
		}
		
		if ($view == 'item') {
			// child product URLs
			$parent = isset($query['parent']) ? $query['parent'] : false;
			
			if ($view == $mView && intval($id) > 0 && intval($id) == $mId) {
				unset($query['id']);
				unset($query['cid']);
			} else if ((($mView == 'items' || $mView == 'archived') || $mView === null) && intval($id) > 0) {
				if (intval($cid) != intval($mCatid)) {
					//$segments[] = (int)$cid == 0 ? 'all' : DJCatalogHelperRoute::formatAlias($cid);
					$segments[] = (int)$cid == 0 ? 'all' : DJCatalogHelperRoute::formatAlias($cid, 'category');
				}
				//$segments[] = DJCatalogHelperRoute::formatAlias($id);
				if ($parent) {
					$segments[] = DJCatalogHelperRoute::formatAlias($parent, 'item');
					unset($query['parent']);
				}
				$segments[] = DJCatalogHelperRoute::formatAlias($id, 'item');
				unset($query['id']);
				unset($query['cid']);
			}
			
			if ($layout && $layout != 'default' && $layout != $mLayout) {
				
				$segments[] = $layout;
				unset($query['layout']);
				
				if (($layout == 'preview' || $layout == 'print' || $layout == 'contact') && isset($query['tmpl'])) {
					unset($query['tmpl']);
				}
			}
		}
		
		if ($view == 'items' || $view == 'archived') {
			if ($cid === null) {
				$cid = '0';
			}
			if (intval($cid) != intval($mCatid) /*|| $mCatid === null*/) {
				//$segments[] = (int)$cid == 0 ? 'all' : DJCatalogHelperRoute::formatAlias($cid);
				$segments[] = (int)$cid == 0 ? 'all' : DJCatalogHelperRoute::formatAlias($cid, 'category');
			} else if (@$menuItem->id == $default_menu->id && (int)$cid == 0 && (isset($query['cm']) || isset($query['ind']) || (int)$mProdid != (int)$pid) ) {
				$segments[] = 'all';
			}
			unset($query['cid']);
			
			if ( (isset($query['pid']) && $query['pid'] === '') ||  (intval($mProdid) == intval($pid) /*&& (intval($mProdid) > 0 || intval($pid) > 0)*/)) {
				unset($query['pid']);
			}
		}
		
		if ($view == 'producers') {
			if ($pcid === null) {
				$pcid = '0';
			}
			if (intval($pcid) != intval($mProdCatid) /*|| $mCatid === null*/) {
				//$segments[] = (int)$cid == 0 ? 'all' : DJCatalogHelperRoute::formatAlias($cid);
				$segments[] = (int)$pcid == 0 ? 'all' : DJCatalogHelperRoute::formatAlias($pcid, 'producercategory');
			}
			unset($query['pcid']);
		}
		
		if ($view == 'producer') {
			if (!($view == $mView && intval($pid) > 0 && intval($pid) == $mProdid) && $mView != 'producer') {
				$segments[] = DJCatalogHelperRoute::formatAlias($pid, 'producer');
			}
			unset($query['pid']);
		}
		
		if ($view == 'itemform') {
			if (intval($id) > 0) {
				$segments[] = $id;
				unset($query['id']);
			}
		}
		
		if ($view == 'question') {
			if (intval($qid) > 0) {
				$segments[] = $qid;
				unset($query['qid']);
			}
		}
		
		if ($view == 'order') {
			if (intval($oid) > 0) {
				$segments[] = $oid;
				unset($query['oid']);
			}
		}
		
		if ($view == 'orderform') {
			if (intval($id) > 0) {
				$segments[] = $id;
				unset($query['id']);
			}
		}
		
		if ($view == 'questionform') {
			if (intval($id) > 0) {
				$segments[] = $id;
				unset($query['id']);
			}
		}
		
		if ($view == 'map') {
			if ($cid === null) {
				$cid = '0';
			}
			if (intval($cid) != intval($mCatid) /*|| $mCatid === null*/) {
				$segments[] = (int)$cid == 0 ? 'all' : DJCatalogHelperRoute::formatAlias($cid);
			} else if (@$menuItem->id == $default_menu->id && (int)$cid == 0 && (int)$mProdid != (int)$pid)  {
				$segments[] = 'all';
			}
			unset($query['cid']);
			
			if ( (isset($query['pid']) && $query['pid'] === '') ||  (intval($mProdid) == intval($pid) /*&& (intval($mProdid) > 0 || intval($pid) > 0)*/)) {
				unset($query['pid']);
			}
		}
		
		if ($view == 'cart') {
			if ($layout && $layout != 'default' /*&& $layout != 'onestep'*/ && $layout != $mLayout) {
				$segments[] = $layout;
			}
			if (isset($query['layout'])) {
				unset($query['layout']);
			}
			
			if ($layout == 'onestep') {
				if (isset($query['type'])) {
					$segments[] = $query['type'];
					unset($query['type']);
				}
			}
		}
	}
	
	if (!empty($segments[0]) && array_key_exists($segments[0], $component_views)) {
		$segments[0] = $component_views[$segments[0]];
	}
	
	return $segments;
}

function DJCatalog2ParseRoute(&$segments) {
	$app	= JFactory::getApplication();
	$menu	= $app->getMenu();
	$activemenu = $menu->getActive();
	$db = JFactory::getDBO();
	$params = JComponentHelper::getParams( 'com_djcatalog2' );
	$advanced = (int)$params->get('seo_advanced', 0);
	
	$catalogViews = array('item', 'items', 'producer', 'itemform', 'myitems', 'producers', 'cart', 'checkout', 'orders', 'order', 'query', 'map', 'archived', 'questions', 'question', 'compare', 'configurable', 'orderform', 'questionform', 'addresses', 'address', 'wishes');
	
	$component_views = array('items' => 'items', 'item' => 'item',
		'producer' => 'producer', 'myitems' => 'myitems',
		'itemform' => 'itemform', 'producers' => 'producers',
		'cart'=>'cart', 'orders'=>'orders', 'order'=>'order',
		'checkout'=>'checkout', 'query'=>'query', 'map' => 'map',
		'archived' => 'archived', 'questions' => 'questions', 'question' => 'question', 'compare' => 'compare', 'orderform' => 'orderform', 'questionform' => 'questionform', 'addresses' => 'addresses', 'address' => 'address', 'configurable' => 'configurable', 'wishes' => 'wishes');
	
	foreach($component_views as $view_name => $seotag) {
		$view_alias = $params->get('seo_'.$view_name.'_view', $view_name);
		$view_alias = JApplicationHelper::stringURLSafe(trim((string)$view_alias));
		if (count($segments)) {
			if ($segments[0] == $view_alias || str_replace(':', '-', $segments[0]) == $view_alias) {
				$segments[0] = $view_name;
				break;
			}
		}
	}
	$query=array();
	
	if (!empty($activemenu) && is_array($activemenu->query)) {
		foreach($activemenu->query as $k=>$v) {
			$query[$k] = $v;
		}
	}
	
	if (count($segments)) {
		if (!in_array($segments[0], $catalogViews)) {
			if ($activemenu) {
				$temp=array();
				$temp[0] = $activemenu->query['view'];
				switch ($temp[0]) {
					case 'item' : {
						$temp[1] = @$activemenu->query['id'];
						foreach ($segments as $k=>$v) {
							$temp[$k+1] = $v;
						}
						break;
					}
					case 'items' :
					case 'archived': {
						$temp[1] = @$activemenu->query['cid'];
						
						if (count($segments) == 1) {
							if ($segments[0] == 'all') {
								$temp[0] = 'items';
								$temp[1] = 0;
							} else {
								$tempQuery = is_array($activemenu) && isset($activemenu->query) ? $activemenu->query : array();
								$tempSegment = DJCatalogHelperRoute::parseAlias($segments[0], 'category', $tempQuery);
								if ($advanced) {
									if ($tempSegment) {
										$temp[1] = $segments[0];
									} else {
										$temp[0] = 'item';
									}
								} else {
									$parts = explode(':', $tempSegment, 2);
									$id = $parts[0];
									$alias = isset($parts[1]) ? $parts[1] : null;
									if ((int)$id > 0) {
										$user	= JFactory::getUser();
										$groups	= $user->getAuthorisedViewLevels();
										$categories = Djc2Categories::getInstance(array('state'=>'1', 'access' => $groups));
										$category = $categories->get((int)$id);
										if (!empty($category) && ($category->alias == $alias || empty($alias))) {
											$temp[1] = $segments[0];
										} else {
											$temp[0] = 'item';
										}
									} else if (!$alias || ($id == 0 && strpos($id, $tempSegment) !== 0)) {
										// this should always result in 404
										$temp[0] = 'item';
										$temp[1] = 0;
									}
								}
							}
						} else {
							$temp[0] = 'item';
						}
						
						foreach ($segments as $k=>$v) {
							$temp[$k+1] = $v;
						}
						break;
					}
					case 'producer' : {
						$temp[1] = @$activemenu->query['pid'];
						foreach ($segments as $k=>$v) {
							$temp[$k+1] = $v;
						}
						break;
					}
					case 'myitems' : {
						//$temp[1] = @$activemenu->query['id'];
						foreach ($segments as $k=>$v) {
							$temp[$k+1] = $v;
						}
						break;
					}
					case 'producers' : {
						$temp[1] = @$activemenu->query['pcid'];
						
						if (count($segments) == 1) {
							if ($segments[0] == 'all') {
								$temp[0] = 'producers';
								$temp[1] = 0;
							} else {
								$tempQuery = is_array($activemenu) && isset($activemenu->query) ? $activemenu->query : array();
								$tempSegment = DJCatalogHelperRoute::parseAlias($segments[0], 'producercategory', $tempQuery);
								if ($advanced) {
									if ($tempSegment) {
										$temp[1] = $segments[0];
									} else {
										$temp[0] = 'producer';
									}
								} else {
									$parts = explode(':', $tempSegment, 2);
									$id = $parts[0];
									$alias = isset($parts[1]) ? $parts[1] : null;
									if ((int)$id > 0) {
										$user	= JFactory::getUser();
										$groups	= $user->getAuthorisedViewLevels();
										$categories = Djc2ProducerCategories::getInstance(array('state'=>'1', 'access' => $groups));
										$category = $categories->get((int)$id);
										if (!empty($category) && ($category->alias == $alias || empty($alias))) {
											$temp[1] = $segments[0];
										} else {
											$temp[0] = 'producer';
										}
									} else if (!$alias || ($id == 0 && strpos($id, $tempSegment) !== 0)) {
										// this should always result in 404
										$temp[0] = 'producer';
										$temp[1] = 0;
									}
								}
							}
						} else {
							$temp[0] = 'producer';
						}
						
						foreach ($segments as $k=>$v) {
							$temp[$k+1] = $v;
						}
						break;
					}
					case 'itemform' : {
						foreach ($segments as $k=>$v) {
							$temp[$k+1] = $v;
						}
						break;
					}
					case 'orderform' : {
						foreach ($segments as $k=>$v) {
							$temp[$k+1] = $v;
						}
						break;
					}
					case 'questionform' : {
						foreach ($segments as $k=>$v) {
							$temp[$k+1] = $v;
						}
						break;
					}
					case 'cart' :
					case 'checkout' :
					case 'query' :
					case 'orders' :
					case 'order' :
					case 'questions' :
					case 'question' : {
						foreach ($segments as $k=>$v) {
							$temp[$k+1] = $v;
						}
						break;
					}
					case 'map' : {
						$temp[1] = @$activemenu->query['cid'];
						foreach ($segments as $k=>$v) {
							$temp[$k+1] = $v;
						}
						break;
					}
				}
				
				$segments = $temp;
			}
		}
		if (isset($segments[0])) {
			switch($segments[0]) {
				case 'items':
				case 'archived': {
					$query['view'] = $segments[0];
					if (isset($segments[1]) && $segments[1] != '') {
						//$query['cid']=($segments[1] == 'all') ? 0 : DJCatalogHelperRoute::parseAlias($segments[1]);
						$query['cid']=($segments[1] == 'all') ? 0 : DJCatalogHelperRoute::parseAlias($segments[1], 'category', $query);
					}
					break;
				}
				case 'itemstable': {
					$query['view'] = 'itemstable';
					if (isset($segments[1])) {
						//$query['cid']=($segments[1] == 'all') ? 0 :DJCatalogHelperRoute::parseAlias($segments[1]);
						$query['cid']=($segments[1] == 'all') ? 0 :DJCatalogHelperRoute::parseAlias($segments[1], 'category', $query);
					}
					break;
				}
				case 'item': {
					$query['view'] = 'item';
					
					// checking if last segment matches layout
					$layout = null;
					$layouts = array('preview', 'print', 'contact');
					$scount = count($segments);
					if (in_array($segments[$scount-1], $layouts)) {
						$layout = $segments[$scount-1];
						unset($segments[$scount-1]);
						
						if ($layout == 'preview' || $layout == 'print' || $layout == 'contact') {
							$query['tmpl'] = 'component';
						}
					}
					
					if (count($segments) > 2) {
						if (count($segments) > 3) {
							if (isset($segments[1]) && $segments[1] != '') {
								//$query['cid']=($segments[1] == 'all') ? 0 : DJCatalogHelperRoute::parseAlias($segments[1]);
								$query['cid']=($segments[1] == 'all') ? 0 : DJCatalogHelperRoute::parseAlias($segments[1], 'category', $query);
							}
							
							if (isset($segments[2])) {
								$query['parent']= DJCatalogHelperRoute::parseAlias($segments[2], 'item', $query);
							}
							
							if (isset($segments[3])) {
								$query['id']= DJCatalogHelperRoute::parseAlias($segments[3], 'variant', $query);
							}
							
						} else if (count($segments) == 3) {
							if (isset($segments[1])) {
								$query['parent']= DJCatalogHelperRoute::parseAlias($segments[1], 'item', $query);
							}
							
							if (isset($segments[2])) {
								$query['id']= DJCatalogHelperRoute::parseAlias($segments[2], 'variant', $query);
							}
						} else {
							if (isset($segments[2])) {
								$query['id']= DJCatalogHelperRoute::parseAlias($segments[2], 'item', $query);
							}
						}
						
					} else if (isset($segments[1])) {
						//$query['id']=  DJCatalogHelperRoute::parseAlias($segments[1]);
						$query['id']=  DJCatalogHelperRoute::parseAlias($segments[1], 'item', $query);
						if ($activemenu && $activemenu->query['option'] == 'com_djcatalog2' && $activemenu->query['view'] == 'items' && !empty($activemenu->query['cid'])) {
							$query['cid'] = $activemenu->query['cid'];
						}
					}
					
					if ($layout) {
						$query['layout'] = $layout;
					}
					
					break;
				}
				case 'producer': {
					$query['view'] = 'producer';
					if (isset($segments[1])) {
						//$query['pid']=$segments[1];
						$query['pid']=  DJCatalogHelperRoute::parseAlias($segments[1], 'producer', $query);
					}
					break;
				}
				case 'itemform': {
					$query['view'] = 'itemform';
					if (isset($segments[1])) {
						$query['id']=$segments[1];
					}
					break;
				}
				case 'myitems': {
					$query['view'] = 'myitems';
					break;
				}
				case 'producers': {
					$query['view'] = $segments[0];
					if (isset($segments[1]) && $segments[1] != '') {
						//$query['cid']=($segments[1] == 'all') ? 0 : DJCatalogHelperRoute::parseAlias($segments[1]);
						$query['pcid']=($segments[1] == 'all') ? 0 : DJCatalogHelperRoute::parseAlias($segments[1], 'producercategory', $query);
					}
					break;
				}
				case 'cart': {
					if (count($segments) == 3) {
						$query['layout'] = $segments[1];
						$query['type'] = $segments[2];
					}  else if (count($segments) == 2) {
						$query['layout'] = $segments[1];
					} else if (count($segments) == 1) {
						$query['layout'] = $segments[0];
					}
					$query['view'] = 'cart';
					break;
				}
				case 'orders': {
					$query['view'] = 'orders';
					break;
				}
				case 'checkout': {
					if ($params->get('cart_one_step')) {
						$query['type'] = 'checkout';
						$query['view'] = 'cart';
						$query['layout'] = 'onestep';
					} else {
						$query['view'] = 'checkout';
					}
					break;
				}
				case 'query': {
					if ($params->get('cart_one_step')) {
						$query['type'] = 'query';
						$query['view'] = 'cart';
						$query['layout'] = 'onestep';
					} else {
						$query['view'] = 'query';
					}
					break;
				}
				case 'order': {
					$query['view'] = 'order';
					if (isset($segments[1])) {
						$query['oid']=$segments[1];
					}
					if (isset($segments[2])) {
						$query['layout']=$segments[2];
					}
					break;
				}
				case 'orderform': {
					$query['view'] = 'orderform';
					if (isset($segments[1])) {
						$query['id']=$segments[1];
					}
					break;
				}
				case 'questions': {
					$query['view'] = 'questions';
					break;
				}
				case 'question': {
					$query['view'] = 'question';
					if (isset($segments[1])) {
						$query['qid']=$segments[1];
					}
					break;
				}
				case 'questionform': {
					$query['view'] = 'questionform';
					if (isset($segments[1])) {
						$query['id']=$segments[1];
					}
					break;
				}
				case 'map': {
					$query['view'] = 'map';
					if (isset($segments[1])) {
						$query['cid']=($segments[1] == 'all') ? 0 : DJCatalogHelperRoute::parseAlias($segments[1]);
					}
					break;
				}
				default: {
					$query['view'] = $segments[0];
				}
			}
		}
	}
	
	if (version_compare(JVERSION, '4', 'ge')) {
		$segments = [];
	}
	
	return $query;
}


class DJCatalog2Router implements JComponentRouterInterface
{
	
	public function build(&$query)
	{
		return DJCatalog2BuildRoute($query);
	}
	
	/*
	 * This function take a category id and finds the path from that category to the root of the category tree
	 * The path returned from getPath() is an associative array of key = category id, value = id:alias
	 * If no valid category is found from the passed-in category id then null is returned.
	 */
	
	public function parse(&$segments)
	{
		return DJCatalog2ParseRoute($segments);
	}
	
	public function preprocess($query)
	{
		static $currentLang = null;
		if (JLanguageAssociations::isEnabled())
		{
			$app  = JFactory::getApplication();
			$sitemenu = $app->getMenu();
			if (!isset($query['lang']))
			{
				return $query;
			}
			
			$lang = $query['lang'];
			
			if ($query['lang'] == $currentLang)
			{
				return $query;
			}
			
			if (!isset($query['Itemid']))  // we're currently on /component type of URL
			{
				// use the home page for the URL's language
				$home = $sitemenu->getItems(array('language','home'), array($lang, true));
				if ($home)
				{
					$query['Itemid'] = $home[0]->id;
				}
				return $query;
			}
			
			$itemid = (!empty($query['Itemid'])) ? $query['Itemid'] : false;
			
			if ($itemid) {
				// ensure the menuitem for Itemid has the correct language
				$thismenuitem = $sitemenu->getItem($itemid);
				if (!empty($thismenuitem)) {
					$thismenuitemLang = $thismenuitem->language;
					if ($thismenuitemLang == $lang)
					{
						$currentLang = $thismenuitemLang;
					}
					if ($thismenuitemLang == $lang || $thismenuitemLang == '*')
					{
						return $query;
					}
					
					// if not, try to find an associated menuitem with the correct language
					$associations = JLanguageAssociations::getAssociations('com_menus', '#__menu', 'com_menus.item', $itemid, 'id', '', '');
					
					if (isset($associations[$lang]))
					{
						$query['Itemid'] = (int) $associations[$lang]->id;
						return $query;
					}
					else // use the home page for that language (if it's set)
					{
						$home = $sitemenu->getItems(array('language','home'), array($lang, true));
						if ($home)
						{
							$query['Itemid'] = $home[0]->id;
						}
					}
				}
			}
		}
		return $query;
	}
}