improve gitUtil

Author: maxxbw

lib/utils/FileUtil.java

@@ -155,20 +155,20 @@ public static Boolean copyFile2File(File srcFile, File dstFile) {
 		return true;
 	}
 
-	public static boolean deleteDirectory(String dirPath) {
-		File directoryToBeDeleted = new File(dirPath);
+	public static boolean deleteDirectory(Path dirPath) {
+		File directoryToBeDeleted = dirPath.toFile();
 		File[] allContents = directoryToBeDeleted.listFiles();
 		if (allContents != null) {
 			for (File file : allContents) {
-				deleteDirectory(file.getAbsolutePath());
+				deleteDirectory(file.toPath());
 			}
 		}
 		return directoryToBeDeleted.delete();
 	}
 
-	public static void copyDirectory(String srcDirStr, String destDirStr) {
-		File source = new File(srcDirStr);
-		File dest = new File(destDirStr);
+	public static void copyDirectory(Path srcDirPath, Path destDirPath) {
+		File source = srcDirPath.toFile();
+		File dest = destDirPath.toFile();
 		try {
 			FileUtils.copyDirectory(source, dest);
 		} catch (IOException e) {
@@ -263,7 +263,7 @@ public static List<String> readFileToLineList(String filePath) {
 	public static void createFolder(String dirStr, Boolean deleteIfExist) {
 		File dir = new File(dirStr);
 		if (deleteIfExist && dir.exists()) {
-			deleteDirectory(dir.getAbsolutePath());
+			deleteDirectory(dir.toPath());
 			dir.mkdirs();
 		} else if (!dir.exists()) {
 			dir.mkdirs();
@@ -280,11 +280,11 @@ public static boolean deleteDirectory(File directoryToBeDeleted) {
 		return directoryToBeDeleted.delete();
 	}
 
-	public static List<File> findFilePathofSpecifcTypeRecusive(String tarDir, String extension) {
+	public static List<File> findFilePathofSpecifcTypeRecusive(Path tarDir, String extension) {
 
 		List<File> fileList = new ArrayList<File>();
 		try {
-			Files.walk(Paths.get(tarDir)).filter(Files::isRegularFile).forEach((f) -> {
+			Files.walk(tarDir).filter(Files::isRegularFile).forEach((f) -> {
 				String filepath = f.toString();
 				if (filepath.endsWith(extension))
 					// System.out.println(file + " found!");

lib/utils/GitUtil.java

@@ -17,7 +17,7 @@ public static Boolean clone(String repoName, String usrName, Path targetDir) {
 		}
 		targetDir.toFile().mkdirs();
 
-		String cmd = "timeout 300 git clone https://github.com/" + repoName + "/" + usrName + " " + targetDir;
+		String cmd = "timeout 600 git clone https://github.com/" + repoName + "/" + usrName + " " + targetDir;
 
 		ProcessUtil.ProcessReporter pr = ProcessUtil.executeCMD(cmd, null, targetDir, 0);
 		if (pr.exitCode == 0) {
@@ -169,13 +169,18 @@ public static Boolean checkout(Path repoDir, String com, Boolean ifForce) {
 		ProcessUtil.ProcessReporter pr = new ProcessUtil.ProcessReporter();
 
 		if (ifForce) {
-			String resetCMD = "timeout 300 git reset --hard";
+			String resetCMD = "timeout 600 git reset --hard";
 			pr = ProcessUtil.executeCMD(resetCMD, null, repoDir, 0);
 		}
 
+		String defaultBranchNameString = getDefaultBranch(repoDir);
+		if (defaultBranchNameString == null || defaultBranchNameString == "") {
+			defaultBranchNameString = "master";
+		}
+
 		String checkoutCMD = null;
 		if (com == null) { // checkout latest if null
-			checkoutCMD = "timeout 300 git checkout master";
+			checkoutCMD = "timeout 300 git checkout " + defaultBranchNameString;
 		} else {
 			checkoutCMD = "timeout 300 git checkout " + com;
 		}
@@ -199,4 +204,13 @@ public static Boolean checkout(Path repoDir, String com, Boolean ifForce) {
 		}
 	}
 
+	private static String getDefaultBranch(Path gitDirPath) {
+		// TODO Auto-generated method stub
+		String cmdString = "git symbolic-ref refs/remotes/origin/HEAD";
+		ProcessUtil.ProcessReporter pr = ProcessUtil.executeCMD(cmdString, null, gitDirPath, 0);
+		// output refs/remotes/origin/master
+		return pr.out.trim().split("refs/remotes/origin/")[1].trim();
+
+	}
+
 }