Mirocow/yii2-elasticsearch

<?php
namespace mirocow\elasticsearch\components\queries\helpers;

/**
 * Class ScriptHelper
 * @package mirocow\elasticsearch\components\queries\helpers
 *
 * @see https://www.elastic.co/guide/en/elasticsearch/painless/5.6/painless-examples.html
 * @see https://docs.oracle.com/javase/specs/jls/se8/html/index.html
 */
class ScriptHelper extends QueryHelper
{
    /**
     * @see https://www.elastic.co/guide/en/elasticsearch/reference/current/search-request-sort.html#_script_based_sorting
     *
     * @example doc['column'].values.size()
     * @param string $column
     * @param int $direction
     * @return array
     */
    public static function sort($script = '', int $direction = SORT_ASC, $language = 'painless') :array
    {
        if(is_string($script)){
            $script = [
                "source" => $script,
            ];
        }

        if(empty($script['lang'])){
            $script['lang'] = $language;
        }

        return [
            '_script' => (object) [
                'type' => 'number',
                'script' => $script,
                'order' => $direction === SORT_DESC ? 'desc' : 'asc',
            ]
        ];
    }

    /**
     * @see https://www.elastic.co/guide/en/elasticsearch/reference/5.6/query-dsl-script-query.html
     * @see https://www.elastic.co/guide/en/elasticsearch/painless/5.6/painless-specification.html
     * @see https://www.elastic.co/guide/en/elasticsearch/painless/5.6/painless-examples.html
     * @see https://www.elastic.co/guide/en/elasticsearch/reference/5.6/modules-scripting-expression.html
     * @see https://www.elastic.co/guide/en/elasticsearch/reference/5.6/modules-scripting-groovy.html
     * @param string $script
     * @param array $params
     * @return array
     */
    public static function query($script = '', $params = [], $language = 'painless') :array
    {
        if(is_string($script)){
            $script = [
                "source" => $script,
            ];
        }

        if(empty($script['lang'])){
            $script['lang'] = $language;
        }

        if(empty($script['params']) && !empty($params)){
            $script['params'] = (object) $params;
        }

        $body = [
            'script' => (object) $script
        ];

        return $body;

    }
}