Swift: make C++ code generation language agnostic

Author: redsun82

swift/BUILD.bazel

@@ -5,7 +5,13 @@ load("//misc/bazel:pkg_runfiles.bzl", "pkg_runfiles")
 
 filegroup(
     name = "schema",
-    srcs = ["schema.py"] + glob(["*.dbscheme"]),
+    srcs = ["schema.py"],
+    visibility = ["//swift:__subpackages__"],
+)
+
+filegroup(
+    name = "schema_includes",
+    srcs = glob(["*.dbscheme"]),
     visibility = ["//swift:__subpackages__"],
 )
 

swift/codegen/codegen.py

@@ -60,12 +60,19 @@ def _parse_args() -> argparse.Namespace:
         p.add_argument("--generated-registry",
                        help="registry file containing information about checked-in generated code"),
     ]
+    p.add_argument("--script-name",
+                   help="script name to put in header comments of generated files. By default, the path of this "
+                        "script relative to the root directory")
+    p.add_argument("--trap-library",
+                   help="path to the trap library from an include directory, required if generating C++ trap bindings"),
     p.add_argument("--ql-format", action="store_true", default=True,
                    help="use codeql to autoformat QL files (which is the default)")
     p.add_argument("--no-ql-format", action="store_false", dest="ql_format", help="do not format QL files")
     p.add_argument("--codeql-binary", default="codeql", help="command to use for QL formatting (default %(default)s)")
     p.add_argument("--force", "-f", action="store_true",
-                   help="generate all files without skipping unchanged files and overwriting modified ones"),
+                   help="generate all files without skipping unchanged files and overwriting modified ones")
+    p.add_argument("--use-current-directory", action="store_true",
+                   help="do not consider paths as relative to --root-dir or the configuration directory")
     opts = p.parse_args()
     if opts.configuration_file is not None:
         with open(opts.configuration_file) as config:
@@ -75,16 +82,15 @@ def _parse_args() -> argparse.Namespace:
                     setattr(opts, flag, getattr(defaults, flag))
         if opts.root_dir is None:
             opts.root_dir = opts.configuration_file.parent
-    if opts.root_dir is None:
-        p.error("Either --configuration-file or --root-dir must be provided, or a codegen.conf file must be in a "
-                "containing directory")
     if not opts.generate:
         p.error("Nothing to do, specify --generate")
-    # absolutize all paths relative to --root-dir
+    # absolutize all paths
     for arg in path_arguments:
         path = getattr(opts, arg.dest)
         if path is not None:
-            setattr(opts, arg.dest, opts.root_dir / path)
+            setattr(opts, arg.dest, _abspath(path) if opts.use_current_directory else (opts.root_dir / path))
+    if not opts.script_name:
+        opts.script_name = paths.exe_file.relative_to(opts.root_dir)
     return opts
 
 
@@ -102,7 +108,7 @@ def run():
         log_level = logging.INFO
     logging.basicConfig(format="{levelname} {message}", style='{', level=log_level)
     for target in opts.generate:
-        generate(target, opts, render.Renderer(opts.root_dir))
+        generate(target, opts, render.Renderer(opts.script_name, opts.root_dir))
 
 
 if __name__ == "__main__":

swift/codegen/generators/cppgen.py

@@ -95,4 +95,5 @@ def generate(opts, renderer):
     out = opts.cpp_output
     for dir, classes in processor.get_classes().items():
         renderer.render(cpp.ClassList(classes, opts.schema,
-                                      include_parent=bool(dir)), out / dir / "TrapClasses")
+                                      include_parent=bool(dir),
+                                      trap_library=opts.trap_library), out / dir / "TrapClasses")

swift/codegen/generators/dbschemegen.py

@@ -125,8 +125,8 @@ def generate(opts, renderer):
 
     data = schemaloader.load_file(input)
 
-    dbscheme = Scheme(src=input.relative_to(opts.root_dir),
-                      includes=get_includes(data, include_dir=input.parent, root_dir=opts.root_dir),
+    dbscheme = Scheme(src=input.name,
+                      includes=get_includes(data, include_dir=input.parent, root_dir=input.parent),
                       declarations=get_declarations(data))
 
     renderer.render(dbscheme, out)

swift/codegen/generators/trapgen.py

@@ -72,6 +72,7 @@ def generate(opts, renderer):
     assert opts.cpp_output
     tag_graph = {}
     out = opts.cpp_output
+    trap_library = opts.trap_library
 
     traps = {pathlib.Path(): []}
     for e in dbschemeloader.iterload(opts.dbscheme):
@@ -84,7 +85,8 @@ def generate(opts, renderer):
 
     for dir, entries in traps.items():
         dir = dir or pathlib.Path()
-        renderer.render(cpp.TrapList(entries, opts.dbscheme), out / dir / "TrapEntries")
+        relative_gen_dir = pathlib.Path(*[".." for _ in dir.parents])
+        renderer.render(cpp.TrapList(entries, opts.dbscheme, trap_library, relative_gen_dir), out / dir / "TrapEntries")
 
     tags = []
     for tag in toposort_flatten(tag_graph):

swift/codegen/lib/cpp.py

@@ -1,3 +1,4 @@
+import pathlib
 import re
 from dataclasses import dataclass, field
 from typing import List, ClassVar
@@ -110,6 +111,8 @@ class TrapList:
     extensions = ["h", "cpp"]
     traps: List[Trap]
     source: str
+    trap_library_dir: pathlib.Path
+    gen_dir: pathlib.Path
 
 
 @dataclass
@@ -156,4 +159,5 @@ class ClassList:
 
     classes: List[Class]
     source: str
+    trap_library: str
     include_parent: bool = False

swift/codegen/lib/render.py

@@ -25,13 +25,10 @@ class Error(Exception):
 class Renderer:
     """ Template renderer using mustache templates in the `templates` directory """
 
-    def __init__(self, root_dir: pathlib.Path):
+    def __init__(self, generator: pathlib.Path, root_dir: pathlib.Path):
         self._r = pystache.Renderer(search_dirs=str(paths.templates_dir), escape=lambda u: u)
         self._root_dir = root_dir
-        try:
-            self._generator = self._get_path(paths.exe_file)
-        except ValueError:
-            self._generator = paths.exe_file.name
+        self._generator = generator
 
     def _get_path(self, file: pathlib.Path):
         return file.relative_to(self._root_dir)
@@ -63,7 +60,7 @@ def _do_write(self, mnemonic: str, contents: str, output: pathlib.Path):
 
     def manage(self, generated: typing.Iterable[pathlib.Path], stubs: typing.Iterable[pathlib.Path],
                registry: pathlib.Path, force: bool = False) -> "RenderManager":
-        return RenderManager(self._root_dir, generated, stubs, registry, force)
+        return RenderManager(self._generator, self._root_dir, generated, stubs, registry, force)
 
 
 class RenderManager(Renderer):
@@ -88,10 +85,10 @@ class Hashes:
         pre: str
         post: typing.Optional[str] = None
 
-    def __init__(self, root_dir: pathlib.Path, generated: typing.Iterable[pathlib.Path],
+    def __init__(self, generator: pathlib.Path, root_dir: pathlib.Path, generated: typing.Iterable[pathlib.Path],
                  stubs: typing.Iterable[pathlib.Path],
                  registry: pathlib.Path, force: bool = False):
-        super().__init__(root_dir)
+        super().__init__(generator, root_dir)
         self._registry_path = registry
         self._force = force
         self._hashes = {}

swift/codegen/templates/cpp_classes_h.mustache

@@ -6,8 +6,8 @@
 #include <optional>
 #include <vector>
 
-#include "swift/extractor/trap/TrapLabel.h"
-#include "swift/extractor/trap/TrapTagTraits.h"
+#include "{{trap_library}}/TrapLabel.h"
+#include "{{trap_library}}/TrapTagTraits.h"
 #include "./TrapEntries.h"
 {{#include_parent}}
 #include "../TrapClasses.h"

swift/codegen/templates/trap_traps_h.mustache

@@ -5,9 +5,9 @@
 #include <iostream>
 #include <string>
 
-#include "swift/extractor/trap/TrapLabel.h"
-#include "swift/extractor/trap/TrapTagTraits.h"
-#include "swift/extractor/trap/generated/TrapTags.h"
+#include "{{trap_library_dir}}/TrapLabel.h"
+#include "{{trap_library_dir}}/TrapTagTraits.h"
+#include "{{gen_dir}}/TrapTags.h"
 
 namespace codeql {
 {{#traps}}

swift/codegen/test/test_dbschemegen.py

@@ -30,7 +30,7 @@ def func(classes, null=None):
 
 def test_empty(generate):
     assert generate([]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[],
     )
@@ -43,10 +43,10 @@ def test_includes(input, opts, generate):
         write(opts.schema.parent / i, i + " data")
 
     assert generate([]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[
             dbscheme.SchemeInclude(
-                src=schema_dir / i,
+                src=pathlib.Path(i),
                 data=i + " data",
             ) for i in includes
         ],
@@ -58,7 +58,7 @@ def test_empty_final_class(generate, dir_param):
     assert generate([
         schema.Class("Object", group=dir_param.input),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Table(
@@ -78,7 +78,7 @@ def test_final_class_with_single_scalar_field(generate, dir_param):
             schema.SingleProperty("foo", "bar"),
         ]),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Table(
@@ -98,7 +98,7 @@ def test_final_class_with_single_class_field(generate, dir_param):
             schema.SingleProperty("foo", "Bar"),
         ]),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Table(
@@ -118,7 +118,7 @@ def test_final_class_with_optional_field(generate, dir_param):
             schema.OptionalProperty("foo", "bar"),
         ]),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Table(
@@ -146,7 +146,7 @@ def test_final_class_with_repeated_field(generate, property_cls, dir_param):
             property_cls("foo", "bar"),
         ]),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Table(
@@ -174,7 +174,7 @@ def test_final_class_with_predicate_field(generate, dir_param):
             schema.PredicateProperty("foo"),
         ]),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Table(
@@ -205,7 +205,7 @@ def test_final_class_with_more_fields(generate, dir_param):
             schema.PredicateProperty("six"),
         ]),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Table(
@@ -259,7 +259,7 @@ def test_empty_class_with_derived(generate):
         schema.Class(name="Left", bases=["Base"]),
         schema.Class(name="Right", bases=["Base"]),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Union(
@@ -290,7 +290,7 @@ def test_class_with_derived_and_single_property(generate, dir_param):
         schema.Class(name="Left", bases=["Base"]),
         schema.Class(name="Right", bases=["Base"]),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Union(
@@ -330,7 +330,7 @@ def test_class_with_derived_and_optional_property(generate, dir_param):
         schema.Class(name="Left", bases=["Base"]),
         schema.Class(name="Right", bases=["Base"]),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Union(
@@ -370,7 +370,7 @@ def test_class_with_derived_and_repeated_property(generate, dir_param):
         schema.Class(name="Left", bases=["Base"]),
         schema.Class(name="Right", bases=["Base"]),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Union(
@@ -432,7 +432,7 @@ def test_null_class(generate):
             bases=["Base"],
         ),
     ], null="Null") == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Union(
@@ -514,7 +514,7 @@ def test_ipa_classes_ignored(generate):
         schema.Class(name="B", ipa=schema.IpaInfo(from_class="A")),
         schema.Class(name="C", ipa=schema.IpaInfo(on_arguments={"x": "A"})),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[],
     )
@@ -526,7 +526,7 @@ def test_ipa_derived_classes_ignored(generate):
         schema.Class(name="B", bases=["A"], ipa=schema.IpaInfo()),
         schema.Class(name="C", bases=["A"]),
     ]) == dbscheme.Scheme(
-        src=schema_file,
+        src=schema_file.name,
         includes=[],
         declarations=[
             dbscheme.Union("@a", ["@c"]),