memories/lib/Controller/AdminController.php

273 lines
9.6 KiB
PHP

<?php
declare(strict_types=1);
/**
* @copyright Copyright (c) 2022 Varun Patil <radialapps@gmail.com>
* @author Varun Patil <radialapps@gmail.com>
* @license AGPL-3.0-or-later
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* 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
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
namespace OCA\Memories\Controller;
use OCA\Memories\AppInfo\Application;
use OCA\Memories\Exceptions;
use OCA\Memories\Service\BinExt;
use OCA\Memories\Settings\SystemConfig;
use OCA\Memories\Util;
use OCP\AppFramework\Http;
use OCP\AppFramework\Http\JSONResponse;
class AdminController extends GenericApiController
{
/**
* @AdminRequired
*/
public function getSystemConfig(): Http\Response
{
return Util::guardEx(function () {
$config = [];
foreach (SystemConfig::DEFAULTS as $key => $default) {
$config[$key] = $this->config->getSystemValue($key, $default);
}
return new JSONResponse($config, Http::STATUS_OK);
});
}
/**
* @AdminRequired
*/
public function setSystemConfig(string $key, mixed $value): Http\Response
{
return Util::guardEx(function () use ($key, $value) {
// Make sure not running in read-only mode
if ($this->config->getSystemValue('memories.readonly', false)) {
throw Exceptions::Forbidden('Cannot change settings in readonly mode');
}
// Assign config with type checking
SystemConfig::set($key, $value);
// If changing vod settings, kill any running go-vod instances
if (str_starts_with($key, 'memories.vod.')) {
try {
BinExt::startGoVod();
} catch (\Exception $e) {
error_log('Failed to start go-vod: '.$e->getMessage());
}
}
return new JSONResponse([], Http::STATUS_OK);
});
}
/**
* @AdminRequired
*
* @UseSession
*/
public function getSystemStatus(): Http\Response
{
return Util::guardEx(function () {
$config = \OC::$server->get(\OCP\IConfig::class);
$index = \OC::$server->get(\OCA\Memories\Service\Index::class);
// Build status array
$status = [];
// Check exiftool version
$exiftoolNoLocal = SystemConfig::get('memories.exiftool_no_local');
$status['exiftool'] = $this->getExecutableStatus(
static fn () => BinExt::getExiftoolPBin(),
static fn () => BinExt::testExiftool(),
!$exiftoolNoLocal,
!$exiftoolNoLocal,
);
// Check for system perl
/** @psalm-suppress ForbiddenCode */
$status['perl'] = $this->getExecutableStatus(
trim(shell_exec('which perl') ?: '/bin/perl'),
static fn (string $p) => BinExt::testSystemPerl($p),
);
// Check number of indexed files
$status['indexed_count'] = $index->getIndexedCount();
// Automatic indexing stats
$jobStart = (int) $config->getAppValue(Application::APPNAME, 'last_index_job_start', (string) 0);
$status['last_index_job_start'] = $jobStart ? time() - $jobStart : 0; // Seconds ago
$status['last_index_job_duration'] = (float) $config->getAppValue(Application::APPNAME, 'last_index_job_duration', (string) 0);
$status['last_index_job_status'] = $config->getAppValue(Application::APPNAME, 'last_index_job_status', 'Indexing has not been run yet');
$status['last_index_job_status_type'] = $config->getAppValue(Application::APPNAME, 'last_index_job_status_type', 'warning');
// Check supported preview mimes
$status['mimes'] = $index->getPreviewMimes($index->getAllMimes());
// Check for PHP Imagick
$status['imagick'] = class_exists('\Imagick') ? \Imagick::getVersion()['versionString'] : false;
// Check for bad encryption module
$status['bad_encryption'] = \OCA\Memories\Util::isEncryptionEnabled();
// Get GIS status
$places = \OC::$server->get(\OCA\Memories\Service\Places::class);
try {
$status['gis_type'] = $places->detectGisType();
$status['gis_count'] = $places->geomCount();
} catch (\Exception $e) {
$status['gis_type'] = $e->getMessage();
}
// Check for FFmpeg for preview generation
/** @psalm-suppress ForbiddenCode */
$status['ffmpeg_preview'] = $this->getExecutableStatus(
SystemConfig::get('preview_ffmpeg_path')
?: trim(shell_exec('which ffmpeg') ?: ''),
static fn ($p) => BinExt::testFFmpeg($p, 'ffmpeg'),
);
// Check ffmpeg and ffprobe binaries for transcoding
$status['ffmpeg'] = $this->getExecutableStatus(
SystemConfig::get('memories.vod.ffmpeg'),
static fn ($p) => BinExt::testFFmpeg($p, 'ffmpeg'),
);
$status['ffprobe'] = $this->getExecutableStatus(
SystemConfig::get('memories.vod.ffprobe'),
static fn ($p) => BinExt::testFFmpeg($p, 'ffprobe'),
);
// Check go-vod binary
$extGoVod = SystemConfig::get('memories.vod.external');
$status['govod'] = $this->getExecutableStatus(
static fn () => BinExt::getGoVodBin(),
static fn () => BinExt::testStartGoVod(),
!$extGoVod,
!$extGoVod,
);
// Check for VA-API device
$devPath = '/dev/dri/renderD128';
if (!file_exists($devPath)) {
$status['vaapi_dev'] = 'not_found';
} elseif (!is_readable($devPath)) {
$status['vaapi_dev'] = 'not_readable';
} else {
$status['vaapi_dev'] = 'ok';
}
// Action token
$status['action_token'] = $this->actionToken(true);
return new JSONResponse($status, Http::STATUS_OK);
});
}
/**
* @AdminRequired
*
* @UseSession
*/
public function placesSetup(?string $actiontoken): Http\Response
{
if (!$actiontoken || $this->actionToken() !== $actiontoken) {
return new JSONResponse(['error' => 'Invalid action token. Refresh the memories admin page.'], Http::STATUS_BAD_REQUEST);
}
// Reset action token
$this->actionToken(true);
return Util::guardExDirect(static function (Http\IOutput $out) {
try {
// Set PHP timeout to infinite
set_time_limit(0);
// Send headers for long-running request
$out->setHeader('Content-Type: text/plain');
$out->setHeader('X-Accel-Buffering: no');
$out->setHeader('Cache-Control: no-cache');
$out->setHeader('Connection: keep-alive');
$out->setHeader('Content-Length: 0');
$places = \OC::$server->get(\OCA\Memories\Service\Places::class);
$datafile = $places->downloadPlanet();
$places->importPlanet($datafile);
$places->recalculateAll();
$out->setOutput("Places set up successfully.\n");
} catch (\Exception $e) {
$out->setOutput('Failed: '.$e->getMessage()."\n");
}
});
}
/**
* Get the status of an executable.
*
* @param (\Closure():string)|string $path Path to the executable
* @param null|(\Closure(string):string) $testFunction Function to test the executable
* @param bool $testIfFile Test if the path is a file
* @param bool $testIfExecutable Test if the path is executable
*/
private function getExecutableStatus(
\Closure|string $path,
null|\Closure $testFunction = null,
bool $testIfFile = true,
bool $testIfExecutable = true,
): string {
if ($path instanceof \Closure) {
try {
$path = $path();
} catch (\Exception $e) {
return 'test_fail:'.$e->getMessage();
}
}
if ($testIfFile && !is_file($path)) {
return 'not_found';
}
if ($testIfExecutable && !is_executable($path)) {
return 'not_executable';
}
if ($testFunction) {
try {
return 'test_ok:'.$testFunction($path);
} catch (\Exception $e) {
return 'test_fail:'.$e->getMessage();
}
}
return 'ok';
}
private function actionToken(bool $set = false): string
{
$session = \OC::$server->get(\OCP\ISession::class);
if (!$set) {
return $session->get('memories_action_token');
}
$token = bin2hex(random_bytes(32));
$session->set('memories_action_token', $token);
return $token;
}
}