generateAddColumnAlterTableSQL method

AlterTableSQL generateAddColumnAlterTableSQL(String table, String fieldName, TypeInfo fieldType, { List<EntityField>? entityFieldAnnotations, })

Implementation

AlterTableSQL generateAddColumnAlterTableSQL(
  String table,
  String fieldName,
  TypeInfo fieldType, {
  List<EntityField>? entityFieldAnnotations,
}) {
  var q = dialect.elementQuote;
  var columnName = normalizeColumnName(fieldName);
  var fieldSQLType = typeToSQLType(
    fieldType,
    columnName,
    entityFieldAnnotations: entityFieldAnnotations,
  );

  var comment = '${fieldType.toString(withT: false)} $fieldName';

  MapEntry<String, MapEntry<String, String>>? entityType;
  String? refTable;
  String? refColumn;

  if (fieldSQLType == null) {
    entityType = entityTypeToSQLType(
      fieldType,
      columnName,
      entityFieldAnnotations: entityFieldAnnotations,
    );
    if (entityType != null) {
      fieldSQLType = entityType.value.value;

      refTable = entityType.key;
      refColumn = entityType.value.key;

      comment += ' @ $refTable.$refColumn';
    }
  }

  if (fieldSQLType == null) {
    var enumType = enumTypeToSQLType(
      fieldType.type,
      columnName,
      entityFieldAnnotations: entityFieldAnnotations,
    );
    if (enumType != null) {
      var type = enumType.key;
      var values = enumType.value;

      fieldSQLType = _buildEnumSQLType(
        type,
        fieldSQLType,
        values,
        q,
        columnName,
      );

      comment += ' enum(${values.join(', ')})';
    }
  }

  var columnEntry = SQLEntry(
    'ADD',
    ' ADD COLUMN $q$columnName$q $fieldSQLType',
    comment: comment,
  );

  List<CreateIndexSQL>? indexes;
  List<AlterTableSQL>? constraints;

  if (entityFieldAnnotations != null && entityFieldAnnotations.isNotEmpty) {
    if (entityFieldAnnotations.hasUnique) {
      var constrainUniqueName = '${table}__${columnName}__unique';

      var uniqueEntry = SQLEntry(
        'CONSTRAINT',
        ' ADD CONSTRAINT $q$constrainUniqueName$q UNIQUE ($q$columnName$q)',
        columns: [
          SQLColumn(
            table,
            columnName,
            referenceTable: refTable!,
            referenceColumn: refColumn!,
          ),
        ],
      );

      constraints = [
        AlterTableSQL(dialect, table, [uniqueEntry]),
      ];
    } else if (entityFieldAnnotations.hasIndexed) {
      var indexName = '${table}__${columnName}__idx';
      indexes = [CreateIndexSQL(dialect, table, columnName, indexName)];
    }
  }

  if (entityType != null) {
    var fieldName = columnName;
    var ref = entityType;

    var refTableName = ref.key;
    var refField = ref.value.key.toLowerCase();

    var constrainName = '${table}__${columnName}__fkey';

    constraints ??= [];

    var constraintEntry = SQLEntry(
      'CONSTRAINT',
      ' ADD CONSTRAINT $q$constrainName$q FOREIGN KEY ($q$columnName$q) REFERENCES $q$refTableName$q($q$refField$q)',
      comment: '$fieldName @ $refTableName.$refField',
      columns: [
        SQLColumn(
          table,
          columnName,
          referenceTable: refTableName,
          referenceColumn: refField,
        ),
      ],
    );

    constraints.add(AlterTableSQL(dialect, table, [constraintEntry]));
  }

  var alterTableSQL = AlterTableSQL(
    dialect,
    table,
    [columnEntry],
    indexes: indexes,
    constraints: constraints,
  );

  return alterTableSQL;
}