com.dbobjekts.codegen.writer.AliasCodeBuilder.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.codegen.metadata.DBCatalogDefinition
import com.dbobjekts.metadata.SchemaAndTable
import com.dbobjekts.metadata.TableAliasesBuilder

class AliasCodeBuilder(val catalog: DBCatalogDefinition) {
  private val strBuilder = StringBuilder()

  fun createFileSource(): String {

    strBuilder.append("package ${catalog.packageName}\n")

    catalog.schemas.flatMap {it.tables} .forEach { strBuilder.append("import ${it.fullyQualifiedClassName()}\n") }

    strBuilder.append(createObjectBody())
    return strBuilder.toString()
  }

  fun createObjectBody(): String {
    val schemaAndTables: List = catalog.schemas.flatMap { schema -> schema.tables.map { SchemaAndTable(schema.schemaName, it.tableName) } }

    val builder = TableAliasesBuilder()
    builder.add(schemaAndTables)
    val aliases = builder.build()

    val aliasesLines = schemaAndTables.map { st -> "    override val ${aliases.aliasForSchemaAndTable(st)} = ${st.table.metaDataObjectName}" }
    val aliasesInterfaceLines = schemaAndTables.map { st -> "    val ${aliases.aliasForSchemaAndTable(st)} : ${st.table.metaDataObjectName}" }


      return """
/**  
 * Auto-generated metadata object. Provides a list of unique aliases for each table across all schemas in the catalog. These are used in the SQL that db-Objekts generates.
 *
 * Do not edit this file manually! Always use [com.dbobjekts.codegen.CodeGenerator] when the metadata model is no longer in sync with the database.
 *
 * You can import the individual values of in this object in your code to use them as convenient shortcuts.
 *
 * ```
 * import ${catalog.packageName}.Aliases.e
 * [...]
 * transaction.select(e.name)
 *```
 */        
object Aliases : HasAliases {
${aliasesLines.joinToString("\n")}
}

/**  
 * Auto-generated metadata object. Interface for a list of unique aliases for each table across all schemas in the catalog.
 *
 * Do not edit this file manually! Always use [com.dbobjekts.codegen.CodeGenerator] when the metadata model is no longer in sync with the database.
 *
 * Any class can extends this interface and delegate to the Aliases object. This imports all the shortcuts without the need for separate manual import statements 
 *
 * ```
 * import ${catalog.packageName}.Aliases
 * import ${catalog.packageName}.HasAliases
 * [...]
 * class AcmeDbInteractions : HasAliases by Aliases {
 *  transaction.select(e.name, a.id, ea.kind)
 * }
 *```
 */ 
interface HasAliases {
${aliasesInterfaceLines.joinToString("\n")}
}
     """

  }

}