icinga2/lib/icinga/user.cpp

147 lines
4.3 KiB
C++

/******************************************************************************
* Icinga 2 *
* Copyright (C) 2012 Icinga Development Team (http://www.icinga.org/) *
* *
* This program is free software; you can redistribute it and/or *
* modify it under the terms of the GNU General Public License *
* as published by the Free Software Foundation; either version 2 *
* of the License, or (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program; if not, write to the Free Software Foundation *
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. *
******************************************************************************/
#include "icinga/user.h"
#include "icinga/usergroup.h"
#include "base/dynamictype.h"
#include "base/utility.h"
#include <boost/smart_ptr/make_shared.hpp>
using namespace icinga;
REGISTER_TYPE(User);
User::User(const Dictionary::Ptr& serializedUpdate)
: DynamicObject(serializedUpdate)
{
RegisterAttribute("display_name", Attribute_Config, &m_DisplayName);
RegisterAttribute("macros", Attribute_Config, &m_Macros);
RegisterAttribute("groups", Attribute_Config, &m_Groups);
RegisterAttribute("enable_notifications", Attribute_Config, &m_EnableNotifications);
RegisterAttribute("notification_period", Attribute_Config, &m_NotificationPeriod);
RegisterAttribute("notification_type_filter", Attribute_Config, &m_NotificationTypeFilter);
RegisterAttribute("notification_state_filter", Attribute_Config, &m_NotificationStateFilter);
RegisterAttribute("last_notification", Attribute_Replicated, &m_LastNotification);
}
User::~User(void)
{
UserGroup::InvalidateMembersCache();
}
void User::OnAttributeChanged(const String& name)
{
ASSERT(!OwnsLock());
if (name == "groups")
UserGroup::InvalidateMembersCache();
}
User::Ptr User::GetByName(const String& name)
{
DynamicObject::Ptr configObject = DynamicObject::GetObject("User", name);
return dynamic_pointer_cast<User>(configObject);
}
String User::GetDisplayName(void) const
{
if (!m_DisplayName.IsEmpty())
return m_DisplayName;
else
return GetName();
}
Array::Ptr User::GetGroups(void) const
{
return m_Groups;
}
Dictionary::Ptr User::GetMacros(void) const
{
return m_Macros;
}
bool User::GetEnableNotifications(void) const
{
if (m_EnableNotifications.IsEmpty())
return true;
else
return m_EnableNotifications;
}
void User::SetEnableNotifications(bool enabled)
{
m_EnableNotifications = enabled;
Touch("enable_notifications");
}
TimePeriod::Ptr User::GetNotificationPeriod(void) const
{
return TimePeriod::GetByName(m_NotificationPeriod);
}
unsigned long User::GetNotificationTypeFilter(void) const
{
if (m_NotificationTypeFilter.IsEmpty())
return ~(unsigned long)0; /* All states. */
else
return m_NotificationTypeFilter;
}
unsigned long User::GetNotificationStateFilter(void) const
{
if (m_NotificationStateFilter.IsEmpty())
return ~(unsigned long)0; /* All states. */
else
return m_NotificationStateFilter;
}
void User::SetLastNotification(double ts)
{
m_LastNotification = ts;
Touch("last_notification");
}
double User::GetLastNotification(void) const
{
return m_LastNotification;
}
bool User::ResolveMacro(const String& macro, const Dictionary::Ptr& cr, String *result) const
{
if (macro == "CONTACTNAME") {
*result = GetName();
return true;
} else if (macro == "CONTACTALIAS") {
*result = GetDisplayName();
return true;
} else {
Dictionary::Ptr macros = GetMacros();
if (macros && macros->Contains(macro)) {
*result = macros->Get(macro);
return true;
}
return false;
}
}