public void readSetup() {
Config config = new Config("props/superD.properties");
Scanner in = new Scanner(System.in);
// prompt for algorithm
System.out.println("Which hashing algorithm would you like to use?");
System.out.println("We recommend SHA-256 for 32-bit machines and SHA-512 for 64-bit machines");
System.out.print("Enter your choice: ");
String input;
input = in.nextLine();
config.setConfig("HASH_ALGO", input, false);
// prompt for ROOT_DIL
System.out.println("Please enter a delimiter for root paths");
System.out.println("Try to use something that won't appear in the path");
System.out.println("We recommend ;;");
System.out.print("Enter choice: ");
input = in.nextLine();
config.setConfig("ROOT_DIL", input, false);
// PROMPT FOR DIRECTORIES
System.out.println(
"Please enter all directories you would like scanned on one line separated by " + input);
System.out.print("Please enter: ");
input = in.nextLine();
config.setConfig("ROOT", input, false);
}
public static void setup() {
// CREATE WALKER OBJECT
Walker walker = new Walker(BUFFER);
// load program properties
Config config = new Config("props/superD.properties");
// list of directories to scan
ArrayList<File> rootDirs = new ArrayList<File>(1);
// Load in all directories to scan from properties file into rootDirs ArrayList
try {
String dil = config.getConfig("ROOT_DIL");
String rootDirList = config.getConfig("ROOT");
List<String> rootDirListArr = Arrays.asList(rootDirList.split(dil));
for (int i = 0; i < rootDirListArr.size(); i++) {
rootDirs.add(new File(rootDirListArr.get(i)));
}
} catch (ConfigException e) {
log.fatal("Failed to read config file!", e);
}
// Call Walker.walk() on all the directories in rootDirs
try {
for (int i = 0; i < rootDirs.size(); i++) {
// make sure that it is a directory
if (rootDirs.get(i).isDirectory()) {
walker.walk(rootDirs.get(i));
} else {
log.debug(
rootDirs.get(i).toString()
+ " Appears to not be a directory; skipping to next in list");
}
}
// TODO replace generic Exception with specific exception(s)
} catch (Exception e) {
e.printStackTrace();
}
}
public void driver() {
// get instance of MilliTimer() for benchmarking
MilliTimer timer = new MilliTimer();
// start timer
timer.startTimer();
// get instance of other helper objects
Config config = new Config("props/superD.properties");
config.loadConfig("props/log4j.properties");
log.info("\n\n");
log.info("Starting program!");
// CREATE DATABASE
H2 db = null;
try { // TODO get rid of this try/catch -- it covers the entire method
// and swallows everything that does not catch. not good...
db = Database.getInstance();
// Create CHECKDUPES OBJ
CheckDupes check = new CheckDupes();
// CONNECT TO DATABASE
try {
db.openConnection();
} catch (ClassNotFoundException | SQLException e) {
log.fatal("Failed to open database connection! Check config file!", e);
}
// LOAD DATABASE TABLES
Schema s = new Schema();
String sqlSchema = s.getSchema();
PreparedStatement psSchema = db.getConnection().prepareStatement(sqlSchema);
try {
log.info("Running schema update on db: " + db.getDbPath());
psSchema.execute();
log.info("Schema update complete!");
} catch (Exception e) {
e.printStackTrace();
}
try {
db.closeConnection();
} catch (SQLException e) {
log.warn("Failed to close database connection!", e);
}
// INSTEAD OF COMMAND LINE ARGUMENTS,
// PROMPT FOR VARIOUS SETTINGS IF
// USER WANTS TO DO THIS INSTEAD OF
// PROP FILE
// TODO need at least 1 cmd line argument to signal if user pompt is wanted or not...
// TODO otherwise automated runs are not possible even with props file used.
Scanner in = new Scanner(System.in);
System.out.println("Would you like to read from prop file or enter options now?");
System.out.print("Enter 1 to enter configuration now, 2 to use existing prop file: ");
int choice = 2;
try {
choice = in.nextInt();
} catch (Exception e) { // TODO find specific exceptions thrown by Scanner(System.in)
System.out.println("Invalid input! Proceeding using property file"); // TODO log out
}
if (choice == 1) {
readSetup();
}
// END PROMPT FOR COMMAND LINE ARGUMENTS
// Run Setup() to do main logic, make calls to Walk()
setup();
// DATABASE now filled with all file hashes; time to look for duplicates
check.checkDupes();
// ALL DONE! stop timer
timer.stopTimer();
log.info("Total Runtime: " + timer.getTime());
} catch (
Exception
e) { // TODO get rid of this to narrow try/catch scope for improved exception
// handling/recovery. log out
e.printStackTrace();
}
}
