Updates core code to use App(db)

main
Yasen Pramatarov 2026-01-20 22:44:47 +02:00
parent 71c25c778f
commit 138ea70185
5 changed files with 52 additions and 28 deletions

View File

@ -1,10 +1,10 @@
<?php <?php
use App\App;
use App\Core\NullLogger; use App\Core\NullLogger;
class RateLimiter { class RateLimiter {
public $db; public $db;
private $database;
/** @var mixed NullLogger (or PSR-3 logger) or plugin Log */ /** @var mixed NullLogger (or PSR-3 logger) or plugin Log */
private $logger; private $logger;
public $maxAttempts = 5; // Maximum login attempts public $maxAttempts = 5; // Maximum login attempts
@ -27,12 +27,13 @@ class RateLimiter {
]; ];
/** /**
* @param mixed $database Database object
* @param mixed $logger Optional NullLogger (or PSR-3 logger) or plugin Log * @param mixed $logger Optional NullLogger (or PSR-3 logger) or plugin Log
*/ */
public function __construct($database, $logger = null) { public function __construct($logger = null) {
$this->database = $database; $db = App::db();
$this->db = $database->getConnection(); // Extract PDO connection from Database object
$this->db = ($db instanceof PDO) ? $db : $db->getConnection();
// Initialize logger (plugin Log if present or NullLogger otherwise) // Initialize logger (plugin Log if present or NullLogger otherwise)
if ($logger !== null) { if ($logger !== null) {
$this->logger = $logger; $this->logger = $logger;

View File

@ -1,5 +1,7 @@
<?php <?php
use App\App;
/** /**
* class User * class User
* *
@ -33,7 +35,7 @@ class User {
require_once __DIR__ . '/ratelimiter.php'; require_once __DIR__ . '/ratelimiter.php';
require_once __DIR__ . '/twoFactorAuth.php'; require_once __DIR__ . '/twoFactorAuth.php';
$this->rateLimiter = new RateLimiter($database); $this->rateLimiter = new RateLimiter();
$this->twoFactorAuth = new TwoFactorAuthentication($database); $this->twoFactorAuth = new TwoFactorAuthentication($database);
} }

View File

@ -2,6 +2,8 @@
namespace App\Core; namespace App\Core;
use App\App;
class Maintenance class Maintenance
{ {
// Keep it simple: store the flag within the app directory // Keep it simple: store the flag within the app directory
@ -13,10 +15,11 @@ class Maintenance
return true; return true;
} }
// Prefer DB settings if available in the current request // Prefer DB settings if available in the current request
if (isset($GLOBALS['db'])) { $db = App::db();
if ($db) {
try { try {
require_once __DIR__ . '/Settings.php'; require_once __DIR__ . '/Settings.php';
$settings = new Settings($GLOBALS['db']); $settings = new Settings($db);
return $settings->get('maintenance_enabled', '0') === '1'; return $settings->get('maintenance_enabled', '0') === '1';
} catch (\Throwable $e) { } catch (\Throwable $e) {
// fall back to file flag // fall back to file flag
@ -27,10 +30,11 @@ class Maintenance
public static function enable(string $message = ''): bool public static function enable(string $message = ''): bool
{ {
if (isset($GLOBALS['db'])) { $db = App::db();
if ($db) {
try { try {
require_once __DIR__ . '/Settings.php'; require_once __DIR__ . '/Settings.php';
$settings = new Settings($GLOBALS['db']); $settings = new Settings($db);
$ok1 = $settings->set('maintenance_enabled', '1'); $ok1 = $settings->set('maintenance_enabled', '1');
$ok2 = $settings->set('maintenance_message', $message); $ok2 = $settings->set('maintenance_message', $message);
return $ok1 && $ok2; return $ok1 && $ok2;
@ -48,10 +52,11 @@ class Maintenance
public static function disable(): bool public static function disable(): bool
{ {
if (isset($GLOBALS['db'])) { $db = App::db();
if ($db) {
try { try {
require_once __DIR__ . '/Settings.php'; require_once __DIR__ . '/Settings.php';
$settings = new Settings($GLOBALS['db']); $settings = new Settings($db);
$ok1 = $settings->set('maintenance_enabled', '0'); $ok1 = $settings->set('maintenance_enabled', '0');
// keep last message for reference, optional to clear // keep last message for reference, optional to clear
return $ok1; return $ok1;
@ -74,10 +79,11 @@ class Maintenance
if ($envMsg) { if ($envMsg) {
return trim($envMsg); return trim($envMsg);
} }
if (isset($GLOBALS['db'])) { $db = App::db();
if ($db) {
try { try {
require_once __DIR__ . '/Settings.php'; require_once __DIR__ . '/Settings.php';
$settings = new Settings($GLOBALS['db']); $settings = new Settings($db);
return (string)$settings->get('maintenance_message', ''); return (string)$settings->get('maintenance_message', '');
} catch (\Throwable $e) { } catch (\Throwable $e) {
// ignore and fall back to file flag // ignore and fall back to file flag

View File

@ -167,9 +167,9 @@ class PluginManager
return false; return false;
} }
// Use global DB and get PDO connection // Use App API to get database connection
$db = $GLOBALS['db']; $db = \App\App::db();
$pdo = $db->getConnection(); $pdo = ($db instanceof \PDO) ? $db : $db->getConnection();
try { try {
// Update or insert plugin setting in database // Update or insert plugin setting in database
@ -213,9 +213,15 @@ class PluginManager
return false; return false;
} }
// Use global DB and get PDO connection // Use App API to get database connection
$db = $GLOBALS['db']; $db = \App\App::db();
$pdo = $db->getConnection();
// If database unavailable, fallback to manifest
if (!$db) {
return self::$catalog[$plugin]['meta']['enabled'] ?? false;
}
$pdo = ($db instanceof \PDO) ? $db : $db->getConnection();
try { try {
$stmt = $pdo->prepare('SELECT `value` FROM settings WHERE `key` = :key LIMIT 1'); $stmt = $pdo->prepare('SELECT `value` FROM settings WHERE `key` = :key LIMIT 1');
@ -226,7 +232,8 @@ class PluginManager
return $result && $result['value'] === '1'; return $result && $result['value'] === '1';
} catch (\PDOException $e) { } catch (\PDOException $e) {
app_log('error', 'PluginManager::isEnabled failed for ' . $plugin . ': ' . $e->getMessage(), ['scope' => 'plugin']); app_log('error', 'PluginManager::isEnabled failed for ' . $plugin . ': ' . $e->getMessage(), ['scope' => 'plugin']);
return false; // Fallback to manifest on database error
return self::$catalog[$plugin]['meta']['enabled'] ?? false;
} }
} }
@ -276,36 +283,43 @@ class PluginManager
return false; return false;
} }
global $db; $db = \App\App::db();
if (!$db instanceof PDO) { if (!$db) {
app_log('error', 'PluginManager::purge: Database connection not available', ['scope' => 'plugin']); app_log('error', 'PluginManager::purge: Database connection not available', ['scope' => 'plugin']);
return false; return false;
} }
$pdo = ($db instanceof \PDO) ? $db : $db->getConnection();
try { try {
// First disable the plugin // First disable the plugin
self::setEnabled($plugin, false); self::setEnabled($plugin, false);
// Remove plugin settings // Remove plugin settings
$stmt = $db->prepare('DELETE FROM settings WHERE `key` LIKE :pattern'); $stmt = $pdo->prepare('DELETE FROM settings WHERE `key` LIKE :pattern');
$stmt->execute([':pattern' => 'plugin_enabled_' . $plugin]); $stmt->execute([':pattern' => 'plugin_enabled_' . $plugin]);
// Drop plugin-specific tables (user_pro_* tables for this plugin) // Drop plugin-specific tables (user_pro_* tables for this plugin)
$stmt = $db->prepare('SHOW TABLES LIKE "user_pro_%"'); $stmt = $pdo->prepare('SHOW TABLES LIKE "user_pro_%"');
$stmt->execute(); $stmt->execute();
$tables = $stmt->fetchAll(PDO::FETCH_COLUMN, 0); $tables = $stmt->fetchAll(\PDO::FETCH_COLUMN, 0);
// Disable foreign key checks temporarily to allow table drops
$pdo->exec('SET FOREIGN_KEY_CHECKS=0');
foreach ($tables as $table) { foreach ($tables as $table) {
// Check if this table belongs to the plugin by checking its migration file // Check if this table belongs to the plugin by checking its migration file
$migrationFile = self::$catalog[$plugin]['path'] . '/migrations/create_' . $plugin . '_tables.sql'; $migrationFile = self::$catalog[$plugin]['path'] . '/migrations/create_' . $plugin . '_tables.sql';
if (file_exists($migrationFile)) { if (file_exists($migrationFile)) {
$migrationContent = file_get_contents($migrationFile); $migrationContent = file_get_contents($migrationFile);
if (strpos($migrationContent, $table) !== false) { if (strpos($migrationContent, $table) !== false) {
$db->exec("DROP TABLE IF EXISTS `$table`"); $pdo->exec("DROP TABLE IF EXISTS `$table`");
app_log('info', 'PluginManager::purge: Dropped table ' . $table . ' for plugin ' . $plugin, ['scope' => 'plugin']); app_log('info', 'PluginManager::purge: Dropped table ' . $table . ' for plugin ' . $plugin, ['scope' => 'plugin']);
} }
} }
} }
// Re-enable foreign key checks
$pdo->exec('SET FOREIGN_KEY_CHECKS=1');
app_log('info', 'PluginManager::purge: Successfully purged plugin ' . $plugin, ['scope' => 'plugin']); app_log('info', 'PluginManager::purge: Successfully purged plugin ' . $plugin, ['scope' => 'plugin']);
return true; return true;

View File

@ -23,7 +23,8 @@ try {
// Initialize RateLimiter // Initialize RateLimiter
require_once '../app/classes/ratelimiter.php'; require_once '../app/classes/ratelimiter.php';
$rateLimiter = new RateLimiter($db); $rateLimiter = new RateLimiter();
// Get user IP // Get user IP
require_once '../app/helpers/ip_helper.php'; require_once '../app/helpers/ip_helper.php';
$user_IP = getUserIP(); $user_IP = getUserIP();