immich/server/src/sql-tools/processors/unique-constraint.processor.ts

import { ConstraintType, Processor } from 'src/sql-tools/types';

export const processUniqueConstraints: Processor = (ctx, items) => {
  for (const {
    item: { object, options },
  } of items.filter((item) => item.type === 'uniqueConstraint')) {
    const table = ctx.getTableByObject(object);
    if (!table) {
      ctx.warnMissingTable('@Unique', object);
      continue;
    }

    const tableName = table.name;
    const columnNames = options.columns;

    table.constraints.push({
      type: ConstraintType.UNIQUE,
      name: options.name || ctx.getNameFor({ type: 'unique', tableName, columnNames }),
      tableName,
      columnNames,
      synchronize: options.synchronize ?? true,
    });
  }

  // column level constraints
  for (const {
    type,
    item: { object, propertyName, options },
  } of items.filter((item) => item.type === 'column' || item.type === 'foreignKeyColumn')) {
    const { table, column } = ctx.getColumnByObjectAndPropertyName(object, propertyName);
    if (!table) {
      ctx.warnMissingTable('@Column', object);
      continue;
    }

    if (!column) {
      // should be impossible since they are created in `column.processor.ts`
      ctx.warnMissingColumn('@Column', object, propertyName);
      continue;
    }

    if (type === 'column' && !options.primary && (options.unique || options.uniqueConstraintName)) {
      const uniqueConstraintName =
        options.uniqueConstraintName ||
        ctx.getNameFor({
          type: 'unique',
          tableName: table.name,
          columnNames: [column.name],
        });

      table.constraints.push({
        type: ConstraintType.UNIQUE,
        name: uniqueConstraintName,
        tableName: table.name,
        columnNames: [column.name],
        synchronize: options.synchronize ?? true,
      });
    }
  }
};