From a437c3298ac85c10e15645a48d38ad3347cca8be Mon Sep 17 00:00:00 2001
From: Nieraj Singh <nierajsingh@vmware.com>
Date: Fri, 16 Oct 2020 13:56:29 -0700
Subject: [PATCH] Added more methods to fetch list of libs and version for a
 Java project

---
 .../commons/java/SpringProjectUtil.java       | 62 ++++++++++++++++++-
 1 file changed, 60 insertions(+), 2 deletions(-)

diff --git a/headless-services/commons/commons-java/src/main/java/org/springframework/ide/vscode/commons/java/SpringProjectUtil.java b/headless-services/commons/commons-java/src/main/java/org/springframework/ide/vscode/commons/java/SpringProjectUtil.java
index 12b7bcfa1..eda145d58 100644
--- a/headless-services/commons/commons-java/src/main/java/org/springframework/ide/vscode/commons/java/SpringProjectUtil.java
+++ b/headless-services/commons/commons-java/src/main/java/org/springframework/ide/vscode/commons/java/SpringProjectUtil.java
@@ -11,6 +11,11 @@
 package org.springframework.ide.vscode.commons.java;
 
 import java.io.File;
+import java.util.List;
+import java.util.Optional;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+import java.util.stream.Collectors;
 
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -18,6 +23,8 @@ import org.slf4j.LoggerFactory;
 public class SpringProjectUtil {
 
 	public static final Logger log = LoggerFactory.getLogger(SpringProjectUtil.class);
+	
+	private static final Pattern MAJOR_MINOR_VERSION = Pattern.compile("(\\d+\\.)(\\d+)");
 
 	public static boolean isSpringProject(IJavaProject jp) {
 		return hasSpecificLibraryOnClasspath(jp, "spring-core", true);
@@ -30,8 +37,8 @@ public class SpringProjectUtil {
 	public static boolean hasBootActuators(IJavaProject jp) {
 		return hasSpecificLibraryOnClasspath(jp, "spring-boot-actuator-", true);
 	}
-
-	private static boolean hasSpecificLibraryOnClasspath(IJavaProject jp, String libraryNamePrefix, boolean onlyLibs) {
+	
+	public static boolean hasSpecificLibraryOnClasspath(IJavaProject jp, String libraryNamePrefix, boolean onlyLibs) {
 		try {
 			IClasspath cp = jp.getClasspath();
 			if (cp!=null) {
@@ -43,6 +50,57 @@ public class SpringProjectUtil {
 		return false;
 	}
 
+	public static File getLibraryOnClasspath(IJavaProject jp, String libraryNamePrefix) {
+		try {
+			IClasspath cp = jp.getClasspath();
+			if (cp!=null) {
+				boolean onlyLibs = true;
+				Optional<File> found = IClasspathUtil.getBinaryRoots(cp, (cpe) -> !cpe.isSystem()).stream().filter(cpe -> isEntry(cpe, libraryNamePrefix, onlyLibs)).findFirst();
+				if (found.isPresent()) {
+					return found.get();
+				}
+			}
+		} catch (Exception e) {
+			log.error("Failed to get library for project '" + jp.getElementName() + "' that start with prefix: " + libraryNamePrefix, e);
+		}
+		return null;
+	}
+	
+	public static String getMajMinVersion(IJavaProject jp, String libraryNamePrefix) {
+		if (jp != null) {
+			File libraryOnClasspath = getLibraryOnClasspath(jp, libraryNamePrefix);
+			if (libraryOnClasspath != null) {
+				String name = libraryOnClasspath.getName();
+				return getMajMinVersion(name);
+			}
+		}
+		return null;
+	}
+
+	public static String getMajMinVersion(String name) {
+		Matcher matcher = MAJOR_MINOR_VERSION.matcher(name);
+		if (matcher.find()) {
+			int start = matcher.start();
+			int end = matcher.end();
+			return name.substring(start, end);
+		}
+		return null;
+	}
+
+	public static List<File> getLibrariesOnClasspath(IJavaProject jp, String libraryNamePrefix) {
+		try {
+			IClasspath cp = jp.getClasspath();
+			if (cp!=null) {
+				boolean onlyLibs = true;
+				List<File> libs = IClasspathUtil.getBinaryRoots(cp, (cpe) -> !cpe.isSystem()).stream().filter(cpe -> isEntry(cpe, libraryNamePrefix, onlyLibs)).collect(Collectors.toList());
+				return libs;
+			}
+		} catch (Exception e) {
+			log.error("Failed to get list of libraries for project '" + jp.getElementName() + "' that start with prefix: " + libraryNamePrefix, e);
+		}
+		return null;
+	}
+
 	private static boolean isEntry(File cpe, String libNamePrefix, boolean onlyLibs) {
 		String name = cpe.getName();
 		return name.startsWith(libNamePrefix) && (!onlyLibs || name.endsWith(".jar"));