|
|
@@ -1,3 +1,4 @@
|
|
|
+#include <algorithm>
|
|
|
#include <sstream>
|
|
|
#include "Util/JSON.h"
|
|
|
#include "Util/Logger.h"
|
|
|
@@ -7,17 +8,21 @@
|
|
|
namespace PresenceDetection {
|
|
|
namespace UniFi {
|
|
|
|
|
|
-Device::Device(const std::string& hostname, const std::string& username, const std::string& password, const std::string& cookieFile, const std::string& inventoryURL, const std::string& target) :
|
|
|
+Device::Device(const std::string& hostname, int port, const std::string& username, const std::string& password, const std::string& cookieFile, int timeout, const std::string& inventoryURL, const std::string& target, const std::vector<Util::Device>& devices) :
|
|
|
m_loggedIn(false),
|
|
|
m_hostname(hostname),
|
|
|
+ m_port(port),
|
|
|
m_username(username),
|
|
|
m_password(password),
|
|
|
m_cookieFile(cookieFile),
|
|
|
+ m_timeout(timeout),
|
|
|
m_inventoryURL(inventoryURL),
|
|
|
m_target(target),
|
|
|
- m_offlineTimeout(120)
|
|
|
+ m_staticDevices(devices)
|
|
|
{
|
|
|
- UpdateDevicesFromInventory();
|
|
|
+ if (!m_inventoryURL.empty())
|
|
|
+ UpdateDevicesFromInventory();
|
|
|
+ ClearDevices();
|
|
|
Start();
|
|
|
}
|
|
|
|
|
|
@@ -29,23 +34,28 @@ Device::~Device()
|
|
|
void Device::Start()
|
|
|
{
|
|
|
m_deviceTimer.StartContinuous(5000, static_cast<std::function<void()>>(std::bind(&Device::UpdatePresentDevices, this)));
|
|
|
- m_inventoryTimer.StartContinuous(300000, static_cast<std::function<void()>>(std::bind(&Device::UpdateDevicesFromInventory, this)));
|
|
|
+ if (!m_inventoryURL.empty())
|
|
|
+ m_inventoryTimer.StartContinuous(300000, static_cast<std::function<void()>>(std::bind(&Device::UpdateDevicesFromInventory, this)));
|
|
|
}
|
|
|
|
|
|
void Device::Stop()
|
|
|
{
|
|
|
m_deviceTimer.Stop();
|
|
|
+ if (!m_inventoryURL.empty())
|
|
|
+ m_inventoryTimer.Stop();
|
|
|
}
|
|
|
|
|
|
void Device::Wait()
|
|
|
{
|
|
|
m_deviceTimer.Wait();
|
|
|
+ if (!m_inventoryURL.empty())
|
|
|
+ m_inventoryTimer.Stop();
|
|
|
}
|
|
|
|
|
|
bool Device::Login()
|
|
|
{
|
|
|
std::stringstream url;
|
|
|
- url << "https://" << m_hostname << "/api/login";
|
|
|
+ url << "https://" << m_hostname << ":" << m_port << "/api/login";
|
|
|
|
|
|
Util::JSON json;
|
|
|
json["password"] = m_password;
|
|
|
@@ -83,7 +93,7 @@ bool Device::Login()
|
|
|
void Device::Logout()
|
|
|
{
|
|
|
std::stringstream url;
|
|
|
- url << "https://" << m_hostname << "/logout";
|
|
|
+ url << "https://" << m_hostname << ":" << m_port << "/logout";
|
|
|
|
|
|
std::lock_guard<std::mutex> lock(m_mutex);
|
|
|
m_loggedIn = false;
|
|
|
@@ -100,6 +110,40 @@ void Device::Logout()
|
|
|
}
|
|
|
}
|
|
|
|
|
|
+void Device::ClearDevices()
|
|
|
+{
|
|
|
+ for (std::vector<std::string>::iterator it = m_devices.begin(); it != m_devices.end(); ++it)
|
|
|
+ {
|
|
|
+ try
|
|
|
+ {
|
|
|
+ SendStateChange(false, *it);
|
|
|
+ }
|
|
|
+ catch (const std::exception& e)
|
|
|
+ {
|
|
|
+ std::stringstream ss;
|
|
|
+ ss << "UniFi::Device::ClearDevices() - Error: " << e.what() << std::endl;
|
|
|
+ Util::Logger::Log(Util::Logger::Severity::Error, ss.str());
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ for (std::vector<Util::Device>::iterator it = m_staticDevices.begin(); it != m_staticDevices.end(); ++it)
|
|
|
+ {
|
|
|
+ if (it->HasWifiMac())
|
|
|
+ {
|
|
|
+ try
|
|
|
+ {
|
|
|
+ SendStateChange(false, it->WifiMac());
|
|
|
+ }
|
|
|
+ catch (const std::exception& e)
|
|
|
+ {
|
|
|
+ std::stringstream ss;
|
|
|
+ ss << "UniFi::Device::ClearDevices() - Error: " << e.what() << std::endl;
|
|
|
+ Util::Logger::Log(Util::Logger::Severity::Error, ss.str());
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
void Device::UpdateDevicesFromInventory()
|
|
|
{
|
|
|
try
|
|
|
@@ -137,7 +181,7 @@ void Device::UpdatePresentDevices()
|
|
|
return;
|
|
|
|
|
|
std::stringstream url;
|
|
|
- url << "https://" << m_hostname << "/api/s/default/stat/sta";
|
|
|
+ url << "https://" << m_hostname << ":" << m_port << "/api/s/default/stat/sta";
|
|
|
|
|
|
std::time_t timeStamp = std::time(nullptr);
|
|
|
std::vector<std::string> presentDevices;
|
|
|
@@ -161,17 +205,31 @@ void Device::UpdatePresentDevices()
|
|
|
for (auto& device : json["data"])
|
|
|
{
|
|
|
std::string macAddress = device["mac"];
|
|
|
+ std::transform(macAddress.begin(), macAddress.end(), macAddress.begin(), ::tolower);
|
|
|
int lastSeen = device["last_seen"];
|
|
|
|
|
|
- if ((timeStamp - lastSeen) < m_offlineTimeout)
|
|
|
+ if (std::find(m_devices.begin(), m_devices.end(), macAddress) != m_devices.end())
|
|
|
{
|
|
|
- if (std::find(m_devices.begin(), m_devices.end(), macAddress) != m_devices.end())
|
|
|
+ if ((timeStamp - lastSeen) < m_timeout)
|
|
|
{
|
|
|
if (std::find(m_presentDevices.begin(), m_presentDevices.end(), macAddress) == m_presentDevices.end())
|
|
|
addedDevices.push_back(macAddress);
|
|
|
presentDevices.push_back(macAddress);
|
|
|
}
|
|
|
}
|
|
|
+
|
|
|
+ for (std::vector<Util::Device>::iterator it = m_staticDevices.begin(); it != m_staticDevices.end(); ++it)
|
|
|
+ {
|
|
|
+ if ((timeStamp - lastSeen) < m_timeout)
|
|
|
+ {
|
|
|
+ if (it->HasWifiMac() && it->WifiMac() == macAddress)
|
|
|
+ {
|
|
|
+ if (std::find(m_presentDevices.begin(), m_presentDevices.end(), macAddress) == m_presentDevices.end())
|
|
|
+ addedDevices.push_back(macAddress);
|
|
|
+ presentDevices.push_back(macAddress);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
catch (const std::exception& e)
|
|
|
@@ -208,13 +266,60 @@ void Device::SendStateChange(bool present, const std::string& macAddress)
|
|
|
ss << "UniFi: " << sign << " " << macAddress;
|
|
|
Util::Logger::Log(Util::Logger::Severity::Info, ss.str());
|
|
|
|
|
|
-
|
|
|
- try
|
|
|
+ if (!m_target.empty())
|
|
|
{
|
|
|
- m_httpClient.GetUrlSilent(url.str());
|
|
|
+ std::stringstream url;
|
|
|
+ url << m_target << "/UniFi/" << sign << "/" << macAddress;
|
|
|
+
|
|
|
+ try
|
|
|
+ {
|
|
|
+ m_httpClient.GetUrlSilent(url.str());
|
|
|
+ }
|
|
|
+ catch (const std::exception& e)
|
|
|
+ {
|
|
|
+ std::stringstream ss;
|
|
|
+ ss << "UniFi::Device::SendStateChange() - Error: " << e.what() << std::endl;
|
|
|
+ Util::Logger::Log(Util::Logger::Severity::Error, ss.str());
|
|
|
+ }
|
|
|
}
|
|
|
- catch (const std::exception& e)
|
|
|
+
|
|
|
+ for (std::vector<Util::Device>::iterator it = m_staticDevices.begin(); it != m_staticDevices.end(); ++it)
|
|
|
{
|
|
|
+ if (it->HasWifiMac() && it->WifiMac() == macAddress)
|
|
|
+ {
|
|
|
+ it->SetWifiState(present);
|
|
|
+
|
|
|
+ std::vector<std::string> urls;
|
|
|
+
|
|
|
+ if (present)
|
|
|
+ {
|
|
|
+ if (!it->GetBluetoothState() && it->HasOnlineURL())
|
|
|
+ urls.push_back(it->OnlineURL());
|
|
|
+ if (it->HasWifiOnlineURL())
|
|
|
+ urls.push_back(it->WifiOnlineURL());
|
|
|
+ }
|
|
|
+ else if (!present)
|
|
|
+ {
|
|
|
+ if (!it->GetBluetoothState() && it->HasOfflineURL())
|
|
|
+ urls.push_back(it->OfflineURL());
|
|
|
+ if (it->HasWifiOfflineURL())
|
|
|
+ urls.push_back(it->WifiOfflineURL());
|
|
|
+ }
|
|
|
+
|
|
|
+ for (auto& url : urls)
|
|
|
+ {
|
|
|
+ try
|
|
|
+ {
|
|
|
+ m_httpClient.GetUrlSilent(url);
|
|
|
+ }
|
|
|
+ catch (const std::exception& e)
|
|
|
+ {
|
|
|
+ std::stringstream ss;
|
|
|
+ ss << "UniFi::Device::SendStateChange() - Error: " << e.what() << std::endl;
|
|
|
+ Util::Logger::Log(Util::Logger::Severity::Error, ss.str());
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
|