1 files changed, 112 insertions, 0 deletions
diff --git a/src/ClientInfo.hpp b/src/ClientInfo.hpp
new file mode 100644
index 0000000..8099b59
--- /dev/null
+++ b/src/ClientInfo.hpp
@@ -0,0 +1,112 @@
+#pragma once
+
+#include <QDateTime>
+#include <QString>
+
+enum ClientInterface {
+    Auto = 0,
+    Clonk = 1,
+    IRC = 1 << 1,
+    Management = 1 << 2
+};
+
+
+class ManagementConnection {
+public:
+    ManagementConnection() {}
+    ManagementConnection(QTcpSocket* socket, const QString& name = "") : socket(socket), name(name) {}
+
+    QTcpSocket* socket = 0;
+    QString name = "";
+};
+
+class ClientInfo
+{
+
+public:
+    ClientInterface interface = Clonk;
+    QString nick = "";
+
+    int CUID = 0;
+    QString pcName = "";
+    bool activated = false;
+
+    QList<QDateTime> antiFloodList;
+
+    QString target = "";
+    ManagementConnection management;
+
+
+    static inline ClientInfo ircClient(QString nick, QString target = "")
+    {
+        ClientInfo ret;
+        ret.interface = IRC;
+        ret.nick = nick;
+        if(target.isEmpty())
+        {
+            target = nick;
+        }
+        ret.target = target;
+        return ret;
+    }
+
+    static inline ClientInfo clonkClient(QString nick, QString pcName, int CUID, bool activated = false)
+    {
+        ClientInfo ret;
+        ret.interface = Clonk;
+        ret.nick = nick;
+        ret.pcName = pcName;
+        ret.CUID = CUID;
+        ret.activated = activated;
+        return ret;
+    }
+
+    static inline ClientInfo managementClient(ManagementConnection conn)
+    {
+        ClientInfo ret;
+        ret.interface = Management;
+        ret.management = conn;
+        ret.nick = conn.name;
+        return ret;
+    }
+
+    static inline ClientInfo autoClient()
+    {
+        ClientInfo ret;
+        ret.interface = Auto;
+        return ret;
+    }
+
+    inline bool operator==(const ClientInfo& other) const
+    {
+        return other.interface == interface && other.nick == nick && (interface == Clonk ? other.pcName == pcName && other.CUID == CUID : true);
+    }
+
+    inline bool operator!=(const ClientInfo& other) const
+    {
+        return !operator==(other);
+    }
+
+    inline bool operator<(const ClientInfo& other) const
+    {
+        return toString() < other.toString();
+    }
+
+    inline QString toString() const
+    {
+        return interface == Auto ? "~auto~" : interface == Management ? management.name + " {CLI}" : (!nick.isEmpty() ? nick + (interface == Clonk ? " (" + pcName + ")" : " [IRC]") : "");
+    }
+
+    inline bool floodCheck(int maxCount, int floodTimeSecs, QDateTime newDateTime = QDateTime::currentDateTime())
+    {
+        foreach(const QDateTime& dateTime, antiFloodList)
+        {
+            if(dateTime.secsTo(newDateTime) > floodTimeSecs)
+            {
+                antiFloodList.removeAll(dateTime);
+            }
+        }
+        antiFloodList.push_back(newDateTime);
+        return antiFloodList.size() > maxCount;
+    }
+};