@mkven/samples-generation 0.1.0

package/dist/generator/clickhouse-generator.js

@@ -0,0 +1,452 @@
+import { createClient } from "@clickhouse/client";
+import { BaseDataGenerator } from "./base-generator.js";
+import { escapeClickHouseIdentifier } from "./escape.js";
+import { getLookupTableName } from "./utils.js";
+function columnTypeToClickHouse(column) {
+    const baseType = (() => {
+        switch (column.type) {
+            case "integer":
+                return "Int32";
+            case "bigint":
+                return "Int64";
+            case "float":
+                return "Float64";
+            case "string":
+                return "String";
+            case "boolean":
+                return "Bool";
+            case "datetime":
+                return "DateTime";
+            case "date":
+                return "Date";
+        }
+    })();
+    return column.nullable ? `Nullable(${baseType})` : baseType;
+}
+/**
+ * Wrap an expression to return NULL with given probability (ClickHouse)
+ */
+function wrapWithNullCheck(expr, nullProbability) {
+    if (nullProbability <= 0)
+        return expr;
+    if (nullProbability >= 1)
+        return "NULL";
+    // rand() returns UInt32 (0 to 4294967295), divide to get 0-1 range
+    return `if(rand() / 4294967295.0 < ${String(nullProbability)}, NULL, ${expr})`;
+}
+/**
+ * Convert a generator config to a ClickHouse SQL expression
+ */
+export function generatorToClickHouseExpr(gen, seqExpr) {
+    switch (gen.kind) {
+        case "sequence": {
+            const start = gen.start ?? 1;
+            const step = gen.step ?? 1;
+            return `(${String(start)} - 1 + ${seqExpr} * ${String(step)})`;
+        }
+        case "randomInt":
+            return `toInt32(randUniform(${String(gen.min)}, ${String(gen.max + 1)}))`;
+        case "randomFloat": {
+            const precision = gen.precision ?? 2;
+            return `round(randUniform(${String(gen.min)}, ${String(gen.max)}), ${String(precision)})`;
+        }
+        case "randomString": {
+            const len = gen.length;
+            return `randomPrintableASCII(${String(len)})`;
+        }
+        case "choice": {
+            const values = gen.values;
+            const arr = values.map((v) => typeof v === "string" ? `'${v}'` : String(v));
+            return `[${arr.join(", ")}][toUInt32(rand() % ${String(arr.length)}) + 1]`;
+        }
+        case "choiceByLookup": {
+            // For ClickHouse, we'll use a subquery that loads the array once
+            // The CTE approach doesn't work well, so we use arrayElement
+            const cteName = getLookupTableName(gen.values);
+            return `${cteName}_arr[toUInt32(rand() % length(${cteName}_arr)) + 1]`;
+        }
+        case "constant": {
+            const val = gen.value;
+            return typeof val === "string" ? `'${val}'` : String(val);
+        }
+        case "datetime": {
+            const from = gen.from ?? new Date("2020-01-01");
+            const to = gen.to ?? new Date();
+            const fromTs = Math.floor(from.getTime() / 1000);
+            const toTs = Math.floor(to.getTime() / 1000);
+            return `toDateTime(${String(fromTs)} + rand() % ${String(toTs - fromTs)})`;
+        }
+        case "uuid":
+            return "generateUUIDv4()";
+    }
+}
+export class ClickHouseDataGenerator extends BaseDataGenerator {
+    config;
+    name = "clickhouse";
+    client = null;
+    constructor(config) {
+        super();
+        this.config = config;
+    }
+    connect() {
+        this.client = createClient({
+            url: `http://${this.config.host}:${String(this.config.port)}`,
+            username: this.config.username,
+            password: this.config.password,
+            database: this.config.database,
+            // Increase timeout for large data generation (6 hours for 10B+ rows)
+            request_timeout: 21_600_000,
+        });
+        return Promise.resolve();
+    }
+    async disconnect() {
+        if (this.client) {
+            await this.client.close();
+            this.client = null;
+        }
+    }
+    getClient() {
+        if (!this.client) {
+            throw new Error("Not connected to ClickHouse");
+        }
+        return this.client;
+    }
+    async createTable(table) {
+        const client = this.getClient();
+        const columns = table.columns
+            .map((col) => `${escapeClickHouseIdentifier(col.name)} ${columnTypeToClickHouse(col)}`)
+            .join(", ");
+        // Find the first column for ORDER BY (ClickHouse requires it for MergeTree)
+        const firstColumn = table.columns[0];
+        if (!firstColumn) {
+            throw new Error("Table must have at least one column");
+        }
+        const orderByColumn = escapeClickHouseIdentifier(firstColumn.name);
+        const escapedTableName = escapeClickHouseIdentifier(table.name);
+        await client.command({
+            query: `CREATE TABLE IF NOT EXISTS ${escapedTableName} (${columns}) ENGINE = MergeTree() ORDER BY ${orderByColumn}`,
+        });
+    }
+    async truncateTable(tableName) {
+        const client = this.getClient();
+        await client.command({
+            query: `TRUNCATE TABLE ${escapeClickHouseIdentifier(tableName)}`,
+        });
+    }
+    async dropTable(tableName) {
+        const client = this.getClient();
+        await client.command({
+            query: `DROP TABLE IF EXISTS ${escapeClickHouseIdentifier(tableName)}`,
+        });
+    }
+    async generateNative(table, rowCount, startSequence) {
+        const client = this.getClient();
+        const escapedTableName = escapeClickHouseIdentifier(table.name);
+        // Collect choiceByLookup arrays for WITH clause
+        // ClickHouse syntax: WITH [values] AS name
+        const arrayDefs = [];
+        for (const col of table.columns) {
+            if (col.generator.kind === "choiceByLookup") {
+                const gen = col.generator;
+                const arrName = `${getLookupTableName(gen.values)}_arr`;
+                const valuesLiteral = gen.values
+                    .map((v) => `'${v.replace(/'/g, "\\'")}'`)
+                    .join(", ");
+                arrayDefs.push(`[${valuesLiteral}] AS ${arrName}`);
+            }
+        }
+        // Build column list and expressions
+        const columns = table.columns.map((c) => escapeClickHouseIdentifier(c.name));
+        const seqExpr = `(number + ${String(startSequence)})`;
+        const expressions = table.columns.map((col) => {
+            let expr = generatorToClickHouseExpr(col.generator, seqExpr);
+            // Apply null probability if specified
+            if (col.nullable && col.nullProbability && col.nullProbability > 0) {
+                expr = wrapWithNullCheck(expr, col.nullProbability);
+            }
+            return expr;
+        });
+        // Build WITH clause if we have lookup arrays
+        const withClause = arrayDefs.length > 0 ? `${arrayDefs.join(", ")} ` : "";
+        const insertSql = `
+      INSERT INTO ${escapedTableName} (${columns.join(", ")})
+      ${withClause.length > 0 ? `WITH ${withClause}` : ""}SELECT ${expressions.join(", ")}
+      FROM numbers(${String(rowCount)})
+    `;
+        await client.command({ query: insertSql });
+    }
+    async queryRows(tableName, limit = 100) {
+        const client = this.getClient();
+        const result = await client.query({
+            query: `SELECT * FROM ${escapeClickHouseIdentifier(tableName)} LIMIT {limit:UInt32}`,
+            query_params: { limit },
+            format: "JSONEachRow",
+        });
+        return result.json();
+    }
+    async countRows(tableName) {
+        const client = this.getClient();
+        const result = await client.query({
+            query: `SELECT COUNT(*) as count FROM ${escapeClickHouseIdentifier(tableName)}`,
+            format: "JSONEachRow",
+        });
+        const rows = await result.json();
+        const firstRow = rows[0];
+        if (!firstRow) {
+            return 0;
+        }
+        return Number(firstRow.count);
+    }
+    async getMaxValue(tableName, columnName) {
+        const client = this.getClient();
+        const result = await client.query({
+            query: `SELECT MAX(${escapeClickHouseIdentifier(columnName)}) as max_val FROM ${escapeClickHouseIdentifier(tableName)}`,
+            format: "JSONEachRow",
+        });
+        const rows = await result.json();
+        const firstRow = rows[0];
+        if (!firstRow?.max_val) {
+            return null;
+        }
+        return Number(firstRow.max_val);
+    }
+    async getTableSize(tableName) {
+        const client = this.getClient();
+        const result = await client.query({
+            query: `SELECT total_bytes FROM system.tables WHERE database = {database:String} AND name = {table:String}`,
+            query_params: { database: this.config.database, table: tableName },
+            format: "JSONEachRow",
+        });
+        const rows = await result.json();
+        const firstRow = rows[0];
+        if (!firstRow) {
+            return null;
+        }
+        return Number(firstRow.total_bytes);
+    }
+    async optimize(tableName) {
+        const client = this.getClient();
+        // OPTIMIZE TABLE FINAL merges all parts into one for MergeTree engines
+        await client.command({
+            query: `OPTIMIZE TABLE ${escapeClickHouseIdentifier(tableName)} FINAL`,
+            clickhouse_settings: {
+                wait_end_of_query: 1,
+            },
+        });
+    }
+    async applyTransformations(tableName, transformations) {
+        if (transformations.length === 0)
+            return;
+        const client = this.getClient();
+        const escapedTable = escapeClickHouseIdentifier(tableName);
+        const setClauses = [];
+        for (const t of transformations) {
+            switch (t.kind) {
+                case "template": {
+                    const escapedCol = escapeClickHouseIdentifier(t.column);
+                    // Replace {column_name} with column references
+                    let expr = `'${t.template.replace(/'/g, "\\'")}'`;
+                    const refs = t.template.match(/\{([^}]+)\}/g) ?? [];
+                    for (const ref of refs) {
+                        const colName = ref.slice(1, -1);
+                        const colRef = escapeClickHouseIdentifier(colName);
+                        // Replace {col} with concatenation
+                        expr = expr.replace(`'{${colName}}'`, `' || toString(${colRef}) || '`);
+                        expr = expr.replace(`{${colName}}`, `' || toString(${colRef}) || '`);
+                    }
+                    // Clean up empty string concatenations
+                    expr = expr.replace(/^'' \|\| /, "").replace(/ \|\| ''$/, "");
+                    expr = expr.replace(/' \|\| '' \|\| '/g, "' || '");
+                    if (t.lowercase) {
+                        expr = `lower(${expr})`;
+                    }
+                    setClauses.push(`${escapedCol} = ${expr}`);
+                    break;
+                }
+                case "mutate": {
+                    const escapedCol = escapeClickHouseIdentifier(t.column);
+                    // Random string mutation using ClickHouse functions
+                    const { probability, operations } = t;
+                    // rand() returns UInt32, divide to get 0-1
+                    const randomPos = `toUInt32(floor(rand() / 4294967295.0 * length(${escapedCol}))) + 1`;
+                    // Build mutation expressions for each operation
+                    const mutationExprs = {
+                        replace: `concat(substring(${escapedCol}, 1, ${randomPos} - 1), 'X', substring(${escapedCol}, ${randomPos} + 1))`,
+                        delete: `concat(substring(${escapedCol}, 1, ${randomPos} - 1), substring(${escapedCol}, ${randomPos} + 1))`,
+                        insert: `concat(substring(${escapedCol}, 1, ${randomPos}), 'X', substring(${escapedCol}, ${randomPos} + 1))`,
+                    };
+                    // Build expression that randomly picks from available operations
+                    let mutateExpr;
+                    if (operations.length === 1) {
+                        // We know operations[0] exists since length === 1
+                        mutateExpr = mutationExprs[operations[0]]; // eslint-disable-line @typescript-eslint/no-non-null-assertion
+                    }
+                    else {
+                        // Use multiIf with rand() to pick operation
+                        const conditions = [];
+                        for (let i = 0; i < operations.length - 1; i++) {
+                            const threshold = (i + 1) / operations.length;
+                            // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                            const op = operations[i];
+                            conditions.push(`rand() / 4294967295.0 < ${String(threshold)}, ${mutationExprs[op]}`);
+                        }
+                        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                        const lastOp = operations[operations.length - 1];
+                        conditions.push(mutationExprs[lastOp]);
+                        mutateExpr = `multiIf(${conditions.join(", ")})`;
+                    }
+                    setClauses.push(`${escapedCol} = if(rand() / 4294967295.0 < ${String(probability)}, ${mutateExpr}, ${escapedCol})`);
+                    break;
+                }
+                case "lookup": {
+                    // ClickHouse ALTER TABLE UPDATE does not support correlated subqueries
+                    // We need to use the table swap approach: CREATE AS + INSERT SELECT + RENAME
+                    // Handle lookup transformations separately
+                    await this.applyLookupTransformation(tableName, t);
+                    // Don't add to setClauses - already handled
+                    break;
+                }
+                case "swap": {
+                    // Swap handled separately via applySwapTransformation
+                    // ClickHouse evaluates each rand() separately, so we need table swap approach
+                    // to ensure same random value is used for both columns
+                    break;
+                }
+            }
+        }
+        // Handle swap transformations separately for ClickHouse (need same random for both columns)
+        const swapTransformations = transformations.filter((t) => t.kind === "swap");
+        if (swapTransformations.length > 0) {
+            // Batch all swaps into a single table swap operation
+            await this.applySwapTransformations(tableName, swapTransformations);
+        }
+        // Only run ALTER TABLE UPDATE if there are SET clauses
+        if (setClauses.length === 0)
+            return;
+        // ClickHouse uses ALTER TABLE UPDATE syntax
+        const updateSql = `ALTER TABLE ${escapedTable} UPDATE ${setClauses.join(", ")} WHERE 1`;
+        await client.command({
+            query: updateSql,
+            clickhouse_settings: {
+                wait_end_of_query: 1,
+                mutations_sync: "2", // Wait for all replicas
+            },
+        });
+    }
+    /**
+     * Apply multiple swap transformations in a single table swap operation.
+     * Each swap gets its own random value to ensure independent swap decisions.
+     */
+    async applySwapTransformations(tableName, swaps) {
+        const client = this.getClient();
+        const escapedTable = escapeClickHouseIdentifier(tableName);
+        // Create unique suffix for temp tables
+        const uniqueSuffix = `${String(Date.now())}_${Math.random().toString(36).slice(2, 8)}`;
+        const tempTable = escapeClickHouseIdentifier(`${tableName}_swap_temp_${uniqueSuffix}`);
+        const oldTable = escapeClickHouseIdentifier(`${tableName}_swap_old_${uniqueSuffix}`);
+        // Get table structure
+        const structResult = await client.query({
+            query: `SHOW CREATE TABLE ${escapedTable}`,
+            format: "TabSeparated",
+        });
+        const createStatement = await structResult.text();
+        // Create temp table with same structure
+        const tempCreateSql = createStatement.replace(new RegExp(`CREATE TABLE [^\\s]*(${tableName.replace(/[.*+?^${}()|[\]\\]/g, "\\$&")})[^\\s]*`), `CREATE TABLE ${tempTable}`);
+        await client.command({ query: tempCreateSql });
+        // Get all columns
+        const colsResult = await client.query({
+            query: `SELECT name FROM system.columns WHERE database = currentDatabase() AND table = '${tableName}'`,
+            format: "JSONEachRow",
+        });
+        const colsData = await colsResult.json();
+        const allColumns = colsData.map((c) => escapeClickHouseIdentifier(c.name));
+        // Build swap info map: column -> { otherColumn, probability, randVar }
+        const swapInfo = new Map();
+        swaps.forEach((swap, i) => {
+            const col1 = escapeClickHouseIdentifier(swap.column1);
+            const col2 = escapeClickHouseIdentifier(swap.column2);
+            const prob = String(swap.probability);
+            const randVar = `_swap_rand_${String(i)}`;
+            swapInfo.set(col1, { otherCol: col2, prob, randVar });
+            swapInfo.set(col2, { otherCol: col1, prob, randVar });
+        });
+        // Build SELECT with swap logic
+        const selectColumns = allColumns.map((col) => {
+            const info = swapInfo.get(col);
+            if (info) {
+                return `if(${info.randVar} < ${info.prob}, ${info.otherCol}, ${col}) as ${col}`;
+            }
+            return col;
+        });
+        // Build random variable definitions for subquery
+        const randVars = swaps
+            .map((_, i) => `rand() / 4294967295.0 as _swap_rand_${String(i)}`)
+            .join(", ");
+        // Use subquery to compute random values once per row
+        const insertSql = `
+      INSERT INTO ${tempTable}
+      SELECT ${selectColumns.join(", ")}
+      FROM (
+        SELECT *, ${randVars}
+        FROM ${escapedTable}
+      )
+    `;
+        await client.command({ query: insertSql });
+        // Swap tables
+        await client.command({
+            query: `RENAME TABLE ${escapedTable} TO ${oldTable}, ${tempTable} TO ${escapedTable}`,
+        });
+        // Drop old table
+        await client.command({ query: `DROP TABLE ${oldTable}` });
+    }
+    /**
+     * Apply a lookup transformation using ClickHouse's table swap approach.
+     * Since ClickHouse doesn't support correlated subqueries in ALTER TABLE UPDATE,
+     * we use: CREATE TABLE new -> INSERT SELECT with JOIN -> RENAME swap
+     */
+    async applyLookupTransformation(tableName, t) {
+        const client = this.getClient();
+        const escapedTable = escapeClickHouseIdentifier(tableName);
+        const escapedCol = escapeClickHouseIdentifier(t.column);
+        const fromTable = escapeClickHouseIdentifier(t.fromTable);
+        const fromCol = escapeClickHouseIdentifier(t.fromColumn);
+        const targetJoinCol = escapeClickHouseIdentifier(t.joinOn.targetColumn);
+        const lookupJoinCol = escapeClickHouseIdentifier(t.joinOn.lookupColumn);
+        // Use unique suffix to avoid conflicts with concurrent runs
+        const uniqueSuffix = `${String(Date.now())}_${Math.random().toString(36).slice(2, 8)}`;
+        const tempTableName = `${tableName}_lookup_temp_${uniqueSuffix}`;
+        const oldTableName = `${tableName}_lookup_old_${uniqueSuffix}`;
+        const escapedTempTable = escapeClickHouseIdentifier(tempTableName);
+        const escapedOldTable = escapeClickHouseIdentifier(oldTableName);
+        // 1. Create new table with same structure
+        await client.command({
+            query: `CREATE TABLE ${escapedTempTable} AS ${escapedTable}`,
+            clickhouse_settings: { wait_end_of_query: 1 },
+        });
+        // 2. INSERT SELECT with LEFT JOIN, using REPLACE to update the lookup column
+        // The REPLACE syntax replaces the column value in the SELECT *
+        const insertSql = `
+      INSERT INTO ${escapedTempTable}
+      SELECT
+        t.* REPLACE (coalesce(s.${fromCol}, t.${escapedCol}) AS ${escapedCol})
+      FROM ${escapedTable} t
+      LEFT JOIN ${fromTable} s ON t.${targetJoinCol} = s.${lookupJoinCol}
+    `;
+        await client.command({
+            query: insertSql,
+            clickhouse_settings: { wait_end_of_query: 1 },
+        });
+        // 3. RENAME tables to swap (atomic operation)
+        // target -> target_old, target_new -> target
+        await client.command({
+            query: `RENAME TABLE ${escapedTable} TO ${escapedOldTable}, ${escapedTempTable} TO ${escapedTable}`,
+            clickhouse_settings: { wait_end_of_query: 1 },
+        });
+        // 4. Drop the old table
+        await client.command({
+            query: `DROP TABLE IF EXISTS ${escapedOldTable}`,
+            clickhouse_settings: { wait_end_of_query: 1 },
+        });
+    }
+}
+//# sourceMappingURL=clickhouse-generator.js.map

package/dist/generator/clickhouse-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"clickhouse-generator.js","sourceRoot":"","sources":["../../src/generator/clickhouse-generator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAyB,MAAM,oBAAoB,CAAC;AAWzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,0BAA0B,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAUhD,SAAS,sBAAsB,CAAC,MAAoB;IAClD,MAAM,QAAQ,GAAG,CAAC,GAAW,EAAE;QAC7B,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;YACpB,KAAK,SAAS;gBACZ,OAAO,OAAO,CAAC;YACjB,KAAK,QAAQ;gBACX,OAAO,OAAO,CAAC;YACjB,KAAK,OAAO;gBACV,OAAO,SAAS,CAAC;YACnB,KAAK,QAAQ;gBACX,OAAO,QAAQ,CAAC;YAClB,KAAK,SAAS;gBACZ,OAAO,MAAM,CAAC;YAChB,KAAK,UAAU;gBACb,OAAO,UAAU,CAAC;YACpB,KAAK,MAAM;gBACT,OAAO,MAAM,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,EAAE,CAAC;IACL,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,YAAY,QAAQ,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC;AAC9D,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,IAAY,EAAE,eAAuB;IAC9D,IAAI,eAAe,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IACtC,IAAI,eAAe,IAAI,CAAC;QAAE,OAAO,MAAM,CAAC;IACxC,mEAAmE;IACnE,OAAO,8BAA8B,MAAM,CAAC,eAAe,CAAC,WAAW,IAAI,GAAG,CAAC;AACjF,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,yBAAyB,CACvC,GAAoB,EACpB,OAAe;IAEf,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;QACjB,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC;YAC7B,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC;YAC3B,OAAO,IAAI,MAAM,CAAC,KAAK,CAAC,UAAU,OAAO,MAAM,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC;QACjE,CAAC;QACD,KAAK,WAAW;YACd,OAAO,uBAAuB,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,MAAM,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC;QAC5E,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,MAAM,SAAS,GAAG,GAAG,CAAC,SAAS,IAAI,CAAC,CAAC;YACrC,OAAO,qBAAqB,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC;QAC5F,CAAC;QACD,KAAK,cAAc,CAAC,CAAC,CAAC;YACpB,MAAM,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC;YACvB,OAAO,wBAAwB,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC;QAChD,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC;YAC1B,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAC3B,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAC7C,CAAC;YACF,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,uBAAuB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC;QAC7E,CAAC;QACD,KAAK,gBAAgB,CAAC,CAAC,CAAC;YACtB,iEAAiE;YACjE,6DAA6D;YAC7D,MAAM,OAAO,GAAG,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC/C,OAAO,GAAG,OAAO,iCAAiC,OAAO,aAAa,CAAC;QACzE,CAAC;QACD,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC;YACtB,OAAO,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,GAAG,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC5D,CAAC;QACD,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC;YAChD,MAAM,EAAE,GAAG,GAAG,CAAC,EAAE,IAAI,IAAI,IAAI,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;YAC7C,OAAO,cAAc,MAAM,CAAC,MAAM,CAAC,eAAe,MAAM,CAAC,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC;QAC7E,CAAC;QACD,KAAK,MAAM;YACT,OAAO,kBAAkB,CAAC;IAC9B,CAAC;AACH,CAAC;AAED,MAAM,OAAO,uBAAwB,SAAQ,iBAAiB;IAIxC;IAHX,IAAI,GAAG,YAAY,CAAC;IACrB,MAAM,GAA4B,IAAI,CAAC;IAE/C,YAAoB,MAAwB;QAC1C,KAAK,EAAE,CAAC;QADU,WAAM,GAAN,MAAM,CAAkB;IAE5C,CAAC;IAED,OAAO;QACL,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC;YACzB,GAAG,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;YAC7D,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,qEAAqE;YACrE,eAAe,EAAE,UAAU;SAC5B,CAAC,CAAC;QACH,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAC1B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACrB,CAAC;IACH,CAAC;IAEO,SAAS;QACf,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAkB;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO;aAC1B,GAAG,CACF,CAAC,GAAG,EAAE,EAAE,CACN,GAAG,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,sBAAsB,CAAC,GAAG,CAAC,EAAE,CAC3E;aACA,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,4EAA4E;QAC5E,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACrC,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;QACD,MAAM,aAAa,GAAG,0BAA0B,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACnE,MAAM,gBAAgB,GAAG,0BAA0B,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEhE,MAAM,MAAM,CAAC,OAAO,CAAC;YACnB,KAAK,EAAE,8BAA8B,gBAAgB,KAAK,OAAO,mCAAmC,aAAa,EAAE;SACpH,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,SAAiB;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,MAAM,CAAC,OAAO,CAAC;YACnB,KAAK,EAAE,kBAAkB,0BAA0B,CAAC,SAAS,CAAC,EAAE;SACjE,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,SAAiB;QAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,MAAM,CAAC,OAAO,CAAC;YACnB,KAAK,EAAE,wBAAwB,0BAA0B,CAAC,SAAS,CAAC,EAAE;SACvE,CAAC,CAAC;IACL,CAAC;IAES,KAAK,CAAC,cAAc,CAC5B,KAAkB,EAClB,QAAgB,EAChB,aAAqB;QAErB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,gBAAgB,GAAG,0BAA0B,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEhE,gDAAgD;QAChD,2CAA2C;QAC3C,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAChC,IAAI,GAAG,CAAC,SAAS,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;gBAC5C,MAAM,GAAG,GAAG,GAAG,CAAC,SAAS,CAAC;gBAC1B,MAAM,OAAO,GAAG,GAAG,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC;gBACxD,MAAM,aAAa,GAAG,GAAG,CAAC,MAAM;qBAC7B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC;qBACzC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACd,SAAS,CAAC,IAAI,CAAC,IAAI,aAAa,QAAQ,OAAO,EAAE,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;QAED,oCAAoC;QACpC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CACtC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,CACnC,CAAC;QACF,MAAM,OAAO,GAAG,aAAa,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC;QACtD,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YAC5C,IAAI,IAAI,GAAG,yBAAyB,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YAC7D,sCAAsC;YACtC,IAAI,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,eAAe,IAAI,GAAG,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;gBACnE,IAAI,GAAG,iBAAiB,CAAC,IAAI,EAAE,GAAG,CAAC,eAAe,CAAC,CAAC;YACtD,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,6CAA6C;QAC7C,MAAM,UAAU,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QAE1E,MAAM,SAAS,GAAG;oBACF,gBAAgB,KAAK,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;QACnD,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;qBACpE,MAAM,CAAC,QAAQ,CAAC;KAChC,CAAC;QAEF,MAAM,MAAM,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,SAAiB,EAAE,KAAK,GAAG,GAAG;QAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;YAChC,KAAK,EAAE,iBAAiB,0BAA0B,CAAC,SAAS,CAAC,uBAAuB;YACpF,YAAY,EAAE,EAAE,KAAK,EAAE;YACvB,MAAM,EAAE,aAAa;SACtB,CAAC,CAAC;QACH,OAAO,MAAM,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,SAAiB;QAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;YAChC,KAAK,EAAE,iCAAiC,0BAA0B,CAAC,SAAS,CAAC,EAAE;YAC/E,MAAM,EAAE,aAAa;SACtB,CAAC,CAAC;QACH,MAAM,IAAI,GAAwB,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACzB,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,CAAC,CAAC;QACX,CAAC;QACD,OAAO,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,WAAW,CACf,SAAiB,EACjB,UAAkB;QAElB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;YAChC,KAAK,EAAE,cAAc,0BAA0B,CAAC,UAAU,CAAC,qBAAqB,0BAA0B,CAAC,SAAS,CAAC,EAAE;YACvH,MAAM,EAAE,aAAa;SACtB,CAAC,CAAC;QACH,MAAM,IAAI,GAAiC,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC/D,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACzB,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,CAAC;YACvB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,SAAiB;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;YAChC,KAAK,EAAE,oGAAoG;YAC3G,YAAY,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE;YAClE,MAAM,EAAE,aAAa;SACtB,CAAC,CAAC;QACH,MAAM,IAAI,GAA8B,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC5D,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACzB,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,SAAiB;QAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,uEAAuE;QACvE,MAAM,MAAM,CAAC,OAAO,CAAC;YACnB,KAAK,EAAE,kBAAkB,0BAA0B,CAAC,SAAS,CAAC,QAAQ;YACtE,mBAAmB,EAAE;gBACnB,iBAAiB,EAAE,CAAC;aACrB;SACF,CAAC,CAAC;IACL,CAAC;IAES,KAAK,CAAC,oBAAoB,CAClC,SAAiB,EACjB,eAAiC;QAEjC,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEzC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,YAAY,GAAG,0BAA0B,CAAC,SAAS,CAAC,CAAC;QAC3D,MAAM,UAAU,GAAa,EAAE,CAAC;QAEhC,KAAK,MAAM,CAAC,IAAI,eAAe,EAAE,CAAC;YAChC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;gBACf,KAAK,UAAU,CAAC,CAAC,CAAC;oBAChB,MAAM,UAAU,GAAG,0BAA0B,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;oBACxD,+CAA+C;oBAC/C,IAAI,IAAI,GAAG,IAAI,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC;oBAClD,MAAM,IAAI,GAAG,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;oBACpD,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;wBACvB,MAAM,OAAO,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;wBACjC,MAAM,MAAM,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;wBACnD,mCAAmC;wBACnC,IAAI,GAAG,IAAI,CAAC,OAAO,CACjB,KAAK,OAAO,IAAI,EAChB,iBAAiB,MAAM,QAAQ,CAChC,CAAC;wBACF,IAAI,GAAG,IAAI,CAAC,OAAO,CACjB,IAAI,OAAO,GAAG,EACd,iBAAiB,MAAM,QAAQ,CAChC,CAAC;oBACJ,CAAC;oBACD,uCAAuC;oBACvC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;oBAC9D,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,QAAQ,CAAC,CAAC;oBACnD,IAAI,CAAC,CAAC,SAAS,EAAE,CAAC;wBAChB,IAAI,GAAG,SAAS,IAAI,GAAG,CAAC;oBAC1B,CAAC;oBACD,UAAU,CAAC,IAAI,CAAC,GAAG,UAAU,MAAM,IAAI,EAAE,CAAC,CAAC;oBAC3C,MAAM;gBACR,CAAC;gBACD,KAAK,QAAQ,CAAC,CAAC,CAAC;oBACd,MAAM,UAAU,GAAG,0BAA0B,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;oBACxD,oDAAoD;oBACpD,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC;oBACtC,2CAA2C;oBAC3C,MAAM,SAAS,GAAG,iDAAiD,UAAU,SAAS,CAAC;oBAEvF,gDAAgD;oBAChD,MAAM,aAAa,GAAsC;wBACvD,OAAO,EAAE,oBAAoB,UAAU,QAAQ,SAAS,yBAAyB,UAAU,KAAK,SAAS,QAAQ;wBACjH,MAAM,EAAE,oBAAoB,UAAU,QAAQ,SAAS,oBAAoB,UAAU,KAAK,SAAS,QAAQ;wBAC3G,MAAM,EAAE,oBAAoB,UAAU,QAAQ,SAAS,qBAAqB,UAAU,KAAK,SAAS,QAAQ;qBAC7G,CAAC;oBAEF,iEAAiE;oBACjE,IAAI,UAAkB,CAAC;oBACvB,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBAC5B,kDAAkD;wBAClD,UAAU,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,+DAA+D;oBAC7G,CAAC;yBAAM,CAAC;wBACN,4CAA4C;wBAC5C,MAAM,UAAU,GAAa,EAAE,CAAC;wBAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;4BAC/C,MAAM,SAAS,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,UAAU,CAAC,MAAM,CAAC;4BAC9C,oEAAoE;4BACpE,MAAM,EAAE,GAAG,UAAU,CAAC,CAAC,CAAE,CAAC;4BAC1B,UAAU,CAAC,IAAI,CACb,2BAA2B,MAAM,CAAC,SAAS,CAAC,KAAK,aAAa,CAAC,EAAE,CAAC,EAAE,CACrE,CAAC;wBACJ,CAAC;wBACD,oEAAoE;wBACpE,MAAM,MAAM,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAE,CAAC;wBAClD,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC;wBACvC,UAAU,GAAG,WAAW,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;oBACnD,CAAC;oBAED,UAAU,CAAC,IAAI,CACb,GAAG,UAAU,iCAAiC,MAAM,CAAC,WAAW,CAAC,KAAK,UAAU,KAAK,UAAU,GAAG,CACnG,CAAC;oBACF,MAAM;gBACR,CAAC;gBACD,KAAK,QAAQ,CAAC,CAAC,CAAC;oBACd,uEAAuE;oBACvE,6EAA6E;oBAC7E,2CAA2C;oBAC3C,MAAM,IAAI,CAAC,yBAAyB,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;oBACnD,4CAA4C;oBAC5C,MAAM;gBACR,CAAC;gBACD,KAAK,MAAM,CAAC,CAAC,CAAC;oBACZ,sDAAsD;oBACtD,8EAA8E;oBAC9E,uDAAuD;oBACvD,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;QAED,4FAA4F;QAC5F,MAAM,mBAAmB,GAAG,eAAe,CAAC,MAAM,CAChD,CAAC,CAAC,EAA2B,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAClD,CAAC;QAEF,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnC,qDAAqD;YACrD,MAAM,IAAI,CAAC,wBAAwB,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;QACtE,CAAC;QAED,uDAAuD;QACvD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEpC,4CAA4C;QAC5C,MAAM,SAAS,GAAG,eAAe,YAAY,WAAW,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;QACxF,MAAM,MAAM,CAAC,OAAO,CAAC;YACnB,KAAK,EAAE,SAAS;YAChB,mBAAmB,EAAE;gBACnB,iBAAiB,EAAE,CAAC;gBACpB,cAAc,EAAE,GAAG,EAAE,wBAAwB;aAC9C;SACF,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,wBAAwB,CACpC,SAAiB,EACjB,KAA2B;QAE3B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,YAAY,GAAG,0BAA0B,CAAC,SAAS,CAAC,CAAC;QAE3D,uCAAuC;QACvC,MAAM,YAAY,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;QACvF,MAAM,SAAS,GAAG,0BAA0B,CAC1C,GAAG,SAAS,cAAc,YAAY,EAAE,CACzC,CAAC;QACF,MAAM,QAAQ,GAAG,0BAA0B,CACzC,GAAG,SAAS,aAAa,YAAY,EAAE,CACxC,CAAC;QAEF,sBAAsB;QACtB,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;YACtC,KAAK,EAAE,qBAAqB,YAAY,EAAE;YAC1C,MAAM,EAAE,cAAc;SACvB,CAAC,CAAC;QACH,MAAM,eAAe,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,CAAC;QAElD,wCAAwC;QACxC,MAAM,aAAa,GAAG,eAAe,CAAC,OAAO,CAC3C,IAAI,MAAM,CACR,wBAAwB,SAAS,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,UAAU,CACnF,EACD,gBAAgB,SAAS,EAAE,CAC5B,CAAC;QACF,MAAM,MAAM,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC;QAE/C,kBAAkB;QAClB,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;YACpC,KAAK,EAAE,mFAAmF,SAAS,GAAG;YACtG,MAAM,EAAE,aAAa;SACtB,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAC,IAAI,EAAoB,CAAC;QAC3D,MAAM,UAAU,GAAI,QAA+B,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAC5D,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,CACnC,CAAC;QAEF,uEAAuE;QACvE,MAAM,QAAQ,GAAG,IAAI,GAAG,EAGrB,CAAC;QACJ,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;YACxB,MAAM,IAAI,GAAG,0BAA0B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACtD,MAAM,IAAI,GAAG,0BAA0B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACtD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACtC,MAAM,OAAO,GAAG,cAAc,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1C,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;YACtD,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,+BAA+B;QAC/B,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YAC3C,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC/B,IAAI,IAAI,EAAE,CAAC;gBACT,OAAO,MAAM,IAAI,CAAC,OAAO,MAAM,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,QAAQ,KAAK,GAAG,QAAQ,GAAG,EAAE,CAAC;YAClF,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,iDAAiD;QACjD,MAAM,QAAQ,GAAG,KAAK;aACnB,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,uCAAuC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;aACjE,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,qDAAqD;QACrD,MAAM,SAAS,GAAG;oBACF,SAAS;eACd,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC;;oBAEnB,QAAQ;eACb,YAAY;;KAEtB,CAAC;QACF,MAAM,MAAM,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAE3C,cAAc;QACd,MAAM,MAAM,CAAC,OAAO,CAAC;YACnB,KAAK,EAAE,gBAAgB,YAAY,OAAO,QAAQ,KAAK,SAAS,OAAO,YAAY,EAAE;SACtF,CAAC,CAAC;QAEH,iBAAiB;QACjB,MAAM,MAAM,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,cAAc,QAAQ,EAAE,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,yBAAyB,CACrC,SAAiB,EACjB,CAAuB;QAEvB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,YAAY,GAAG,0BAA0B,CAAC,SAAS,CAAC,CAAC;QAC3D,MAAM,UAAU,GAAG,0BAA0B,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;QACxD,MAAM,SAAS,GAAG,0BAA0B,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,0BAA0B,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACzD,MAAM,aAAa,GAAG,0BAA0B,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACxE,MAAM,aAAa,GAAG,0BAA0B,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAExE,4DAA4D;QAC5D,MAAM,YAAY,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;QACvF,MAAM,aAAa,GAAG,GAAG,SAAS,gBAAgB,YAAY,EAAE,CAAC;QACjE,MAAM,YAAY,GAAG,GAAG,SAAS,eAAe,YAAY,EAAE,CAAC;QAC/D,MAAM,gBAAgB,GAAG,0BAA0B,CAAC,aAAa,CAAC,CAAC;QACnE,MAAM,eAAe,GAAG,0BAA0B,CAAC,YAAY,CAAC,CAAC;QAEjE,0CAA0C;QAC1C,MAAM,MAAM,CAAC,OAAO,CAAC;YACnB,KAAK,EAAE,gBAAgB,gBAAgB,OAAO,YAAY,EAAE;YAC5D,mBAAmB,EAAE,EAAE,iBAAiB,EAAE,CAAC,EAAE;SAC9C,CAAC,CAAC;QAEH,6EAA6E;QAC7E,+DAA+D;QAC/D,MAAM,SAAS,GAAG;oBACF,gBAAgB;;kCAEF,OAAO,OAAO,UAAU,QAAQ,UAAU;aAC/D,YAAY;kBACP,SAAS,WAAW,aAAa,QAAQ,aAAa;KACnE,CAAC;QACF,MAAM,MAAM,CAAC,OAAO,CAAC;YACnB,KAAK,EAAE,SAAS;YAChB,mBAAmB,EAAE,EAAE,iBAAiB,EAAE,CAAC,EAAE;SAC9C,CAAC,CAAC;QAEH,8CAA8C;QAC9C,6CAA6C;QAC7C,MAAM,MAAM,CAAC,OAAO,CAAC;YACnB,KAAK,EAAE,gBAAgB,YAAY,OAAO,eAAe,KAAK,gBAAgB,OAAO,YAAY,EAAE;YACnG,mBAAmB,EAAE,EAAE,iBAAiB,EAAE,CAAC,EAAE;SAC9C,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,MAAM,CAAC,OAAO,CAAC;YACnB,KAAK,EAAE,wBAAwB,eAAe,EAAE;YAChD,mBAAmB,EAAE,EAAE,iBAAiB,EAAE,CAAC,EAAE;SAC9C,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/generator/escape.d.ts

@@ -0,0 +1,18 @@
+/**
+ * Escape identifier for PostgreSQL using double quotes.
+ * Uses pg-format library for battle-tested escaping.
+ */
+export declare function escapePostgresIdentifier(name: string): string;
+/**
+ * Escape identifier for ClickHouse using backticks.
+ * Doubles any existing backticks in the name.
+ */
+export declare function escapeClickHouseIdentifier(name: string): string;
+/**
+ * Escape identifier for Trino using double quotes.
+ * Always quotes the identifier (unlike Postgres, Trino's $ suffix for
+ * metadata tables requires quoting).
+ * Doubles any existing double quotes in the name.
+ */
+export declare function escapeTrinoIdentifier(name: string): string;
+//# sourceMappingURL=escape.d.ts.map

package/dist/generator/escape.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"escape.d.ts","sourceRoot":"","sources":["../../src/generator/escape.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,wBAAgB,wBAAwB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAE7D;AAED;;;GAGG;AACH,wBAAgB,0BAA0B,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAE/D;AAED;;;;;GAKG;AACH,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAE1D"}

package/dist/generator/escape.js

@@ -0,0 +1,25 @@
+import format from "pg-format";
+/**
+ * Escape identifier for PostgreSQL using double quotes.
+ * Uses pg-format library for battle-tested escaping.
+ */
+export function escapePostgresIdentifier(name) {
+    return format.ident(name);
+}
+/**
+ * Escape identifier for ClickHouse using backticks.
+ * Doubles any existing backticks in the name.
+ */
+export function escapeClickHouseIdentifier(name) {
+    return `\`${name.replace(/`/g, "``")}\``;
+}
+/**
+ * Escape identifier for Trino using double quotes.
+ * Always quotes the identifier (unlike Postgres, Trino's $ suffix for
+ * metadata tables requires quoting).
+ * Doubles any existing double quotes in the name.
+ */
+export function escapeTrinoIdentifier(name) {
+    return `"${name.replace(/"/g, '""')}"`;
+}
+//# sourceMappingURL=escape.js.map

package/dist/generator/escape.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"escape.js","sourceRoot":"","sources":["../../src/generator/escape.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,WAAW,CAAC;AAE/B;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CAAC,IAAY;IACnD,OAAO,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;AAC5B,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,0BAA0B,CAAC,IAAY;IACrD,OAAO,KAAK,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC;AAC3C,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,qBAAqB,CAAC,IAAY;IAChD,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC;AACzC,CAAC"}

package/dist/generator/index.d.ts

@@ -0,0 +1,9 @@
+export type { DataGenerator, TableConfig, ColumnConfig, GenerateOptions, GenerateResult, GeneratedRow, ColumnType, GeneratorConfig, SequenceGenerator, RandomIntGenerator, RandomFloatGenerator, RandomStringGenerator, ChoiceGenerator, ConstantGenerator, DatetimeGenerator, UuidGenerator, ChoiceByLookupGenerator, Transformation, TemplateTransformation, MutateTransformation, LookupTransformation, TransformationBatch, TransformResult, } from "./types.js";
+export { formatBytes, formatDuration, getLookupTableName } from "./utils.js";
+export { BaseDataGenerator } from "./base-generator.js";
+export { escapePostgresIdentifier, escapeClickHouseIdentifier, escapeTrinoIdentifier, } from "./escape.js";
+export { PostgresDataGenerator, type PostgresConfig, } from "./postgres-generator.js";
+export { ClickHouseDataGenerator, type ClickHouseConfig, } from "./clickhouse-generator.js";
+export { SQLiteDataGenerator, type SQLiteConfig } from "./sqlite-generator.js";
+export { TrinoDataGenerator, type TrinoConfig } from "./trino-generator.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/generator/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/generator/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,aAAa,EACb,WAAW,EACX,YAAY,EACZ,eAAe,EACf,cAAc,EACd,YAAY,EACZ,UAAU,EACV,eAAe,EACf,iBAAiB,EACjB,kBAAkB,EAClB,oBAAoB,EACpB,qBAAqB,EACrB,eAAe,EACf,iBAAiB,EACjB,iBAAiB,EACjB,aAAa,EACb,uBAAuB,EACvB,cAAc,EACd,sBAAsB,EACtB,oBAAoB,EACpB,oBAAoB,EACpB,mBAAmB,EACnB,eAAe,GAChB,MAAM,YAAY,CAAC;AAEpB,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAE7E,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EACL,wBAAwB,EACxB,0BAA0B,EAC1B,qBAAqB,GACtB,MAAM,aAAa,CAAC;AAErB,OAAO,EACL,qBAAqB,EACrB,KAAK,cAAc,GACpB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,uBAAuB,EACvB,KAAK,gBAAgB,GACtB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,mBAAmB,EAAE,KAAK,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAC/E,OAAO,EAAE,kBAAkB,EAAE,KAAK,WAAW,EAAE,MAAM,sBAAsB,CAAC"}

package/dist/generator/index.js

@@ -0,0 +1,8 @@
+export { formatBytes, formatDuration, getLookupTableName } from "./utils.js";
+export { BaseDataGenerator } from "./base-generator.js";
+export { escapePostgresIdentifier, escapeClickHouseIdentifier, escapeTrinoIdentifier, } from "./escape.js";
+export { PostgresDataGenerator, } from "./postgres-generator.js";
+export { ClickHouseDataGenerator, } from "./clickhouse-generator.js";
+export { SQLiteDataGenerator } from "./sqlite-generator.js";
+export { TrinoDataGenerator } from "./trino-generator.js";
+//# sourceMappingURL=index.js.map

package/dist/generator/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/generator/index.ts"],"names":[],"mappings":"AA0BA,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAE7E,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EACL,wBAAwB,EACxB,0BAA0B,EAC1B,qBAAqB,GACtB,MAAM,aAAa,CAAC;AAErB,OAAO,EACL,qBAAqB,GAEtB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,uBAAuB,GAExB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,mBAAmB,EAAqB,MAAM,uBAAuB,CAAC;AAC/E,OAAO,EAAE,kBAAkB,EAAoB,MAAM,sBAAsB,CAAC"}

package/dist/generator/postgres-generator.d.ts

@@ -0,0 +1,33 @@
+import type { TableConfig, GeneratedRow, GeneratorConfig, Transformation } from "./types.js";
+import { BaseDataGenerator } from "./base-generator.js";
+export interface PostgresConfig {
+    host: string;
+    port: number;
+    database: string;
+    username: string;
+    password: string;
+}
+/**
+ * Convert a generator config to a Postgres SQL expression
+ */
+export declare function generatorToPostgresExpr(gen: GeneratorConfig, seqExpr: string): string;
+export declare class PostgresDataGenerator extends BaseDataGenerator {
+    private config;
+    readonly name = "postgres";
+    private sql;
+    constructor(config: PostgresConfig);
+    connect(): Promise<void>;
+    disconnect(): Promise<void>;
+    private getSql;
+    createTable(table: TableConfig): Promise<void>;
+    truncateTable(tableName: string): Promise<void>;
+    dropTable(tableName: string): Promise<void>;
+    protected generateNative(table: TableConfig, rowCount: number, startSequence: number): Promise<void>;
+    queryRows(tableName: string, limit?: number): Promise<GeneratedRow[]>;
+    countRows(tableName: string): Promise<number>;
+    getMaxValue(tableName: string, columnName: string): Promise<number | null>;
+    getTableSize(tableName: string): Promise<number | null>;
+    optimize(tableName: string): Promise<void>;
+    protected applyTransformations(tableName: string, transformations: Transformation[]): Promise<void>;
+}
+//# sourceMappingURL=postgres-generator.d.ts.map

package/dist/generator/postgres-generator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"postgres-generator.d.ts","sourceRoot":"","sources":["../../src/generator/postgres-generator.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,WAAW,EACX,YAAY,EAEZ,eAAe,EAEf,cAAc,EAGf,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAIxD,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;CAClB;AA8BD;;GAEG;AACH,wBAAgB,uBAAuB,CACrC,GAAG,EAAE,eAAe,EACpB,OAAO,EAAE,MAAM,GACd,MAAM,CA4CR;AAED,qBAAa,qBAAsB,SAAQ,iBAAiB;IAI9C,OAAO,CAAC,MAAM;IAH1B,QAAQ,CAAC,IAAI,cAAc;IAC3B,OAAO,CAAC,GAAG,CAAoB;gBAEX,MAAM,EAAE,cAAc;IAI1C,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAWlB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAOjC,OAAO,CAAC,MAAM;IAOR,WAAW,CAAC,KAAK,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IAe9C,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAK/C,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;cAKjC,cAAc,CAC5B,KAAK,EAAE,WAAW,EAClB,QAAQ,EAAE,MAAM,EAChB,aAAa,EAAE,MAAM,GACpB,OAAO,CAAC,IAAI,CAAC;IAgEV,SAAS,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,SAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAMlE,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAU7C,WAAW,CACf,SAAS,EAAE,MAAM,EACjB,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAWnB,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAWvD,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;cAMhC,oBAAoB,CAClC,SAAS,EAAE,MAAM,EACjB,eAAe,EAAE,cAAc,EAAE,GAChC,OAAO,CAAC,IAAI,CAAC;CAkHjB"}