owncloud/core

<?php
/**
 * @author Thomas Müller <thomas.mueller@tmit.eu>
 *
 * @copyright Copyright (c) 2018, ownCloud GmbH
 * @license AGPL-3.0
 *
 * This code is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License, version 3,
 * along with this program.  If not, see <http://www.gnu.org/licenses/>
 *
 */
namespace OC\Migration;

use OC\BackgroundJob\JobList;
use OC\BackgroundJob\TimedJob;
use OC\NeedsUpdateException;
use OC\Repair;
use OC_App;
use OCP\BackgroundJob\IJobList;
use OCP\ILogger;
use Symfony\Component\EventDispatcher\EventDispatcher;

/**
 * Class BackgroundRepair
 *
 * @package OC\Migration
 */
class BackgroundRepair extends TimedJob {
    /** @var IJobList */
    private $jobList;

    /** @var ILogger */
    private $logger;

    /** @var EventDispatcher */
    private $dispatcher;

    public function setDispatcher(EventDispatcher $dispatcher) {
        $this->dispatcher = $dispatcher;
    }
    /**
     * run the job, then remove it from the job list
     *
     * @param JobList $jobList
     * @param ILogger $logger
     */
    public function execute($jobList, ILogger $logger = null) {
        // add an interval of 15 mins
        $this->setInterval(15*60);

        $this->jobList = $jobList;
        $this->logger = $logger;
        parent::execute($jobList, $logger);
    }

    /**
     * @param array $argument
     * @throws \Exception
     * @throws \OC\NeedsUpdateException
     */
    protected function run($argument) {
        if (!isset($argument['app']) || !isset($argument['step'])) {
            // remove the job - we can never execute it
            $this->jobList->remove($this, $this->argument);
            return;
        }
        $app = $argument['app'];

        try {
            $this->loadApp($app);
        } catch (NeedsUpdateException $ex) {
            // as long as the app is not yet done with it's offline migration
            // we better not start with the live migration
            return;
        }

        $step = $argument['step'];
        $repair = new Repair([], $this->dispatcher);
        try {
            $repair->addStep($step);
        } catch (\Exception $ex) {
            $this->logger->logException($ex, [
                'app' => 'migration'
            ]);

            // remove the job - we can never execute it
            $this->jobList->remove($this, $this->argument);
            return;
        }

        // execute the repair step
        $repair->run();

        // remove the job once executed successfully
        $this->jobList->remove($this, $this->argument);
    }

    /**
     * @codeCoverageIgnore
     * @param $app
     * @throws NeedsUpdateException
     */
    protected function loadApp($app) {
        OC_App::loadApp($app);
    }
}