mirror of https://github.com/Icinga/icinga2.git
104 lines
2.6 KiB
C++
104 lines
2.6 KiB
C++
/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
|
|
|
|
#include "icinga/user.hpp"
|
|
#include "icinga/user-ti.cpp"
|
|
#include "icinga/usergroup.hpp"
|
|
#include "icinga/notification.hpp"
|
|
#include "icinga/usergroup.hpp"
|
|
#include "base/objectlock.hpp"
|
|
#include "base/exception.hpp"
|
|
|
|
using namespace icinga;
|
|
|
|
REGISTER_TYPE(User);
|
|
|
|
void User::OnConfigLoaded()
|
|
{
|
|
ObjectImpl<User>::OnConfigLoaded();
|
|
|
|
SetTypeFilter(FilterArrayToInt(GetTypes(), Notification::GetTypeFilterMap(), ~0));
|
|
SetStateFilter(FilterArrayToInt(GetStates(), Notification::GetStateFilterMap(), ~0));
|
|
}
|
|
|
|
void User::OnAllConfigLoaded()
|
|
{
|
|
ObjectImpl<User>::OnAllConfigLoaded();
|
|
|
|
UserGroup::EvaluateObjectRules(this);
|
|
|
|
Array::Ptr groups = GetGroups();
|
|
|
|
if (groups) {
|
|
groups = groups->ShallowClone();
|
|
|
|
ObjectLock olock(groups);
|
|
|
|
for (const String& name : groups) {
|
|
UserGroup::Ptr ug = UserGroup::GetByName(name);
|
|
|
|
if (ug)
|
|
ug->ResolveGroupMembership(this, true);
|
|
}
|
|
}
|
|
}
|
|
|
|
void User::Stop(bool runtimeRemoved)
|
|
{
|
|
ObjectImpl<User>::Stop(runtimeRemoved);
|
|
|
|
Array::Ptr groups = GetGroups();
|
|
|
|
if (groups) {
|
|
ObjectLock olock(groups);
|
|
|
|
for (const String& name : groups) {
|
|
UserGroup::Ptr ug = UserGroup::GetByName(name);
|
|
|
|
if (ug)
|
|
ug->ResolveGroupMembership(this, false);
|
|
}
|
|
}
|
|
}
|
|
|
|
void User::AddGroup(const String& name)
|
|
{
|
|
std::unique_lock<std::mutex> lock(m_UserMutex);
|
|
|
|
Array::Ptr groups = GetGroups();
|
|
|
|
if (groups && groups->Contains(name))
|
|
return;
|
|
|
|
if (!groups)
|
|
groups = new Array();
|
|
|
|
groups->Add(name);
|
|
}
|
|
|
|
TimePeriod::Ptr User::GetPeriod() const
|
|
{
|
|
return TimePeriod::GetByName(GetPeriodRaw());
|
|
}
|
|
|
|
void User::ValidateStates(const Lazy<Array::Ptr>& lvalue, const ValidationUtils& utils)
|
|
{
|
|
ObjectImpl<User>::ValidateStates(lvalue, utils);
|
|
|
|
int filter = FilterArrayToInt(lvalue(), Notification::GetStateFilterMap(), 0);
|
|
|
|
if (filter == -1 || (filter & ~(StateFilterUp | StateFilterDown | StateFilterOK | StateFilterWarning | StateFilterCritical | StateFilterUnknown)) != 0)
|
|
BOOST_THROW_EXCEPTION(ValidationError(this, { "states" }, "State filter is invalid."));
|
|
}
|
|
|
|
void User::ValidateTypes(const Lazy<Array::Ptr>& lvalue, const ValidationUtils& utils)
|
|
{
|
|
ObjectImpl<User>::ValidateTypes(lvalue, utils);
|
|
|
|
int filter = FilterArrayToInt(lvalue(), Notification::GetTypeFilterMap(), 0);
|
|
|
|
if (filter == -1 || (filter & ~(NotificationDowntimeStart | NotificationDowntimeEnd | NotificationDowntimeRemoved |
|
|
NotificationCustom | NotificationAcknowledgement | NotificationProblem | NotificationRecovery |
|
|
NotificationFlappingStart | NotificationFlappingEnd)) != 0)
|
|
BOOST_THROW_EXCEPTION(ValidationError(this, { "types" }, "Type filter is invalid."));
|
|
}
|