[client] merged the session initializing stuff in ThriftActions.initSession

1 files changed, 81 insertions, 0 deletions

diff --git a/dozentenmodul/src/main/java/org/openslx/dozmod/thrift/ThriftActions.java b/dozentenmodul/src/main/java/org/openslx/dozmod/thrift/ThriftActions.java
index e31a2852..ddac7c67 100644
--- a/dozentenmodul/src/main/java/org/openslx/dozmod/thrift/ThriftActions.java
+++ b/dozentenmodul/src/main/java/org/openslx/dozmod/thrift/ThriftActions.java
@@ -23,13 +23,20 @@ import org.openslx.bwlp.thrift.iface.LecturePermissions;
 import org.openslx.bwlp.thrift.iface.LectureRead;
 import org.openslx.bwlp.thrift.iface.LectureSummary;
 import org.openslx.bwlp.thrift.iface.LectureWrite;
+import org.openslx.bwlp.thrift.iface.SatelliteServer.Client;
 import org.openslx.bwlp.thrift.iface.TAuthorizationException;
+import org.openslx.bwlp.thrift.iface.TInvocationException;
 import org.openslx.bwlp.thrift.iface.TransferInformation;
 import org.openslx.bwlp.thrift.iface.UserInfo;
+import org.openslx.bwlp.thrift.iface.WhoamiInfo;
+import org.openslx.dozmod.App;
 import org.openslx.dozmod.Config;
+import org.openslx.dozmod.Config.SavedSession;
+import org.openslx.dozmod.authentication.Authenticator.AuthenticationData;
 import org.openslx.dozmod.filetransfer.AsyncHashGenerator;
 import org.openslx.dozmod.filetransfer.DownloadTask;
 import org.openslx.dozmod.filetransfer.UploadTask;
+import org.openslx.dozmod.gui.GraphicalCertHandler;
 import org.openslx.dozmod.gui.Gui;
 import org.openslx.dozmod.gui.Gui.GuiCallable;
 import org.openslx.dozmod.gui.MainWindow;
@@ -46,6 +53,80 @@ public class ThriftActions {
 
 	/* *******************************************************************************	
 	 * 
+	 *                                  LOGIN
+	 *
+	 * Login methods
+	 * 
+	 ********************************************************************************/
+	/**
+	 * @param frame to show user feedback on
+	 * @param data AuthenticationData as received from a successfull login, or null if trying to resume a saved sessions
+	 * @return true if initiliazing the session worked, false otherwise
+	 */
+	public static boolean initSession(final Frame frame, AuthenticationData data) {
+		Client client = null;
+		WhoamiInfo whoami = null;
+		String address = null;
+		String satToken = null;
+		String masterToken = null;
+		if (data == null) {
+			// in session resume
+			SavedSession session = Config.getSavedSession();
+			if (session != null) {
+				address = session.address;
+				satToken = session.token;
+				masterToken = session.masterToken;
+
+			} else {
+				// fail
+				return false;
+			}
+		} else {
+			// after login
+			address = "132.230.8.113";
+			satToken = data.satelliteToken;
+			masterToken = data.masterToken;
+		}
+		// try to get a new client
+		client = ThriftManager.getNewSatelliteClient(GraphicalCertHandler.getSslContext(address),
+				address, App.THRIFT_SSL_PORT, App.THRIFT_TIMEOUT_MS);
+		if (client == null)
+			return false;
+
+		// all good, try to get the whoami info
+		try {
+			whoami = client.whoami(satToken);
+		} catch (TAuthorizationException e) {
+			Gui.showMessageBox(frame,
+					"Authentifizierung erfolgreich, der Satellit verweigert jedoch die Verbindung.\n\n"
+							+ "Grund: " + e.number.toString() + " (" + e.message + ")", MessageType.ERROR,
+					null, null);
+			return false;
+		} catch (TInvocationException e) {
+			Gui.showMessageBox(
+					frame,
+					"Authentifizierung erfolgreich, bei der Kommunikation mit dem Satelliten trat jedoch ein interner Server-Fehler auf.",
+					MessageType.ERROR, LOGGER, e);
+			return false;
+		} catch (Exception e) {
+			Gui.showMessageBox(
+					frame,
+					"Authentifizierung erfolgreich, aber der Satellit akzeptiert das Sitzungstoken nicht.",
+					MessageType.ERROR, LOGGER, e);
+			return false;
+		}
+		
+		if (whoami != null) {
+			Session.initialize(whoami, address, satToken, masterToken);
+			ThriftManager.setSatelliteAddress(
+					GraphicalCertHandler.getSslContext(Session.getSatelliteAddress()),
+					Session.getSatelliteAddress(), App.THRIFT_SSL_PORT, App.THRIFT_TIMEOUT_MS);
+			return true;
+		}
+		return false;
+	}
+	/* *******************************************************************************	
+	 * 
 	 *                              IMAGE CREATION
 	 *
 	 * Creates a base image with the given name