121 lines
2.5 KiB
C++
121 lines
2.5 KiB
C++
#ifndef _USERMAN_H
|
|
#define _USERMAN_H
|
|
|
|
#include <string>
|
|
#include <map>
|
|
#include <vector>
|
|
#include <set>
|
|
#include <sstream>
|
|
#include <mutex>
|
|
#include "definitions.h"
|
|
#include "configfilemanager.h"
|
|
|
|
class userManager
|
|
{
|
|
private:
|
|
std::map<std::string, bool> users;
|
|
std::mutex mtx;
|
|
|
|
public:
|
|
|
|
static std::string getScanCommandFromUser(const std::string&, configfilemanager& cfm);
|
|
|
|
void addUserFromLogLine(std::string &line)
|
|
{
|
|
addUser(splitString(line, '|').at(USER_LOG_LOCATION));
|
|
}
|
|
|
|
void addUser(std::string &user)
|
|
{
|
|
std::lock_guard<std::mutex> lock(mtx);
|
|
if (users.count(user) == 0)
|
|
{
|
|
users[user] = false;
|
|
}
|
|
}
|
|
|
|
void removeUser(std::string &user)
|
|
{
|
|
std::lock_guard<std::mutex> lock(mtx);
|
|
users.erase(user);
|
|
}
|
|
|
|
bool isContains(std::string &user)
|
|
{
|
|
std::lock_guard<std::mutex> lock(mtx);
|
|
return users.count(user) == 1;
|
|
}
|
|
|
|
void setUserFlagged(std::string &user)
|
|
{
|
|
std::lock_guard<std::mutex> lock(mtx);
|
|
if (users.count(user) == 1)
|
|
{
|
|
users[user] = true;
|
|
}
|
|
}
|
|
|
|
void setUserUnflagged(std::string &user)
|
|
{
|
|
std::lock_guard<std::mutex> lock(mtx);
|
|
if (users.count(user) == 1)
|
|
{
|
|
users[user] = false;
|
|
}
|
|
}
|
|
|
|
void unflagAllUsers()
|
|
{
|
|
std::lock_guard<std::mutex> lock(mtx);
|
|
for (std::map<std::string, bool>::iterator it = users.begin(); it != users.end(); ++it)
|
|
{
|
|
it->second = false;
|
|
}
|
|
}
|
|
|
|
std::set<std::string> getUsers()
|
|
{
|
|
std::set<std::string> ret;
|
|
|
|
std::lock_guard<std::mutex> lock(mtx);
|
|
|
|
for (std::map<std::string, bool>::iterator it = users.begin(); it != users.end(); ++it)
|
|
{
|
|
ret.insert(it->first);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
std::set<std::string> getFlaggedUsers()
|
|
{
|
|
std::set<std::string> ret;
|
|
|
|
std::lock_guard<std::mutex> lock(mtx);
|
|
|
|
for (std::map<std::string, bool>::iterator it = users.begin(); it != users.end(); ++it)
|
|
{
|
|
if (it->second)
|
|
{
|
|
ret.insert(it->first);
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool isAnybodyFlagged()
|
|
{
|
|
std::lock_guard<std::mutex> lock(mtx);
|
|
|
|
for (std::map<std::string, bool>::iterator it = users.begin(); it != users.end(); ++it)
|
|
{
|
|
if (it->second)
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
};
|
|
|
|
#endif // _USERMAN_H
|