From 33c88ec9e4560bdc78023fdb0963fb48ab907e0f Mon Sep 17 00:00:00 2001
From: Moxie Marlinspike <moxie@thoughtcrime.org>
Date: Fri, 6 Jan 2017 12:23:56 -0800
Subject: [PATCH] Move stats

// FREEBIE
---
 .../textsecuregcm/auth/AccountAuthenticator.java       | 10 ++++++++++
 .../websocket/AuthenticatedConnectListener.java        |  8 --------
 2 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/src/main/java/org/whispersystems/textsecuregcm/auth/AccountAuthenticator.java b/src/main/java/org/whispersystems/textsecuregcm/auth/AccountAuthenticator.java
index 73a711165..20b41ee67 100644
--- a/src/main/java/org/whispersystems/textsecuregcm/auth/AccountAuthenticator.java
+++ b/src/main/java/org/whispersystems/textsecuregcm/auth/AccountAuthenticator.java
@@ -27,6 +27,7 @@ import org.whispersystems.textsecuregcm.storage.Account;
 import org.whispersystems.textsecuregcm.storage.AccountsManager;
 import org.whispersystems.textsecuregcm.storage.Device;
 import org.whispersystems.textsecuregcm.util.Constants;
+import org.whispersystems.textsecuregcm.util.Util;
 
 import static com.codahale.metrics.MetricRegistry.name;
 import io.dropwizard.auth.AuthenticationException;
@@ -67,6 +68,7 @@ public class AccountAuthenticator implements Authenticator<BasicCredentials, Acc
       if (device.get().getAuthenticationCredentials().verify(basicCredentials.getPassword())) {
         authenticationSucceededMeter.mark();
         account.get().setAuthenticatedDevice(device.get());
+        updateLastSeen(account.get(), device.get());
         return account;
       }
 
@@ -76,4 +78,12 @@ public class AccountAuthenticator implements Authenticator<BasicCredentials, Acc
       return Optional.absent();
     }
   }
+
+  private void updateLastSeen(Account account, Device device) {
+    if (device.getLastSeen() != Util.todayInMillis()) {
+      device.setLastSeen(Util.todayInMillis());
+      accountsManager.update(account);
+    }
+  }
+
 }
diff --git a/src/main/java/org/whispersystems/textsecuregcm/websocket/AuthenticatedConnectListener.java b/src/main/java/org/whispersystems/textsecuregcm/websocket/AuthenticatedConnectListener.java
index c4a2de18f..64149932e 100644
--- a/src/main/java/org/whispersystems/textsecuregcm/websocket/AuthenticatedConnectListener.java
+++ b/src/main/java/org/whispersystems/textsecuregcm/websocket/AuthenticatedConnectListener.java
@@ -55,7 +55,6 @@ public class AuthenticatedConnectListener implements WebSocketConnectListener {
                                                                        context.getClient());
 
     pubSubManager.publish(info, PubSubMessage.newBuilder().setType(PubSubMessage.Type.CONNECTED).build());
-    updateLastSeen(account, device);
     pubSubManager.subscribe(address, connection);
 
     context.addListener(new WebSocketSessionContext.WebSocketEventListener() {
@@ -66,12 +65,5 @@ public class AuthenticatedConnectListener implements WebSocketConnectListener {
       }
     });
   }
-
-  private void updateLastSeen(Account account, Device device) {
-    if (device.getLastSeen() != Util.todayInMillis()) {
-      device.setLastSeen(Util.todayInMillis());
-      accountsManager.update(account);
-    }
-  }
 }