Store last-login in server's xml configuration

BearWare · Oct 17, 2024 · 6ff42b5 · 6ff42b5
1 parent 9f6de6a
commit 6ff42b5
Show file tree

Hide file tree

Showing 8 changed files with 52 additions and 3 deletions.
diff --git a/Library/TeamTalkJNI/test/dk/bearware/TeamTalkTestCase.java b/Library/TeamTalkJNI/test/dk/bearware/TeamTalkTestCase.java
@@ -40,6 +40,7 @@
 
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNotEquals;
 import static org.junit.Assert.assertTrue;
 
 @FixMethodOrder(MethodSorters.NAME_ASCENDING)
@@ -4494,6 +4495,30 @@ public void testUserStateMediaFile() {
         assertEquals("User is not streaming", UserState.USERSTATE_NONE, (msg.user.uUserState & UserState.USERSTATE_MEDIAFILE_AUDIO));
     }
 
+    @Test
+    public void testUserAcountLastLogin() {
+        final String USERNAME = "tt_test", PASSWORD = "tt_test", NICKNAME = "jUnit - " + getTestMethodName();
+        int USERRIGHTS = UserRight.USERRIGHT_TRANSMIT_MEDIAFILE | UserRight.USERRIGHT_CREATE_TEMPORARY_CHANNEL;
+        makeUserAccount(NICKNAME, USERNAME, PASSWORD, USERRIGHTS);
+
+        TeamTalkBase client = newClientInstance();
+        connect(client);
+        login(client, NICKNAME, USERNAME, PASSWORD);
+
+        UserAccount first_login_account = new UserAccount();
+        assertTrue("get account", client.getMyUserAccount(first_login_account));
+        //assertEquals("1970/01/01 00:00", first_login_account.szLastLoginTime);
+
+        assertTrue("disconnect", client.disconnect());
+        connect(client);
+        login(client, NICKNAME, USERNAME, PASSWORD);
+
+        UserAccount second_login_account = new UserAccount();
+        assertTrue("get account again", client.getMyUserAccount(second_login_account));
+        //assertNotEquals("1970/01/01 00:00", second_login_account.szLastLoginTime);
+        assertNotEquals(first_login_account.szLastLoginTime, second_login_account.szLastLoginTime);
+    }
+
     /* cannot test output levels since a user is muted by sound system after decoding and callback.
 
     @Test

diff --git a/Library/TeamTalkLib/bin/ttsrv/ServerXML.cpp b/Library/TeamTalkLib/bin/ttsrv/ServerXML.cpp
@@ -1318,6 +1318,7 @@ namespace teamtalk{
         PutInteger(userElement, "userdata", user.userdata);
         PutString(userElement, "init-channel", UnicodeToUtf8(user.init_channel).c_str());
         PutString(userElement, "modified-time", DateToString(user.lastupdated.sec()).c_str());
+        PutString(userElement, "last-login-time", DateToString(user.lastlogin.sec()).c_str());
         TiXmlElement opchanElement("channel-operator");
         for(intset_t::const_iterator i=user.auto_op_channels.begin();
             i!=user.auto_op_channels.end();i++)
@@ -1386,6 +1387,9 @@ namespace teamtalk{
         GetInteger(userElement, "audiocodec-bps-limit", bpslimit);
         GetString(userElement, "modified-time", tmp);
         user.lastupdated = StringToDate(tmp);
+        tmp.clear();
+        GetString(userElement, "last-login-time", tmp);
+        user.lastlogin = StringToDate(tmp);
 
         if(b)
         {
@@ -1440,6 +1444,8 @@ namespace teamtalk{
             user = int_user;
             if(user.usertype == USERTYPE_ADMIN)
                 user.userrights = USERRIGHT_ALL;
+
+            UpdateLastLogin(user);
             return true;
         }
         return false;
@@ -1479,6 +1485,14 @@ namespace teamtalk{
         return false;
     }
 
+    void ServerXML::UpdateLastLogin(const UserAccount& user)
+    {
+        RemoveUser(UnicodeToUtf8(user.username).c_str());
+        UserAccount updateduser = user;
+        updateduser.lastlogin = ACE_OS::gettimeofday();
+        AddNewUser(updateduser);
+    }
+
     /******* </users> ******/
 
     /********** files in static channels **************/

diff --git a/Library/TeamTalkLib/bin/ttsrv/ServerXML.h b/Library/TeamTalkLib/bin/ttsrv/ServerXML.h
@@ -159,6 +159,7 @@ namespace teamtalk {
         bool GetNextUser(int index, UserAccount& user);
         bool AuthenticateUser(UserAccount& user);
         bool GetUser(const std::string& username, UserAccount& user);
+        void UpdateLastLogin(const UserAccount& user);
         /****** </users> *****/
 
         /********** <serverbans>  ************/

diff --git a/Library/TeamTalkLib/teamtalk/Commands.cpp b/Library/TeamTalkLib/teamtalk/Commands.cpp
@@ -388,6 +388,7 @@ namespace teamtalk {
         GetProperty(properties, TT_AUTOOPCHANNELS, useraccount.auto_op_channels);
         GetProperty(properties, TT_AUDIOBPSLIMIT, useraccount.audiobpslimit);
         GetProperty(properties, TT_MODIFIEDTIME, useraccount.lastupdated);
+        GetProperty(properties, TT_LASTLOGINTIME, useraccount.lastlogin);
 
         vector<int> flood;
         if(GetProperty(properties, TT_CMDFLOOD, flood))

diff --git a/Library/TeamTalkLib/teamtalk/Commands.h b/Library/TeamTalkLib/teamtalk/Commands.h
@@ -33,7 +33,7 @@
 #include <set>
 #include <map>
 
-#define TEAMTALK_PROTOCOL_VERSION ACE_TEXT("5.13")
+#define TEAMTALK_PROTOCOL_VERSION ACE_TEXT("5.14")
 
 /* parameter names */
 #define TT_USERID ACE_TEXT("userid")
@@ -142,6 +142,7 @@
 #define TT_TOTMEDIAFILE ACE_TEXT("mediafiletot") // v5.12
 #define TT_TRANSFERKEY ACE_TEXT("filetxkey") // v5.13
 #define TT_BANOWNER ACE_TEXT("owner") // v5.13
+#define TT_LASTLOGINTIME ACE_TEXT("lastlogin") // v5.14
 
 //    Client ---> Server
 //    -------------------------

diff --git a/Library/TeamTalkLib/teamtalk/Common.cpp b/Library/TeamTalkLib/teamtalk/Common.cpp
@@ -46,7 +46,6 @@ namespace teamtalk
     UserAccount::UserAccount()
     {
         lastupdated = ACE_OS::gettimeofday();
-        lastlogin = ACE_OS::gettimeofday();
     }
 
     bool UserAccount::IsWebLogin() const

diff --git a/Library/TeamTalkLib/teamtalk/server/ServerNode.cpp b/Library/TeamTalkLib/teamtalk/server/ServerNode.cpp
@@ -33,7 +33,6 @@
 #include <ace/OS_NS_sys_socket.h>
 
 #include <stack>
-#include <queue>
 #include <memory>
 #include <vector>
 #include <algorithm>
@@ -2741,6 +2740,14 @@ ErrorMsg ServerNode::UserLogin(int userid, const ACE_TString& username,
         m_srvguard->OnUserLogin(*user);
     }
 
+    // update last login
+    if (IsAutoSaving())
+    {
+        auto err = m_srvguard->SaveConfiguration(*user, *this);
+        if (err.success() && (m_properties.logevents & SERVERLOGEVENT_SERVER_SAVECONFIG))
+            m_srvguard->OnSaveConfiguration(user.get());
+    }
+
     return ErrorMsg(TT_CMDERR_SUCCESS);
 }
 

diff --git a/Library/TeamTalkLib/teamtalk/server/ServerUser.cpp b/Library/TeamTalkLib/teamtalk/server/ServerUser.cpp
@@ -1310,6 +1310,7 @@ void AppendUserAccount(const UserAccount& useraccount, ACE_TString& command)
     AppendProperty(TT_AUDIOBPSLIMIT, useraccount.audiobpslimit, command);
     AppendProperty(TT_CMDFLOOD, useraccount.abuse.toParam(), command);
     AppendProperty(TT_MODIFIEDTIME, useraccount.lastupdated, command);
+    AppendProperty(TT_LASTLOGINTIME, useraccount.lastlogin, command);
 }
 
 void ServerUser::DoAccepted(const UserAccount& useraccount)