<?php

/*
 * Copyright 2005 - 2025 Centreon (https://www.centreon.com/)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * For more information : contact@centreon.com
 *
 */

// file centreon.config.php may not exist in test environment
$configFile = realpath(__DIR__ . '/../../config/centreon.config.php');
if ($configFile !== false) {
    require_once $configFile;
}

require_once __DIR__ . '/centreonDB.class.php';
require_once realpath(__DIR__ . '/centreonDBInstance.class.php');
require_once __DIR__ . '/../include/common/common-Func.php';

/**
 * Class
 *
 * @class CentreonExternalCommand
 * @description This class allows the user to send external commands to Nagios
 */
class CentreonExternalCommand
{
    /** @var array */
    public $localhostTab = [];

    /** @var int */
    public $debug = 0;

    /** @var CentreonDB */
    protected $DB;

    /** @var CentreonDB */
    protected $DBC;

    /** @var array */
    protected $cmdTab = [];

    /** @var array */
    protected $pollerTab;

    /** @var array */
    protected $actions = [];

    /** @var CentreonGMT */
    protected $GMT;

    /** @var string|null */
    protected $userAlias;

    /** @var int|mixed|string|null */
    protected $userId;

    /**
     * CentreonExternalCommand constructor
     */
    public function __construct()
    {
        global $centreon;

        $rq = "SELECT id FROM `nagios_server` WHERE localhost = '1'";
        $DBRES = CentreonDBInstance::getDbCentreonInstance()->query($rq);
        while ($row = $DBRES->fetchRow()) {
            $this->localhostTab[$row['id']] = '1';
        }
        $DBRES->closeCursor();

        $this->setExternalCommandList();

        // Init GMT classes
        $this->GMT = new CentreonGMT();
        $this->GMT->getMyGMTFromSession(session_id());

        if (! is_null($centreon)) {
            $this->userId = $centreon->user->get_id();
            $this->userAlias = $centreon->user->get_alias();
        }
    }

    /**
     * @param $newUserId
     *
     * @return void
     */
    public function setUserId($newUserId): void
    {
        $this->userId = $newUserId;
    }

    /**
     * @param $newUserAlias
     *
     * @return void
     */
    public function setUserAlias($newUserAlias): void
    {
        $this->userAlias = $newUserAlias;
    }

    /**
     * Write command in Nagios or Centcore Pipe.
     *
     * @return int
     */
    public function write()
    {
        global $centreon;

        $varlib = ! defined('_CENTREON_VARLIB_') ? '/var/lib/centreon' : _CENTREON_VARLIB_;

        $str_remote = '';
        $return_remote = 0;

        foreach ($this->cmdTab as $key => $cmd) {
            $cmd = str_replace('"', '', $cmd);
            $cmd = str_replace("\n", '<br>', $cmd);
            $cmd = '[' . time() . '] ' . $cmd . "\n";
            $str_remote .= 'EXTERNALCMD:' . $this->pollerTab[$key] . ':' . $cmd;
        }

        if ($str_remote != '') {
            if ($this->debug) {
                echo "COMMAND BEFORE SEND: {$str_remote}";
            }
            $result = file_put_contents($varlib . '/centcore/' . microtime(true) . '-externalcommand.cmd', $str_remote, FILE_APPEND);
            $return_remote = ($result !== false) ? 0 : 1;
        }

        $this->cmdTab = [];
        $this->pollerTab = [];

        return $return_remote;
    }

    /**
     * @param $command
     * @param $poller
     *
     * @return void
     */
    public function setProcessCommand($command, $poller): void
    {
        if ($this->debug) {
            echo "POLLER: {$poller}<br>";
            echo "COMMAND: {$command}<br>";
        }

        $this->cmdTab[] = $command;
        $this->pollerTab[] = $poller;
    }

    /**
     * Get poller id where the host is hosted
     *
     * @param null $host
     *
     * @throws PDOException
     * @return int
     * @internal param $pearDB
     * @internal param $host_name
     */
    public function getPollerID($host = null)
    {
        if (! isset($host)) {
            return 0;
        }
        $db = CentreonDBInstance::getDbCentreonStorageInstance();
        // Check if $host is an id or a name
        if (preg_match("/^\d+$/", $host)) {
            $statement = $db->prepare(
                <<<'SQL'
                    SELECT instance_id
                    FROM hosts
                    WHERE hosts.host_id = :host_id
                      AND hosts.enabled = '1'
                    SQL
            );
            $statement->bindValue(':host_id', (int) $host, PDO::PARAM_INT);
            $statement->execute();
        } else {
            $statement = $db->prepare(
                <<<'SQL'
                    SELECT instance_id
                    FROM hosts
                    WHERE hosts.name = :host_name
                      AND hosts.enabled = '1' LIMIT 1
                    SQL
            );
            $statement->bindValue(':host_name', $host);
            $statement->execute();
        }
        $row = $statement->fetchRow();

        return $row['instance_id'] ?? 0;
    }

    /**
     * get list of external commands
     *
     * @return array
     */
    public function getExternalCommandList()
    {
        return $this->actions;
    }

    // Schedule check

    /**
     * @param $hostName
     *
     * @throws PDOException
     */
    public function scheduleForcedCheckHost($hostName): void
    {
        $pollerId = $this->getPollerID($hostName);

        $this->setProcessCommand(
            'SCHEDULE_FORCED_HOST_CHECK;' . $hostName . ';' . time(),
            $pollerId
        );

        $this->write();
    }

    /**
     * @param $hostName
     * @param $serviceDescription
     *
     * @throws PDOException
     */
    public function scheduleForcedCheckService($hostName, $serviceDescription): void
    {
        $pollerId = $this->getPollerID($hostName);

        $this->setProcessCommand(
            'SCHEDULE_FORCED_SVC_CHECK;' . $hostName . ';' . $serviceDescription . ';' . time(),
            $pollerId
        );

        $this->write();
    }

    // Acknowledgement

    /**
     * @param $hostName
     * @param $sticky
     * @param $notify
     * @param $persistent
     * @param $author
     * @param $comment
     *
     * @throws PDOException
     */
    public function acknowledgeHost(
        $hostName,
        $sticky,
        $notify,
        $persistent,
        $author,
        $comment,
    ): void {
        $pollerId = $this->getPollerID($hostName);

        $this->setProcessCommand(
            'ACKNOWLEDGE_HOST_PROBLEM;' . $hostName . ';'
            . $sticky . ';' . $notify . ';' . $persistent . ';' . $author . ';' . $comment,
            $pollerId
        );

        $this->write();
    }

    /**
     * @param $hostName
     * @param $serviceDescription
     * @param $sticky
     * @param $notify
     * @param $persistent
     * @param $author
     * @param $comment
     *
     * @throws PDOException
     */
    public function acknowledgeService(
        $hostName,
        $serviceDescription,
        $sticky,
        $notify,
        $persistent,
        $author,
        $comment,
    ): void {
        $pollerId = $this->getPollerID($hostName);

        $this->setProcessCommand(
            'ACKNOWLEDGE_SVC_PROBLEM;' . $hostName . ';' . $serviceDescription . ';'
            . $sticky . ';' . $notify . ';' . $persistent . ';' . $author . ';' . $comment,
            $pollerId
        );

        $this->write();
    }

    /**
     * Delete acknowledgement.
     *
     * @param string $type (HOST/SVC)
     * @param array $hosts
     *
     * @throws PDOException
     */
    public function deleteAcknowledgement($type, $hosts = []): void
    {
        foreach (array_keys($hosts) as $name) {
            $res = preg_split("/\;/", $name);
            $oName = $res[0];
            $pollerId = $this->getPollerID($oName);
            if ($type === 'SVC') {
                $oName .= ';' . $res[1];
            }
            $this->setProcessCommand('REMOVE_' . $type . '_ACKNOWLEDGEMENT;' . $oName, $pollerId);
        }
        $this->write();
    }

    /**
     * Delete downtimes.
     *
     * @param string $type
     * @param array $hosts
     *
     * @throws PDOException
     */
    public function deleteDowntime($type, $hosts = []): void
    {
        foreach ($hosts as $key => $value) {
            $res = preg_split("/\;/", $key);
            $pollerId = $this->getPollerID($res[0]);
            $downtimeInternalId = (int) $res[1];
            $this->setProcessCommand('DEL_' . $type . '_DOWNTIME;' . $downtimeInternalId, $pollerId);
        }
        $this->write();
    }

    /**
     * Add a host downtime
     *
     * @param string $host
     * @param string $comment
     * @param string $start
     * @param string $end
     * @param int $persistant
     * @param null $duration
     * @param bool $withServices
     * @param string $hostOrCentreonTime
     *
     * @throws PDOException
     */
    public function addHostDowntime(
        $host,
        $comment,
        $start,
        $end,
        $persistant,
        $duration = null,
        $withServices = false,
        $hostOrCentreonTime = '0',
    ): void {
        global $centreon;

        if (is_null($centreon)) {
            global $oreon;
            $centreon = $oreon;
        }

        if (! isset($persistant) || ! in_array($persistant, ['0', '1'])) {
            $persistant = '0';
        }

        if ($hostOrCentreonTime == '0') {
            $timezoneId = $this->GMT->getMyGTMFromUser($this->userId);
        } else {
            $timezoneId = $this->GMT->getUTCLocationHost($host);
        }
        $timezone = $this->GMT->getActiveTimezone($timezoneId);
        $start_time = $this->getDowntimeTimestampFromDate($start, $timezone, true);
        $end_time = $this->getDowntimeTimestampFromDate($end, $timezone, false);

        if ($end_time == $start_time) {
            return;
        }

        // Get poller for this host
        $poller_id = $this->getPollerID($host);

        // Send command
        if (! isset($duration)) {
            $duration = $end_time - $start_time;
        }
        $finalHostName = '';
        if (! is_numeric($host)) {
            $finalHostName .= $host;
        } else {
            $finalHostName .= getMyHostName($host);
        }
        $this->setProcessCommand(
            'SCHEDULE_HOST_DOWNTIME;' . $finalHostName . ';' . $start_time . ';' . $end_time
            . ';' . $persistant . ';0;' . $duration . ';' . $this->userAlias . ';' . $comment,
            $poller_id
        );
        if ($withServices === true) {
            $this->setProcessCommand(
                'SCHEDULE_HOST_SVC_DOWNTIME;' . $finalHostName . ';' . $start_time . ';' . $end_time
                . ';' . $persistant . ';0;' . $duration . ';' . $this->userAlias . ';' . $comment,
                $poller_id
            );
        }
        $this->write();
    }

    /**
     * Add Service Downtime
     *
     * @param string $host
     * @param string $service
     * @param string $comment
     * @param string $start
     * @param string $end
     * @param int $persistant
     * @param null $duration
     * @param string $hostOrCentreonTime
     *
     * @throws PDOException
     */
    public function addSvcDowntime(
        $host,
        $service,
        $comment,
        $start,
        $end,
        $persistant,
        $duration = null,
        $hostOrCentreonTime = '0',
    ): void {
        global $centreon;

        if (is_null($centreon)) {
            global $oreon;
            $centreon = $oreon;
        }

        if (! isset($persistant) || ! in_array($persistant, ['0', '1'])) {
            $persistant = '0';
        }

        if ($hostOrCentreonTime == '0') {
            $timezoneId = $this->GMT->getMyGTMFromUser($this->userId);
        } else {
            $timezoneId = $this->GMT->getUTCLocationHost($host);
        }

        $timezone = $this->GMT->getActiveTimezone($timezoneId);
        $start_time = $this->getDowntimeTimestampFromDate($start, $timezone, true);
        $end_time = $this->getDowntimeTimestampFromDate($end, $timezone, false);

        if ($end_time == $start_time) {
            return;
        }

        // Get poller for this host
        $poller_id = $this->getPollerID($host);

        // Send command
        if (! isset($duration)) {
            $duration = $end_time - $start_time;
        }
        $finalHostName = '';
        if (! is_numeric($host)) {
            $finalHostName .= $host;
        } else {
            $finalHostName .= getMyHostName($host);
        }
        $finalServiceName = '';
        if (! is_numeric($service)) {
            $finalServiceName .= $service;
        } else {
            $finalServiceName .= getMyServiceName($service);
        }
        $this->setProcessCommand(
            'SCHEDULE_SVC_DOWNTIME;' . $finalHostName . ';' . $finalServiceName . ';' . $start_time
            . ';' . $end_time . ';' . $persistant . ';0;' . $duration . ';' . $this->userAlias
            . ';' . $comment,
            $poller_id
        );
        $this->write();
    }

    /**
     * set list of external commands
     *
     * @return void
     */
    private function setExternalCommandList(): void
    {
        // Services Actions
        $this->actions['service_checks'][0] = 'ENABLE_SVC_CHECK';
        $this->actions['service_checks'][1] = 'DISABLE_SVC_CHECK';

        $this->actions['service_notifications'][0] = 'ENABLE_SVC_NOTIFICATIONS';
        $this->actions['service_notifications'][1] = 'DISABLE_SVC_NOTIFICATIONS';

        $this->actions['service_acknowledgement'][0] = 'ACKNOWLEDGE_SVC_PROBLEM';
        $this->actions['service_disacknowledgement'][0] = 'REMOVE_SVC_ACKNOWLEDGEMENT';

        $this->actions['service_schedule_check'][0] = 'SCHEDULE_SVC_CHECK';
        $this->actions['service_schedule_check'][1] = 'SCHEDULE_FORCED_SVC_CHECK';
        $this->actions['service_schedule_forced_check'][0] = 'SCHEDULE_FORCED_SVC_CHECK';

        $this->actions['service_schedule_downtime'][0] = 'SCHEDULE_SVC_DOWNTIME';

        $this->actions['service_comment'][0] = 'ADD_SVC_COMMENT';

        $this->actions['service_event_handler'][0] = 'ENABLE_SVC_EVENT_HANDLER';
        $this->actions['service_event_handler'][1] = 'DISABLE_SVC_EVENT_HANDLER';

        $this->actions['service_flap_detection'][0] = 'ENABLE_SVC_FLAP_DETECTION';
        $this->actions['service_flap_detection'][1] = 'DISABLE_SVC_FLAP_DETECTION';

        $this->actions['service_passive_checks'][0] = 'ENABLE_PASSIVE_SVC_CHECKS';
        $this->actions['service_passive_checks'][1] = 'DISABLE_PASSIVE_SVC_CHECKS';

        $this->actions['service_submit_result'][0] = 'PROCESS_SERVICE_CHECK_RESULT';

        $this->actions['service_obsess'][0] = 'START_OBSESSING_OVER_SVC';
        $this->actions['service_obsess'][1] = 'STOP_OBSESSING_OVER_SVC';

        // Hosts Actions
        $this->actions['host_checks'][0] = 'ENABLE_HOST_CHECK';
        $this->actions['host_checks'][1] = 'DISABLE_HOST_CHECK';

        $this->actions['host_passive_checks'][0] = 'ENABLE_PASSIVE_HOST_CHECKS';
        $this->actions['host_passive_checks'][1] = 'DISABLE_PASSIVE_HOST_CHECKS';

        $this->actions['host_notifications'][0] = 'ENABLE_HOST_NOTIFICATIONS';
        $this->actions['host_notifications'][1] = 'DISABLE_HOST_NOTIFICATIONS';

        $this->actions['host_acknowledgement'][0] = 'ACKNOWLEDGE_HOST_PROBLEM';
        $this->actions['host_disacknowledgement'][0] = 'REMOVE_HOST_ACKNOWLEDGEMENT';

        $this->actions['host_schedule_check'][0] = 'SCHEDULE_HOST_SVC_CHECKS';
        $this->actions['host_schedule_check'][1] = 'SCHEDULE_FORCED_HOST_SVC_CHECKS';
        $this->actions['host_schedule_forced_check'][0] = 'SCHEDULE_FORCED_HOST_SVC_CHECKS';

        $this->actions['host_schedule_downtime'][0] = 'SCHEDULE_HOST_DOWNTIME';

        $this->actions['host_comment'][0] = 'ADD_HOST_COMMENT';

        $this->actions['host_event_handler'][0] = 'ENABLE_HOST_EVENT_HANDLER';
        $this->actions['host_event_handler'][1] = 'DISABLE_HOST_EVENT_HANDLER';

        $this->actions['host_flap_detection'][0] = 'ENABLE_HOST_FLAP_DETECTION';
        $this->actions['host_flap_detection'][1] = 'DISABLE_HOST_FLAP_DETECTION';

        $this->actions['host_checks_for_services'][0] = 'ENABLE_HOST_SVC_CHECKS';
        $this->actions['host_checks_for_services'][1] = 'DISABLE_HOST_SVC_CHECKS';

        $this->actions['host_notifications_for_services'][0] = 'ENABLE_HOST_SVC_NOTIFICATIONS';
        $this->actions['host_notifications_for_services'][1] = 'DISABLE_HOST_SVC_NOTIFICATIONS';

        $this->actions['host_obsess'][0] = 'START_OBSESSING_OVER_HOST';
        $this->actions['host_obsess'][1] = 'STOP_OBSESSING_OVER_HOST';

        // Global Nagios External Commands
        $this->actions['global_shutdown'][0] = 'SHUTDOWN_PROGRAM';
        $this->actions['global_shutdown'][1] = 'SHUTDOWN_PROGRAM';

        $this->actions['global_restart'][0] = 'RESTART_PROGRAM';
        $this->actions['global_restart'][1] = 'RESTART_PROGRAM';

        $this->actions['global_notifications'][0] = 'ENABLE_NOTIFICATIONS';
        $this->actions['global_notifications'][1] = 'DISABLE_NOTIFICATIONS';

        $this->actions['global_service_checks'][0] = 'START_EXECUTING_SVC_CHECKS';
        $this->actions['global_service_checks'][1] = 'STOP_EXECUTING_SVC_CHECKS';

        $this->actions['global_service_passive_checks'][0] = 'START_ACCEPTING_PASSIVE_SVC_CHECKS';
        $this->actions['global_service_passive_checks'][1] = 'STOP_ACCEPTING_PASSIVE_SVC_CHECKS';

        $this->actions['global_host_checks'][0] = 'START_EXECUTING_HOST_CHECKS';
        $this->actions['global_host_checks'][1] = 'STOP_EXECUTING_HOST_CHECKS';

        $this->actions['global_host_passive_checks'][0] = 'START_ACCEPTING_PASSIVE_HOST_CHECKS';
        $this->actions['global_host_passive_checks'][1] = 'STOP_ACCEPTING_PASSIVE_HOST_CHECKS';

        $this->actions['global_event_handler'][0] = 'ENABLE_EVENT_HANDLERS';
        $this->actions['global_event_handler'][1] = 'DISABLE_EVENT_HANDLERS';

        $this->actions['global_flap_detection'][0] = 'ENABLE_FLAP_DETECTION';
        $this->actions['global_flap_detection'][1] = 'DISABLE_FLAP_DETECTION';

        $this->actions['global_service_obsess'][0] = 'START_OBSESSING_OVER_SVC_CHECKS';
        $this->actions['global_service_obsess'][1] = 'STOP_OBSESSING_OVER_SVC_CHECKS';

        $this->actions['global_host_obsess'][0] = 'START_OBSESSING_OVER_HOST_CHECKS';
        $this->actions['global_host_obsess'][1] = 'STOP_OBSESSING_OVER_HOST_CHECKS';

        $this->actions['global_perf_data'][0] = 'ENABLE_PERFORMANCE_DATA';
        $this->actions['global_perf_data'][1] = 'DISABLE_PERFORMANCE_DATA';
    }

    // Downtime

    /**
     * @param $date
     * @param $timezone
     * @param $start
     *
     * @throws Exception
     * @return int
     */
    private function getDowntimeTimestampFromDate($date = 'now', $timezone = '', $start = true)
    {
        $inputDate = new DateTime($date . ' GMT');
        $dateTime = new DateTime($date, new DateTimeZone($timezone));

        // Winter to summer dst
        $dateTime2 = clone $dateTime;
        $dateTime2->setTimestamp($dateTime2->getTimestamp());

        if ($dateTime2->format('H') != $inputDate->format('H')) {
            $hour = $inputDate->format('H');
            $dateTime->setTime($hour, '00');

            return $dateTime->getTimestamp();
        }

        // Summer to winter dst
        $dateTime3 = clone $dateTime;
        $dateTime3->sub(new DateInterval('PT1H'));
        if ($dateTime3->format('H:m') === $dateTime->format('H:m')) {
            if ($start) {
                return $dateTime->getTimestamp() - 3600;
            }

            return $dateTime->getTimestamp();
        }

        $dateTime4 = clone $dateTime;
        $dateTime4->add(new DateInterval('PT1H'));
        if ($dateTime4->format('H:m') === $dateTime->format('H:m')) {
            if ($start) {
                return $dateTime->getTimestamp();
            }

            return $dateTime->getTimestamp() + 3600;
        }

        return $dateTime->getTimestamp();
    }
}
