import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.net.Socket;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class ServerThread extends Thread {
	public static Socket socket;
	public static Scanner scanner;
	public static int size;
	public static long sizeR;
	public static String request;
	public static String requestM;
	public static String fileName;
	public static String fileLocation;
	public static Path path = null;
	public static String listString = "";

	ServerThread(Socket socket) {
		this.socket = socket;
	}

	public void run() {
		try {
			while (true) {
				InputStream is = socket.getInputStream();
				OutputStream os = socket.getOutputStream();
				PrintWriter pw = new PrintWriter(os, true);
				BufferedReader br = new BufferedReader(new InputStreamReader(is));
				String request = br.readLine();
				if (request != null) {
					System.out.println("Client request: " + request);
					processRequest(request, pw, is);
				}
			}
		} catch (IOException e) {
			e.printStackTrace();
		}
	}

	public void processRequest(String request, PrintWriter pw, InputStream is) throws IOException {
		String[] parts = request.split("-");
		requestM = parts[0];
		fileName = parts[1];
		String sizeS = parts[2];
		size = Integer.valueOf(sizeS);
		System.out.println("File" + fileName);

		if (requestM.equals("savefile")) {
			File f = new File("C:\\Users\\Pacia\\Desktop\\JAVA\\" + fileName);
//			File f = new File("C:\\Users\\Dominika\\Desktop\\JAVA2\\" + fileName);
			if(f.exists()){
			    System.out.println("file exists");
				pw.println("File exists.");
				pw.flush();
			}
			else{
				pw.println("notexist");
				pw.flush();
				receiveFile(fileLocation, size);
				System.out.println("File saved.");
				pw.println("File saved.");
				pw.flush();
			}

		} else if (requestM.equals("deletefile")) {
			deleteFiles(fileLocation);
			System.out.println("File deleted.");
			pw.println("File deleted.");
			pw.flush();
		}else if (requestM.equals("zipfile")) {
			zipFile(fileLocation);
			System.out.println("File ziped.");
			pw.println("File ziped.");
			pw.flush();
		}	else if (requestM.equals("showarchive")) {
			listOfArchive(new File("C:\\Users\\Pacia\\Desktop\\JAVA"));
	//		listOfArchive(new File("C:\\Users\\Dominika\\Desktop\\JAVA2"));
			System.out.println("List shown");
			pw.println("" + listString);
			pw.flush();
		} else if (requestM.equals("returnfile")) {
			sendSaveRequest(fileName);
			returnFile(fileLocation);
			System.out.println("File returned");
			pw.println("File returned.");
			pw.flush();
		} else {
			System.out.println("Option doesn't exist");
		}
	}

	private void sendSaveRequest(String fileName) throws IOException {
		fileLocation = ("C:\\Users\\Pacia\\Desktop\\JAVA\\" + fileName);
		File file = new File(fileLocation);
		sizeR = file.length();
		String sizeS = String.valueOf(sizeR);

		System.out.println("name:" + fileName);
		System.out.println("size:" + sizeS);

		OutputStream os = socket.getOutputStream();
		PrintWriter pw = new PrintWriter(os, true);
		pw.println(fileName + "-" + sizeS);
		pw.flush();
	}

	public void receiveFile(String fileLocation, int size) throws IOException {
		System.out.println("Saving file...");
		fileLocation = ("C:\\Users\\Pacia\\Desktop\\JAVA\\" + fileName);
		//fileLocation = ("C:\\Users\\Dominika\\Desktop\\JAVA2\\" + fileName);
		DataInputStream dis = new DataInputStream(socket.getInputStream());
		FileOutputStream fos = new FileOutputStream(fileLocation);
		byte[] buffer = new byte[size];

		System.out.println("rozmiar" + size);
		System.out.println("Nazwa" + fileLocation);
		int read = 0;
		int totalRead = 0;
		int remaining = size;
		try {
			while ((read = dis.read(buffer, 0, Math.min(buffer.length, remaining))) > 0) {
				totalRead += read;
				remaining -= read;
				fos.write(buffer, 0, read);
			}
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}

		fos.close();
		// dis.close();
	}

	public static boolean deleteFiles(String fileLocation) {
		System.out.println("Called deleteFiles");
//		fileLocation = ("C:\\Users\\Pacia\\Desktop\\JAVA\\" + fileName);
		fileLocation = ("C:\\Users\\Dominika\\Desktop\\JAVA2\\" + fileName);
		File file = new File(fileLocation);

		if (file.delete()) {
			System.out.println("File deleted successfully");
		} else {
			System.out.println("Failed to delete the file");
		}
		return true;
	}

	public static List<File> listOfArchive(File catalog) {
		System.out.println(catalog.getAbsoluteFile());
		List<File> listOfFiles = new ArrayList<File>();
		ArrayList<String> list = new ArrayList<String>();
		System.out.println("List of files in archive");

		if (catalog.isDirectory()) {
			String[] subNote = catalog.list();
			for (String filename : subNote) {
				listOfFiles.add(new File(catalog.getAbsoluteFile() + "//" + filename));
				System.out.println("" + filename);
				list.add("" + filename);
			}
		}
		for (String s : list) {
			listString += s + "\t";
		}
		return listOfFiles;
	}

	public static void returnFile(String fileLocation) throws IOException {

		FileInputStream fileInputStream = null;
		BufferedInputStream bufferedInputStream = null;
		OutputStream outputStream = null;

		try {
			System.out.println("Waiting for client to receiver...");
			try {
				int partCounter = 1;
				int sizeOfFiles = 1024 * 1024;// 1MB
				byte[] buffer = new byte[sizeOfFiles];
			//	fileLocation = ("C:\\Users\\Pacia\\Desktop\\JAVA\\" + fileName);
				fileLocation = ("C:\\Users\\Dominika\\Desktop\\JAVA2\\" + fileName);
				File file = new File(fileLocation);
//				fileName = file.getName();
				int count = 0;

				try (FileInputStream fis = new FileInputStream(file);
						BufferedInputStream bis = new BufferedInputStream(fis)) {
					int bytesAmount = 0;

					while ((bytesAmount = bis.read(buffer)) > 0) {
						String filePartName = String.format(fileName + "." + partCounter);
						File newFile = new File(file.getParent(), filePartName);
						try (FileOutputStream out = new FileOutputStream(newFile)) {
							out.write(buffer, 0, bytesAmount);
							count++;
						}
						byte[] byteArray = new byte[(int) newFile.length()];
						fileInputStream = new FileInputStream(newFile);
						bufferedInputStream = new BufferedInputStream(fileInputStream);
						bufferedInputStream.read(byteArray, 0, byteArray.length); // copied file into byteArray

						// sending file through socket
						outputStream = socket.getOutputStream();
						System.out.println("Sending " + fileLocation + "( size: " + byteArray.length + " bytes)");
						outputStream.write(byteArray, 0, byteArray.length);

						outputStream.flush();
						System.out.println("Done.");
						partCounter++;
					}

				}

			} finally {
				if (bufferedInputStream != null)
					bufferedInputStream.close();
				if (outputStream != null)
					bufferedInputStream.close();
			}
		} catch (IOException e) {
			e.printStackTrace();
		}
	}

	public boolean zipFile(String fileLocation) throws IOException {

	//	fileLocation = ("C:\\Users\\Dominika\\Desktop\\JAVA2\\" + fileName);
		fileLocation = ("C:\\Users\\Pacia\\Desktop\\JAVA\\" + fileName);
		File file = new File(fileLocation);
		FileInputStream in = new FileInputStream(file);

		// out put file
		String fileNameZip = "C:\\Users\\Dominika\\Desktop\\JAVA2\\" + file.getName() + ".zip";
		System.out.println("Plik zip: " + fileNameZip);
		ZipOutputStream out = new ZipOutputStream(new FileOutputStream(fileNameZip));

		// name the file inside the zip file
		out.putNextEntry(new ZipEntry(file.getName()));

		byte[] b = new byte[1024];
		int count;

		while ((count = in.read(b)) > 0) {
			out.write(b, 0, count);
		}
		out.close();
		in.close();
		return true;
	}
}