Merge pull request github#9198 from github/redsun82/swift-self-contained-cpp-code-gen

Swift: make C++ code generation more self-contained

Author: redsun82

.github/workflows/swift-codegen.yml

@@ -25,8 +25,7 @@ jobs:
           git diff --exit-code --stat HEAD
       - name: Generate C++ files
         run: |
-          bazel run //swift/codegen:trapgen -- --cpp-output=$PWD/swift-generated-headers
-          bazel run //swift/codegen:cppgen -- --cpp-output=$PWD/swift-generated-headers
+          bazel run //swift/codegen:cppcodegen -- --cpp-output=$PWD/swift-generated-headers
       - uses: actions/upload-artifact@v3
         with:
           name: swift-generated-headers

swift/codegen/BUILD.bazel

@@ -14,41 +14,22 @@ filegroup(
 
 py_binary(
     name = "codegen",
-    srcs = glob(
-        ["*.py"],
-        exclude = ["trapgen.py"],
-    ),
+    srcs = ["codegen.py"],
     visibility = ["//swift/codegen/test:__pkg__"],
-    deps = ["//swift/codegen/lib"],
+    deps = ["//swift/codegen/generators"],
 )
 
 # as opposed to the above, that is meant to only be run with bazel run,
 # we need to be precise with data dependencies of this which is meant be run during build
 py_binary(
-    name = "trapgen",
-    srcs = ["trapgen.py"],
-    data = [
-        "//swift:dbscheme",
-        "//swift/codegen/templates:trap",
-    ],
-    visibility = ["//swift:__subpackages__"],
-    deps = [
-        "//swift/codegen/lib",
-        requirement("toposort"),
-    ],
-)
-
-py_binary(
-    name = "cppgen",
-    srcs = ["cppgen.py"],
+    name = "cppcodegen",
+    srcs = ["cppcodegen.py"],
     data = [
         ":schema",
         ":schema_includes",
         "//swift/codegen/templates:cpp",
+        "//swift/codegen/templates:trap",
     ],
     visibility = ["//swift:__subpackages__"],
-    deps = [
-        "//swift/codegen/lib",
-        requirement("toposort"),
-    ],
+    deps = ["//swift/codegen/generators"],
 )

swift/codegen/codegen.py

@@ -1,9 +1,7 @@
 #!/usr/bin/env python3
 """ Driver script to run all checked in code generation """
 
-from lib import generator
-import dbschemegen
-import qlgen
+from swift.codegen.generators import generator, dbschemegen, qlgen
 
 if __name__ == "__main__":
     generator.run(dbschemegen, qlgen)

swift/codegen/cppcodegen.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python3
+""" Driver script to run all cpp code generation """
+
+from swift.codegen.generators import generator, dbschemegen, trapgen, cppgen
+
+if __name__ == "__main__":
+    generator.run(dbschemegen, trapgen, cppgen)

swift/codegen/generators/BUILD.bazel

@@ -0,0 +1,11 @@
+load("@swift_codegen_deps//:requirements.bzl", "requirement")
+
+py_library(
+    name = "generators",
+    srcs = glob(["*.py"]),
+    visibility = ["//swift/codegen:__subpackages__"],
+    deps = [
+        "//swift/codegen/lib",
+        requirement("toposort"),
+    ],
+)

swift/codegen/cppgen.py renamed to swift/codegen/generators/cppgen.py

@@ -4,7 +4,8 @@
 import inflection
 from toposort import toposort_flatten
 
-from swift.codegen.lib import cpp, generator, schema
+from swift.codegen.lib import cpp, schema
+from swift.codegen.generators import generator
 
 
 def _get_type(t: str, trap_affix: str) -> str:

swift/codegen/dbschemegen.py renamed to swift/codegen/generators/dbschemegen.py

@@ -3,7 +3,8 @@
 
 import inflection
 
-from swift.codegen.lib import paths, schema, generator
+from swift.codegen.lib import schema
+from swift.codegen.generators import generator
 from swift.codegen.lib.dbscheme import *
 
 log = logging.getLogger(__name__)

swift/codegen/lib/generator.py renamed to swift/codegen/generators/generator.py

@@ -5,7 +5,8 @@
 import sys
 from typing import Set
 
-from . import options, render, paths
+from swift.codegen.lib import render, paths
+from swift.codegen.generators import options
 
 
 def _parse(tags: Set[str]) -> argparse.Namespace:

swift/codegen/lib/options.py renamed to swift/codegen/generators/options.py

@@ -5,7 +5,7 @@
 import pathlib
 from typing import Set
 
-from . import paths
+from swift.codegen.lib import paths
 
 
 def _init_options():

swift/codegen/qlgen.py renamed to swift/codegen/generators/qlgen.py

@@ -6,7 +6,8 @@
 
 import inflection
 
-from swift.codegen.lib import schema, paths, generator, ql
+from swift.codegen.lib import schema, ql
+from swift.codegen.generators import generator
 
 log = logging.getLogger(__name__)