From: Richard Millet <remillet@gmail.com>
Date: Thu, 29 Aug 2019 23:53:10 +0000 (-0700)
Subject: DRYD-736: Added post-auth code to update lastlogin timestamp to users table.
X-Git-Url: https://git.aero2k.de/?a=commitdiff_plain;h=0ad9621208b8810071173b3b2ce6c86761acd931;p=tmp%2Fjakarta-migration.git

DRYD-736: Added post-auth code to update lastlogin timestamp to users table.
---

diff --git a/services/authentication/service/src/main/java/org/collectionspace/authentication/CSpaceAuthenticationSuccessEvent.java b/services/authentication/service/src/main/java/org/collectionspace/authentication/CSpaceAuthenticationSuccessEvent.java
index a325cd053..f83cfdb31 100644
--- a/services/authentication/service/src/main/java/org/collectionspace/authentication/CSpaceAuthenticationSuccessEvent.java
+++ b/services/authentication/service/src/main/java/org/collectionspace/authentication/CSpaceAuthenticationSuccessEvent.java
@@ -1,14 +1,97 @@
 package org.collectionspace.authentication;
 
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+import javax.security.auth.login.AccountException;
+import javax.security.auth.login.AccountNotFoundException;
+
+import org.collectionspace.authentication.realm.db.CSpaceDbRealm;
+import org.postgresql.util.PSQLState;
 import org.springframework.context.ApplicationListener;
 import org.springframework.security.authentication.event.AuthenticationSuccessEvent;
+import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
 
 public class CSpaceAuthenticationSuccessEvent implements ApplicationListener<AuthenticationSuccessEvent> {
+	
+	final private static String UPDATE_USER_SQL =
+			"UPDATE users SET lastlogin = now() WHERE username = ?";
 
 	@Override
 	public void onApplicationEvent(AuthenticationSuccessEvent event) {
 		// TODO Auto-generated method stub
 		System.out.println(); //org.springframework.security.authentication.UsernamePasswordAuthenticationToken@8a633e91: Principal: org.collectionspace.authentication.CSpaceUser@b122ec20: Username: admin@core.collectionspace.org; Password: [PROTECTED]; Enabled: true; AccountNonExpired: true; credentialsNonExpired: true; AccountNonLocked: true; Granted Authorities: ROLE_1_TENANT_ADMINISTRATOR,ROLE_SPRING_ADMIN; Credentials: [PROTECTED]; Authenticated: true; Details: {grant_type=password, username=admin@core.collectionspace.org}; Granted Authorities: ROLE_1_TENANT_ADMINISTRATOR, ROLE_SPRING_ADMIN
+		String userName = null;
+		CSpaceDbRealm cspaceDbRealm = new CSpaceDbRealm();
+		
+		if (event.getSource() instanceof UsernamePasswordAuthenticationToken) {
+			UsernamePasswordAuthenticationToken eventSource = (UsernamePasswordAuthenticationToken)event.getSource();
+			if (eventSource.getPrincipal() instanceof CSpaceUser) {
+				CSpaceUser cspaceUser = (CSpaceUser) eventSource.getPrincipal();
+				userName = cspaceUser.getUsername();
+				try {
+					setLastLogin(cspaceDbRealm, userName);
+				} catch (Exception e) {
+					// TODO Auto-generated catch block
+					e.printStackTrace();
+				}
+				System.out.println();
+			}
+		}
 	}
+	
+	private void setLastLogin(CSpaceDbRealm cspaceDbRealm, String username) throws AccountException {
+        Connection conn = null;
+        PreparedStatement ps = null;
+        ResultSet rs = null;
+        try {
+            conn = cspaceDbRealm.getConnection();
+            // Get the salt
+
+            ps = conn.prepareStatement(UPDATE_USER_SQL);
+            ps.setString(1, username);
+            rs = ps.executeQuery();
+            if (rs.next() == false) {
+                throw new AccountNotFoundException("No matching username found");
+            }
+        } catch (SQLException ex) {
+        	// Assuming PostgreSQL
+            if (PSQLState.UNDEFINED_COLUMN.getState().equals(ex.getSQLState())) {
+            	System.err.println("'USERS' table is missing 'lastlogin' column.");
+            } else {
+                AccountException ae = new AccountException("Authentication query failed: " + ex.getLocalizedMessage());
+                ae.initCause(ex);
+                throw ae;
+            }
+        } catch (AccountNotFoundException ex) {
+            throw ex;
+        } catch (Exception ex) {
+            AccountException ae = new AccountException("Unknown Exception");
+            ae.initCause(ex);
+            throw ae;
+        } finally {
+            if (rs != null) {
+                try {
+                    rs.close();
+                } catch (SQLException e) {
+                }
+            }
+            if (ps != null) {
+                try {
+                    ps.close();
+                } catch (SQLException e) {
+                }
+            }
+            if (conn != null) {
+                try {
+                    conn.close();
+                } catch (SQLException ex) {
+                }
+            }
+        }
+    }
+
 
 }
diff --git a/services/authentication/service/src/main/java/org/collectionspace/authentication/realm/db/CSpaceDbRealm.java b/services/authentication/service/src/main/java/org/collectionspace/authentication/realm/db/CSpaceDbRealm.java
index 620457111..814ec72fa 100644
--- a/services/authentication/service/src/main/java/org/collectionspace/authentication/realm/db/CSpaceDbRealm.java
+++ b/services/authentication/service/src/main/java/org/collectionspace/authentication/realm/db/CSpaceDbRealm.java
@@ -77,6 +77,8 @@ import org.slf4j.LoggerFactory;
  * @author 
  */
 public class CSpaceDbRealm implements CSpaceRealm {
+	public static String DEFAULT_DATASOURCE_NAME = "CspaceDS";
+	
     private Logger logger = LoggerFactory.getLogger(CSpaceDbRealm.class);
     
     private String datasourceName;
@@ -128,6 +130,10 @@ public class CSpaceDbRealm implements CSpaceRealm {
 	protected long getDelayBetweenAttemptsMillis() {
 		return this.delayBetweenAttemptsMillis;
 	}
+	
+	public CSpaceDbRealm() {
+        datasourceName = DEFAULT_DATASOURCE_NAME;
+	}
     
     /**
      * CSpace Database Realm
@@ -136,7 +142,7 @@ public class CSpaceDbRealm implements CSpaceRealm {
     public CSpaceDbRealm(Map<String, ?> options) {
         datasourceName = (String) options.get("dsJndiName");
         if (datasourceName == null) {
-            datasourceName = "java:/DefaultDS";
+            datasourceName = DEFAULT_DATASOURCE_NAME;
         }
         Object tmp = options.get("principalsQuery");
         if (tmp != null) {
@@ -172,7 +178,6 @@ public class CSpaceDbRealm implements CSpaceRealm {
             logger.trace("rolesQuery=" + rolesQuery);
             logger.trace("suspendResume=" + suspendResume);
         }
-
     }
 
     @Override
@@ -445,7 +450,7 @@ public class CSpaceDbRealm implements CSpaceRealm {
      * it will retry for the next 'getMaxRetrySeconds()' seconds.  If it is unable to get the connection then it will timeout and
      * throw an exception.
      */
-    private Connection getConnection() throws Exception {
+    public Connection getConnection() throws Exception {
         Connection result = null;
 		boolean failed = true;
 		Exception lastException = null;