SQL-IN-Groovy Datagrip extractor

SQL-IN-Groovy.sql.groovy

/*
 * 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() }
 */

SEPARATOR = ", "
QUOTE = "'"

Map<String, List<Object>> map = new LinkedHashMap<>()

ROWS.each { row ->
    COLUMNS.each { column ->
        value = FORMATTER.format(row, column)
        map.computeIfAbsent((String) column.name(), { k -> new LinkedList<>() })
            .add(FORMATTER.format(row, column))
    }
}

firstCol = true
map.entrySet().each { e ->
    if (!firstCol) {
        OUT.append("\n  AND ")
    }
    firstRow = true
    OUT.append(e.key)
        .append(" IN (")
    e.getValue().each { value ->
        def q = !value.isNumber()
        OUT.append(firstRow ? "" : SEPARATOR)
            .append(q ? QUOTE : "")
            .append(value.replace(QUOTE, QUOTE + QUOTE))
            .append(q ? QUOTE : "")
        firstRow = false
    }
    OUT.append(")")
    firstCol = false
}