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
ContactGroupsTable()9 ContactGroupsTable::ContactGroupsTable()
10 {
11 AddColumns(this);
12 }
13
AddColumns(Table * table,const String & prefix,const Column::ObjectAccessor & objectAccessor)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
GetName() const22 String ContactGroupsTable::GetName() const
23 {
24 return "contactgroups";
25 }
26
GetPrefix() const27 String ContactGroupsTable::GetPrefix() const
28 {
29 return "contactgroup";
30 }
31
FetchRows(const AddRowFunction & addRowFn)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
NameAccessor(const Value & row)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
AliasAccessor(const Value & row)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
MembersAccessor(const Value & row)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 }
75