<?php

namespace ggbn;

use Yii;
use Solarium;

/*
 * This QueryManager handles all SOLR-Queries
 */
class SOLRQueryManager {
	
	// place here all queries for the field results:
	// use the correct field var names of the index doc (should match with model field)!
	public static function getPartialQuery($field, $value, $filters, $details) {
		// echo "getPartialQuery";
		if ($value != "\(\)" && $value != "coming\ soon") {
			switch ($field) {
				case 'fullScientificName' :
					return $field . '_nc:' . $value . '';
					break;
				case 'kingdom' :
					if ($value == "Not\ assigned")
						return '(regnum:' . $value . ' OR N/A OR null)';
					else
						return 'regnum:' . $value . '';
					break;
				case 'phylumName' :
					if ($value == "Not\ assigned")
						return '(phylum:' . $value . ' OR N/A OR null)';
					else
						return 'phylum:' . $value . '';
					break;
				case 'orderName' :
					if ($value == "Not\ assigned")
						return '(ordo:' . $value . ' OR N/A OR null)';
					else
						return 'ordo:' . $value . '';
					break;
				case 'familyName' :
					if ($value == "Not\ assigned")
						return '(familia:' . $value . ' OR N/A OR null)';
					else
						return 'familia:' . $value . '';
					break;
				case 'genusName' :
					if ($value == "Not\ assigned")
						return '(genus:' . $value . ' OR N/A OR null)';
					else
						return 'genus:' . $value . '';
					break;
				case 'className' :
					if ($value == "Not\ assigned")
						return '(classis:' . $value . ' OR N/A OR null)';
					else
						return 'classis:' . $value . '';
					break;
				case 'geneticlocus' :
					return $field . ':"' . $value . '"';
					break;
				case 'collectioncode' :
					return $field . ':"' . $value . '"';
					break;
				case 'institutioncode' :
					return $field . ':"' . $value . '"';
					break;
					return $field . ':"' . $value . '"';
					break;
				case 'locality' :
					return $field . '_nc:*' . $value . '*';
					break;
				case 'collectors' :
					return $field . '_nc:*' . $value . '*';
					break;
				case 'collectornumber' :
					return 'collectornumbers_nc:' . $value;
					break;
				case 'cites' :
					if ($value == "include")
						return "citesStatus:*";
					if ($value == "exclude")
						return "citesStatus:N/A";
					if ($value == "citesOnly")
						return "citesStatus:I*";
					break;
				case 'sampleavailability' :
					if ($value == "only")
						return " NOT disposition_clean:missing AND NOT disposition_clean:consumed ";
					if ($value == "all")
						return "*:*";
					break;
				case 'sequenceaccessionIdentifier' :
					return 'sequenceaccessionIdentifier_nc:' . $value;
					break;
				case 'unitID' :
					// case 'sample' :
					if ($value == '*') {
						return 'unitID_nc:*';
					} else {
						// if (! isset ( $_GET ['collectioncode'] ) or $_GET ['collectioncode'] != "") {
						// return $field . ':"' . $value . '"';
						// } else {
						if ($details)
							return '(' . $field . '_nc:' . $value . ' OR unitaccessionnumber:' . $value . ' OR trick17ID:' . $value . ')';
						else
							return '(' . $field . '_nc:' . $value . ' OR unitaccessionnumber_nc:' . $value . ' OR trick17ID:' . $value . ')';
						// return '(' . $field . ':"' . $value . '" OR unitaccessionnumber:"' . $value . '"*)';
						// }
					}
					// return "country:Germany"; }
					;
					break;
				case 'hasImage' :
					return $field . ":" . $value;
					break;
				case 'hasTypestatus' :
					return $field . ":" . $value;
					break;
				// lists
				// they alway deliver a value so we have to check
				// if it is the "nothing choosen value"
				// then we won't need a query
				case 'country' :
					return $field . ":" . $value;
					break;
				case 'recordbasis' :
					return $field . ":" . $value;
					break;
				case 'isocountrycode' :
					return $field . ":" . $value;
					break;
				case 'continent' :
					return $field . ":" . $value;
					break;
				case 'sea' :
					return $field . ":" . $value;
					break;
				case 'ocean' :
					return $field . ":" . $value;
					break;
				case 'sampletype' :
					return $field . ":" . $value;
					break;
				case 'institution' :
					// if ($value == $model->getSelectionListNonChosen () or $value == '0') {
					if ($value == "---" or $value == '0') { // Gabis Version
						return false;
					} else {
						return $field . ":\"" . $value . "\"";
					}
					break;
				// fields with that come as a pair and provide a range for the value
				case 'latitudeFrom' :
					// TODO this has to go to form validation!!!
					return SOLRQueryManager::handleCoordinates ( $filters );
					break;
				case 'latitudeTo' :
					// TODO this has to go to form validation!!!
					return SOLRQueryManager::handleCoordinates ( $filters );
					break;
				
				case 'longitudeFrom' :
					// TODO this has to go to form validation!!!
					return SOLRQueryManager::handleCoordinates ( $filters );
					break;
				case 'longitudeTo' :
					// TODO this has to go to form validation!!!
					return SOLRQueryManager::handleCoordinates ( $filters );
					break;
				case 'collectionYearFrom' :
					// TODO this has to go to form validation!!!
					return SOLRQueryManager::handleYear ( $filters );
					break;
				// collection year
				case 'collectionYearTo' :
					// TODO this has to go to form validation!!!
					return SOLRQueryManager::handleYear ( $filters );
					break;
				default : // this should never happen
					return $field . ":" . $value;
					break;
			}
		}
		// return $query[$field];
	}
	static function handleCoordinates($filters) {
		$latf = NULL;
		$latt = NULL;
		$longf = NULL;
		$longt = NULL;
		if (isset ( $filters ['latitudeTo'] ))
			$latt = str_replace ( ",", ".", $filters ['latitudeTo'] );
		if (isset ( $filters ['latitudeFrom'] ))
			$latf = str_replace ( ",", ".", $filters ['latitudeFrom'] );
		if (isset ( $filters ['longitudeFrom'] ))
			$longf = str_replace ( ",", ".", $filters ['longitudeFrom'] );
		if (isset ( $filters ['longitudeTo'] ))
			$longt = str_replace ( ",", ".", $filters ['longitudeTo'] );
		
		if ($latf == NULL && $latt == NULL) {
			$latf = - 90;
			$latt = 90;
		} elseif ($latf == NULL && $latt != NULL) {
			$latf = $latt - 0.10;
			$latf = - 90;
		} elseif ($latf != NULL && $latt == NULL) {
			$latt = $latf + 0.10;
			$latt = 90;
		}
		
		if ($longf == NULL && $longt == NULL) {
			$longf = - 180;
			$longt = 180;
		} elseif ($longf == NULL && $longt != NULL) {
			$longf = $longt - 0.10;
			$longf = - 180;
		} elseif ($longf != NULL && $longt == NULL) {
			$longt = $longf + 0.10;
			$longt = 180;
		}
		
		return "coordinates:[" . $latf . "," . $longf . " TO " . $latt . "," . $longt . "]";
	}
	static function handleYear($filters) {
		$yearf = 0;
		$yeart = date ( 'Y' );
		if (isset ( $filters ['collectionYearFrom'] ))
			$yearf = str_replace ( ",", ".", $filters ['collectionYearFrom'] );
		if (isset ( $filters ['collectionYearTo'] ))
			$yeart = str_replace ( ",", ".", $filters ['collectionYearTo'] );
		
		return "gatheringyear:[" . $yearf . " TO " . $yeart . "]";
	}
	public static $client;
	/**
	 * this fun creates an index query for the search request
	 *
	 * @param unknown $model        	
	 * @param unknown $selectionLists        	
	 * @param number $pageSize        	
	 * @return SolrQuery - if any value in the form was set, else return false
	 */
	public static function createSearchQuery($config) {
		
		// create a client instance
		SOLRQueryManager::$client = new Solarium\Client ( $config );
		// set the adapter to http
		SOLRQueryManager::$client->setAdapter ( 'Solarium\Core\Client\Adapter\Http' );
		
		SOLRQueryManager::$client->getPlugin ( 'postbigrequest' );
		// get a select query instance
		$query = SOLRQueryManager::$client->createSelect ();
		
		return $query;
	}
	public static function fillAndExecuteQuery($query, $queryString, $fq, $filters, $details) {
		 Yii::info ( "fillAndExecuteQuery " . $queryString . " and fq: " . $fq );
		
		// foreach ( $filters as $k => $v )
		// Yii::info ( "fillAndExecuteQuery request param " . $k . ", " . $v );
		$query->setQuery ( $queryString );
		// Note: SOLR starts counting with row 0, page 2 needs to start with row 50 not 51!
		if (isset ( $_GET ['page'] )) {
			$query->setStart ( Yii::$app->params ['listPerPage'] * ($_GET ['page'] - 1) );
		} else {
			$query->setStart ( 0 );
		}
		
		$query->setRows ( Yii::$app->params ['maxLines'] );
		$filterq = [ ];
		if (isset ( $filters )) {
			if (sizeof ( $filters ) == 0)
				$query->createFilterQuery ( 'myfilter' )->setQuery ( $fq );
			else
				foreach ( $filters as $key => $value ) {
					// Yii::info ( "hop " . $key . "---" . $value );
					$value = str_replace ( " ", "\ ", $value );
					$value = str_replace ( "(", "\(", $value );
					$value = str_replace ( ")", "\)", $value );
					$value = str_replace ( ":", "\:", $value );
					$filterq [] = SOLRQueryManager::getPartialQuery ( $key, $value, $filters, $details );
					
					 Yii::info ( "filterquery " . implode(" GLUE ",$filterq ));
					
					// }
				}
		}
		if (isset ( $filterq )) {
			if (is_array ( $fq )) {
				$query->createFilterQuery ( 'myfilter_' )->setQuery ( $fq [0] . implode ( " AND ", $filterq ) );
				if (sizeof ( $fq ) > 1) {
					for($i = 1; $i < sizeof ( $fq ); $i ++)
						$query->createFilterQuery ( 'myfilter' . $i . '_' )->setQuery ( $fq [$i] );
				}
			} else
				$query->createFilterQuery ( 'myfilter_' )->setQuery ( $fq . implode ( " AND ", $filterq ) );
		}
		
		$debug = $query->getDebug ();
		Yii::info ( 'Go query '); 
// 		try {
			$resultset = SOLRQueryManager::$client->select ( $query );
			$debugResult = $resultset->getDebug ();
			
			Yii::info ( 'Querystring: ' . $debugResult->getQueryString () );
			Yii::info ( 'Parsed query: ' . $debugResult->getParsedQuery () );
			Yii::info ( 'Query parser: ' . $debugResult->getQueryParser ());
			Yii::info ( 'Other query: ' . $debugResult->getOtherQuery ()  );
			
// 		} catch ( Solarium\Exception\UnexpectedValueException $e ) {
// 			$resultset = SOLRQueryManager::$client->select ( $query );
// 		}
		
		return $resultset;
	}
	public static function browseQuery($query, $queryString, $fqs, $page, $facetNames) {
		Yii::info ( "browseQuery " . $queryString );
		
		$query->setQuery ( $queryString );
		// Note: SOLR starts counting with row 0, page 2 needs to start with row 50 not 51!
		if ($page) {
			$query->setStart ( $page - 1 )->setRows ( Yii::$app->params ['listPerPage'] );
		} else {
			$query->setStart ( 0 )->setRows ( Yii::$app->params ['listPerPage'] );
		}
		
		$facetSet = $query->getFacetSet ();
		$facetSet->setLimit ( - 1 );
		$facetSet->setMinCount ( 1 );
		$facetSet->setSort ( "false" );
		
		$filtering = $fqs;
		
		$i = 0;
		if (sizeof ( $fqs ) > 0) {
			foreach ( $fqs as $fq ) {
				$query->createFilterQuery ( 'myfilter' . $i )->setQuery ( $fq );
				foreach ( $facetNames as $fname )
					if (sizeof ( explode ( $fname, $fq, - 1 ) ) > 0) {
						$facetSet->createFacetQuery ( $fname . $i )->setQuery ( $filtering [$i] );
					}
				$i ++;
			}
		} else {
			$fq = "*:*";
			$query->createFilterQuery ( 'myfilter' )->setQuery ( $fq );
		}
		
		// create a facet field instance and set options
		foreach ( $facetNames as $fname )
			$facetSet->createFacetField ( $fname )->setField ( $fname );
		
		$resultset = SOLRQueryManager::$client->select ( $query );
		
		return $resultset;
	}
	public static function apiQuery($query, $queryString, $fqs, $facetNames, $outputfields) {
		Yii::info ( "apiQuery " . $queryString . " " . implode ( " -- ", $facetNames ) );
		
		$query->setQuery ( $queryString );
		$query->setStart ( 0 );
		
		$query->setFields ( [ 
				$outputfields 
		] );
		
		$facetSet = $query->getFacetSet ();
		$facetSet->setLimit ( - 1 );
		$facetSet->setMinCount ( 1 );
		$facetSet->setSort ( "false" );
		
		$filtering = $fqs;
		// MARHCE PAS
		
		// foreach ( $facetNames as $fname )
		// $facetSet->createFacetField ( $fname )->setField ( $fname );
		
		$query->createFilterQuery ( 'myfilter' )->setQuery ( $fqs );
		
		// create a facet field instance and set options
		foreach ( $facetNames as $fname )
			$facetSet->createFacetField ( $fname )->setField ( $fname );
		
		$resultset = SOLRQueryManager::$client->select ( $query );
		return $resultset;
	}
	public static function treeQuery($query, $queryString, $fname) {
		// Yii::info ( "treeQuery " . $queryString );
		$query->setQuery ( $queryString );
		$query->setStart ( 0 )->setRows ( Yii::$app->params ['listPerPage'] );
		
		$facetSet = $query->getFacetSet ();
		$facetSet->setLimit ( - 1 );
		$facetSet->setMinCount ( 1 );
		$facetSet->setSort ( "false" );
		
		$fq = "*:*";
		$query->createFilterQuery ( 'myfilter' )->setQuery ( $fq );
		
		// create a facet field instance and set options
		$facetSet->createFacetField ( $fname )->setField ( $fname );
		
		$resultset = SOLRQueryManager::$client->select ( $query );
		
		return $resultset;
	}
	public static function loanQuery($query, $queryString, $fqs) {
		$query->setQuery ( $queryString );
		// print ($queryString);
		// Note: SOLR starts counting with row 0, page 2 needs to start with row 50 not 51!
		if (isset ( $_GET ['page'] )) {
			$query->setStart ( Yii::$app->params ['listPerPage'] * ($_GET ['page'] - 1) );
		} else {
			$query->setStart ( 0 );
		}
		
		$query->setRows ( Yii::$app->params ['maxLines'] );
		// generate query string:
		$string = 'tripleidstoreid:(' . implode ( $fqs, ' OR ' ) . ')';
		$query->createFilterQuery ( 'filter1' )->setQuery ( 'tripleidstoreid:(' . implode ( $fqs, ' OR ' ) . ')' ); // anf\FChrungszeichen!
		                                                                                                            // $debug = $query->getDebug ();
		$resultset = SOLRQueryManager::$client->select ( $query );
		// $debugResult = $resultset->getDebug ();
		// Yii::info ( 'Querystring: ' . $debugResult->getParsedQuery () );
		// Yii::info ( 'NumFound: ' . $resultset->getNumFound () );
		return $resultset;
	}
	
	/**
	 *
	 * @param SolrQuery $criteria        	
	 * @return SolrObject
	 */
	public static function getConfigDetails() {
		$config = array (
				'endpoint' => array (
						'bgbm-web01' => array (
								'host' => Yii::$app->params ['solrhost'],
								'port' => 8080,
								'path' => Yii::$app->params ['solrdetails'] 
						) 
				) 
		);
		return $config;
	}
	public static function getConfigSearch() {
		$config = array (
				'endpoint' => array (
						'bgbm-web01' => array (
								'host' => Yii::$app->params ['solrhost'],
								'port' => 8080,
								'path' => Yii::$app->params ['solrsearch'] 
						) 
				) 
		);
		return $config;
	}
	public static function getConfigPreview() {
		$config = array (
				'endpoint' => array (
						'bgbm-web01' => array (
								'host' => Yii::$app->params ['solrhost'],
								'port' => 8080,
								'path' => Yii::$app->params ['solrpreview'] 
						) 
				) 
		);
		return $config;
	}
	public static function getConfigStats() {
		$config = array (
				'endpoint' => array (
						'bgbm-web01' => array (
								'host' => Yii::$app->params ['solrhost'],
								'port' => 8080,
								'path' => Yii::$app->params ['solrstats'] 
						) 
				) 
		);
		return $config;
	}
	public static function getConfigBackbone() {
		$config = array (
				'endpoint' => array (
						'bgbm-web01' => array (
								'host' => Yii::$app->params ['solrhost'],
								'port' => 8080,
								'path' => '/solr/backbone' 
						) 
				) 
		);
		return $config;
	}
	public static function getConfigColTree() {
		$config = array (
				'endpoint' => array (
						'bgbm-web01' => array (
								'host' => Yii::$app->params ['solrhost'],
								'port' => 8080,
								'path' => '/solr/colTree' 
						) 
				) 
		);
		return $config;
	}
	public static function getConfigWebsites() {
		$config = array (
				'endpoint' => array (
						'bgbm-web01' => array (
								'host' => Yii::$app->params ['solrhost'],
								'port' => 8080,
								'path' => Yii::$app->params ['solrwebsites'] 
						) 
				) 
		);
		return $config;
	}
}
?>