From bd0da338a7430e992a05e8af15c4bdd616ffb979 Mon Sep 17 00:00:00 2001
From: kske <kai@kske.dev>
Date: Sat, 28 Mar 2020 15:32:24 +0100
Subject: [PATCH] Added user and message loading and message posting

---
 src/main/java/envoy/client/data/Chat.java     |  29 ++-
 src/main/java/envoy/client/data/LocalDB.java  |   2 +-
 src/main/java/envoy/client/ui/ChatScene.fxml  |  22 ++-
 .../envoy/client/ui/ChatSceneController.java  | 126 ++++++++++++-
 .../java/envoy/client/ui/MessageListCell.java |   3 +-
 src/main/java/envoy/client/ui/Startup.java    | 173 +++++++++---------
 .../java/envoy/client/ui/UserListCell.java    |   6 +-
 .../envoy/client/ui/container/ChatWindow.java |  30 +--
 8 files changed, 252 insertions(+), 139 deletions(-)

diff --git a/src/main/java/envoy/client/data/Chat.java b/src/main/java/envoy/client/data/Chat.java
index 162b256..6f0715c 100644
--- a/src/main/java/envoy/client/data/Chat.java
+++ b/src/main/java/envoy/client/data/Chat.java
@@ -2,9 +2,10 @@ package envoy.client.data;
 
 import java.io.IOException;
 import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.List;
 
 import envoy.client.net.WriteProxy;
-import envoy.client.ui.list.Model;
 import envoy.data.Message;
 import envoy.data.Message.MessageStatus;
 import envoy.data.User;
@@ -23,12 +24,12 @@ import envoy.event.MessageStatusChangeEvent;
  * @author Kai S. K. Engelbart
  * @since Envoy Client v0.1-alpha
  */
-public class Chat implements Serializable {
+public final class Chat implements Serializable {
 
-	private static final long serialVersionUID = 0L;
+	private final User			recipient;
+	private final List<Message>	messages	= new ArrayList<>();
 
-	private final User				recipient;
-	private final Model<Message>	model	= new Model<>();
+	private static final long serialVersionUID = 1L;
 
 	/**
 	 * Provides the list of messages that the recipient receives.<br>
@@ -39,14 +40,6 @@ public class Chat implements Serializable {
 	 */
 	public Chat(User recipient) { this.recipient = recipient; }
 
-	/**
-	 * Appends a message to the bottom of this chat
-	 *
-	 * @param message the message to append
-	 * @since Envoy Client v0.1-alpha
-	 */
-	public void appendMessage(Message message) { model.add(message); }
-
 	/**
 	 * Sets the status of all chat messages received from the recipient to
 	 * {@code READ} starting from the bottom and stopping once a read message is
@@ -59,8 +52,8 @@ public class Chat implements Serializable {
 	 * @since Envoy Client v0.3-alpha
 	 */
 	public void read(WriteProxy writeProxy) throws IOException {
-		for (int i = model.size() - 1; i >= 0; --i) {
-			final Message m = model.get(i);
+		for (int i = messages.size() - 1; i >= 0; --i) {
+			final Message m = messages.get(i);
 			if (m.getSenderID() == recipient.getID()) if (m.getStatus() == MessageStatus.READ) break;
 			else {
 				m.setStatus(MessageStatus.READ);
@@ -74,13 +67,13 @@ public class Chat implements Serializable {
 	 *         the status {@code READ}
 	 * @since Envoy Client v0.3-alpha
 	 */
-	public boolean isUnread() { return !model.isEmpty() && model.get(model.size() - 1).getStatus() != MessageStatus.READ; }
+	public boolean isUnread() { return !messages.isEmpty() && messages.get(messages.size() - 1).getStatus() != MessageStatus.READ; }
 
 	/**
 	 * @return all messages in the current chat
-	 * @since Envoy Client v0.1-alpha
+	 * @since Envoy Client v0.1-beta
 	 */
-	public Model<Message> getModel() { return model; }
+	public List<Message> getMessages() { return messages; }
 
 	/**
 	 * @return the recipient of a message
diff --git a/src/main/java/envoy/client/data/LocalDB.java b/src/main/java/envoy/client/data/LocalDB.java
index 0980289..1e36307 100644
--- a/src/main/java/envoy/client/data/LocalDB.java
+++ b/src/main/java/envoy/client/data/LocalDB.java
@@ -154,7 +154,7 @@ public abstract class LocalDB {
 	 */
 	public Message getMessage(long id) {
 		for (Chat c : chats)
-			for (Message m : c.getModel())
+			for (Message m : c.getMessages())
 				if (m.getID() == id) return m;
 		return null;
 	}
diff --git a/src/main/java/envoy/client/ui/ChatScene.fxml b/src/main/java/envoy/client/ui/ChatScene.fxml
index cc8b797..43af60d 100644
--- a/src/main/java/envoy/client/ui/ChatScene.fxml
+++ b/src/main/java/envoy/client/ui/ChatScene.fxml
@@ -29,16 +29,20 @@
 			vgrow="SOMETIMES" />
 	</rowConstraints>
 	<children>
-		<ListView fx:id="userList" prefHeight="211.0" prefWidth="300.0"
-			GridPane.rowIndex="1" />
-		<Label text="Label" />
-		<Button id="settingsButton" mnemonicParsing="false"
-			text="Settings" GridPane.columnIndex="1" />
+		<ListView fx:id="userList" prefHeight="211.0"
+			prefWidth="300.0" GridPane.rowIndex="1"
+			onMouseClicked="#userListClicked" />
+		<Label text="Select a contact to chat with" fx:id="contactLabel" />
+		<Button fx:id="settingsButton" mnemonicParsing="false"
+			text="Settings" GridPane.columnIndex="1"
+			onAction="#settingsButtonClicked" />
 		<ListView fx:id="messageList" prefHeight="257.0"
 			prefWidth="300.0" GridPane.columnIndex="1" GridPane.rowIndex="1" />
-		<Button id="postButton" mnemonicParsing="false" text="Post"
-			GridPane.columnIndex="1" GridPane.rowIndex="2" />
-		<TextArea prefHeight="200.0" prefWidth="200.0"
-			GridPane.rowIndex="2" />
+		<Button fx:id="postButton" mnemonicParsing="false" text="Post"
+			GridPane.columnIndex="1" GridPane.rowIndex="2"
+			onAction="#postButtonClicked" disable="true" />
+		<TextArea fx:id="messageTextArea" prefHeight="200.0"
+			prefWidth="200.0" GridPane.rowIndex="2"
+			onKeyReleased="#messageTextUpdated" />
 	</children>
 </GridPane>
diff --git a/src/main/java/envoy/client/ui/ChatSceneController.java b/src/main/java/envoy/client/ui/ChatSceneController.java
index d809d60..cff16bf 100644
--- a/src/main/java/envoy/client/ui/ChatSceneController.java
+++ b/src/main/java/envoy/client/ui/ChatSceneController.java
@@ -1,11 +1,22 @@
 package envoy.client.ui;
 
+import java.io.IOException;
+import java.util.logging.Level;
 import java.util.logging.Logger;
 
+import envoy.client.data.Chat;
+import envoy.client.data.LocalDB;
+import envoy.client.net.Client;
+import envoy.client.net.WriteProxy;
+import envoy.data.Message;
+import envoy.data.MessageBuilder;
+import envoy.data.User;
 import envoy.util.EnvoyLog;
-import javafx.event.ActionEvent;
+import javafx.collections.FXCollections;
 import javafx.fxml.FXML;
-import javafx.scene.control.ListView;
+import javafx.scene.control.*;
+import javafx.scene.input.KeyCode;
+import javafx.scene.input.KeyEvent;
 
 /**
  * Project: <strong>envoy-client</strong><br>
@@ -18,26 +29,121 @@ import javafx.scene.control.ListView;
 public final class ChatSceneController {
 
 	@FXML
-	private ListView messageList;
+	private Label contactLabel;
 
 	@FXML
-	private ListView userList;
+	private ListView<Message> messageList;
+
+	@FXML
+	private ListView<User> userList;
+
+	@FXML
+	private Button postButton;
+
+	@FXML
+	private Button settingsButton;
+
+	@FXML
+	private TextArea messageTextArea;
+
+	private LocalDB		localDB;
+	private Client		client;
+	private WriteProxy	writeProxy;
+
+	private Chat currentChat;
 
 	private static final Logger logger = EnvoyLog.getLogger(ChatSceneController.class);
 
 	@FXML
-	public void initialize() {
+	private void initialize() {
 		messageList.setCellFactory(listView -> new MessageListCell());
 		userList.setCellFactory(listView -> new UserListCell());
 	}
 
-	@FXML
-	public void postButtonClicked(ActionEvent e) {
-		logger.info("Post Button clicked.");
+	void initializeData(LocalDB localDB, Client client, WriteProxy writeProxy) {
+		this.localDB	= localDB;
+		this.client		= client;
+		this.writeProxy	= writeProxy;
+
+		// TODO: handle offline mode
+		userList.getItems().addAll(client.getContacts().getContacts());
+		// userList.getItems().addAll(localDB.getChats().stream().map(Chat::getRecipient).collect(Collectors.toList()));
 	}
 
 	@FXML
-	public void settingsButtonClicked(ActionEvent e) {
-		logger.info("Settings Button clicked.");
+	private void userListClicked() {
+		final User user = userList.getSelectionModel().getSelectedItem();
+		if (user != null && (currentChat == null || user != currentChat.getRecipient())) {
+			contactLabel.setText(user.getName());
+
+			// Swap observable list
+			if (currentChat != null) {
+				currentChat.getMessages().clear();
+				currentChat.getMessages().addAll(messageList.getItems());
+			}
+
+			// LEON: JFC <===> JAVA FRIED CHICKEN <=/=> Java Foundation Classes
+
+			// Load the chat or create a new one and add it to the LocalDB
+			currentChat = localDB.getChats()
+				.stream()
+				.filter(c -> c.getRecipient() == user)
+				.findAny()
+				.orElseGet(() -> { var chat = new Chat(user); localDB.getChats().add(chat); return chat; });
+			messageList.setItems(FXCollections.observableArrayList(currentChat.getMessages()));
+		}
+	}
+
+	@FXML
+	private void postButtonClicked() { postMessage(); }
+
+	@FXML
+	private void settingsButtonClicked() { logger.info("Settings Button clicked."); }
+
+	@FXML
+	private void messageTextUpdated(KeyEvent e) {
+		if (e.getCode() == KeyCode.ENTER) postMessage();
+		else postButton.setDisable(messageTextArea.getText().isBlank());
+	}
+
+	/**
+	 * Sends a new message to the server based on the text entered in the
+	 * messageTextArea.
+	 *
+	 * @since Envoy Client v0.1-beta
+	 */
+	private void postMessage() {
+
+		// Create and send message
+		sendMessage(new MessageBuilder(localDB.getUser().getID(), currentChat.getRecipient().getID(), localDB.getIDGenerator())
+			.setText(messageTextArea.getText().strip())
+			.build());
+
+		// Clear text field and disable post button
+		messageTextArea.setText("");
+		postButton.setDisable(true);
+	}
+
+	/**
+	 * Sends a message to the server.
+	 *
+	 * @param message the message to send
+	 * @since Envoy Client v0.1-beta
+	 */
+	private void sendMessage(Message message) {
+		try {
+
+			// Send message
+			writeProxy.writeMessage(message);
+
+			// Add message to LocalDB and update UI
+			messageList.getItems().add(message);
+
+			// Request a new ID generator if all IDs were used
+			if (!localDB.getIDGenerator().hasNext()) client.requestIdGenerator();
+
+		} catch (IOException e) {
+			logger.log(Level.SEVERE, "Error sending message", e);
+		}
 	}
 }
diff --git a/src/main/java/envoy/client/ui/MessageListCell.java b/src/main/java/envoy/client/ui/MessageListCell.java
index 86d37e2..21c216d 100644
--- a/src/main/java/envoy/client/ui/MessageListCell.java
+++ b/src/main/java/envoy/client/ui/MessageListCell.java
@@ -21,6 +21,7 @@ public class MessageListCell extends ListCell<Message> {
 	@Override
 	protected void updateItem(Message message, boolean empty) {
 		super.updateItem(message, empty);
-		if (!empty && message != null) { setGraphic(new Label(message.getText())); }
+		if (!empty && message != null) setGraphic(new Label(message.getText()));
+		else setGraphic(null);
 	}
 }
diff --git a/src/main/java/envoy/client/ui/Startup.java b/src/main/java/envoy/client/ui/Startup.java
index 2f2bbd7..c392281 100644
--- a/src/main/java/envoy/client/ui/Startup.java
+++ b/src/main/java/envoy/client/ui/Startup.java
@@ -1,12 +1,21 @@
 package envoy.client.ui;
 
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.util.Properties;
 import java.util.logging.Level;
 import java.util.logging.Logger;
 
+import javax.swing.JOptionPane;
+
 import envoy.client.data.*;
 import envoy.client.net.Client;
 import envoy.client.net.WriteProxy;
+import envoy.client.ui.container.LoginDialog;
 import envoy.data.Message;
+import envoy.data.User.UserStatus;
+import envoy.exception.EnvoyException;
 import envoy.util.EnvoyLog;
 import javafx.application.Application;
 import javafx.fxml.FXMLLoader;
@@ -25,9 +34,9 @@ import javafx.stage.Stage;
  */
 public final class Startup extends Application {
 
-	private LocalDB		localDB;
-	private Client		client;
-	private WriteProxy	writeProxy;
+	private LocalDB			localDB;
+	private Client			client;
+	private WriteProxy		writeProxy;
 	private Cache<Message>	cache;
 
 	private static final ClientConfig	config	= ClientConfig.getInstance();
@@ -38,87 +47,81 @@ public final class Startup extends Application {
 	 */
 	@Override
 	public void init() throws Exception {
-		// try {
-		// // Load the configuration from client.properties first
-		// Properties properties = new Properties();
-		// properties.load(Startup.class.getClassLoader().getResourceAsStream("client.properties"));
-		// config.load(properties);
-		//
-		// // Override configuration values with command line arguments
-		// String[] args = getParameters().getRaw().toArray(new String[0]);
-		// if (args.length > 0) config.load(args);
-		//
-		// // Check if all mandatory configuration values have been initialized
-		// if (!config.isInitialized()) throw new EnvoyException("Configuration is not
-		// fully initialized");
-		// } catch (Exception e) {
-		// JOptionPane.showMessageDialog(null, "Error loading configuration values:\n" +
-		// e, "Configuration error", JOptionPane.ERROR_MESSAGE);
-		// e.printStackTrace();
-		// System.exit(1);
-		// }
-		//
-		// // Setup logger for the envoy package
-		// EnvoyLog.initialize(config);
-		// EnvoyLog.attach("envoy");
-		// EnvoyLog.setFileLevelBarrier(config.getFileLevelBarrier());
-		// EnvoyLog.setConsoleLevelBarrier(config.getConsoleLevelBarrier());
-		//
-		// // Initialize the local database
-		// if (config.isIgnoreLocalDB()) {
-		// localDB = new TransientLocalDB();
-		// JOptionPane.showMessageDialog(null,
-		// "Ignoring local database.\nMessages will not be saved!",
-		// "Local database warning",
-		// JOptionPane.WARNING_MESSAGE);
-		// } else try {
-		// localDB = new PersistentLocalDB(new File(config.getHomeDirectory(),
-		// config.getLocalDB().getPath()));
-		// } catch (IOException e3) {
-		// logger.log(Level.SEVERE, "Could not initialize local database", e3);
-		// JOptionPane.showMessageDialog(null, "Could not initialize local database!\n"
-		// + e3, "Local database error", JOptionPane.ERROR_MESSAGE);
-		// System.exit(1);
-		// return;
-		// }
-		//
-		// // Initialize client and unread message cache
-		// client = new Client();
-		// cache = new Cache<>();
-		//
-		// // Try to connect to the server
-		// new LoginDialog(client, localDB, cache);
-		//
-		// // Set client user in local database
-		// localDB.setUser(client.getSender());
-		//
-		// // Initialize chats in local database
-		// try {
-		// localDB.initializeUserStorage();
-		// localDB.loadUserData();
-		// } catch (FileNotFoundException e) {
-		// // The local database file has not yet been created, probably first login
-		// } catch (Exception e) {
-		// e.printStackTrace();
-		// JOptionPane.showMessageDialog(null,
-		// "Error while loading local database: " + e + "\nChats will not be stored
-		// locally.",
-		// "Local DB error",
-		// JOptionPane.WARNING_MESSAGE);
-		// }
-		//
-		// // Initialize write proxy
-		// writeProxy = client.createWriteProxy(localDB);
-		//
-		// if (client.isOnline()) {
-		//
-		// // Save all users to the local database and flush cache
-		// localDB.setUsers(client.getUsers());
-		// writeProxy.flushCache();
-		// } else
-		// // Set all contacts to offline mode
-		// localDB.getUsers().values().stream().filter(u -> u !=
-		// localDB.getUser()).forEach(u -> u.setStatus(UserStatus.OFFLINE));
+		try {
+			// Load the configuration from client.properties first
+			Properties properties = new Properties();
+			properties.load(Startup.class.getClassLoader().getResourceAsStream("client.properties"));
+			config.load(properties);
+
+			// Override configuration values with command line arguments
+			String[] args = getParameters().getRaw().toArray(new String[0]);
+			if (args.length > 0) config.load(args);
+
+			// Check if all mandatory configuration values have been initialized
+			if (!config.isInitialized()) throw new EnvoyException("Configuration is not fully initialized");
+		} catch (Exception e) {
+			JOptionPane.showMessageDialog(null, "Error loading configuration values:\n" + e, "Configuration error", JOptionPane.ERROR_MESSAGE);
+			e.printStackTrace();
+			System.exit(1);
+		}
+
+		// Setup logger for the envoy package
+		EnvoyLog.initialize(config);
+		EnvoyLog.attach("envoy");
+		EnvoyLog.setFileLevelBarrier(config.getFileLevelBarrier());
+		EnvoyLog.setConsoleLevelBarrier(config.getConsoleLevelBarrier());
+
+		// Initialize the local database
+		if (config.isIgnoreLocalDB()) {
+			localDB = new TransientLocalDB();
+			JOptionPane.showMessageDialog(null,
+					"Ignoring local database.\nMessages will not be saved!",
+					"Local database warning",
+					JOptionPane.WARNING_MESSAGE);
+		} else try {
+			localDB = new PersistentLocalDB(new File(config.getHomeDirectory(), config.getLocalDB().getPath()));
+		} catch (IOException e3) {
+			logger.log(Level.SEVERE, "Could not initialize local database", e3);
+			JOptionPane.showMessageDialog(null, "Could not initialize local database!\n" + e3, "Local database error", JOptionPane.ERROR_MESSAGE);
+			System.exit(1);
+			return;
+		}
+
+		// Initialize client and unread message cache
+		client	= new Client();
+		cache	= new Cache<>();
+
+		// Try to connect to the server
+		new LoginDialog(client, localDB, cache);
+
+		// Set client user in local database
+		localDB.setUser(client.getSender());
+
+		// Initialize chats in local database
+		try {
+			localDB.initializeUserStorage();
+			localDB.loadUserData();
+		} catch (FileNotFoundException e) {
+			// The local database file has not yet been created, probably first login
+		} catch (Exception e) {
+			e.printStackTrace();
+			JOptionPane.showMessageDialog(null,
+					"Error while loading local database: " + e + "\nChats will not be stored locally.",
+					"Local DB error",
+					JOptionPane.WARNING_MESSAGE);
+		}
+
+		// Initialize write proxy
+		writeProxy = client.createWriteProxy(localDB);
+
+		if (client.isOnline()) {
+
+			// Save all users to the local database and flush cache
+			localDB.setUsers(client.getUsers());
+			writeProxy.flushCache();
+		} else
+			// Set all contacts to offline mode
+			localDB.getUsers().values().stream().filter(u -> u != localDB.getUser()).forEach(u -> u.setStatus(UserStatus.OFFLINE));
 	}
 
 	/**
@@ -134,6 +137,8 @@ public final class Startup extends Application {
 		stage.setTitle("Envoy");
 		stage.getIcons().add(new Image(getClass().getResourceAsStream("/icons/envoy_logo.png")));
 		stage.setScene(chatScene);
+
+		loader.<ChatSceneController>getController().initializeData(localDB, client, writeProxy);
 		stage.show();
 
 		// Relay unread messages from cache
diff --git a/src/main/java/envoy/client/ui/UserListCell.java b/src/main/java/envoy/client/ui/UserListCell.java
index fa35ce9..f62f98a 100644
--- a/src/main/java/envoy/client/ui/UserListCell.java
+++ b/src/main/java/envoy/client/ui/UserListCell.java
@@ -1,7 +1,9 @@
 package envoy.client.ui;
 
 import envoy.data.User;
+import javafx.scene.control.Label;
 import javafx.scene.control.ListCell;
+import javafx.scene.layout.VBox;
 
 /**
  * Project: <strong>envoy-client</strong><br>
@@ -20,7 +22,9 @@ public class UserListCell extends ListCell<User> {
 	protected void updateItem(User user, boolean empty) {
 		super.updateItem(user, empty);
 		if (!empty && user != null) {
-
+			final Label	name	= new Label(user.getName());
+			final Label	status	= new Label(user.getStatus().toString());
+			setGraphic(new VBox(name, status));
 		}
 	}
 }
diff --git a/src/main/java/envoy/client/ui/container/ChatWindow.java b/src/main/java/envoy/client/ui/container/ChatWindow.java
index e3dfd47..eb92fc4 100644
--- a/src/main/java/envoy/client/ui/container/ChatWindow.java
+++ b/src/main/java/envoy/client/ui/container/ChatWindow.java
@@ -259,7 +259,7 @@ public class ChatWindow extends JFrame {
 					textPane.setText(currentChat.getRecipient().getName());
 
 					// Update model and scroll down
-					messageList.setModel(currentChat.getModel());
+					// messageList.setModel(currentChat.getModel());
 					scrollPane.setChatOpened(true);
 
 					messageList.synchronizeModel();
@@ -413,7 +413,7 @@ public class ChatWindow extends JFrame {
 		EventBus.getInstance().register(MessageCreationEvent.class, evt -> {
 			Message	message	= evt.get();
 			Chat	chat	= localDB.getChats().stream().filter(c -> c.getRecipient().getID() == message.getSenderID()).findFirst().get();
-			chat.appendMessage(message);
+			// chat.appendMessage(message);
 
 			// Read message and update UI if in current chat
 			if (chat == currentChat) readCurrentChat();
@@ -428,18 +428,18 @@ public class ChatWindow extends JFrame {
 			final MessageStatus	status	= evt.get();
 
 			for (Chat c : localDB.getChats())
-				for (Message m : c.getModel())
-					if (m.getID() == id) {
-
-						// Update message status
-						m.setStatus(status);
-
-						// Update model and scroll down if current chat
-						if (c == currentChat) {
-							messageList.setModel(currentChat.getModel());
-							scrollPane.setChatOpened(true);
-						} else messageList.synchronizeModel();
-					}
+				// for (Message m : c.getModel())
+				// if (m.getID() == id) {
+				//
+				// // Update message status
+				// m.setStatus(status);
+				//
+				// // Update model and scroll down if current chat
+				// if (c == currentChat) {
+				// messageList.setModel(currentChat.getModel());
+				// scrollPane.setChatOpened(true);
+				// } else messageList.synchronizeModel();
+				// }
 
 			revalidate();
 			repaint();
@@ -584,7 +584,7 @@ public class ChatWindow extends JFrame {
 			writeProxy.writeMessage(message);
 
 			// Add message to PersistentLocalDB and update UI
-			currentChat.appendMessage(message);
+			// currentChat.appendMessage(message);
 
 			// Update UI
 			revalidate();