Complete the project

2026-06-20 19:17:07 +03:30
parent 240eac0504
commit 610b5a1364
5 changed files with 314 additions and 68 deletions
@@ -1,29 +1,138 @@
 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.
+import com.university.chat.Common.ChatMessage;
+import com.university.chat.Common.FileMessage;
+import com.university.chat.Common.MessageType;

-        while (true){
-            try {
-                // TODO: Program loop — read a line from the console and act on it:
-                //   - "/msg <user> <text>"        -> build & send a PRIVATE_MESSAGE
-                //   - "/users"                    -> build & send a USER_LIST request
-                //   - "/sendfile <user> <path>"   -> 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());
+import java.io.*;
+import java.net.Socket;
+import java.net.UnknownHostException;
+import java.util.Scanner;
+
+public class chatClient
+{
+    private static Socket socket;
+    private static ObjectOutputStream out;
+    private static ObjectInputStream in;
+    private static String username;
+
+    public static void main()
+    {
+        Scanner scanner = new Scanner(System.in);
+
+        try
+        {
+            socket = new Socket("localhost", 5000);
+            out = new ObjectOutputStream(socket.getOutputStream());
+            in = new ObjectInputStream(socket.getInputStream());
+
+            System.out.print("Enter your username: ");
+            username = scanner.nextLine();
+
+            ChatMessage loginMsg = new ChatMessage(MessageType.LOGIN, username, null, "");
+            out.writeObject(loginMsg);
+            out.flush();
+
+            ServerListener listener = new ServerListener(in);
+            Thread thread = new Thread(listener);
+            thread.start();
+
+            System.out.println("Connected!");
+
+            while (true)
+            {
+                try
+                {
+                    String input = scanner.nextLine();
+
+                    if (input.startsWith("/msg "))
+                    {
+                        String[] parts = input.split(" ", 3);
+                        if (parts.length >= 3)
+                        {
+                            String receiver = parts[1];
+                            String content = parts[2];
+                            ChatMessage msg = new ChatMessage(MessageType.PRIVATE_MESSAGE, username, receiver, content);
+                            out.writeObject(msg);
+                            out.flush();
+                        }
+                        else
+                        {
+                            System.out.println("Usage: /msg <username> <message>");
+                        }
+
+                    }
+                    else if (input.trim().equals("/users"))
+                    {
+                        ChatMessage req = new ChatMessage(MessageType.USER_LIST, username, null, "");
+                        out.writeObject(req);
+                        out.flush();
+
+                    }
+                    else if (input.startsWith("/sendfile "))
+                    {
+                        String[] parts = input.split(" ", 3);
+                        if (parts.length >= 3)
+                        {
+                            String receiver = parts[1];
+                            String filePath = parts[2];
+                            sendFile(receiver, filePath);
+                        }
+                        else
+                        {
+                            System.out.println("Usage: /sendfile <username> <filepath>");
+                        }
+
+                    }
+                    else
+                    {
+                        if (!input.isEmpty())
+                        {
+                            ChatMessage msg = new ChatMessage(MessageType.PUBLIC_MESSAGE, username, null, input);
+                            out.writeObject(msg);
+                            out.flush();
+                        }
+                    }
+                }
+                catch (Exception e)
+                {
+                    System.out.println("command failed: " + e.getMessage());
+                }
            }
        }
+        catch (UnknownHostException e)
+        {
+            throw new RuntimeException(e);
+        }
+        catch (IOException e)
+        {
+            throw new RuntimeException(e);
+        }
+    }
+
+    private static void sendFile(String receiver, String filePath) throws IOException
+    {
+        File file = new File(filePath);
+        if (!file.exists())
+        {
+            System.out.println("file not found : " + filePath);
+            return;
+        }
+
+        long fileSize = file.length();
+        byte[] data = new byte[(int) fileSize];
+
+        try (FileInputStream fis = new FileInputStream(file))
+        {
+            fis.read(data);
+        }
+
+        TransferProgress progress = new TransferProgress(fileSize);
+        progress.update(fileSize);
+
+        FileMessage fileMsg = new FileMessage(username, receiver, file.getName(), data);
+        out.writeObject(fileMsg);
+        out.flush();
+        System.out.println("Send file : " + file.getName());
    }
 }