mirror of
https://github.com/github/codeql.git
synced 2025-12-19 18:33:16 +01:00
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
""" generator options, categorized by tags """
|
|
|
|
import argparse
|
|
import collections
|
|
import pathlib
|
|
from typing import Set
|
|
|
|
from . import paths
|
|
|
|
|
|
def _init_options():
|
|
Option("--verbose", "-v", action="store_true")
|
|
Option("--swift-dir", type=_abspath, default=paths.swift_dir)
|
|
Option("--schema", tags=["schema"], type=_abspath, default=paths.swift_dir / "codegen/schema.yml")
|
|
Option("--dbscheme", tags=["dbscheme"], type=_abspath, default=paths.swift_dir / "ql/lib/swift.dbscheme")
|
|
Option("--ql-output", tags=["ql"], type=_abspath, default=paths.swift_dir / "ql/lib/codeql/swift/generated")
|
|
Option("--ql-stub-output", tags=["ql"], type=_abspath, default=paths.swift_dir / "ql/lib/codeql/swift/elements")
|
|
Option("--codeql-binary", tags=["ql"], default="codeql")
|
|
Option("--cpp-output", tags=["cpp"], type=_abspath, required=True)
|
|
Option("--cpp-namespace", tags=["cpp"], default="codeql")
|
|
Option("--trap-affix", tags=["cpp"], default="Trap")
|
|
Option("--cpp-include-dir", tags=["cpp"], required=True)
|
|
|
|
|
|
def _abspath(x):
|
|
return pathlib.Path(x).resolve()
|
|
|
|
|
|
_options = collections.defaultdict(list)
|
|
|
|
|
|
class Option:
|
|
def __init__(self, *args, tags=None, **kwargs):
|
|
tags = tags or []
|
|
self.args = args
|
|
self.kwargs = kwargs
|
|
if tags:
|
|
for t in tags:
|
|
_options[t].append(self)
|
|
else:
|
|
_options["*"].append(self)
|
|
|
|
def add_to(self, parser: argparse.ArgumentParser):
|
|
parser.add_argument(*self.args, **self.kwargs)
|
|
|
|
|
|
_init_options()
|
|
|
|
|
|
def get(tags: Set[str]):
|
|
""" get options marked by `tags`
|
|
|
|
Options tagged by wildcard '*' are always returned
|
|
"""
|
|
# use specifically tagged options + those tagged with wildcard *
|
|
return (o for tag in ('*',) + tuple(tags) for o in _options[tag])
|