Use the ert plugin mechanism to install forward models

pyproject.toml

@@ -64,6 +64,9 @@ docs = [
 [project.entry-points."everest"]
 everest-models = "everest_models.everest_hooks"
 
+[project.entry-points."ert"]
+everest_models_forward_models = "everest_models.forward_models"
+
 [project.scripts]
 fm_add_templates = "everest_models.jobs.fm_add_templates.cli:main_entry_point"
 fm_drill_date_planner = "everest_models.jobs.fm_drill_date_planner.cli:main_entry_point"

src/everest_models/everest_hooks.py

@@ -7,7 +7,6 @@
 
 import logging
 import pathlib
-import sys
 from importlib import import_module, resources
 from typing import Any, Dict, List, Sequence, Type
 
@@ -24,9 +23,7 @@
 
 logger = logging.getLogger(__name__)
 
-FORWARD_MODEL_DIR = "forward_models"
-PACKAGE = "everest_models"
-JOBS = f"{PACKAGE}.jobs"
+JOBS = "everest_models.jobs"
 
 
 def _get_jobs():
@@ -42,16 +39,7 @@ def get_forward_models() -> List[Dict[str, str]]:
         - {name: forward_model, path: /path/to/forward_model}
         - ...
     """
-    if sys.version_info.minor >= 9:
-        jobs = resources.files(PACKAGE) / FORWARD_MODEL_DIR  # type: ignore
-    else:
-        with resources.path(PACKAGE, FORWARD_MODEL_DIR) as fd:
-            jobs = fd
-
-    return [
-        {"name": (job_name := job.lstrip("fm_")), "path": str(jobs / job_name)}
-        for job in _get_jobs()
-    ]
+    return []
 
 
 @hookimpl

src/everest_models/forward_models.py

@@ -0,0 +1,38 @@
+from importlib import resources
+from importlib.util import find_spec
+from typing import Final
+
+_HAVE_ERT: Final = find_spec("ert") is not None
+
+if _HAVE_ERT:
+    import ert
+    from ert import ForwardModelStepDocumentation, ForwardModelStepPlugin
+
+    def build_forward_model_step_plugin(executable_name: str):
+        forward_model_name = executable_name.lstrip("fm_")
+        class_name = "".join(
+            x.capitalize() for x in forward_model_name.lower().split("_")
+        )
+        return type(
+            class_name,
+            (ForwardModelStepPlugin,),
+            {
+                "__init__": lambda x: ForwardModelStepPlugin.__init__(
+                    x, name=forward_model_name, command=[executable_name]
+                ),
+                "documentation": lambda: ForwardModelStepDocumentation(
+                    category="everest.everest_models",
+                    source_package="everest_models",
+                    source_function_name=class_name,
+                    description=f"The {forward_model_name} forward model.",
+                ),
+            },
+        )
+
+    @ert.plugin(name="everest_models")
+    def installable_forward_model_steps():
+        return [
+            build_forward_model_step_plugin(job)
+            for job in resources.contents("everest_models.jobs")
+            if job.startswith("fm_")
+        ]