Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion cli/detectors/detector.go
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
205 changes: 205 additions & 0 deletions cli/detectors/java.go
Original file line number Diff line number Diff line change
@@ -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 ""
}
145 changes: 145 additions & 0 deletions cli/detectors/java_test.go
Original file line number Diff line number Diff line change
@@ -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())
}
}
Loading
Loading