immich/mobile/lib/infrastructure/entities/asset_face.entity.dart

import 'package:drift/drift.dart';
import 'package:immich_mobile/infrastructure/entities/person.entity.dart';
import 'package:immich_mobile/infrastructure/entities/remote_asset.entity.dart';
import 'package:immich_mobile/infrastructure/utils/datetime_clamp.type.dart';
import 'package:immich_mobile/infrastructure/utils/drift_default.mixin.dart';

@TableIndex.sql('CREATE INDEX IF NOT EXISTS idx_asset_face_person_id ON asset_face_entity (person_id)')
@TableIndex.sql('CREATE INDEX IF NOT EXISTS idx_asset_face_asset_id ON asset_face_entity (asset_id)')
@TableIndex.sql('''
CREATE INDEX IF NOT EXISTS idx_asset_face_visible_person
ON asset_face_entity (person_id, asset_id)
WHERE is_visible = 1 AND deleted_at IS NULL
''')
class AssetFaceEntity extends Table with DriftDefaultsMixin {
  const AssetFaceEntity();

  TextColumn get id => text()();

  TextColumn get assetId => text().references(RemoteAssetEntity, #id, onDelete: KeyAction.cascade)();

  TextColumn get personId => text().nullable().references(PersonEntity, #id, onDelete: KeyAction.setNull)();

  IntColumn get imageWidth => integer()();

  IntColumn get imageHeight => integer()();

  IntColumn get boundingBoxX1 => integer()();

  IntColumn get boundingBoxY1 => integer()();

  IntColumn get boundingBoxX2 => integer()();

  IntColumn get boundingBoxY2 => integer()();

  TextColumn get sourceType => text()();

  BoolColumn get isVisible => boolean().withDefault(const Constant(true))();

  DateTimeColumn get deletedAt => customType(clampedDateTime).nullable()();

  @override
  Set<Column> get primaryKey => {id};
}