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

import 'package:drift/drift.dart';
import 'package:immich_mobile/domain/models/local_album.model.dart';
import 'package:immich_mobile/infrastructure/entities/local_album.entity.drift.dart';
import 'package:immich_mobile/infrastructure/entities/local_asset.entity.dart';
import 'package:immich_mobile/infrastructure/utils/drift_default.mixin.dart';

class LocalAlbumEntity extends Table with DriftDefaultsMixin {
  const LocalAlbumEntity();

  TextColumn get id => text()();
  TextColumn get name => text()();
  DateTimeColumn get updatedAt => dateTime().withDefault(currentDateAndTime)();
  IntColumn get assetCount => integer().withDefault(const Constant(0))();
  TextColumn get thumbnailId => text()
      .nullable()
      .references(LocalAssetEntity, #localId, onDelete: KeyAction.setNull)();
  IntColumn get backupSelection => intEnum<BackupSelection>()();
  BoolColumn get isAll => boolean().withDefault(const Constant(false))();

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

extension LocalAlbumEntityX on LocalAlbumEntityData {
  LocalAlbum toDto() {
    return LocalAlbum(
      id: id,
      name: name,
      updatedAt: updatedAt,
      thumbnailId: thumbnailId,
      backupSelection: backupSelection,
      isAll: isAll,
    );
  }
}