source: trunk/Modules/Notify/Notify.php@ 887

<?php

define('CONTACT_CATEGORY_EMAIL', 4);

class ModuleNotify extends AppModule
{
  public array $Checks;

  function __construct(System $System)
  {
    parent::__construct($System);
    $this->Name = 'Notify';
    $this->Version = '1.0';
    $this->Creator = 'Chronos';
    $this->License = 'GNU/GPL';
    $this->Description = 'Send notification messages to selected users';
    $this->Dependencies = array('User', 'RSS');
    $this->Checks = array();
  }

  function DoStart(): void
  {
    $this->System->FormManager->RegisterClass('NotifyUser', array(
      'Title' => 'Upozornění uživatelé',
      'Table' => 'NotifyUser',
      'Items' => array(
        'User' => array('Type' => 'TUser', 'Caption' => 'Uživatel', 'Default' => ''),
        'Contact' => array('Type' => 'TContact', 'Caption' => 'Kontakt', 'Default' => ''),
        'Period' => array('Type' => 'Integer', 'Caption' => 'Interval', 'Default' => '60'),
      ),
    ));
    $this->System->FormManager->RegisterClass('NotifyCategory', array(
      'Title' => 'Kategorie upozornění',
      'Table' => 'NotifyCategory',
      'Items' => array(
        'Name' => array('Type' => 'String', 'Caption' => 'Jméno', 'Default' => ''),
        'SysName' => array('Type' => 'String', 'Caption' => 'Systémové jméno', 'Default' => ''),
      ),
    ));
    $this->System->RegisterCommandLine('notify', 'Perform notifications processing.', array($this, 'RunCheck'));
    ModuleRSS::Cast($this->System->GetModule('RSS'))->RegisterRSS(array('Title' => 'Notify log',
    'Channel' => 'notifylog', 'Callback' => array($this, 'ShowLogRSS'),
    'Permission' => array('Module' => 'Notify', 'Operation' => 'RSS')));
  }

  function RegisterCheck(string $Name, callable $Callback): void
  {
    if (array_key_exists($Name, $this->Checks))
      throw new Exception('Check function "'.$Name.'" already registered.');
    $this->Checks[$Name] = array('Callback' => $Callback);
  }

  function UnregisterCheck(string $Name): void
  {
    if (!array_key_exists($Name, $this->Checks))
      throw new Exception('Check function "'.$Name.'" not registered.');
    unset($this->Checks[$Name]);
  }

  function Check(): string
  {
    $Output = '';
    $Content = '';
    $Title = '';

    // Get output from checks
    $DbResult2 = $this->Database->query('SELECT `Id`, `Name`, `SysName` FROM `NotifyCategory`');
    while ($Category = $DbResult2->fetch_assoc())
    {
      if (array_key_exists($Category['SysName'], $this->Checks))
      {
        $Status = call_user_func($this->Checks[$Category['SysName']]['Callback']);
        if ($Status['Report'] != '')
        {
          $Output .= 'Kategorie: '.$Category['Name'].":\n";
          $Content .= 'Kategorie: '.$Category['Name']."<br>\n";
          $Content .= $Status['Report'];
        }
        if (strlen($Title) > 0) $Title .= ', ';
        $Title .= $Status['ShortTitle'].':'.$Status['Count'];
      }
    }

    $Time = time();

    // Send content to users
    $DbResult = $this->Database->query('SELECT `NotifyUser`.`Id`, `NotifyUser`.`LastTime`, `User`.`Name`, `Contact`.`Value`, `Contact`.`Category` FROM `NotifyUser` '.
      'LEFT JOIN `User` ON `User`.`Id` = `NotifyUser`.`User` '.
      'LEFT JOIN `Contact` ON `Contact`.`Id` = `NotifyUser`.`Contact`');
    while ($User = $DbResult->fetch_assoc())
    {
      $Output .= 'User '.$User['Name'].'<br/>';

      $this->Database->update('NotifyUser', '`Id`='.$User['Id'], array('LastTime' => TimeToMysqlDateTime($Time)));

      if (($User['Category'] == CONTACT_CATEGORY_EMAIL) and ($Content != ''))
      {
        $Mail = new Mail();
        $Mail->Subject = $Title;
        $Mail->AddTo($User['Value'], $User['Name']);
        $Mail->AddBody(strip_tags($Content), 'text/plain');
        $Mail->AddBody('<style>
table { border-collapse: collapse; }
table, th, td { border: 1px solid black; }
td { padding: 5px; }
        </style>'.$Content, 'text/html');
        $Mail->Send();
        $Output .= 'Sent email to '.$User['Value'].' ('.$User['Name'].')<br/>';
        $Output .= strip_tags(str_replace("</", " </", str_replace('<br/>', "\n", $Content)));
      }
    }

    if ($Content != '')
    {
      $this->Database->insert('NotifyLog', array(
        'Time' => TimeToMysqlDateTime($Time),
        'Title' => $Title,
        'Content' => $Content)
      );
    }

    return $Output;
  }

  function RunCheck(array $Parameters): void
  {
    RepeatFunction(30, array($this, 'Check'));
  }

  function DoInstall(): void
  {
    $this->Database->query('CREATE TABLE IF NOT EXISTS `NotifyCategory` (
      `Id` int(11) NOT NULL AUTO_INCREMENT,
      `Name` varchar(255) NOT NULL,
      `SysName` varchar(255) NOT NULL,
      PRIMARY KEY (`Id`)
    ) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8;');

    $this->Database->query("INSERT INTO `NotifyCategory` (`Id`, `Name`, `SysName`) VALUES
    (1, 'Dostupnost zařízení (ping)', 'NetworkReachability'),
    (2, 'Dostupnost URL', 'URL'),
    (3, 'Minimální úroveň signálu', 'WirelessSignal'),
    (4, 'Dostupnost síťového portu', 'NetworkPort'),
    (5, 'Minimální odezva', 'NetworkLatency'),
    (6, 'Minimální propustnost', 'NetworkBandwidth');");

    $this->Database->query('
  CREATE TABLE IF NOT EXISTS `NotifyUser` (
  `Id` int(11) NOT NULL AUTO_INCREMENT,
  `User` int(11) NOT NULL,
  `Contact` int(11) NOT NULL,
  `Period` int(11) NOT NULL,
  PRIMARY KEY (`Id`),
  KEY `User` (`User`),
  KEY `Contact` (`Contact`)
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8 AUTO_INCREMENT=1;');
    $this->Database->query('ALTER TABLE `NotifyUser`
ADD CONSTRAINT `NotifyUser_ibfk_1` FOREIGN KEY (`User`) REFERENCES `User` (`Id`),
ADD CONSTRAINT `NotifyUser_ibfk_2` FOREIGN KEY (`Contact`) REFERENCES `Contact` (`Id`);');
  }

  function DoUninstall(): void
  {
    $this->Database->query('DROP TABLE `NotifyUser`');
    $this->Database->query('DROP TABLE `NotifyCategory`');
  }

  function ShowLogRSS(): string
  {
    $this->ClearPage = true;
    $this->FormatHTML = false;
    Header('Content-Type: text/xml');
    $Count = 100;

    $Output = '';
    $Items = array();
    $sql = 'SELECT Title,Content, UNIX_TIMESTAMP(`Time`) AS `Time` FROM `NotifyLog`'.
      ' ORDER BY `Time` DESC LIMIT '.$Count;
    $DbResult = $this->System->Database->query($sql);
    while ($Line = $DbResult->fetch_assoc())
    {
      $Items[] = array
      (
        'Title' => $Line['Title'],
        'Link' => 'http://'.$this->System->Config['Web']['Host'].$this->System->Link('/Notify.php'),
        'Description' => $Line['Content'],
        'Time' => $Line['Time'],
      );
    }

    $RSS = new RSS();
    $RSS->Title = $this->System->Config['Web']['Title'].' - Záznamy upozornění';
    $RSS->Link = 'http://'.$this->System->Config['Web']['Host'].'/';
    $RSS->Description = 'Záznam upozornění '.$this->System->Config['Web']['Description'];
    $RSS->WebmasterEmail = $this->System->Config['Web']['AdminEmail'];
    $RSS->Items = $Items;
    return $RSS->Generate();
  }

  static function Cast(AppModule $AppModule): ModuleNotify
  {
    if ($AppModule instanceof ModuleNotify)
    {
      return $AppModule;
    }
    throw new Exception('Expected ModuleNotify type but got '.gettype($AppModule));
  }
}