hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2026-05-03 12:45:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Kotlin: Generate type aliases for dbscheme

Browse Source
This commit is contained in:
Ian Lynagh
2021-08-09 23:31:27 +01:00
parent dbef421204
commit 5f991653c1
2 changed files with 20 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
java/kotlin-extractor/generate_dbscheme.py
View File

@@ -13,6 +13,7 @@ with open('../ql/src/config/semmlecode.dbscheme', 'r') as f:
dbscheme = re.sub(r'/\*.*?\*/', '', dbscheme, flags=re.DOTALL)
dbscheme = re.sub(r'//[^\r\n]*/', '', dbscheme)
type_aliases = {}
type_hierarchy = {}
with open('src/main/kotlin/KotlinExtractorDbScheme.kt', 'w') as kt:
@@ -38,10 +39,14 @@ with open('src/main/kotlin/KotlinExtractorDbScheme.kt', 'w') as kt:
dbscheme,
flags=re.DOTALL):
type_hierarchy[name] = type_hierarchy.get(name, set())
for typ in re.findall(r'@(\w+)', unions):
s = type_hierarchy.get(typ, set())
s.add(name)
type_hierarchy[typ] = s
typs = re.findall(r'@(\w+)', unions)
if len(typs) == 1:
type_aliases[name] = typs[0]
else:
for typ in typs:
s = type_hierarchy.get(typ, set())
s.add(name)
type_hierarchy[typ] = s
# tables
for relname, body in re.findall('\n([\w_]+)(\([^)]*\))',
@@ -84,10 +89,13 @@ with open('src/main/kotlin/KotlinExtractorDbScheme.kt', 'w') as kt:
kt.write('}\n')
for typ in sorted(type_hierarchy):
kt.write('sealed interface Db' + upperFirst(typ))
names = sorted(type_hierarchy[typ])
if names:
kt.write(': ')
kt.write(', '.join(map(lambda name: 'Db' + upperFirst(name), names)))
kt.write(' {}\n')
if typ in type_aliases:
kt.write('typealias Db' + upperFirst(typ) + ' = Db' + upperFirst(type_aliases[typ]) + '\n')
else:
kt.write('sealed interface Db' + upperFirst(typ))
names = sorted(type_hierarchy[typ])
if names:
kt.write(': ')
kt.write(', '.join(map(lambda name: 'Db' + upperFirst(name), names)))
kt.write('\n')

4
java/kotlin-extractor/src/main/kotlin/KotlinExtractorExtension.kt
View File

@@ -57,12 +57,12 @@ class TrapWriter (
fun writeTrap(trap: String) {
file.write(trap)
}
fun getLocation(startOffset: Int, endOffset: Int): Label<DbLocation_default> {
fun getLocation(startOffset: Int, endOffset: Int): Label<DbLocation> {
val startLine = fileEntry.getLineNumber(startOffset) + 1
val startColumn = fileEntry.getColumnNumber(startOffset) + 1
val endLine = fileEntry.getLineNumber(endOffset) + 1
val endColumn = fileEntry.getColumnNumber(endOffset)
val id: Label<DbLocation_default> = getFreshLabel()
val id: Label<DbLocation> = getFreshLabel()
val fileId: Label<DbFile> = getLabelFor(fileLabel)
writeTrap("$id = @\"loc,{$fileId},$startLine,$startColumn,$endLine,$endColumn\"\n")
writeLocations_default(id, fileId, startLine, startColumn, endLine, endColumn)