ElggUpgrade.php

Go to the documentation of this file.

<?php
use Elgg\Exceptions\InvalidArgumentException as ElggInvalidArgumentException;
use Elgg\Traits\TimeUsing;
use Elgg\Upgrade\Batch;

class ElggUpgrade extends ElggObject {

    use TimeUsing;
    
    private $requiredProperties = [
        'id',
        'title',
        'description',
        'class',
    ];

    public function initializeAttributes() {
        parent::initializeAttributes();

        $this->attributes['subtype'] = 'elgg_upgrade';

        // unowned
        $this->attributes['container_guid'] = 0;
        $this->attributes['owner_guid'] = 0;
    }

    public function setCompleted() {
        $this->setStartTime(); // to make sure a start time is present
        $this->setCompletedTime();
        $this->is_completed = true;

        elgg_trigger_event('complete', 'upgrade', $this);
    }

    public function isCompleted() {
        return (bool) $this->is_completed;
    }

    public function setID($id) {
        $this->id = $id;
    }

    public function setClass($class) {
        $this->class = $class;
    }

    public function isAsynchronous() {
        return !is_subclass_of($this->class, \Elgg\Upgrade\SystemUpgrade::class);
    }

    public function getBatch() {
        try {
            $batch = _elgg_services()->upgradeLocator->getBatch($this->class);
        } catch (ElggInvalidArgumentException $ex) {
            // only report error if the upgrade still needs to run
            $loglevel = $this->isCompleted() ? 'INFO' : 'ERROR';
            elgg_log($ex->getMessage(), $loglevel);
            
            return false;
        }

        // check version before shouldBeSkipped() so authors can get immediate feedback on an invalid batch.
        $version = $batch->getVersion();

        // Version must be in format yyyymmddnn
        if (preg_match("/^[0-9]{10}$/", $version) == 0) {
            elgg_log("Upgrade $this->class returned an invalid version: $version");
            return false;
        }

        return $batch;
    }

    public function setCompletedTime($time = null) {
        if (!is_int($time)) {
            $time = $this->getCurrentTime()->getTimestamp();
        }

        return $this->completed_time = $time;
    }

    public function getCompletedTime() {
        return $this->completed_time;
    }
    
    public function reset() {
        unset($this->is_completed);
        unset($this->completed_time);
        unset($this->processed);
        unset($this->offset);
        unset($this->start_time);
    }
    
    public function setStartTime($time = null) {
        if (!is_int($time)) {
            $time = $this->getCurrentTime()->getTimestamp();
        }
        
        if (isset($this->start_time)) {
            return $this->start_time;
        }
        
        return $this->start_time = $time;
    }
    
    public function getStartTime() {
        return (int) $this->start_time;
    }

    public function save() : bool {
        if (!isset($this->is_completed)) {
            $this->is_completed = false;
        }

        foreach ($this->requiredProperties as $prop) {
            if (!$this->$prop) {
                throw new UnexpectedValueException("ElggUpgrade objects must have a value for the {$prop} property.");
            }
        }

        return parent::save();
    }

    public function __set($name, $value) {
        if (array_key_exists($name, $this->attributes)) {
            parent::__set($name, $value);
        } else {
            $this->setPrivateSetting($name, $value);
        }
    }

    public function __get($name) {
        // See if its in our base attribute
        if (array_key_exists($name, $this->attributes)) {
            return parent::__get($name);
        }

        return $this->getPrivateSetting($name);
    }
    
    public function __isset($name) {
        if (array_key_exists($name, $this->attributes)) {
            return parent::__isset($name);
        }
        
        $private_setting = $this->getPrivateSetting($name);
        return !is_null($private_setting);
    }

    public function getDisplayName() {
        return elgg_echo($this->title);
    }
}