Function convertToChartDBDiagram

src/lib/data/sql-import/common.ts:622–1074

(
    parserResult: SQLParserResult,
    sourceDatabaseType: DatabaseType,
    targetDatabaseType: DatabaseType
)

// Convert SQLParserResult to ChartDB Diagram structure
export function convertToChartDBDiagram(
    parserResult: SQLParserResult,
    sourceDatabaseType: DatabaseType,
    targetDatabaseType: DatabaseType
): Diagram {
    // Create a mapping of old table IDs to new ones
    const tableIdMapping = new Map<string, string>();

    // Convert SQL tables to ChartDB tables
    const tables: DBTable[] = parserResult.tables.map((table, index) => {
        const row = Math.floor(index / 4);
        const col = index % 4;
        const tableSpacing = 300;
        const newId = generateId();
        tableIdMapping.set(table.id, newId);

        // Create fields from columns
        const fields: DBField[] = table.columns.map((column) => {
            // Use special case handling for specific database types to ensure correct mapping
            let mappedType: DataType;

            // Detect and handle array types (e.g., int[], text[], varchar[])
            const isArrayType = column.type.endsWith('[]');
            const baseColumnType = isArrayType
                ? column.type.slice(0, -2)
                : column.type;

            // Create a modified column object with the base type for mapping
            const columnForMapping = { ...column, type: baseColumnType };

            // SQLite-specific handling for numeric types
            if (sourceDatabaseType === DatabaseType.SQLITE) {
                const normalizedType = columnForMapping.type.toLowerCase();

                if (normalizedType === 'integer' || normalizedType === 'int') {
                    // Ensure integer types are preserved
                    mappedType = { id: 'integer', name: 'integer' };
                } else if (
                    normalizedType === 'real' ||
                    normalizedType === 'float' ||
                    normalizedType === 'double' ||
                    normalizedType === 'numeric' ||
                    normalizedType === 'decimal'
                ) {
                    // Ensure real types are preserved
                    mappedType = { id: 'real', name: 'real' };
                } else if (normalizedType === 'blob') {
                    // Ensure blob types are preserved
                    mappedType = { id: 'blob', name: 'blob' };
                } else {
                    // Use the standard mapping for other types
                    mappedType = mapSQLTypeToGenericType(
                        columnForMapping.type,
                        sourceDatabaseType
                    );
                }
            }
            // Handle MySQL/MariaDB integer types specifically