Files
codeql/csharp/ql/src/utils/modelconverter/ConvertExtensions.py
2023-11-09 08:34:35 +01:00

47 lines
2.0 KiB
Python

# Tool to generate data extensions files based on the existing models.
# Usage:
# python3 ConvertExtensions.py
# (1) A folder named `csharp/ql/lib/ext` will be created, if it doesn't already exist.
# (2) The converted models will be written to `csharp/ql/lib/ext`. One file for each namespace.
import os
import subprocess
import sys
import tempfile
# Add Models as Data script directory to sys.path.
gitroot = subprocess.check_output(["git", "rev-parse", "--show-toplevel"]).decode("utf-8").strip()
madpath = os.path.join(gitroot, "misc/scripts/models-as-data/")
sys.path.append(madpath)
import helpers
import convert_extensions as extensions
print('Running script to generate data extensions files from the existing MaD models.')
print('Making a dummy database.')
# Configuration
language = "csharp"
workDir = tempfile.mkdtemp()
projectDir = os.path.join(workDir, "project")
dbDir = os.path.join(workDir, "db")
# Make dummy project
helpers.run_cmd(['dotnet', 'new', 'webapp', '-o', projectDir], "Failed to create dummy project.")
# Add nuget packages for all packages where we have models
helpers.run_cmd(['dotnet', 'add', projectDir, 'package', 'Newtonsoft.Json'])
helpers.run_cmd(['dotnet', 'add', projectDir, 'package', 'Microsoft.EntityFrameworkCore'])
helpers.run_cmd(['dotnet', 'add', projectDir, 'package', 'Microsoft.EntityFrameworkCore.Relational'])
helpers.run_cmd(['dotnet', 'add', projectDir, 'package', 'Dapper'])
helpers.run_cmd(['dotnet', 'add', projectDir, 'package', 'ServiceStack'])
helpers.run_cmd(['dotnet', 'add', projectDir, 'package', 'ServiceStack.OrmLite'])
helpers.run_cmd(['dotnet', 'add', projectDir, 'package', 'System.Collections.Immutable'])
helpers.run_cmd(['codeql', 'database', 'create', f'--language={language}', '-c', f'dotnet build {projectDir}', dbDir], "Failed to create dummy database.")
print('Converting data extensions for C#.')
extensions.Converter(language, dbDir).run()
print('Cleanup.')
# Cleanup - delete database.
helpers.remove_dir(dbDir)
print('Done.')