com.dbobjekts.codegen.writer.TableSourcesBuilder.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of db-objekts-core Show documentation
There is a newer version: 0.6.0-RC2
package com.dbobjekts.codegen.writer

import com.dbobjekts.api.PackageName
import com.dbobjekts.api.SchemaName
import com.dbobjekts.api.TableRowData
import com.dbobjekts.codegen.metadata.DBColumnDefinition
import com.dbobjekts.codegen.metadata.DBForeignKeyDefinition
import com.dbobjekts.codegen.metadata.DBTableDefinition
import com.dbobjekts.metadata.Table
import com.dbobjekts.metadata.joins.JoinBase
import com.dbobjekts.metadata.joins.JoinType
import com.dbobjekts.metadata.joins.DerivedJoin
import com.dbobjekts.statement.WriteQueryAccessors
import com.dbobjekts.statement.insert.InsertBuilderBase
import com.dbobjekts.statement.update.HasUpdateBuilder
import com.dbobjekts.statement.update.UpdateBuilderBase

internal class TableSourcesBuilder(
    basePackage: PackageName,
    val model: DBTableDefinition
) {

    val strBuilder = StringBuilder()
    val importLineBuilder = ImportLineBuilder(basePackage)

    /**
     * FKs to parents in a different schema/package need to be imported
     */
    fun build(): String {
        val detailedSourceBuilder = TableDetailsSourceBuilder(model)
        strBuilder.append("package ${model.packageName}\n\n")

        fun generateField(column: DBColumnDefinition) {
            val fk = if (column is DBForeignKeyDefinition) {
                "\n     *\n     * Foreign key to ${column.parentSchema.value}.${column.parentTable.value}.${column.parentColumn.value}"
            } else ""
            strBuilder.appendLine("    /**\n     * Represents db column ${column.schemaName.value}.${column.tableName.value}.${column.columnName}$fk\n     */")
            strBuilder.appendLine("    val ${column.columnName.fieldName} = ${column.asFactoryMethod()}")
        }

        importLineBuilder.addClasses(
            Table::class.java,
            TableRowData::class.java,
            WriteQueryAccessors::class.java,
            HasUpdateBuilder::class.java,
            InsertBuilderBase::class.java,
            UpdateBuilderBase::class.java
        )
        importLineBuilder.add("com.dbobjekts.api.AnyColumn")

        val updateBuilderInterface = HasUpdateBuilder::class.java.simpleName

        importLineBuilder.addImportsForForeignKeys(model.schema, model.foreignKeys())

        val columnNames = model.columns.map {
            importLineBuilder.add(it.column.qualifiedClassName())
            it.columnName.fieldName
        }.joinToString(",")

        strBuilder.appendLine(importLineBuilder.build())
        strBuilder.appendLine()
        val tbl = model.asClassName()
        strBuilder.appendLine(detailedSourceBuilder.sourceForTableComment())
        strBuilder.appendLine("""object $tbl:Table<${tbl}Row>("${model.tableName.value}"), $updateBuilderInterface<${tbl}UpdateBuilder, ${tbl}InsertBuilder> {""".trimMargin())
        model.columns.forEach {
            generateField(it)
        }

        strBuilder.appendLine("    override val columns: List = listOf($columnNames)")
        strBuilder.appendLine(detailedSourceBuilder.sourceForToValue())
        strBuilder.appendLine(detailedSourceBuilder.sourceForMetaDataVal())
        strBuilder.appendLine("}")
        strBuilder.appendLine(detailedSourceBuilder.sourceForBuilderClasses())
        strBuilder.appendLine(detailedSourceBuilder.sourceForRowDataClass())
        return strBuilder.toString()
    }

}