Latex extractor for IntelliJ Ultimate Database Tool

latex.kts

import com.intellij.database.extensions.*

/*
 * Available context bindings:
 *   COLUMNS     List<DataColumn>
 *   ROWS        Iterable<DataRow>
 *   OUT         { append() }
 *   FORMATTER   { format(row, col); formatValue(Object, col) }
 *   TRANSPOSED  Boolean
 * plus ALL_COLUMNS, TABLE, DIALECT
 *
 * where:
 *   DataRow     { rowNumber(); first(); last(); data(): List<Object>; value(column): Object }
 *   DataColumn  { columnNumber(), name() }
 */

val out = bindings["OUT"] as java.lang.Appendable
val columns = bindings["COLUMNS"] as List<DataColumn>
val rows = bindings["ROWS"] as Iterable<DataRow>
val format = bindings["FORMATTER"] as ValueFormatter

out.apply {
    append(
        """
        \begin{table}[]
             \centering
    """.trimIndent()
    )
    append("\\begin{tabular}{${columns.joinToString(separator = "|") { "l" }}}")
    appendLine()
    append(columns.joinToString(separator = " & ", postfix = """\\ \hline""") {
        it.name()
    })

    for (row in rows) {
        appendLine()
        append(columns.joinToString(separator = " & ", postfix = """\\""") {
            format.format(row, it)
        })
    }
    append("""
    \end{tabular}
    \caption{}
    \label{tab:}
\end{table}
        """)
}