Merge pull request #5 from EdgarPsda/v0.5.0/python-java-detection

Add Python and Java project detection

Author: EdgarPsda

cli/detectors/detector.go

@@ -30,7 +30,8 @@ func DetectProject(dir string) (*ProjectInfo, error) {
 	detectors := []Detector{
 		&NodeDetector{},
 		&GoDetector{},
-		// Add more detectors here in the future
+		&PythonDetector{},
+		&JavaDetector{},
 	}
 
 	var bestMatch *ProjectInfo

cli/detectors/java.go

@@ -0,0 +1,205 @@
+// cli/detectors/java.go
+package detectors
+
+import (
+	"bufio"
+	"errors"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+type JavaDetector struct {
+	confidence int
+}
+
+func (d *JavaDetector) Detect(dir string) (*ProjectInfo, error) {
+	// Check for Java project files
+	packageFiles := []string{
+		"pom.xml",
+		"build.gradle",
+		"build.gradle.kts",
+	}
+
+	var foundFile string
+	for _, f := range packageFiles {
+		if fileExists(filepath.Join(dir, f)) {
+			foundFile = f
+			break
+		}
+	}
+
+	if foundFile == "" {
+		d.confidence = 0
+		return nil, errors.New("no Java project file found")
+	}
+
+	deps := collectJavaDeps(filepath.Join(dir, foundFile), foundFile)
+	framework := detectJavaFramework(deps, dir, foundFile)
+
+	d.confidence = 95
+
+	return &ProjectInfo{
+		Language:     "java",
+		Framework:    framework,
+		PackageFile:  foundFile,
+		RootDir:      dir,
+		Dependencies: deps,
+	}, nil
+}
+
+func (d *JavaDetector) Confidence() int {
+	return d.confidence
+}
+
+func detectJavaFramework(deps []string, dir, packageFile string) string {
+	// Check dependencies for known frameworks
+	for _, dep := range deps {
+		lower := strings.ToLower(dep)
+		switch {
+		case strings.Contains(lower, "spring-boot"):
+			return "spring-boot"
+		case strings.Contains(lower, "quarkus"):
+			return "quarkus"
+		case strings.Contains(lower, "micronaut"):
+			return "micronaut"
+		case strings.Contains(lower, "jakarta.ee") || strings.Contains(lower, "javax.servlet"):
+			return "jakarta-ee"
+		case strings.Contains(lower, "dropwizard"):
+			return "dropwizard"
+		}
+	}
+
+	// Also check the file content directly for Spring Boot parent POM
+	if packageFile == "pom.xml" {
+		data, err := os.ReadFile(filepath.Join(dir, packageFile))
+		if err == nil {
+			content := strings.ToLower(string(data))
+			if strings.Contains(content, "spring-boot-starter-parent") ||
+				strings.Contains(content, "spring-boot-starter") {
+				return "spring-boot"
+			}
+		}
+	}
+
+	return ""
+}
+
+func collectJavaDeps(path, packageFile string) []string {
+	switch {
+	case packageFile == "pom.xml":
+		return parsePomXML(path)
+	case strings.HasPrefix(packageFile, "build.gradle"):
+		return parseGradle(path)
+	}
+	return nil
+}
+
+// parsePomXML does a minimal parse to extract dependency artifact IDs from pom.xml
+func parsePomXML(path string) []string {
+	f, err := os.Open(path)
+	if err != nil {
+		return nil
+	}
+	defer f.Close()
+
+	var deps []string
+	scanner := bufio.NewScanner(f)
+	inDependency := false
+	var currentGroup, currentArtifact string
+
+	for scanner.Scan() {
+		line := strings.TrimSpace(scanner.Text())
+
+		if strings.Contains(line, "<dependency>") {
+			inDependency = true
+			currentGroup = ""
+			currentArtifact = ""
+			continue
+		}
+
+		if strings.Contains(line, "</dependency>") {
+			if inDependency && (currentGroup != "" || currentArtifact != "") {
+				dep := currentGroup
+				if currentArtifact != "" {
+					if dep != "" {
+						dep += ":"
+					}
+					dep += currentArtifact
+				}
+				deps = append(deps, dep)
+			}
+			inDependency = false
+			continue
+		}
+
+		if inDependency {
+			if groupID := extractXMLValue(line, "groupId"); groupID != "" {
+				currentGroup = groupID
+			}
+			if artifactID := extractXMLValue(line, "artifactId"); artifactID != "" {
+				currentArtifact = artifactID
+			}
+		}
+	}
+
+	return deps
+}
+
+// parseGradle does a minimal parse to extract dependencies from build.gradle
+func parseGradle(path string) []string {
+	f, err := os.Open(path)
+	if err != nil {
+		return nil
+	}
+	defer f.Close()
+
+	var deps []string
+	scanner := bufio.NewScanner(f)
+
+	for scanner.Scan() {
+		line := strings.TrimSpace(scanner.Text())
+
+		// Match patterns like: implementation 'group:artifact:version'
+		// or: implementation "group:artifact:version"
+		for _, keyword := range []string{"implementation", "api", "compileOnly", "runtimeOnly", "testImplementation"} {
+			if strings.HasPrefix(line, keyword+" ") || strings.HasPrefix(line, keyword+"(") {
+				dep := extractGradleDep(line)
+				if dep != "" {
+					deps = append(deps, dep)
+				}
+			}
+		}
+	}
+
+	return deps
+}
+
+// extractXMLValue extracts the text content of a simple XML element
+func extractXMLValue(line, tag string) string {
+	openTag := "<" + tag + ">"
+	closeTag := "</" + tag + ">"
+
+	startIdx := strings.Index(line, openTag)
+	endIdx := strings.Index(line, closeTag)
+
+	if startIdx >= 0 && endIdx > startIdx {
+		return strings.TrimSpace(line[startIdx+len(openTag) : endIdx])
+	}
+	return ""
+}
+
+// extractGradleDep extracts the dependency coordinate from a Gradle dependency line
+func extractGradleDep(line string) string {
+	// Find the quoted string (single or double)
+	for _, quote := range []string{"'", "\""} {
+		start := strings.Index(line, quote)
+		if start >= 0 {
+			end := strings.Index(line[start+1:], quote)
+			if end >= 0 {
+				return line[start+1 : start+1+end]
+			}
+		}
+	}
+	return ""
+}

cli/detectors/java_test.go

@@ -0,0 +1,145 @@
+package detectors
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+)
+
+func TestJavaDetector_Detect_SpringBoot_Maven(t *testing.T) {
+	dir := t.TempDir()
+
+	pomXML := `<?xml version="1.0" encoding="UTF-8"?>
+<project>
+    <parent>
+        <groupId>org.springframework.boot</groupId>
+        <artifactId>spring-boot-starter-parent</artifactId>
+        <version>3.2.0</version>
+    </parent>
+    <dependencies>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-web</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-data-jpa</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.postgresql</groupId>
+            <artifactId>postgresql</artifactId>
+        </dependency>
+    </dependencies>
+</project>`
+
+	if err := os.WriteFile(filepath.Join(dir, "pom.xml"), []byte(pomXML), 0o644); err != nil {
+		t.Fatalf("failed to write pom.xml: %v", err)
+	}
+
+	d := &JavaDetector{}
+	info, err := d.Detect(dir)
+	if err != nil {
+		t.Fatalf("Detect returned error: %v", err)
+	}
+
+	if info.Language != "java" {
+		t.Errorf("expected Language=java, got %s", info.Language)
+	}
+	if info.Framework != "spring-boot" {
+		t.Errorf("expected Framework=spring-boot, got %s", info.Framework)
+	}
+	if info.PackageFile != "pom.xml" {
+		t.Errorf("expected PackageFile=pom.xml, got %s", info.PackageFile)
+	}
+	if d.Confidence() <= 0 {
+		t.Errorf("expected confidence > 0, got %d", d.Confidence())
+	}
+	if len(info.Dependencies) != 3 {
+		t.Errorf("expected 3 dependencies, got %d", len(info.Dependencies))
+	}
+}
+
+func TestJavaDetector_Detect_Gradle(t *testing.T) {
+	dir := t.TempDir()
+
+	buildGradle := `plugins {
+    id 'java'
+    id 'org.springframework.boot' version '3.2.0'
+}
+
+dependencies {
+    implementation 'org.springframework.boot:spring-boot-starter-web:3.2.0'
+    implementation 'org.springframework.boot:spring-boot-starter-data-jpa:3.2.0'
+    testImplementation 'org.springframework.boot:spring-boot-starter-test:3.2.0'
+    runtimeOnly 'org.postgresql:postgresql:42.7.0'
+}
+`
+
+	if err := os.WriteFile(filepath.Join(dir, "build.gradle"), []byte(buildGradle), 0o644); err != nil {
+		t.Fatalf("failed to write build.gradle: %v", err)
+	}
+
+	d := &JavaDetector{}
+	info, err := d.Detect(dir)
+	if err != nil {
+		t.Fatalf("Detect returned error: %v", err)
+	}
+
+	if info.Language != "java" {
+		t.Errorf("expected Language=java, got %s", info.Language)
+	}
+	if info.Framework != "spring-boot" {
+		t.Errorf("expected Framework=spring-boot, got %s", info.Framework)
+	}
+	if info.PackageFile != "build.gradle" {
+		t.Errorf("expected PackageFile=build.gradle, got %s", info.PackageFile)
+	}
+	if len(info.Dependencies) != 4 {
+		t.Errorf("expected 4 dependencies, got %d", len(info.Dependencies))
+	}
+}
+
+func TestJavaDetector_Detect_Quarkus(t *testing.T) {
+	dir := t.TempDir()
+
+	pomXML := `<?xml version="1.0" encoding="UTF-8"?>
+<project>
+    <dependencies>
+        <dependency>
+            <groupId>io.quarkus</groupId>
+            <artifactId>quarkus-resteasy</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>io.quarkus</groupId>
+            <artifactId>quarkus-hibernate-orm</artifactId>
+        </dependency>
+    </dependencies>
+</project>`
+
+	if err := os.WriteFile(filepath.Join(dir, "pom.xml"), []byte(pomXML), 0o644); err != nil {
+		t.Fatalf("failed to write pom.xml: %v", err)
+	}
+
+	d := &JavaDetector{}
+	info, err := d.Detect(dir)
+	if err != nil {
+		t.Fatalf("Detect returned error: %v", err)
+	}
+
+	if info.Framework != "quarkus" {
+		t.Errorf("expected Framework=quarkus, got %s", info.Framework)
+	}
+}
+
+func TestJavaDetector_Detect_NoProject(t *testing.T) {
+	dir := t.TempDir()
+
+	d := &JavaDetector{}
+	_, err := d.Detect(dir)
+	if err == nil {
+		t.Error("expected error for empty directory, got nil")
+	}
+	if d.Confidence() != 0 {
+		t.Errorf("expected confidence=0, got %d", d.Confidence())
+	}
+}