livestatus/src/HostListColumn.cc

   1 // +------------------------------------------------------------------+
   2 // |             ____ _               _        __  __ _  __           |
   3 // |            / ___| |__   ___  ___| | __   |  \/  | |/ /           |
   4 // |           | |   | '_ \ / _ \/ __| |/ /   | |\/| | ' /            |
   5 // |           | |___| | | |  __/ (__|   <    | |  | | . \            |
   6 // |            \____|_| |_|\___|\___|_|\_\___|_|  |_|_|\_\           |
   7 // |                                                                  |
   8 // | Copyright Mathias Kettner 2014             mk@mathias-kettner.de |
   9 // +------------------------------------------------------------------+
  10 //
  11 // This file is part of Check_MK.
  12 // The official homepage is at http://mathias-kettner.de/check_mk.
  13 //
  14 // check_mk is free software;  you can redistribute it and/or modify it
  15 // under the  terms of the  GNU General Public License  as published by
  16 // the Free Software Foundation in version 2.  check_mk is  distributed
  17 // in the hope that it will be useful, but WITHOUT ANY WARRANTY;  with-
  18 // out even the implied warranty of  MERCHANTABILITY  or  FITNESS FOR A
  19 // PARTICULAR PURPOSE. See the  GNU General Public License for more de-
  20 // tails. You should have  received  a copy of the  GNU  General Public
  21 // License along with GNU Make; see the file  COPYING.  If  not,  write
  22 // to the Free Software Foundation, Inc., 51 Franklin St,  Fifth Floor,
  23 // Boston, MA 02110-1301 USA.
  24
  25 #include "HostListColumn.h"
  26 #include <algorithm>
  27 #include <iterator>
  28 #include "Renderer.h"
  29 #include "Row.h"
  30
  31 #ifdef CMC
  32 #include <unordered_set>
  33 #include "Host.h"
  34 #include "LogEntry.h"
  35 #include "State.h"
  36 #include "cmc.h"
  37 #else
  38 #include "auth.h"
  39 #include "nagios.h"
  40 #endif
  41
  42 void HostListColumn::output(Row row, RowRenderer &r, const contact *auth_user,
  43                             std::chrono::seconds /*timezone_offset*/) const {
  44     ListRenderer l(r);
  45     for (const auto &member : getMembers(row, auth_user)) {
  46         if (_show_state) {
  47             SublistRenderer s(l);
  48             s.output(member.host_name);
  49             s.output(static_cast<int>(member.current_state));
  50             s.output(static_cast<int>(member.has_been_checked));
  51         } else {
  52             l.output(member.host_name);
  53         }
  54     }
  55 }
  56
  57 std::vector<std::string> HostListColumn::getValue(
  58     Row row, const contact *auth_user,
  59     std::chrono::seconds /*timezone_offset*/) const {
  60     auto members = getMembers(row, auth_user);
  61     std::vector<std::string> host_names;
  62     std::transform(members.begin(), members.end(),
  63                    std::back_inserter(host_names),
  64                    [](const auto &member) { return member.host_name; });
  65     return host_names;
  66 };
  67
  68 std::vector<HostListColumn::Member> HostListColumn::getMembers(
  69     Row row, const contact *auth_user) const {
  70     std::vector<Member> members;
  71 #ifdef CMC
  72     if (auto p = columnData<std::unordered_set<Host *>>(row)) {
  73         for (const auto &hst : *p) {
  74             if (auth_user == nullptr || hst->hasContact(_mc, auth_user)) {
  75                 members.emplace_back(
  76                     hst->name(),
  77                     static_cast<HostState>(hst->state()->_current_state),
  78                     hst->state()->_has_been_checked);
  79             }
  80         }
  81     }
  82 #else
  83     if (auto p = columnData<hostsmember *>(row)) {
  84         for (const hostsmember *mem = *p; mem != nullptr; mem = mem->next) {
  85             host *hst = mem->host_ptr;
  86             if (auth_user == nullptr ||
  87                 is_authorized_for(_mc, auth_user, hst, nullptr)) {
  88                 members.emplace_back(hst->name,
  89                                      static_cast<HostState>(hst->current_state),
  90                                      hst->has_been_checked != 0);
  91             }
  92         }
  93     }
  94 #endif
  95     return members;
  96 }