granicus.if.org Git - icinga2/blob - lib/livestatus/contactgroupstable.cpp

   1 /* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
   2
   3 #include "livestatus/contactgroupstable.hpp"
   4 #include "icinga/usergroup.hpp"
   5 #include "base/configtype.hpp"
   6
   7 using namespace icinga;
   8
   9 ContactGroupsTable::ContactGroupsTable()
  10 {
  11         AddColumns(this);
  12 }
  13
  14 void ContactGroupsTable::AddColumns(Table *table, const String& prefix,
  15         const Column::ObjectAccessor& objectAccessor)
  16 {
  17         table->AddColumn(prefix + "name", Column(&ContactGroupsTable::NameAccessor, objectAccessor));
  18         table->AddColumn(prefix + "alias", Column(&ContactGroupsTable::AliasAccessor, objectAccessor));
  19         table->AddColumn(prefix + "members", Column(&ContactGroupsTable::MembersAccessor, objectAccessor));
  20 }
  21
  22 String ContactGroupsTable::GetName() const
  23 {
  24         return "contactgroups";
  25 }
  26
  27 String ContactGroupsTable::GetPrefix() const
  28 {
  29         return "contactgroup";
  30 }
  31
  32 void ContactGroupsTable::FetchRows(const AddRowFunction& addRowFn)
  33 {
  34         for (const UserGroup::Ptr& ug : ConfigType::GetObjectsByType<UserGroup>()) {
  35                 if (!addRowFn(ug, LivestatusGroupByNone, Empty))
  36                         return;
  37         }
  38 }
  39
  40 Value ContactGroupsTable::NameAccessor(const Value& row)
  41 {
  42         UserGroup::Ptr user_group = static_cast<UserGroup::Ptr>(row);
  43
  44         if (!user_group)
  45                 return Empty;
  46
  47         return user_group->GetName();
  48 }
  49
  50 Value ContactGroupsTable::AliasAccessor(const Value& row)
  51 {
  52         UserGroup::Ptr user_group = static_cast<UserGroup::Ptr>(row);
  53
  54         if (!user_group)
  55                 return Empty;
  56
  57         return user_group->GetName();
  58 }
  59
  60 Value ContactGroupsTable::MembersAccessor(const Value& row)
  61 {
  62         UserGroup::Ptr user_group = static_cast<UserGroup::Ptr>(row);
  63
  64         if (!user_group)
  65                 return Empty;
  66
  67         ArrayData result;
  68
  69         for (const User::Ptr& user : user_group->GetMembers()) {
  70                 result.push_back(user->GetName());
  71         }
  72
  73         return new Array(std::move(result));
  74 }