121 lines
3.8 KiB
PHP
121 lines
3.8 KiB
PHP
<?php
|
|
|
|
require_once __DIR__ . '/init.php';
|
|
require_once __DIR__ . '/controllers/general.php';
|
|
|
|
use Appwrite\CLI\Tasks;
|
|
use Utopia\CLI\CLI;
|
|
use Utopia\Database\Validator\Authorization;
|
|
use Utopia\Platform\Service;
|
|
use Utopia\App;
|
|
use Utopia\CLI\Console;
|
|
use Utopia\Cache\Adapter\Redis as RedisCache;
|
|
use Utopia\Cache\Cache;
|
|
use Utopia\Database\Adapter\MariaDB;
|
|
use Utopia\Database\Database;
|
|
use Utopia\Logger\Log;
|
|
use Utopia\Registry\Registry;
|
|
|
|
Authorization::disable();
|
|
|
|
CLI::setResource('register', fn()=>$register);
|
|
|
|
CLI::setResource('db', function (Registry $register) {
|
|
$attempts = 0;
|
|
$max = 10;
|
|
$sleep = 1;
|
|
do {
|
|
try {
|
|
$attempts++;
|
|
$db = $register->get('db');
|
|
break; // leave the do-while if successful
|
|
} catch (\Exception $e) {
|
|
Console::warning("Database not ready. Retrying connection ({$attempts})...");
|
|
if ($attempts >= $max) {
|
|
throw new \Exception('Failed to connect to database: ' . $e->getMessage());
|
|
}
|
|
sleep($sleep);
|
|
}
|
|
} while ($attempts < $max);
|
|
return $db;
|
|
}, ['register']);
|
|
|
|
CLI::setResource('cache', fn ($register) => $register->get('cache'), ['register']);
|
|
|
|
CLI::setResource('dbForConsole', function ($db, $cache) {
|
|
$cache = new Cache(new RedisCache($cache));
|
|
|
|
$database = new Database(new MariaDB($db), $cache);
|
|
$database->setDefaultDatabase(App::getEnv('_APP_DB_SCHEMA', 'appwrite'));
|
|
$database->setNamespace('_console');
|
|
|
|
return $database;
|
|
}, ['db', 'cache']);
|
|
|
|
CLI::setResource('influxdb', function (Registry $register) {
|
|
/** @var InfluxDB\Client $client */
|
|
$client = $register->get('influxdb');
|
|
$attempts = 0;
|
|
$max = 10;
|
|
$sleep = 1;
|
|
|
|
do { // check if telegraf database is ready
|
|
try {
|
|
$attempts++;
|
|
$database = $client->selectDB('telegraf');
|
|
if (in_array('telegraf', $client->listDatabases())) {
|
|
break; // leave the do-while if successful
|
|
}
|
|
} catch (\Throwable$th) {
|
|
Console::warning("InfluxDB not ready. Retrying connection ({$attempts})...");
|
|
if ($attempts >= $max) {
|
|
throw new \Exception('InfluxDB database not ready yet');
|
|
}
|
|
sleep($sleep);
|
|
}
|
|
} while ($attempts < $max);
|
|
return $database;
|
|
}, ['register']);
|
|
|
|
CLI::setResource('logError', function (Registry $register) {
|
|
return function (Throwable $error, string $namespace, string $action) use ($register) {
|
|
$logger = $register->get('logger');
|
|
|
|
if ($logger) {
|
|
$version = App::getEnv('_APP_VERSION', 'UNKNOWN');
|
|
|
|
$log = new Log();
|
|
$log->setNamespace($namespace);
|
|
$log->setServer(\gethostname());
|
|
$log->setVersion($version);
|
|
$log->setType(Log::TYPE_ERROR);
|
|
$log->setMessage($error->getMessage());
|
|
|
|
$log->addTag('code', $error->getCode());
|
|
$log->addTag('verboseType', get_class($error));
|
|
|
|
$log->addExtra('file', $error->getFile());
|
|
$log->addExtra('line', $error->getLine());
|
|
$log->addExtra('trace', $error->getTraceAsString());
|
|
$log->addExtra('detailedTrace', $error->getTrace());
|
|
|
|
$log->setAction($action);
|
|
|
|
$isProduction = App::getEnv('_APP_ENV', 'development') === 'production';
|
|
$log->setEnvironment($isProduction ? Log::ENVIRONMENT_PRODUCTION : Log::ENVIRONMENT_STAGING);
|
|
|
|
$responseCode = $logger->addLog($log);
|
|
Console::info('Usage stats log pushed with status code: ' . $responseCode);
|
|
}
|
|
|
|
Console::warning("Failed: {$error->getMessage()}");
|
|
Console::warning($error->getTraceAsString());
|
|
};
|
|
});
|
|
|
|
$cliPlatform = new Tasks();
|
|
$cliPlatform->init(Service::TYPE_CLI);
|
|
|
|
$cli = $cliPlatform->getCli();
|
|
$cli->run();
|