116 lines
3.8 KiB
Java
116 lines
3.8 KiB
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.SplitMergeTaskExecutor;
|
|
|
|
import java.io.File;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
public class MergeCli {
|
|
|
|
private final 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{
|
|
Runnable mergeTask = new SplitMergeTaskExecutor(
|
|
false,
|
|
getFilesFromStrings(),
|
|
saveTo);
|
|
Thread thread = new Thread(mergeTask);
|
|
thread.setDaemon(true);
|
|
thread.start();
|
|
thread.join();
|
|
}
|
|
private List<File> getFilesFromStrings(){
|
|
ArrayList<File> realFiles = new ArrayList<>();
|
|
for (String splitFileString : splitFiles){
|
|
realFiles.add(new File(splitFileString));
|
|
}
|
|
return realFiles;
|
|
}
|
|
}
|