ns-usbloader/src/main/java/nsusbloader/cli/MergeCli.java

/*
    Copyright 2019-2020 Dmitry Isaenko

    This file is part of NS-USBloader.

    NS-USBloader is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    NS-USBloader is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with NS-USBloader.  If not, see <https://www.gnu.org/licenses/>.
*/
package nsusbloader.cli;

import nsusbloader.Utilities.splitmerge.MergeTask;
import nsusbloader.Utilities.splitmerge.SplitTask;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class MergeCli {

    private String[] arguments;
    private String saveTo;
    private String[] splitFiles;

    MergeCli(String[] arguments) throws InterruptedException, IncorrectSetupException{
        this.arguments = arguments;
        checkArguments();
        parseArguments();
        printFilesForMerge();
        runBackend();
    }

    private void checkArguments() throws IncorrectSetupException{
        if (arguments == null || arguments.length == 0) {
            throw new IncorrectSetupException("No arguments.\n" +
                    "Try 'ns-usbloader -m help' for more information.");
        }

        if (arguments.length == 1){
            if (isHelpDirective(arguments[0])){
                showHelp();
                return;
            }

            throw new IncorrectSetupException("Not enough arguments.\n"
                    + "Try 'ns-usbloader -m help' for more information.");
        }

        this.saveTo = arguments[0];
        File saveToFile = new File(saveTo);
        if (! saveToFile.exists() || saveToFile.isFile()){
            throw new IncorrectSetupException("First argument must be existing directory.");
        }
    }

    private boolean isHelpDirective(String argument){
        return argument.equals("help");
    }

    private void showHelp() throws IncorrectSetupException{
        throw new IncorrectSetupException("Usage:\n"
                + "\tns-usbloader -m <SAVE_TO_DIR> <SPLIT_FILE>...\n"
                + "\n\nOptions:"
                + "\n\tSAVE_TO_DIR\tWhere results should be saved"
                + "\n\tSPLIT_FILE\tOne or more split-files (folders) to merge");
    }

    private void parseArguments() throws IncorrectSetupException{
        List<String> files = new ArrayList<>();
        for (int i = 1; i < arguments.length; i++){
            File file = new File(arguments[i]);
            if (file.isDirectory())
                files.add(file.getAbsolutePath());
        }

        if (files.isEmpty()){
            throw new IncorrectSetupException("No files specified.\n" +
                    "Try 'ns-usbloader -m help' for more information.");
        }

        this.splitFiles = files.toArray(new String[0]);
    }

    private void printFilesForMerge(){
        System.out.println("Next files will be merged:");
        for (String f : this.splitFiles)
            System.out.println("  "+f);
    }

    private void runBackend() throws InterruptedException{
        for (String filePath : splitFiles){
            Runnable mergeTask = new MergeTask(filePath, saveTo);
            Thread thread = new Thread(mergeTask);
            thread.setDaemon(true);
            thread.start();
            thread.join();
        }
    }
}