[wip] Add CDT migrator

conda_forge_tick/make_migrators.py

@@ -64,6 +64,7 @@
     PyPIOrgMigrator,
     Replacement,
     RUCRTCleanup,
+    CDTMigrator,
     StaticLibMigrator,
     StdlibMigrator,
     UpdateCMakeArgsMigrator,

conda_forge_tick/migrators/__init__.py

@@ -39,6 +39,7 @@
 from .pip_wheel_dep import PipWheelMigrator
 from .pypi_org import PyPIOrgMigrator
 from .r_ucrt import RUCRTCleanup
+from .cdt import CDTMigrator
 from .recipe_v1 import CombineV1ConditionsMigrator
 from .replacement import Replacement, MiniReplacement
 from .use_pip import PipMigrator

conda_forge_tick/migrators/cdt.py

@@ -0,0 +1,128 @@
+import re
+import typing
+from typing import Any
+import textwrap
+
+from conda_forge_tick.contexts import ClonedFeedstockContext, FeedstockContext
+from conda_forge_tick.migrators.core import Migrator
+from conda_forge_tick.os_utils import pushd
+
+cdt_mapping = {
+    'libx11-devel': 'xorg-libx11',
+    'libxext-devel': 'xorg-libxext',
+    'libxrender-devel': 'xorg-libxrender',
+    'libdrm-devel': 'libdrm',
+    'libxcomposite-devel': 'xorg-libxcomposite',
+    'libxcursor-devel': 'xorg-libxcursor',
+    'libxdamage': 'xorg-libxdamage',
+    'libxfixes': 'xorg-libxfixes',
+    'libxscrnsaver-devel': 'xorg-libxscrnsaver',
+    'libxtst-devel': 'xorg-libxtst',
+    'libxxf86vm': 'xorg-libxxf86vm',
+    'libselinux-devel': 'libselinux-devel',
+    'xorg-x11-proto-devel': 'xorgproto',
+    'libxrender': 'xorg-libxrender',
+    'libxext': 'xorg-libxext',
+    'libsm-devel': 'xorg-libsm',
+    'mesa-libgbm': 'libgl-devel',
+    'mesa-libgl-devel': 'libgl-devel',
+    'mesa-dri-drivers': 'libgl-devel',
+    'mesa-libegl-devel': 'libgl-devel',
+    'libselinux': 'libselinux',
+    'libglvnd-opengl': 'libglvnd-opengl',
+    'libxcb': 'libxcb',
+    'libxau': 'xorg-libxau',
+    'systemd-devel': 'systemd-devel',
+    'libudev': 'libudev',
+    'libudev-devel': 'libudev-devel',
+    'numactl-devel': 'numactl-devel',
+    'pciutils-devel': 'pciutils-devel',
+    'libxi-devel': 'xorg-libxi',
+    'libxrandr-devel': 'xorg-librandr',
+    'libuuid': 'libuuid',
+    'alsa-lib-devel': 'alsa-lib-devel',
+    'gtk2-devel': 'gtk2-devel'
+}
+
+def _replace_cdts(yaml):
+
+    def replacer(match):
+        package = match.group(1).strip()
+        return cdt_mapping.get(package, package)
+
+    pattern = re.compile(r"\{\{\s*cdt\('([^']+)'\)\s*\}\}")
+    replaced_yaml = pattern.sub(replacer, yaml)
+    return replaced_yaml
+
+class CDTMigrator(Migrator):
+    """Description"""
+
+    name = "CDT Migrator"
+    rerender = True
+    migrator_version = 1
+
+    def filter_not_in_migration(self, attrs, not_bad_str_start=""):
+        if super().filter_not_in_migration(attrs, not_bad_str_start):
+            return True
+
+        # we need to check if there is any cdt package as dependency
+        requirements = attrs.get("requirements", {})
+
+        rq = (
+            requirements.get("build", set())
+            | requirements.get("host", set())
+            | requirements.get("run", set())
+            | requirements.get("test", set())
+        )
+        needed = False
+
+        for k in ctd_mapping.keys():
+            if k in requirements:
+                needed = True
+                break
+
+        return (not needed)
+        #return len(rq & self.packages) == 0
+
+    def migrate(self, recipe_dir: str, attrs: "AttrsTypedDict", **kwargs: Any) -> None:
+        with pushd(recipe_dir):
+            with open("meta.yaml") as fp:
+                yaml = fp.read()
+
+            yaml = _replace_cdts(yaml)
+
+            # Rewrite the recipe file
+            with open("meta.yaml", "w") as fp:
+                fp.write(yaml)
+
+        # TODO: remove duplicate libgl entries
+
+
+    def pr_body(
+        self, feedstock_ctx: ClonedFeedstockContext, add_label_text: bool = True
+    ) -> str:
+        body = super().pr_body(feedstock_ctx)
+        body = body.format(
+            textwrap.dedent(
+                """\
+
+""",
+            )
+        )
+        return body
+
+    def commit_message(self, feedstock_ctx) -> str:
+        return "Migrate CDT dependencies to regular feedstocks"
+
+    def pr_title(self, feedstock_ctx) -> str:
+        return "Migrate CDT dependencies to regular feedstocks"
+
+    def remote_branch(self, feedstock_ctx) -> str:
+        return f"{self.name}-migration-{self.migrator_version}"
+
+    def migrator_uid(self, attrs):
+        if self.name is None:
+            raise ValueError("name is None")
+        n = super().migrator_uid(attrs)
+        n["name"] = self.name
+        return n