Merge pull request #1818 from codeflash-ai/fix/java-init-flow

fix: complete Java init flow (enum, detection, config)

Author: mohamedashrraf222

codeflash/cli_cmds/init_java.py

@@ -456,6 +456,7 @@ def configure_java_project(setup_info: JavaSetupInfo) -> bool:
     source_root = setup_info.module_root_override or detect_java_source_root(curdir)
     test_root = setup_info.test_root_override or detect_java_test_root(curdir)
 
+    config["language"] = "java"
     config["module-root"] = source_root
     config["tests-root"] = test_root
 

codeflash/cli_cmds/init_javascript.py

@@ -35,6 +35,7 @@ class ProjectLanguage(Enum):
     PYTHON = auto()
     JAVASCRIPT = auto()
     TYPESCRIPT = auto()
+    JAVA = auto()
 
 
 class JsPackageManager(Enum):
@@ -86,6 +87,12 @@ def detect_project_language(project_root: Path | None = None) -> ProjectLanguage
     """
     root = project_root or Path.cwd()
 
+    # Java detection (pom.xml or build.gradle is definitive)
+    has_pom = (root / "pom.xml").exists()
+    has_gradle = (root / "build.gradle").exists() or (root / "build.gradle.kts").exists()
+    if has_pom or has_gradle:
+        return ProjectLanguage.JAVA
+
     has_pyproject = (root / "pyproject.toml").exists()
     has_setup_py = (root / "setup.py").exists()
     has_package_json = (root / "package.json").exists()

tests/test_init_javascript.py

@@ -8,6 +8,8 @@
 
 from codeflash.cli_cmds.init_javascript import (
     JsPackageManager,
+    ProjectLanguage,
+    detect_project_language,
     determine_js_package_manager,
     get_package_install_command,
     should_modify_package_json_config,
@@ -344,3 +346,95 @@ def test_collect_js_setup_info_skip_confirm(self, tmp_project: Path, monkeypatch
         assert setup_info.module_root_override is None
         assert setup_info.formatter_override is None
         assert setup_info.git_remote == "origin"
+
+
+class TestDetectProjectLanguage:
+    """Tests for detect_project_language function."""
+
+    def test_detects_java_from_pom_xml(self, tmp_project: Path) -> None:
+        (tmp_project / "pom.xml").write_text("<project/>")
+
+        result = detect_project_language(tmp_project)
+
+        assert result == ProjectLanguage.JAVA
+
+    def test_detects_java_from_build_gradle(self, tmp_project: Path) -> None:
+        (tmp_project / "build.gradle").write_text("")
+
+        result = detect_project_language(tmp_project)
+
+        assert result == ProjectLanguage.JAVA
+
+    def test_detects_java_from_build_gradle_kts(self, tmp_project: Path) -> None:
+        (tmp_project / "build.gradle.kts").write_text("")
+
+        result = detect_project_language(tmp_project)
+
+        assert result == ProjectLanguage.JAVA
+
+    def test_detects_typescript_from_tsconfig(self, tmp_project: Path) -> None:
+        (tmp_project / "tsconfig.json").write_text("{}")
+
+        result = detect_project_language(tmp_project)
+
+        assert result == ProjectLanguage.TYPESCRIPT
+
+    def test_detects_javascript_from_package_json(self, tmp_project: Path) -> None:
+        (tmp_project / "package.json").write_text("{}")
+
+        result = detect_project_language(tmp_project)
+
+        assert result == ProjectLanguage.JAVASCRIPT
+
+    def test_detects_python_from_pyproject_toml(self, tmp_project: Path) -> None:
+        (tmp_project / "pyproject.toml").write_text("")
+
+        result = detect_project_language(tmp_project)
+
+        assert result == ProjectLanguage.PYTHON
+
+    def test_defaults_to_python_for_empty_directory(self, tmp_project: Path) -> None:
+        result = detect_project_language(tmp_project)
+
+        assert result == ProjectLanguage.PYTHON
+
+    def test_java_takes_priority_over_python(self, tmp_project: Path) -> None:
+        (tmp_project / "pom.xml").write_text("<project/>")
+        (tmp_project / "pyproject.toml").write_text("")
+
+        result = detect_project_language(tmp_project)
+
+        assert result == ProjectLanguage.JAVA
+
+    def test_java_takes_priority_over_javascript(self, tmp_project: Path) -> None:
+        (tmp_project / "build.gradle").write_text("")
+        (tmp_project / "package.json").write_text("{}")
+
+        result = detect_project_language(tmp_project)
+
+        assert result == ProjectLanguage.JAVA
+
+    def test_java_takes_priority_over_typescript(self, tmp_project: Path) -> None:
+        (tmp_project / "pom.xml").write_text("<project/>")
+        (tmp_project / "tsconfig.json").write_text("{}")
+
+        result = detect_project_language(tmp_project)
+
+        assert result == ProjectLanguage.JAVA
+
+    def test_javascript_with_js_indicators_over_python(self, tmp_project: Path) -> None:
+        (tmp_project / "package.json").write_text("{}")
+        (tmp_project / "pyproject.toml").write_text("")
+        (tmp_project / "node_modules").mkdir()
+
+        result = detect_project_language(tmp_project)
+
+        assert result == ProjectLanguage.JAVASCRIPT
+
+    def test_python_over_package_json_without_js_indicators(self, tmp_project: Path) -> None:
+        (tmp_project / "package.json").write_text("{}")
+        (tmp_project / "pyproject.toml").write_text("")
+
+        result = detect_project_language(tmp_project)
+
+        assert result == ProjectLanguage.PYTHON