package com.stockid;

import com.sun.tools.javac.util.List;

import java.awt.*;
import java.io.File;
import java.io.IOException;
import java.nio.file.*;
import java.util.*;
import java.util.function.Consumer;

public class Main {

    static Scanner scanner = new Scanner(System.in);

    static String directorySeparator;
    private static String currentDirectory;
    private static String rootDirectory;
    private static final String defPhotoExtension = ".jpg";
    private static final int defPass = 1234;
    private static int exitCode = 0;
    private static boolean canOpenFiles = false;

    private static HashMap<Integer, String> choices;
    private static HashMap<Integer, Consumer<String>> actions;

    private static boolean init() {
        choices = new HashMap<>();
        actions = new HashMap<>();

        OSValidator.dosplayOS();
        if (OSValidator.isWindows()) {
            directorySeparator = "\\";
        } else if (OSValidator.isUnix() || OSValidator.isMac()) {
            directorySeparator = "/";
        } else {
            System.out.println("Error: Your os is not supported by application.");
            return false;
        }

        currentDirectory = directorySeparator;
        rootDirectory = System.getProperty("user.dir") + directorySeparator + "root";
        File file = new File(rootDirectory);
        boolean bool = file.mkdir();
        if (bool) {
            System.out.println("Root directory created successfully");
        } else {
            System.out.println("Root directory already exist");
        }
        choices.put(exitCode++, "Display Files");
        actions.put(exitCode - 1, __ -> displayFiles(rootDirectory + currentDirectory));
        choices.put(exitCode++, "Open File");
        actions.put(exitCode - 1, __ -> openFile());
        choices.put(exitCode++, "Enter directory");
        actions.put(exitCode - 1, __ -> enterDirectory());
        choices.put(exitCode++, "Parent directory");
        actions.put(exitCode - 1, __ -> parentDirectory());
        choices.put(exitCode++, "Take a picture");
        actions.put(exitCode - 1, __ -> createFile());
        choices.put(exitCode++, "Create Folder");
        actions.put(exitCode - 1, __ -> createFolder());
        choices.put(exitCode++, "Delete");
        actions.put(exitCode - 1, __ -> deleteFile());
        choices.put(exitCode++, "Share File");
        actions.put(exitCode - 1, __ -> shareFile());
        choices.put(exitCode++, "Rename File");
        actions.put(exitCode - 1, __ -> renameFile());
        choices.put(exitCode++, "Upload Document");
        actions.put(exitCode - 1, __ -> uploadDocument());
        choices.put(exitCode++, "Enter a key");
        actions.put(exitCode - 1, __ -> enterKey());
        choices.put(exitCode, "Exit");
        return true;
    }

    public static void main(String[] args) {
        if (!init()) {
            return;
        }
        System.out.println(" ______       _________   ______       ______       ___   ___      ________      ______      \n"
                           + "/_____/\\     /________/\\ /_____/\\     /_____/\\     /___/\\/__/\\    /_______/\\    /_____/\\     \n"
                           + "\\::::_\\/_    \\__.::.__\\/ \\:::_ \\ \\    \\:::__\\/     \\::.\\ \\\\ \\ \\   \\__.::._\\/    \\:::_ \\ \\    \n"
                           + " \\:\\/___/\\      \\::\\ \\    \\:\\ \\ \\ \\    \\:\\ \\  __    \\:: \\/_) \\ \\     \\::\\ \\      \\:\\ \\ \\ \\   \n"
                           + "  \\_::._\\:\\      \\::\\ \\    \\:\\ \\ \\ \\    \\:\\ \\/_/\\    \\:. __  ( (     _\\::\\ \\__    \\:\\ \\ \\ \\  \n"
                           + "    /____\\:\\      \\::\\ \\    \\:\\_\\ \\ \\    \\:\\_\\ \\ \\    \\: \\ )  \\ \\   /__\\::\\__/\\    \\:\\/.:| | \n"
                           + "    \\_____\\/       \\__\\/     \\_____\\/     \\_____\\/     \\__\\/\\__\\/   \\________\\/     \\____/_/ \n"
                           + "                                                                                             ");
        int opCount = 0;
        while (pickAction(displayMenu())) {
            opCount++;
        }
        System.out.println("You have performed " + opCount + " operations.");
        scanner.close();
    }

    private static void println(String message) {
        System.out.println(message);
    }

    private static boolean notExistOrDirectory(File file){
        if (!file.exists())
        {
            println("Error: File " + file.getName()+ " does not exist");
            return true;
        }
        if (file.isDirectory())
        {
            println("Error: File " + file.getName()+ " is a directory");
            return true;
        }
        return false;
    }

    private static int countFiles(File file){
        int i = 0;
        File[] contents = file.listFiles();
        if (contents != null)
            for (File f : contents)
            {
                deleteDir(f);
                i++;
            }
        return i;
    }

    private static void deleteDir(File file) {
        File[] contents = file.listFiles();
        if (contents != null)
            for (File f : contents)
                deleteDir(f);
        if (!file.delete())
            println("Error: unable delete "+file.getName());
    }

    private static int displayMenu() {
        int choise = 0;
        do {
            System.out.println("\nList of available actions :");
            for (Integer key : choices.keySet()) {
                System.out.println(key + ") " + choices.get(key));
            }
            try {
                choise = Integer.parseInt(scanner.next());
            } catch (Exception e) {
                System.out.println("Error: Integer expected.\n");
            }
        } while (choise > exitCode);
        return choise;
    }

    private static boolean pickAction(int userChoise) {
        System.out.print("You have chosen : ");
        System.out.println(choices.get(userChoise));
        if (userChoise == exitCode) {
            return false;
        }
        if (actions.get(userChoise) != null) {
            actions.get(userChoise).accept(currentDirectory);
        } else {
            System.out.println("Error: Chosen function has not been implemented yet.\n");
        }
        return true;
    }

    private static void displayFiles(String path) {
        File file = new File(path);
        if (!file.isDirectory()) {
            System.out.println("Error, " + new File(path).getName() + " is not a directory\n");
            return;
        }
        System.out.println("\nPath: " + currentDirectory + "\nFile list:\n".replace(directorySeparator, "/"));
        File[] files = file.listFiles();
        for (File f : Objects.requireNonNull(files)) {
            String fName = "\t" + f.getName();
            if (f.isDirectory()) {
                fName = "d" + fName;
            } else if (f.isFile() && f.getName().contains(".")){
                fName = "f" + fName;
            } else
                fName = "?" + fName;
            System.out.println(fName);
        }
        System.out.println();
    }

    private static void openFile() {
        if(!Desktop.isDesktopSupported())
        {
            println("Error: Desktop is not supported");
            return;
        }
        int res = enterKey();
        while (res==2)
            res = enterKey();
        if(res == -1)
            return;
        displayFiles(rootDirectory+currentDirectory);
        println("Enter file name:");
        String fileName = scanner.next();
        File file = new File(rootDirectory+currentDirectory+fileName);
        if(notExistOrDirectory(file))
            return;
        try {
            Desktop.getDesktop().open(file);
        } catch (IOException e) {
            println("Error: Unable open " + file.getName());
        }
    }

    private static void enterDirectory() {
        System.out.println("Enter directory to enter: ");
        String dirName = scanner.next();
        File file = new File(rootDirectory + currentDirectory + dirName);
        if (!file.exists()) {
            System.out.println("Error: Directory " + dirName + " does not exist.\n");
            return;
        }
        currentDirectory += dirName + directorySeparator;
    }

    private static void parentDirectory() {
        currentDirectory = removeLastDirectory(currentDirectory, directorySeparator);
        displayFiles(rootDirectory + currentDirectory);
    }

    private static void createFile() {
        System.out.print("Pick a name for your new document: ");
        String fileName = scanner.next() + defPhotoExtension;
        File file = new File(rootDirectory, currentDirectory + directorySeparator + fileName);
        if (file.exists()) {
            System.out.println("Error: File already exist");
            return;
        }
        try {
            if (file.createNewFile()) {
                System.out.println("File " + fileName + " created with success");
            }
        } catch (IOException e) {
            System.out.println("Error: unable to create " + fileName);
        }
    }

    private static void createFolder() {
        System.out.println("Enter the name of the desired directory: ");
        String path = rootDirectory + currentDirectory + scanner.next();
        File file = new File(path);
        boolean bool = file.mkdir();
        if (bool) {
            System.out.println("Directory created successfully\n");
        } else {
            System.out.println("Directory already exist");
        }
        displayFiles(file.getParent());
    }

    private static void deleteFile() {
        System.out.print("Enter file name :");
        String fileName = scanner.next();
        File file = new File(rootDirectory + currentDirectory + fileName);
        if (!file.exists()) {
            System.out.println("Error: File " + fileName + " does not exist");
            return;
        }
        if (file.isDirectory()) {
            println("File " + fileName + " is a directory,\nAre you sure you want to delete it along with "
                    + countFiles(file) + " objects inside? (yes/n)");
            String answer = "";
            while (!answer.equals("yes") && !answer.equals("n")) {
                answer = scanner.next();
            }
            if (answer.equals("n")) {
                println("Aborting");
                return;
            }
            deleteDir(file);
            return;
        }
        if (!file.delete()) {
            println("Error: unable to delete " + fileName);
        }
    }

    private static void shareFile() {
        displayFiles(rootDirectory + currentDirectory);
        System.out.println("Chose a file");
        Share.document(rootDirectory + currentDirectory + directorySeparator + scanner.next());
    }

    private static void renameFile() {
        println("Enter file name:");
        File file = new File(rootDirectory+currentDirectory+scanner.next());
        if(!file.exists())
        {
            println("Error: File " + file.getName()+ " does not exist");
            return;
        }
        println("Enter new file name:");
        String fileName = scanner.next();
        while (new File(rootDirectory+currentDirectory+fileName).exists()){
            println("Error, "+fileName+" already exist, choose another name");
            fileName = scanner.next();
        }
        if(!file.renameTo(new File(rootDirectory+currentDirectory+fileName)))
            println("Error, renaming "+file.getName()+" to "+fileName);
    }

    private static void uploadDocument() {
        System.out.print("Enter file path :");
        String filePath = scanner.next();
        File file = new File(filePath);
        if(notExistOrDirectory(file))
            return;
        String fileName = file.getName();
        try {
            while (new File(rootDirectory + currentDirectory + directorySeparator+fileName).exists()){
                println("File "+fileName+" already exists\n"
                        + "Would you like to :\n"
                        + "1) Rename it\n"
                        + "2) Replace it\n"
                        + "3) Abandon");
                int choise = requestInt();
                if (choise==3)
                    return;
                if (choise==1) {
                    println("Enter a new name:");
                    fileName = scanner.next();
                }
            }
            Files.copy(Paths.get(file.getPath()), Paths.get(rootDirectory + currentDirectory + directorySeparator+fileName), StandardCopyOption.REPLACE_EXISTING);
        } catch (IOException e) {
            println("Error: Unable to upload " + file.getName());
        }
    }

    private static int enterKey() {
        if (canOpenFiles)
            return 1;
        println("Enter key code (4 digits, -1 to abandon):");
        int pass = requestInt();
        if (pass == -1)
            return pass;
        if(pass == defPass){
            canOpenFiles = true;
            return 1;
        }else{
            println("Key code error");
            return 2;
        }
    }

    public static Integer requestInt(){
        Integer choise = null;
        while (choise==null){
            try {
                choise = Integer.parseInt(scanner.next());
            } catch (Exception e) {
                System.out.println("Error: Integer expected.\n");
            }
        }
        return choise;
    }

    public static String removeLastDirectory(String path, String directorySeparator)
    {
        String sep = "/";
        if (directorySeparator.equals("\\"))
            sep = "\\\\";
        ArrayList<String>dirs = new ArrayList<>(Arrays.asList(path.split(sep)));
        if(dirs.isEmpty())
        {
            System.out.println("You are at root directory.\n");
        }else {
            System.out.println(dirs.toString());
            System.out.println("separator "+directorySeparator);
            System.out.println("path "+path);

            dirs.remove(dirs.size()-1);
        }
        return dirs.stream().reduce((s1, s2)-> s1 + directorySeparator + s2).orElse("") + directorySeparator;
    }

}

class OSValidator {

    private static String OS = System.getProperty("os.name").toLowerCase();

    static void dosplayOS() {

        System.out.println(OS);

        if (isWindows()) {
            System.out.println("This is Windows");
        } else if (isMac()) {
            System.out.println("This is Mac");
        } else if (isUnix()) {
            System.out.println("This is Unix or Linux");
        } else if (isSolaris()) {
            System.out.println("This is Solaris");
        } else {
            System.out.println("Your OS is not support!!");
        }
    }

    static boolean isWindows() {

        return (OS.contains("win"));

    }

    static boolean isMac() {

        return (OS.contains("mac"));

    }

    static boolean isUnix() {

        return (OS.contains("nix") || OS.contains("nux") || OS.indexOf("aix") > 0 );

    }

    private static boolean isSolaris() {

        return (OS.contains("sunos"));

    }

}

class Share {
    private static ArrayList<String> medias = new ArrayList<>(List.of("Facebook","Instaram", "Mail","Gmail", "SMS", "Go back"));
    static void document(String path)
    {
        int choise = -1;
        File file = new File(path);
        String fileName = new File(path).getName();
        if (!file.exists())
        {
            System.out.println("Error: File " + fileName +" does not exist.\n");
            return;
        }
        if(file.isDirectory())
        {
            System.out.println("Error: File " + fileName+" is a directory.\n");
            return;
        }
        System.out.println("Sharing " + fileName+"\nPick a Media:");
        for (String s : medias) {
            System.out.println(medias.indexOf(s)+") " + s);
        }
        while (choise>medias.size() || choise < 0){
            try {
                choise = Integer.parseInt(Main.scanner.next());
                if(choise>medias.size() || choise < 0){
                    System.out.println("Error: option "+choise+" does not exist\n");
                }
            } catch (Exception e) {
                System.out.println("Error: Integer Expected.\n");
            }
        }
        if (choise!= medias.size()-1)
            System.out.println("Shared "+fileName+" on "+medias.get(choise)+" with success.");
        else
            System.out.println("Aborting");
    }
}