From f147682f189a5991d97ca31bd2c83bfd139f0c7b Mon Sep 17 00:00:00 2001 From: Romina Date: Fri, 19 Jun 2026 23:22:33 +0330 Subject: [PATCH] Sharing chat System --- .../chat/Client/ServerListener.java | 20 --- .../university/chat/Client/chatClient.java | 29 ---- .../university/chat/Server/ChatServer.java | 15 -- .../university/chat/Server/ClientSession.java | 71 ---------- .../chat/client/ServerListener.java | 44 ++++++ .../{Client => client}/TransferProgress.java | 2 +- .../university/chat/client/chatClient.java | 123 ++++++++++++++++ .../chat/{Common => common}/ChatMessage.java | 2 +- .../chat/{Common => common}/FileMessage.java | 2 +- .../chat/{Common => common}/MessageType.java | 2 +- .../university/chat/server/ChatServer.java | 30 ++++ .../university/chat/server/ClientSession.java | 134 ++++++++++++++++++ .../chat/{Server => server}/FileManager.java | 2 +- .../chat/{Server => server}/UserManager.java | 2 +- 14 files changed, 337 insertions(+), 141 deletions(-) delete mode 100644 src/main/java/com/university/chat/Client/ServerListener.java delete mode 100644 src/main/java/com/university/chat/Client/chatClient.java delete mode 100644 src/main/java/com/university/chat/Server/ChatServer.java delete mode 100644 src/main/java/com/university/chat/Server/ClientSession.java create mode 100644 src/main/java/com/university/chat/client/ServerListener.java rename src/main/java/com/university/chat/{Client => client}/TransferProgress.java (96%) create mode 100644 src/main/java/com/university/chat/client/chatClient.java rename src/main/java/com/university/chat/{Common => common}/ChatMessage.java (94%) rename src/main/java/com/university/chat/{Common => common}/FileMessage.java (95%) rename src/main/java/com/university/chat/{Common => common}/MessageType.java (85%) create mode 100644 src/main/java/com/university/chat/server/ChatServer.java create mode 100644 src/main/java/com/university/chat/server/ClientSession.java rename src/main/java/com/university/chat/{Server => server}/FileManager.java (95%) rename src/main/java/com/university/chat/{Server => server}/UserManager.java (94%) diff --git a/src/main/java/com/university/chat/Client/ServerListener.java b/src/main/java/com/university/chat/Client/ServerListener.java deleted file mode 100644 index b609d74..0000000 --- a/src/main/java/com/university/chat/Client/ServerListener.java +++ /dev/null @@ -1,20 +0,0 @@ -package com.university.chat.Client; - -public class ServerListener implements Runnable{ - // TODO: store the ObjectInputStream from the user socket - // (this should be the same input stream the - // chatClient created when connecting) - - @Override - public void run() { - try { - // TODO: In an infinite loop read objects from the server - // - if it's a ChatMessage -> print ": " - // - if it's a FileMessage -> print that a file was received - // (filename + sender), it's already - // saved to disk by the server. - } catch (Exception e){ - System.out.println("Disconnected from server"); - } - } -} diff --git a/src/main/java/com/university/chat/Client/chatClient.java b/src/main/java/com/university/chat/Client/chatClient.java deleted file mode 100644 index dae0bad..0000000 --- a/src/main/java/com/university/chat/Client/chatClient.java +++ /dev/null @@ -1,29 +0,0 @@ -package com.university.chat.Client; - -public class chatClient { - public static void main() { - // TODO: Connecting to the server - // 1. Create a socket and connect to the server - // 2. Create an ObjectOutputStream (out) and ObjectInputStream (in) - // from the socket's streams — output FIRST, then input. - // 2. Get the username, and send a LOGIN ChatMessage with that username - // 3. Start a new Thread running a ServerListener(in) so incoming - // messages are handled concurrently. - - while (true){ - try { - // TODO: Program loop — read a line from the console and act on it: - // - "/msg " -> build & send a PRIVATE_MESSAGE - // - "/users" -> build & send a USER_LIST request - // - "/sendfile " -> read the file into a byte[] - // (you can use TransferProgress - // to show progress) - // and send it as a FileMessage - // - anything else -> send a PUBLIC_MESSAGE - // Remember to flush() the output stream after writeObject(). - } catch (Exception e){ - System.out.println("command failed: " + e.getMessage()); - } - } - } -} diff --git a/src/main/java/com/university/chat/Server/ChatServer.java b/src/main/java/com/university/chat/Server/ChatServer.java deleted file mode 100644 index b2a35ae..0000000 --- a/src/main/java/com/university/chat/Server/ChatServer.java +++ /dev/null @@ -1,15 +0,0 @@ -package com.university.chat.Server; - -public class ChatServer { - // TODO: declare a single shared UserManager instance (static final) - // This MUST be shared by all ClientSession threads so that - // broadcasting and private messaging work correctly. - - public static void main(String[] args) { - // TODO: Create a ServerSocket - - // TODO: In an infinite loop: - // accept an incoming client connection - // make a new thread running ClientSession for each user. - } -} diff --git a/src/main/java/com/university/chat/Server/ClientSession.java b/src/main/java/com/university/chat/Server/ClientSession.java deleted file mode 100644 index f09d64a..0000000 --- a/src/main/java/com/university/chat/Server/ClientSession.java +++ /dev/null @@ -1,71 +0,0 @@ -package com.university.chat.Server; - -import com.university.chat.Common.ChatMessage; -import com.university.chat.Common.FileMessage; - -import java.io.IOException; -import java.net.Socket; -import java.nio.file.Files; - -public class ClientSession implements Runnable { - - private String username; - - public ClientSession(Socket socket, UserManager userManager) { - // TODO : Create an ObjectOutputStream from socket.getOutputStream() - // and an ObjectInputStream from socket.getInputStream(). - } - - @Override - public void run() { - try { - - // TODO: Welcome the user (login step) - // 1. Read the first object sent by the client. - // 2. Check it's a ChatMessage with type LOGIN. - // 3. Extract the username. - // 4. Try to register the user via userManager.addUser(...). - // 5. If the username is taken, send back LOGIN_FAILED and close the socket. - // 6. Otherwise, create the user's folders with FileManager.createUserFolders(...) - // and send back LOGIN_SUCCESS. - - // TODO: Main message loop - // In a loop, call in.readObject(), you can separate messages by their type: - // - if it's a ChatMessage -> call handleChatMessage(msg) - // - if it's a FileMessage -> call handleFileMessage(fileMsg) - // Keep looping until the connection is closed (an exception will be thrown). - - } catch (Exception e) { - System.out.println("Disconnected: " + username); - } finally { - // TODO: Remove the user from UserManager so they no longer - // receive broadcasts or appear in users list - } - } - - - private void handleChatMessage(ChatMessage msg) throws IOException { - switch (msg.getType()) { - case PUBLIC_MESSAGE -> { - // TODO: Broadcast this message to every connected client. - } - case PRIVATE_MESSAGE -> { - // TODO: Forward this message to the receiver user. - } - case USER_LIST -> { - // TODO: Reply to the requester with the list of online users. - } - } - } - - private void handleFileMessage(FileMessage fileMsg) throws IOException { - // Storing the file - var sentPath = FileManager.getSentPath(fileMsg.getSender(), fileMsg.getFilename()); - var recvPath = FileManager.getReceivedPath(fileMsg.getReceiver(), fileMsg.getFilename()); - - Files.write(sentPath, fileMsg.getData()); - Files.write(recvPath, fileMsg.getData()); - - // TODO: Forward the received file-message to the destination user. - } -} \ No newline at end of file diff --git a/src/main/java/com/university/chat/client/ServerListener.java b/src/main/java/com/university/chat/client/ServerListener.java new file mode 100644 index 0000000..d6de17b --- /dev/null +++ b/src/main/java/com/university/chat/client/ServerListener.java @@ -0,0 +1,44 @@ +package com.university.chat.client; +import com.university.chat.common.ChatMessage; +import com.university.chat.common.FileMessage; +import java.io.ObjectInputStream; + +public class ServerListener implements Runnable +{ + private ObjectInputStream in; + + public ServerListener(ObjectInputStream in) {this.in = in;} + + @Override + public void run() + { + try + { + while (true) + { + Object obj = in.readObject(); + + if (obj instanceof ChatMessage) + { + ChatMessage msg = (ChatMessage) obj; + + String sender = msg.getSender(); + String content = msg.getContent(); + + System.out.println(sender + ": " + content); + } + + else if (obj instanceof FileMessage) + { + FileMessage fileMsg = (FileMessage) obj; + + String sender = fileMsg.getSender(); + String fileName = fileMsg.getFilename(); + + System.out.println("File received from " + sender + ": " + fileName); + } + } + } + catch (Exception e) {System.out.println("Disconnected from server");} + } +} \ No newline at end of file diff --git a/src/main/java/com/university/chat/Client/TransferProgress.java b/src/main/java/com/university/chat/client/TransferProgress.java similarity index 96% rename from src/main/java/com/university/chat/Client/TransferProgress.java rename to src/main/java/com/university/chat/client/TransferProgress.java index 0e617a4..9be0d49 100644 --- a/src/main/java/com/university/chat/Client/TransferProgress.java +++ b/src/main/java/com/university/chat/client/TransferProgress.java @@ -1,4 +1,4 @@ -package com.university.chat.Client; +package com.university.chat.client; public class TransferProgress { diff --git a/src/main/java/com/university/chat/client/chatClient.java b/src/main/java/com/university/chat/client/chatClient.java new file mode 100644 index 0000000..1fbf385 --- /dev/null +++ b/src/main/java/com/university/chat/client/chatClient.java @@ -0,0 +1,123 @@ +package com.university.chat.client; +import com.university.chat.common.ChatMessage; +import com.university.chat.common.FileMessage; +import com.university.chat.common.MessageType; +import java.io.*; +import java.net.Socket; +import java.nio.file.Files; +import java.nio.file.Paths; +import java.util.Scanner; + +public class chatClient +{ + public static void main(String[] args) + { + try + { + Socket socket = new Socket("localhost", 12345); + + ObjectOutputStream out = new ObjectOutputStream(socket.getOutputStream()); + out.flush(); + ObjectInputStream in = new ObjectInputStream(socket.getInputStream()); + + Scanner scanner = new Scanner(System.in); + System.out.print("Enter username: "); + String username = scanner.nextLine(); + + ChatMessage loginMsg = new ChatMessage(MessageType.LOGIN, username, null, null); + out.writeObject(loginMsg); + out.flush(); + + new Thread(new ServerListener(in)).start(); + + while (true) + { + try + { + String line = scanner.nextLine(); + + if (line.startsWith("/msg ")) + { + String[] parts = line.split(" ", 3); + if (parts.length < 3) + { + System.out.println("Usage: /msg "); + continue; + } + + String target = parts[1]; + String text = parts[2]; + + ChatMessage msg = new ChatMessage( + MessageType.PRIVATE_MESSAGE, + username, + target, + text + ); + + out.writeObject(msg); + out.flush(); + } + + else if (line.equals("/users")) + { + ChatMessage msg = new ChatMessage( + MessageType.USER_LIST, + username, + null, + null + ); + + out.writeObject(msg); + out.flush(); + } + + else if (line.startsWith("/sendfile ")) + { + String[] parts = line.split(" ", 3); + if (parts.length < 3) + { + System.out.println("Usage: /sendfile "); + continue; + } + + String target = parts[1]; + String path = parts[2]; + + byte[] fileData = Files.readAllBytes(Paths.get(path)); + String fileName = Paths.get(path).getFileName().toString(); + + FileMessage fileMsg = new FileMessage( + username, + target, + fileName, + fileData + ); + + out.writeObject(fileMsg); + out.flush(); + + System.out.println("File sent: " + fileName); + } + + else + { + ChatMessage msg = new ChatMessage( + MessageType.PUBLIC_MESSAGE, + username, + null, + line + ); + + out.writeObject(msg); + out.flush(); + } + + } + catch (Exception e) {System.out.println("command failed: " + e.getMessage());} + } + + } + catch (Exception e) {System.out.println("Connection failed: " + e.getMessage());} + } +} \ No newline at end of file diff --git a/src/main/java/com/university/chat/Common/ChatMessage.java b/src/main/java/com/university/chat/common/ChatMessage.java similarity index 94% rename from src/main/java/com/university/chat/Common/ChatMessage.java rename to src/main/java/com/university/chat/common/ChatMessage.java index c097745..8e7c6de 100644 --- a/src/main/java/com/university/chat/Common/ChatMessage.java +++ b/src/main/java/com/university/chat/common/ChatMessage.java @@ -1,4 +1,4 @@ -package com.university.chat.Common; +package com.university.chat.common; import java.io.Serializable; diff --git a/src/main/java/com/university/chat/Common/FileMessage.java b/src/main/java/com/university/chat/common/FileMessage.java similarity index 95% rename from src/main/java/com/university/chat/Common/FileMessage.java rename to src/main/java/com/university/chat/common/FileMessage.java index 4f90fbf..1647edf 100644 --- a/src/main/java/com/university/chat/Common/FileMessage.java +++ b/src/main/java/com/university/chat/common/FileMessage.java @@ -1,4 +1,4 @@ -package com.university.chat.Common; +package com.university.chat.common; import java.io.Serializable; diff --git a/src/main/java/com/university/chat/Common/MessageType.java b/src/main/java/com/university/chat/common/MessageType.java similarity index 85% rename from src/main/java/com/university/chat/Common/MessageType.java rename to src/main/java/com/university/chat/common/MessageType.java index 093ff8d..e44de4e 100644 --- a/src/main/java/com/university/chat/Common/MessageType.java +++ b/src/main/java/com/university/chat/common/MessageType.java @@ -1,4 +1,4 @@ -package com.university.chat.Common; +package com.university.chat.common; import java.io.Serializable; diff --git a/src/main/java/com/university/chat/server/ChatServer.java b/src/main/java/com/university/chat/server/ChatServer.java new file mode 100644 index 0000000..6b3b18a --- /dev/null +++ b/src/main/java/com/university/chat/server/ChatServer.java @@ -0,0 +1,30 @@ +package com.university.chat.server; +import java.net.ServerSocket; +import java.net.Socket; + +public class ChatServer +{ + public static final UserManager userManager = new UserManager(); + + public static void main(String[] args) + { + int port = 12345; + + try + { + ServerSocket serverSocket = new ServerSocket(port); + System.out.println("Server started on port " + port); + + while (true) + { + Socket clientSocket = serverSocket.accept(); + System.out.println("New client connected"); + + ClientSession session = new ClientSession(clientSocket, userManager); + new Thread(session).start(); + } + + } + catch (Exception e) {e.printStackTrace();} + } +} \ No newline at end of file diff --git a/src/main/java/com/university/chat/server/ClientSession.java b/src/main/java/com/university/chat/server/ClientSession.java new file mode 100644 index 0000000..d647549 --- /dev/null +++ b/src/main/java/com/university/chat/server/ClientSession.java @@ -0,0 +1,134 @@ +package com.university.chat.server; +import com.university.chat.common.ChatMessage; +import com.university.chat.common.FileMessage; +import com.university.chat.common.MessageType; +import java.io.*; +import java.net.Socket; +import java.nio.file.Files; + +public class ClientSession implements Runnable +{ + private Socket socket; + private ObjectInputStream in; + private ObjectOutputStream out; + + private String username; + private UserManager userManager; + + public ClientSession(Socket socket, UserManager userManager) + { + try + { + this.socket = socket; + this.userManager = userManager; + + this.out = new ObjectOutputStream(socket.getOutputStream()); + this.out.flush(); + this.in = new ObjectInputStream(socket.getInputStream()); + + } + catch (IOException e) {System.out.println("Stream error: " + e.getMessage());} + } + + public void send(Object obj) throws IOException + { + out.writeObject(obj); + out.flush(); + } + + @Override + public void run() + { + try + { + ChatMessage loginMsg = (ChatMessage) in.readObject(); + + if (loginMsg.getType() != MessageType.LOGIN) + { + socket.close(); + return; + } + + this.username = loginMsg.getSender(); + + boolean ok = userManager.addUser(username, this); + + if (!ok) + { + send(new ChatMessage( + MessageType.LOGIN_FAILED, + "SERVER", + username, + "Username already taken" + )); + socket.close(); + return; + } + + FileManager.createUserFolders(username); + + send(new ChatMessage( + MessageType.LOGIN_SUCCESS, + "SERVER", + username, + "Welcome " + username + )); + + while (true) + { + Object obj = in.readObject(); + + if (obj instanceof ChatMessage msg) {handleChat(msg);} + + else if (obj instanceof FileMessage fileMsg) {handleFile(fileMsg);} + } + + } + catch (Exception e) {System.out.println("Disconnected: " + username);} + finally + { + if (username != null) {userManager.removeUser(username);} + try { socket.close(); } + catch (Exception ignored) {} + } + } + + private void handleChat(ChatMessage msg) throws IOException + { + switch (msg.getType()) + { + case PUBLIC_MESSAGE -> + { + for (ClientSession session : userManager.getAllSessions()) {session.send(msg);} + } + + case PRIVATE_MESSAGE -> + { + ClientSession target = userManager.getUser(msg.getReceiver()); + if (target != null) {target.send(msg);} + } + + case USER_LIST -> + { + send(new ChatMessage( + MessageType.USER_LIST, + "SERVER", + username, + userManager.listUsers() + )); + } + } + } + + private void handleFile(FileMessage fileMsg) throws IOException + { + var sentPath = FileManager.getSentPath(fileMsg.getSender(), fileMsg.getFilename()); + var recvPath = FileManager.getReceivedPath(fileMsg.getReceiver(), fileMsg.getFilename()); + + Files.write(sentPath, fileMsg.getData()); + Files.write(recvPath, fileMsg.getData()); + + ClientSession target = userManager.getUser(fileMsg.getReceiver()); + if (target != null) {target.send(fileMsg);} + } +} \ No newline at end of file diff --git a/src/main/java/com/university/chat/Server/FileManager.java b/src/main/java/com/university/chat/server/FileManager.java similarity index 95% rename from src/main/java/com/university/chat/Server/FileManager.java rename to src/main/java/com/university/chat/server/FileManager.java index 0c2c715..00759b0 100644 --- a/src/main/java/com/university/chat/Server/FileManager.java +++ b/src/main/java/com/university/chat/server/FileManager.java @@ -1,4 +1,4 @@ -package com.university.chat.Server; +package com.university.chat.server; import java.io.IOException; import java.nio.file.Files; diff --git a/src/main/java/com/university/chat/Server/UserManager.java b/src/main/java/com/university/chat/server/UserManager.java similarity index 94% rename from src/main/java/com/university/chat/Server/UserManager.java rename to src/main/java/com/university/chat/server/UserManager.java index 07cbc2d..13c9812 100644 --- a/src/main/java/com/university/chat/Server/UserManager.java +++ b/src/main/java/com/university/chat/server/UserManager.java @@ -1,4 +1,4 @@ -package com.university.chat.Server; +package com.university.chat.server; import java.util.concurrent.ConcurrentHashMap;