XC_VM/src/Cli/Commands/OndemandCommand.php

<?php

/**
 * OndemandCommand — ondemand command
 *
 * @package XC_VM_CLI_Commands
 * @author  Divarion_D <https://github.com/Divarion-D>
 * @copyright 2025-2026 Vateron Media
 * @link    https://github.com/Vateron-Media/XC_VM
 * @license AGPL-3.0 https://www.gnu.org/licenses/agpl-3.0.html
 */

class OndemandCommand implements CommandInterface {

	public function getName(): string {
		return 'ondemand';
	}

	public function getDescription(): string {
		return 'On-Demand Killer — kill streams with no viewers';
	}

	public function execute(array $rArgs): int {
		if (posix_getpwuid(posix_geteuid())['name'] != 'xc_vm') {
			echo "Please run as XC_VM!\n";
			return 1;
		}

		set_time_limit(0);

		global $db;

		shell_exec('kill -9 $(ps aux | grep -E "ondemand\.php" | grep -v grep | grep -v ' . getmypid() . ' | awk \'{print $2}\')');

		if (!SettingsManager::getAll()['on_demand_instant_off']) {
			echo 'On-Demand - Instant Off setting is disabled.' . "\n";
			return 0;
		}

		if (SettingsManager::getAll()['redis_handler']) {
			RedisManager::ensureConnected();
		}

		$rMainID = ConnectionTracker::getMainID();
		$rLastCheck = null;
		$rInterval = 60;
		$rMD5 = md5_file(__FILE__);

		while (true) {
			if (!$db || !$db->ping() || (SettingsManager::getAll()['redis_handler'] && RedisManager::instance() && !RedisManager::instance()->ping())) {
				break;
			}

			$rCurentMD5Hash = md5_file(__FILE__);
			if (!$rLastCheck || time() - $rLastCheck > $rInterval || $rCurentMD5Hash !== $rMD5) {
				SettingsManager::set(SettingsRepository::getAll(true));
				$rLastCheck = time();
				$rMD5 = $rCurentMD5Hash;
			}

			$rRows = [];

			if (SettingsManager::getAll()['redis_handler'] && RedisManager::instance()) {
				$db->query("SELECT stream_id FROM streams_servers WHERE server_id = ? AND on_demand = 1 AND pid IS NOT NULL AND pid > 0", SERVER_ID);
				$rStreamIDs = $db->get_column();

				$rAttached = [];
				if (!empty($rStreamIDs)) {
					$placeholders = str_repeat('?,', count($rStreamIDs) - 1) . '?';
					$db->query("SELECT stream_id, COUNT(*) AS cnt FROM streams_servers WHERE parent_id = ? AND pid > 0 AND monitor_pid > 0 AND stream_id IN ($placeholders) GROUP BY stream_id", SERVER_ID, ...$rStreamIDs);
					$rAttachedRows = $db->get_rows(true, 'stream_id');
					foreach ($rAttachedRows as $id => $row) {
						$rAttached[$id] = (int) $row['cnt'];
					}
				}

				$rConnections = ConnectionTracker::getStreamConnections($rStreamIDs, false, false);

				foreach ($rStreamIDs as $rStreamID) {
					$rRows[] = [
						'stream_id' => $rStreamID,
						'online_clients' => count($rConnections[$rStreamID][SERVER_ID] ?? []),
						'attached' => $rAttached[$rStreamID] ?? 0
					];
				}
			} else {
				$db->query("SELECT stream_id FROM streams_servers WHERE server_id = ? AND on_demand = 1 AND pid IS NOT NULL AND pid > 0", SERVER_ID);
				$rActive = $db->get_column();

				if (empty($rActive)) {
					usleep(800000);
					continue;
				}

				$placeholders = str_repeat('?,', count($rActive) - 1) . '?';

				$online = [];
				$db->query("SELECT stream_id, COUNT(*) AS cnt FROM lines_live WHERE server_id = ? AND hls_end = 0 AND stream_id IN ($placeholders) GROUP BY stream_id", SERVER_ID, ...$rActive);
				$onlineRows = $db->get_rows(true, 'stream_id');
				foreach ($onlineRows as $id => $row) {
					$online[$id] = (int) $row['cnt'];
				}

				$attached = [];
				$db->query("SELECT stream_id, COUNT(*) AS cnt FROM streams_servers WHERE parent_id = ? AND pid > 0 AND monitor_pid > 0 AND stream_id IN ($placeholders) GROUP BY stream_id", SERVER_ID, ...$rActive);
				$attachedRows = $db->get_rows(true, 'stream_id');
				foreach ($attachedRows as $id => $row) {
					$attached[$id] = (int) $row['cnt'];
				}

				foreach ($rActive as $stream_id) {
					$rRows[] = [
						'stream_id' => $stream_id,
						'online_clients' => $online[$stream_id] ?? 0,
						'attached' => $attached[$stream_id] ?? 0
					];
				}
			}

			foreach ($rRows as $rRow) {
				if ($rRow['online_clients'] > 0 || $rRow['attached'] > 0)
					continue;

				$rStreamID = $rRow['stream_id'];
				$pidFile = STREAMS_PATH . $rStreamID . '_.pid';
				$monitorFile = STREAMS_PATH . $rStreamID . '_.monitor';

				if (!file_exists($pidFile))
					continue;

				$rPID = (int) @file_get_contents($pidFile);
				$rMonitorPID = file_exists($monitorFile) ? (int) @file_get_contents($monitorFile) : 0;

				$rQueue = 0;
				$queueFile = SIGNALS_TMP_PATH . 'queue_' . $rStreamID;
				if (file_exists($queueFile)) {
					$queue = @igbinary_unserialize(@file_get_contents($queueFile)) ?: [];
					foreach ($queue as $pid) {
						if (ProcessManager::isRunning($pid, 'php-fpm'))
							$rQueue++;
					}
				}

				$rAdminQueue = (file_exists(SIGNALS_TMP_PATH . 'admin_' . $rStreamID) && time() - @filemtime(SIGNALS_TMP_PATH . 'admin_' . $rStreamID) <= 30) ? 1 : 0;

				$rStreamAge = 0;
				if (file_exists($pidFile)) {
					$rStreamAge = time() - @filemtime($pidFile);
				}

				if ($rQueue > 0 || $rAdminQueue > 0 || $rStreamAge < 30) {
					continue;
				}

				echo "Killing a stream without viewers: ID $rStreamID\n";

				if ($rMonitorPID > 0)
					@posix_kill($rMonitorPID, 9);
				if ($rPID > 0)
					@posix_kill($rPID, 9);

				@shell_exec('rm -f ' . STREAMS_PATH . $rStreamID . '_*');
				@unlink($queueFile);
				@unlink(SIGNALS_TMP_PATH . 'admin_' . $rStreamID);

				$db->query("UPDATE streams_servers SET bitrate = NULL, current_source = NULL, to_analyze = 0, pid = NULL, stream_started = NULL, stream_info = NULL, audio_codec = NULL, video_codec = NULL, resolution = NULL, compatible = 0, stream_status = 0, monitor_pid = NULL WHERE stream_id = ? AND server_id = ?", $rStreamID, SERVER_ID);

				$db->query("INSERT INTO signals (server_id, cache, time, custom_data) VALUES (?, 1, ?, ?)", $rMainID, time(), json_encode(['type' => 'update_stream', 'id' => $rStreamID]));

				StreamProcess::updateStream($rStreamID);
			}

			usleep(800000);
		}

		if (is_object($db))
			$db->close_mysql();
		shell_exec('(sleep 2; ' . PHP_BIN . ' ' . MAIN_HOME . 'console.php ondemand) > /dev/null 2>&1 &');

		return 0;
	}
}