detect build system

Author: nim65s

README.md

@@ -26,7 +26,7 @@ options:
 
 ## Example
 
-For the [eigenpy](https://github.com/stack-of-tasks/eigenpy) project:
+For the [eigenpy](https://github.com/stack-of-tasks/eigenpy) project, you can add a `dockgen.toml` with:
 ```toml
 [jrl-cmakemodules]
 url = "github:jrl-umi3218"

dockgen/forge.py

@@ -30,12 +30,12 @@ def __init__(
         self.headers = {}
         self.org = None
         self.name = name
+        self.dir = args.work_dir / self.name
         self.slug = (
             "dot"
             if self.forge_type == ForgeType.Dot
             else str(self.forge_type).removesuffix(":")
         )
-
         getattr(self, f"init_{self.slug}")()
 
     def init_github(self):
@@ -45,7 +45,6 @@ def init_github(self):
         self.headers = {"Accept": "application/vnd.github+json"}
         if self.args.token:
             self.headers["Authorization"] = f"Bearer {self.args.token}"
-        self.dir = self.args.work_dir / self.name
         self.dir.mkdir(parents=True, exist_ok=True)
 
         self.api_url = f"https://api.github.com/repos/{self.org}/{self.name}"
@@ -59,7 +58,6 @@ def init_http(self):
         if not self.name:
             err = f"name must be set for {self.url}"
             raise AttributeError(err)
-        self.dir = self.args.work_dir / self.name
         self.tarball = self.url
         if self.dir.exists():
             logger.info("%s already exists, skipping download", self.dir)

dockgen/project.py

@@ -1,13 +1,14 @@
 """Definition of a project to build."""
 
 from enum import StrEnum
+from pathlib import Path
 from tomllib import load
 from typing import TYPE_CHECKING
 
 from .forge import Forge, ForgeType
 
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from .__main__ import Dockgen
 
 
@@ -38,6 +39,7 @@ def __init__(
         apt_deps: list[str] | None = None,
         src_deps: list[str] | None = None,
     ):
+        self.dockgen = dockgen
         self.name = name
         self.url = url
         for forge_type in ForgeType:
@@ -51,28 +53,44 @@ def __init__(
         self.org = org or self.forge.org
         self.version = version or self.forge.version
         self.tarball = tarball or self.forge.tarball
-        self.build_systems = [BuildSystem[b] for b in (build_systems or ["CMake"])]
+        self.build_systems = [BuildSystem[b] for b in (build_systems or self.detect())]
         self.apt_deps = set(apt_deps or [])
         self.src_deps = set(src_deps or [])
 
         if self.url != ".":
             if upstream_dockgen := self.forge.get_file("dockgen.toml"):
-                with upstream_dockgen.open("rb") as f:
-                    for name, conf in load(f).items():
-                        if name == self.name:
-                            tgt = self
-                        elif name in dockgen.projects:
-                            tgt = dockgen.projects[name]
-                        else:
-                            dockgen.projects[name] = Project(
-                                dockgen=dockgen, name=name, **conf
-                            )
-                            dockgen.projects_order.append(name)
-                            continue
+                self.upstream(upstream_dockgen)
 
-                        for k, v in conf.items():
-                            match k:
-                                case "apt_deps":
-                                    tgt.apt_deps |= set(v)
-                                case "src_deps":
-                                    tgt.src_deps |= set(v)
+    def detect(self) -> list[str]:
+        """
+        Detect build systems
+        """
+        return [
+            build_system.name
+            for build_system in BuildSystem
+            if self.forge.get_file(build_system) is not None
+        ]
+
+    def upstream(self, upstream_dockgen: Path):
+        """
+        Process data from upstream dockgen.toml
+        """
+        with upstream_dockgen.open("rb") as f:
+            for name, conf in load(f).items():
+                if name == self.name:
+                    tgt = self
+                elif name in self.dockgen.projects:
+                    tgt = self.dockgen.projects[name]
+                else:
+                    self.dockgen.projects[name] = Project(
+                        dockgen=self.dockgen, name=name, **conf
+                    )
+                    self.dockgen.projects_order.append(name)
+                    continue
+
+                for k, v in conf.items():
+                    match k:
+                        case "apt_deps":
+                            tgt.apt_deps |= set(v)
+                        case "src_deps":
+                            tgt.src_deps |= set(v)

tests/tests.py

@@ -68,7 +68,7 @@ def test_help(self):
             import coverage  # noqa: F401
 
             exe = ["coverage", "run"]
-        except ImportError:
+        except ImportError:  # pragma: no cover
             pass
         output = check_output([*exe, "-m", "dockgen", "-h"], text=True)
         self.assertIn("Generate fresh docker images", output)