public function testGenerate()
 {
     $fileList = array();
     $pollerId = 1;
     CommandRepository::generate($fileList, $pollerId, $this->tmpDir . '/', 'command.cfg', CommandRepository::CHECK_TYPE);
     $this->assertEquals(array('cfg_file' => array($this->tmpDir . '/1/command.cfg')), $fileList);
     $content = file_get_contents($this->tmpDir . '/1/command.cfg');
     /* Remove line with the generate date */
     $lines = explode("\n", $content);
     $lines = preg_grep('/^#\\s+Last.*#$/', $lines, PREG_GREP_INVERT);
     $content = join("\n", $lines);
     $resultContent = file_get_contents(dirname(__DIR__) . '/data/configfiles/command1.cfg');
     $this->assertEquals($resultContent, $content);
     $fileList = array();
     CommandRepository::generate($fileList, $pollerId, $this->tmpDir . '/', 'command.cfg', CommandRepository::NOTIF_TYPE);
     $this->assertEquals(array('cfg_file' => array($this->tmpDir . '/1/command.cfg')), $fileList);
     $content = file_get_contents($this->tmpDir . '/1/command.cfg');
     /* Remove line with the generate date */
     $lines = explode("\n", $content);
     $lines = preg_grep('/^#\\s+Last.*#$/', $lines, PREG_GREP_INVERT);
     $content = join("\n", $lines);
     $resultContent = file_get_contents(dirname(__DIR__) . '/data/configfiles/command2.cfg');
     $this->assertEquals($resultContent, $content);
 }
 /**
  * 
  * @param int $poller_id
  * @param array $filesList
  * @param int $testing
  * @return array
  */
 private static function getContent($poller_id, &$filesList, $testing)
 {
     $di = Di::getDefault();
     /* Get Database Connexion */
     $dbconn = $di->get('db_centreon');
     /* Init Content Array */
     $content = array();
     /* Default values that can be overwritten by template and user */
     $defaultValues = static::getDefaultValues();
     /* Template values that can be overwritten by user */
     $templateValues = static::getTemplateValues($poller_id);
     /* For command name resolution */
     $commandIdFields = static::getCommandIdField();
     /* Get configuration files */
     $content = static::getFilesList($filesList, $content, $testing, $poller_id);
     /* Get values from the table, those are saved by user */
     $query = "SELECT * FROM cfg_engine WHERE poller_id = ?";
     $stmt = $dbconn->prepare($query);
     $stmt->execute(array($poller_id));
     $row = $stmt->fetch(\PDO::FETCH_ASSOC);
     if (false === $row) {
         throw new Exception(sprintf('Could not find parameters for poller %s.', $poller_id));
     }
     $userValues = array();
     foreach ($row as $key => $val) {
         if (!is_null($val) && $val != "") {
             $userValues[$key] = $val;
         }
     }
     /* Overwrite parameter */
     $tmpConf = array_merge($defaultValues, $templateValues);
     $finalConf = array_merge($tmpConf, $userValues);
     /* For object directory */
     $objectDirectives = static::getConfigFiles($poller_id);
     $finalConf = array_merge($finalConf, $objectDirectives);
     /* Add illegal chars */
     $finalConf['illegal_macro_output_chars'] = CustomMacroRepository::$forbidenCHar;
     $finalConf['illegal_object_name_chars'] = CENTREON_ILLEGAL_CHAR_OBJ;
     /* Set real etc path of the poller */
     static::$finalPath = $finalConf['conf_dir'];
     /* Replace path macros */
     foreach ($finalConf as $k => $v) {
         $arr = array();
         if (!is_array($v)) {
             $arr[] = $v;
         } else {
             $arr = $v;
         }
         foreach ($arr as $key => $val) {
             if (preg_match('/%([a-z_]+)%/', $val, $matches)) {
                 $macro = $matches[1];
                 if (isset($finalConf[$macro])) {
                     $finalConf[$macro] = rtrim($finalConf[$macro], '/');
                     if (is_array($finalConf[$k])) {
                         $finalConf[$k][$key] = str_replace("%{$macro}%", $finalConf[$macro], $val);
                     } else {
                         $finalConf[$k] = str_replace("%{$macro}%", $finalConf[$macro], $val);
                     }
                 }
                 if ($macro == 'conf_dir' && $testing) {
                     $finalConf[$k][$key] = str_replace('%conf_dir%', static::$path . "/" . $poller_id, $val);
                 }
             }
         }
     }
     /* Replace commands */
     foreach ($commandIdFields as $fieldName) {
         if (isset($finalConf[$fieldName]) && $finalConf[$fieldName]) {
             $finalConf[$fieldName] = CommandRepository::getCommandName($finalConf[$fieldName]);
         }
     }
     /* Exclude parameters */
     static::unsetParameters($finalConf);
     return $finalConf;
 }
示例#3
0
 /**
  * Generate all object files (host, service, etc...)
  *
  */
 public static function generateObjectsFiles()
 {
     $event = static::$event;
     /* Generate Configuration files */
     CommandRepository::generate(static::$fileList, $event->getPollerId(), static::$path, "check-command.cfg", CommandRepository::CHECK_TYPE);
     $event->setOutput('check-command.cfg');
     CommandRepository::generate(static::$fileList, $event->getPollerId(), static::$path, "misc-command.cfg", CommandRepository::NOTIF_TYPE);
     $event->setOutput('misc-command.cfg');
     ConfigGenerateResourcesRepository::generate(static::$fileList, $event->getPollerId(), static::$path, "resources.cfg");
     $event->setOutput('resources.cfg');
     ConfigGenerateModulesRepository::generate(static::$fileList, $event->getPollerId(), static::$path, $event);
     TimePeriodRepository::generate(static::$fileList, $event->getPollerId(), static::$path, "timeperiods.cfg");
     $event->setOutput('timeperiods.cfg');
     ConnectorRepository::generate(static::$fileList, $event->getPollerId(), static::$path, "connectors.cfg");
     $event->setOutput('connectors.cfg');
     /* Retrieve all extra macros by emitting event */
     $events = Di::getDefault()->get('events');
     $hostMacroEvent = new HostMacroEvent($event->getPollerId());
     $events->emit('centreon-engine.get.macro.host', array($hostMacroEvent));
     $serviceMacroEvent = new ServiceMacroEvent($event->getPollerId());
     $events->emit('centreon-engine.get.macro.service', array($serviceMacroEvent));
     $notificationRulesEvent = new NotificationRuleEvent($event->getPollerId());
     $events->emit('centreon-engine.set.notifications.rules', array($notificationRulesEvent));
     /* Templates config files */
     HostTemplateRepository::generate(static::$fileList, $event->getPollerId(), static::$path, "hostTemplates.cfg", $hostMacroEvent);
     $event->setOutput('hostTemplates.cfg');
     ServicetemplateRepository::generate(static::$fileList, $event->getPollerId(), static::$path, "serviceTemplates.cfg", $serviceMacroEvent);
     $event->setOutput('serviceTemplate.cfg');
     /* Monitoring Resources files */
     HostRepository::generate(static::$fileList, $event->getPollerId(), static::$path, "resources/", $hostMacroEvent, $serviceMacroEvent);
     $event->setOutput('host configuration files');
 }