import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.IOException;

import java.io.InputStreamReader;

import java.io.OutputStreamWriter;

import java.net.Socket;

import java.net.UnknownHostException;

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.time.LocalDateTime;

import java.util.Iterator;

import java.util.Scanner;

import augie.edu.finalProgram.Belgovi.Encrpytion;

import augie.edu.finalProgram.Belgovi.FileStorage;

import augie.edu.finalProgram.Belgovi.MyArrayList;

public class Client {

private Socket socket;

private BufferedReader bufferedReader;

private BufferedWriter bufferedWriter;

private String clientName;

private FileStorage fileStorage;

private Encrpytion encrpytion;

// Constructor

public Client(Socket socket, String clientName) {

try {

this.socket = socket;

this.clientName = clientName;

this.bufferedReader = new BufferedReader(new InputStreamReader(socket.getInputStream()));

this.bufferedWriter = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));

this.fileStorage = new FileStorage(clientName + ".csv");

this.encrpytion = new Encrpytion(1234);

// Check if history of previous conversation exists, and load history

Path path = Paths.get(clientName + ".csv");

if (Files.exists(path)) {

MyArrayList<String[]> history = fileStorage.readfromFile();

Iterator<String[]> iterator = history.iterator();

while (iterator.hasNext()) {

String[] singleLine = iterator.next();

System.out.println(singleLine[0] + ": " + encrpytion.decrpytText(singleLine[2]));

}

}

} catch (Exception e) {

// Disconnect everything

disconnect();

}

}

// Send a message to the server

public void sendMessage() {

try {

bufferedWriter.write(clientName);

bufferedWriter.newLine();

bufferedWriter.flush();

// TODO: Make sure that the client can send messages to the server using GUI

Scanner scanner = new Scanner(System.in);

while (socket.isConnected()) {

// Get datetime to store to file database

LocalDateTime now = LocalDateTime.now();

String message = scanner.nextLine();

bufferedWriter.write(message);

bufferedWriter.newLine();

bufferedWriter.flush();

// Store into file

// Encrpyt Message before file

message = encrpytion.encrpytText(message);

fileStorage.writeToFile(new String[] {clientName, now.toString(), message});

}

} catch (Exception e) {

// Disconnect everything

e.printStackTrace();

disconnect();

}

}

// Listen for messages from the server

// Using a thread to listen for messages

// This uses concurrency

public void listenForMessage() {

new Thread(new Runnable() {

@Override

public void run() {

try {

while (socket.isConnected()) {

// Get datetime to store to file database

LocalDateTime now = LocalDateTime.now();

String message = bufferedReader.readLine();

System.out.println(message);

// Decrpyt Message before storing

String[] messagePart = message.split(": ");

messagePart[1] = encrpytion.encrpytText(messagePart[1]);

// Store message from incoming chat

fileStorage.writeToFile(new String[] {messagePart[0], now.toString(), messagePart[1]});

}

} catch (Exception e) {

// Disconnect everything

disconnect();

}

}

}).start();

}

private void disconnect() {

try {

if (socket != null) {

socket.close();

}

if (bufferedReader != null) {

bufferedReader.close();

}

if (bufferedWriter != null) {

bufferedWriter.close();

}

} catch (Exception e) {

e.printStackTrace();

}

}

// Main method to run the client

public static void main(String[] args) throws UnknownHostException, IOException {

Scanner scanner = new Scanner(System.in);

System.out.println("Enter your name: ");

String clientName = scanner.nextLine();

Socket socket = new Socket("localhost", 8080);

Client client = new Client(socket, clientName);

// These are blocking methods

// The client will not be able to send messages until it receives a message from

// the server

// Since they are sperate threads, the client can send messages and listen for

// messages at the same time

client.listenForMessage();

client.sendMessage();

}

}