Connect cxx_python_extension generated type stub targets to the target graph

Summary: Following up on D88693627, this diff adds `cxx_python_extension` rules to the target graph, treating it in the same way as a Python library target with no sources -- there are presented in the target graph just to be able to provide the deps edge

Reviewed By: connernilsen

Differential Revision: D88773199

fbshipit-source-id: c8355a7036f01ebe5ca4cd3e898d4fa39068f38a

Author: grievejia

prelude/python/sourcedb/pyrefly.bxl

@@ -112,6 +112,14 @@ def _handle_alias(
     else:
         fail("Alias has no actual target".format(target.label))
 
+def _handle_python_extension(
+        entry: dict,
+        attrs: bxl.LazyAttrs,
+        fs: bxl.Filesystem,
+        target: bxl.ConfiguredTargetNode,
+        known_targets: dict[TargetLabel, bxl.ConfiguredTargetNode]) -> None:
+    _set_entry_common_fields(entry, attrs, fs, target, known_targets)
+
 def _handle_genrule(
         entry: dict,
         attrs: bxl.LazyAttrs,
@@ -289,7 +297,6 @@ def _main(ctx: bxl.Context) -> None:
         if x.rule_type not in (
             "forward",
             "prelude//rules.bzl:command_alias",
-            "prelude//rules.bzl:cxx_python_extension",
             "prelude//rules.bzl:export_file",
             "prelude//rules.bzl:filegroup",
         ) and x.label.name != "restricted_bin"
@@ -305,6 +312,8 @@ def _main(ctx: bxl.Context) -> None:
 
         if x.rule_type == "prelude//rules.bzl:alias":
             _handle_alias(entry, attrs, x)
+        elif x.rule_type == "prelude//rules.bzl:cxx_python_extension":
+            _handle_python_extension(entry, attrs, fs, x, known_targets)
         elif x.rule_type == "prelude//rules.bzl:genrule":
             _handle_genrule(entry, attrs, fs, x, known_targets, to_remove, to_build, to_insert, genrule_to_insert)
         elif x.rule_type == "prelude//rules.bzl:python_library":

prelude/python/sourcedb/tests/pyrefly_expected.json

@@ -61,7 +61,8 @@
     "prelude//python/sourcedb/tests/test_files:main": {
       "buildfile_path": "prelude/python/sourcedb/tests/test_files/BUCK",
       "deps": [
-        "prelude//python/sourcedb/tests/test_files:lib"
+        "prelude//python/sourcedb/tests/test_files:lib",
+        "prelude//python/sourcedb/tests/test_files:my_extension_lib"
       ],
       "python_platform": "linux",
       "python_version": "3.12",
@@ -72,6 +73,42 @@
         ]
       }
     },
+    "prelude//python/sourcedb/tests/test_files:my_extension": {
+      "buildfile_path": "prelude/python/sourcedb/tests/test_files/BUCK",
+      "deps": [
+        "prelude//python/sourcedb/tests/test_files:my_extension_stub"
+      ],
+      "python_platform": "linux",
+      "python_version": "3.12",
+      "relative_to": null,
+      "srcs": {}
+    },
+    "prelude//python/sourcedb/tests/test_files:my_extension_lib": {
+      "buildfile_path": "prelude/python/sourcedb/tests/test_files/BUCK",
+      "deps": [
+        "prelude//python/sourcedb/tests/test_files:my_extension"
+      ],
+      "python_platform": "linux",
+      "python_version": "3.12",
+      "relative_to": null,
+      "srcs": {
+        "python.sourcedb.tests.test_files.my_extension": [
+          "prelude/python/sourcedb/tests/test_files/my_extension.py"
+        ]
+      }
+    },
+    "prelude//python/sourcedb/tests/test_files:my_extension_stub": {
+      "buildfile_path": "prelude/python/sourcedb/tests/test_files/BUCK",
+      "deps": [],
+      "python_platform": "linux",
+      "python_version": "3.12",
+      "relative_to": null,
+      "srcs": {
+        "python.sourcedb.tests.test_files.my_extension": [
+          "prelude/python/sourcedb/tests/test_files/my_extension.pyi"
+        ]
+      }
+    },
     "prelude//python/sourcedb/tests/test_files:subdir": {
       "buildfile_path": "prelude/python/sourcedb/tests/test_files/BUCK",
       "deps": [

prelude/python/sourcedb/tests/test_files/BUCK

@@ -7,7 +7,10 @@ source_listing()
 python_library(
     name = "main",
     srcs = ["main.py"],
-    deps = [":lib"],
+    deps = [
+        ":lib",
+        ":my_extension_lib",
+    ],
 )
 
 python_library(
@@ -61,3 +64,21 @@ genrule(
     out = "generated_file.pyi",
     cmd = "touch ${OUT}",
 )
+
+python_library(
+    name = "my_extension_stub",
+    srcs = ["my_extension.pyi"],
+)
+
+cxx_python_extension(
+    name = "my_extension",
+    srcs = ["my_extension.cpp"],
+    base_module = "my_extension",
+    deps = [":my_extension_stub"],
+)
+
+python_library(
+    name = "my_extension_lib",
+    srcs = ["my_extension.py"],
+    deps = [":my_extension"],
+)