una/modules/boonex/elasticsearch/classes/BxElsApi.php

<?php defined('BX_DOL') or die('hack attempt');
/**
 * Copyright (c) UNA, Inc - https://una.io
 * MIT License - https://opensource.org/licenses/MIT
 *
 * @defgroup    ElasticSearch ElasticSearch
 * @ingroup     UnaModules
 *
 * @{
 */

define('BX_ELASTICSEARCH_LOG', 1);

/**
 * ElasticSearch API
 */
class BxElsApi extends BxDol
{
    static $TYPE = '_doc';
    
	protected $_oModule;

    protected $_sError = false;
    protected $_iTimeout = 30;
    protected $_sApiUrl = 'http://localhost:9200';

    protected $_aOprs = array('AND' => array('AG' => 'bool', 'FN' => 'filter'), 'OR' => array('AG' => 'dis_max', 'FN' => 'queries'));

    public function __construct(&$oModule)
    {
        $this->_oModule = $oModule;

        $CNF = &$this->_oModule->_oConfig->CNF;

        $this->_sApiUrl = getParam($CNF['PARAM_API_URL']);
    }

    public function searchSimple($sIndex, $sType, $sTerm) 
    {
        if ($sType) {
            $sQuery = "/{$sIndex}@{$sType}/_search";
            return $this->api($sQuery, array('query' => array('simple_query_string' => array('query' => $sTerm))), 'post');
        } 
        else {
            $this->log('ERROR: searchSimple - search without specifying particular module(type) isn\'t supported', '', BX_LOG_ERR);
        }
    }

    public function searchExtended($sIndex, $sType, $aCondition, $aSelection = array()) 
    {
        if ($sType) {
            $sQuery = "/{$sIndex}@{$sType}/_search";

            $aQuery = array('query' => $this->_prepareConditionsGroup($aCondition));
            if(!empty($aSelection))
                $aQuery = array_merge($aQuery, $aSelection);

            return $this->api($sQuery, $aQuery, 'post');
        }
        else {
            $this->log('ERROR: searchExtended - search without specifying particular module(type) isn\'t supported', BX_LOG_ERR);
        }
    }

    protected function _prepareConditionsGroup($aGrp)
    {
        if(!isset($aGrp['grp']) || !(bool)$aGrp['grp'])
			return $this->_prepareConditionsSingle($aGrp);

        $aResult = array();
    	if(!isset($aGrp['opr'], $aGrp['cnds']) || empty($aGrp['cnds']) || !is_array($aGrp['cnds']))
    		return $aResult;

        $aOpr = $this->_aOprs[$aGrp['opr']];
        $aResult[$aOpr['AG']] = array($aOpr['FN'] => array());

        foreach($aGrp['cnds'] as $aCnd) {
    		$sMethod = '_prepareConditions' . (isset($aCnd['grp']) && (bool)$aCnd['grp'] ? 'Group' : 'Single');
    		$aResultCnd = $this->$sMethod($aCnd);
    		if(!empty($aResultCnd))
    		    $aResult[$aOpr['AG']][$aOpr['FN']][] = $aResultCnd;
    	}

        return $aResult;
    }

    protected function _prepareConditionsSingle($aCnd)
    {
        if(!isset($aCnd['val']) && !(isset($aCnd['val1']) || isset($aCnd['val2'])))
            return array();

        $bTp = !empty($aCnd['tp']);
        $bFld = !empty($aCnd['fld']);
        $bOprIn = !empty($aCnd['opr']) && $aCnd['opr'] == 'IN';

        if($bTp)
            $sType = $aCnd['tp'];
        else if($bFld) {
            if($bOprIn)
                $sType = 'terms';
            else 
                $sType = 'term';
        }
        else 
            $sType = 'simple_query_string';

        $aQuery = array();
        switch ($sType) {
            case 'term':
            case 'terms':
                $aQuery = array($aCnd['fld'] => $aCnd['val']);
                break;

            case 'range':
                $aOpr = array();
                if(!empty($aCnd['opr1']) && isset($aCnd['val1']))
                    $aOpr[$aCnd['opr1']] = $aCnd['val1'];
                if(!empty($aCnd['opr2']) && isset($aCnd['val2']))
                    $aOpr[$aCnd['opr2']] = $aCnd['val2'];

                $aQuery = array($aCnd['fld'] => $aOpr);
                break;

            case 'simple_query_string':
                $aQuery = array('query' => $aCnd['val']);
                break;
        }

        return array($sType => $aQuery);
    }

    public function getData($sIndex, $sType, $iContentId) 
    {
        return $this->api("/{$sIndex}@{$sType}/" . self::$TYPE . "/$iContentId");
    }

    public function indexData($sIndex, $sType, $iContentId, $aData) 
    {
        return $this->api("/{$sIndex}@{$sType}/" . self::$TYPE . "/$iContentId", $aData, 'put');
    }

    public function updateData($sIndex, $sType, $iContentId, $aData)
    {
        return $this->api("/{$sIndex}@{$sType}/" . self::$TYPE . "/$iContentId", $aData, 'post');
    }

    public function deleteData($sIndex, $sType, $iContentId)
    {
        return $this->api("/{$sIndex}@{$sType}/" . self::$TYPE . "/$iContentId", array(), 'delete');
    }

    public function getErrorMsg() 
    {
        return $this->_sError;
    }
    
    public function api($sAction, $aData = array(), $sMetod = 'get', $bJsonResponse = true) 
    {
        if (!$this->_sApiUrl) {
            $this->_sError = 'Host isn\'t defined';
            return null;
        }

        if (!$sAction) {
            $this->_sError = 'Action isn\'t defined';
            return null;
        }        

        $sUrl = bx_append_url_params($this->_sApiUrl . $sAction, array(
            'format' => 'json',
            'human' => false,
        ));

        $sHttpCode = 200;
        $s = $this->curl($sUrl, $aData, $sMetod, $sHttpCode);

        $this->log($s, '', BX_LOG_DEBUG);

        if ($sHttpCode >= 300 || $sHttpCode < 200) {
            $this->_sError = 'Action('.$sAction.') returned status - ' . $sHttpCode;
            return null;
        }

        // return JSON response
        if ($s && $bJsonResponse) {
            if (null === ($a = json_decode($s, true))) {
                $this->_sError = 'Can not decode response';
                return null;
            }
            return $a;
        }
    
        return $s;
    }

    protected function curl ($sUrl, $aData = array(), $sMethod = 'get', &$sHttpCode = null, $aBasicAuth = array()) 
    {
        $rConnect = curl_init();

        curl_setopt($rConnect, CURLOPT_USERAGENT, 'UNA ' . BX_DOL_VERSION);
        curl_setopt($rConnect, CURLOPT_URL, $sUrl);
        curl_setopt($rConnect, CURLOPT_HEADER, NULL === $sHttpCode ? false : true);
        curl_setopt($rConnect, CURLOPT_RETURNTRANSFER, 1);
        curl_setopt($rConnect, CURLOPT_VERBOSE, 0);
        curl_setopt($rConnect, CURLOPT_HEADER, 0);

        curl_setopt($rConnect, CURLOPT_CONNECTTIMEOUT, $this->_iTimeout);
        curl_setopt($rConnect, CURLOPT_TIMEOUT, $this->_iTimeout);

        if (!ini_get('open_basedir'))
            curl_setopt($rConnect, CURLOPT_FOLLOWLOCATION, 1);

        if ($aBasicAuth)
            curl_setopt($rConnect, CURLOPT_USERPWD, $aBasicAuth['user'] . ':' . $aBasicAuth['password']);

        if ('get' != $sMethod)
            curl_setopt($rConnect, CURLOPT_CUSTOMREQUEST, strtoupper($sMethod));

        if ('post' == $sMethod)
            curl_setopt($rConnect, CURLOPT_POST, true);

        if ($aData) {
            curl_setopt($rConnect, CURLOPT_POSTFIELDS, json_encode($aData));
            curl_setopt($rConnect, CURLOPT_HTTPHEADER, array('Content-Type: application/json'));
        }

        $sResult = curl_exec($rConnect);

        if (NULL !== $sHttpCode)
            $sHttpCode = curl_getinfo($rConnect, CURLINFO_HTTP_CODE);

        return $sResult;
    }

    protected function log ($mixed, $sProvider = '', $iLevel = BX_LOG_ERR)
    {
        if (!defined('BX_ELASTICSEARCH_LOG') || !constant('BX_ELASTICSEARCH_LOG'))
            return;
        
		if (is_array($mixed))
			$mixed = var_export($mixed, true);	
		else if (is_object($mixed))
			$mixed = json_encode($mixed);
		
		bx_log('bx_elasticsearch', $mixed, $iLevel);
    }
}

/** @} */