Entities.php

Go to the documentation of this file.

<?php

namespace Elgg\Database;

use Closure;
use Doctrine\DBAL\Query\Expression\CompositeExpression;
use Elgg\Database\Clauses\AnnotationWhereClause;
use Elgg\Database\Clauses\EntityWhereClause;
use Elgg\Database\Clauses\MetadataWhereClause;
use Elgg\Database\Clauses\PrivateSettingWhereClause;
use Elgg\Database\Clauses\RelationshipWhereClause;
use ElggEntity;
use InvalidArgumentException;
use InvalidParameterException;

class Entities extends Repository {

    public function count() {
        $qb = Select::fromTable('entities', 'e');

        $count_expr = $this->options->distinct ? "DISTINCT e.guid" : "*";
        $qb->select("COUNT({$count_expr}) AS total");

        $qb = $this->buildQuery($qb);

        $result = _elgg_services()->db->getDataRow($qb);

        if (empty($result)) {
            return 0;
        }
        
        return (int) $result->total;
    }

    public function calculate($function, $property, $property_type = null) {

        if (!in_array(strtolower($function), QueryBuilder::$calculations)) {
            throw new InvalidArgumentException("'$function' is not a valid numeric function");
        }

        if (!isset($property_type)) {
            if (in_array($property, ElggEntity::$primary_attr_names)) {
                $property_type = 'attribute';
            } else {
                $property_type = 'metadata';
            }
        }

        $qb = Select::fromTable('entities', 'e');

        switch ($property_type) {
            case 'attribute':
                if (!in_array($property, ElggEntity::$primary_attr_names)) {
                    throw new InvalidParameterException("'$property' is not a valid attribute");
                }

                $qb->addSelect("{$function}(e.{$property}) AS calculation");
                break;

            case 'metadata' :
                $alias = $qb->joinMetadataTable('e', 'guid', $property, 'inner', 'n_table');
                $qb->addSelect("{$function}({$alias}.value) AS calculation");
                break;

            case 'annotation' :
                $alias = $qb->joinAnnotationTable('e', 'guid', $property, 'inner', 'n_table');
                $qb->addSelect("{$function}({$alias}.value) AS calculation");
                break;

            case 'private_setting' :
                $alias = $qb->joinPrivateSettingsTable('e', 'guid', $property, 'inner', 'ps');
                $qb->addSelect("{$function}({$alias}.value) AS calculation");
                break;
        }

        $qb = $this->buildQuery($qb);

        $result = _elgg_services()->db->getDataRow($qb);

        // do not cast data as calculations could be used as int, float or string
        return $result->calculation;
    }

    public function get($limit = null, $offset = null, $callback = null) {

        $qb = Select::fromTable('entities', 'e');

        $distinct = $this->options->distinct ? "DISTINCT" : "";
        $qb->select("$distinct e.*");

        $this->expandInto($qb, 'e');

        $qb = $this->buildQuery($qb);

        // Keeping things backwards compatible
        $original_order = elgg_extract('order_by', $this->options->__original_options);
        if (empty($original_order) && $original_order !== false) {
            $qb->addOrderBy('e.time_created', 'desc');
            // also add order by guid, to rely less on internals of MySQL fallback ordering
            $qb->addOrderBy('e.guid', 'desc');
        }

        if ($limit > 0) {
            $qb->setMaxResults((int) $limit);
            $qb->setFirstResult((int) $offset);
        }

        $options = $this->options->getArrayCopy();

        $options['limit'] = (int) $limit;
        $options['offset'] = (int) $offset;
        $options['callback'] = $callback ? : $this->options->callback;
        if (!isset($options['callback'])) {
            $options['callback'] = 'entity_row_to_elggstar';
        }

        unset($options['count']);

        return _elgg_services()->entityTable->fetch($qb, $options);
    }
    
    public function getDates() {
        
        $qb = Select::fromTable('entities', 'e');

        $qb->select("DISTINCT EXTRACT(YEAR_MONTH FROM FROM_UNIXTIME(e.time_created)) AS yearmonth");

        $this->expandInto($qb, 'e');

        $qb = $this->buildQuery($qb);

        $options = $this->options->getArrayCopy();
        unset($options['count']);
        
        $options['callback'] = false;
        
        $results = _elgg_services()->entityTable->fetch($qb, $options);
        
        if (empty($results)) {
            return false;
        }
    
        return array_map(function ($e) {
            return $e->yearmonth;
        }, $results);
    }

    public function execute() {

        if ($this->options->annotation_calculation) {
            $clauses = $this->options->annotation_name_value_pairs;
            if (count($clauses) > 1 && $this->options->annotation_name_value_pairs_operator !== 'OR') {
                throw new \LogicException("Annotation calculation can not be performed on multiple annotation name value pairs merged with AND");
            }

            $clause = array_shift($clauses);

            return $this->calculate($this->options->annotation_calculation, $clause->names, 'annotation');
        } else if ($this->options->metadata_calculation) {
            $clauses = $this->options->metadata_name_value_pairs;
            if (count($clauses) > 1 && $this->options->metadata_name_value_pairs_operator !== 'OR') {
                throw new \LogicException("Metadata calculation can not be performed on multiple metadata name value pairs merged with AND");
            }

            $clause = array_shift($clauses);

            return $this->calculate($this->options->metadata_calculation, $clause->names, 'metadata');
        } else if ($this->options->count) {
            return $this->count();
        } else if ($this->options->batch) {
            return $this->batch($this->options->limit, $this->options->offset, $this->options->callback);
        } else {
            return $this->get($this->options->limit, $this->options->offset, $this->options->callback);
        }
    }

    protected function buildQuery(QueryBuilder $qb) {

        $ands = [];

        foreach ($this->options->joins as $join) {
            $join->prepare($qb, 'e');
        }

        foreach ($this->options->wheres as $where) {
            $ands[] = $where->prepare($qb, 'e');
        }

        $ands[] = $this->buildEntityClause($qb);
        $ands[] = $this->buildPairedMetadataClause($qb, $this->options->metadata_name_value_pairs, $this->options->metadata_name_value_pairs_operator);
        $ands[] = $this->buildPairedMetadataClause($qb, $this->options->search_name_value_pairs, 'OR');
        $ands[] = $this->buildPairedAnnotationClause($qb, $this->options->annotation_name_value_pairs, $this->options->annotation_name_value_pairs_operator);
        $ands[] = $this->buildPairedPrivateSettingsClause($qb, $this->options->private_setting_name_value_pairs, $this->options->private_setting_name_value_pairs_operator);
        $ands[] = $this->buildPairedRelationshipClause($qb, $this->options->relationship_pairs);

        $ands = $qb->merge($ands);

        if (!empty($ands)) {
            $qb->andWhere($ands);
        }

        return $qb;
    }

    protected function buildEntityClause(QueryBuilder $qb) {
        return EntityWhereClause::factory($this->options)->prepare($qb, 'e');
    }

    protected function buildPairedMetadataClause(QueryBuilder $qb, $clauses, $boolean = 'AND') {
        $parts = [];


        foreach ($clauses as $clause) {
            if ($clause instanceof MetadataWhereClause) {
                if (strtoupper($boolean) === 'OR' || count($clauses) === 1) {
                    $joined_alias = $qb->joinMetadataTable('e', 'guid', null, 'inner', 'n_table');
                } else {
                    $joined_alias = $qb->joinMetadataTable('e', 'guid', $clause->names);
                }
                $parts[] = $clause->prepare($qb, $joined_alias);
            }
        }

        return $qb->merge($parts, $boolean);
    }

    protected function buildPairedAnnotationClause(QueryBuilder $qb, $clauses, $boolean = 'AND') {
        $parts = [];

        foreach ($clauses as $clause) {
            if (strtoupper($boolean) === 'OR' || count($clauses) === 1) {
                $joined_alias = $qb->joinAnnotationTable('e', 'guid', null, 'inner', 'n_table');
            } else {
                $joined_alias = $qb->joinAnnotationTable('e', 'guid', $clause->names);
            }
            $parts[] = $clause->prepare($qb, $joined_alias);
        }

        return $qb->merge($parts, $boolean);
    }

    protected function buildPairedPrivateSettingsClause(QueryBuilder $qb, $clauses, $boolean = 'AND') {
        $parts = [];

        foreach ($clauses as $clause) {
            if (strtoupper($boolean) === 'OR' || count($clauses) === 1) {
                $joined_alias = $qb->joinPrivateSettingsTable('e', 'guid', null, 'inner', 'ps');
            } else {
                $joined_alias = $qb->joinPrivateSettingsTable('e', 'guid', $clause->names);
            }
            $parts[] = $clause->prepare($qb, $joined_alias);
        }

        return $qb->merge($parts, $boolean);
    }

    protected function buildPairedRelationshipClause(QueryBuilder $qb, $clauses, $boolean = 'AND') {
        $parts = [];

        foreach ($clauses as $clause) {
            if (strtoupper($boolean) == 'OR' || count($clauses) === 1) {
                $joined_alias = $qb->joinRelationshipTable('e', $clause->join_on, null, $clause->inverse, 'inner', 'r');
            } else {
                $joined_alias = $qb->joinRelationshipTable('e', $clause->join_on, $clause->names, $clause->inverse);
            }
            $parts[] = $clause->prepare($qb, $joined_alias);
        }

        return $qb->merge($parts, $boolean);
    }
}