78 lines
2.4 KiB
Dart
78 lines
2.4 KiB
Dart
import 'package:drift/drift.dart';
|
|
import 'package:immich_mobile/domain/models/user.model.dart';
|
|
import 'package:immich_mobile/domain/models/user_metadata.model.dart';
|
|
import 'package:immich_mobile/infrastructure/entities/auth_user.entity.drift.dart';
|
|
import 'package:immich_mobile/infrastructure/mapper.dart';
|
|
import 'package:immich_mobile/infrastructure/repositories/db.repository.dart';
|
|
import 'package:immich_mobile/infrastructure/repositories/user_metadata.repository.dart';
|
|
|
|
class UserRepository {
|
|
final Drift _db;
|
|
const UserRepository(this._db);
|
|
|
|
Stream<Iterable<User>> getAll() => _db.select(_db.userEntity).map(mapToUser).watch();
|
|
}
|
|
|
|
class DriftAuthUserRepository extends DriftDatabaseRepository {
|
|
final Drift _db;
|
|
const DriftAuthUserRepository(super.db) : _db = db;
|
|
|
|
Future<UserDto?> get(String id) async {
|
|
final user = await _db.managers.authUserEntity.filter((user) => user.id.equals(id)).getSingleOrNull();
|
|
|
|
if (user == null) {
|
|
return null;
|
|
}
|
|
|
|
final query = _db.userMetadataEntity.select()..where((e) => e.userId.equals(id));
|
|
final metadata = await query.map((row) => row.toDto()).get();
|
|
|
|
return user.toDto(metadata);
|
|
}
|
|
|
|
Future<UserDto> upsert(UserDto user) async {
|
|
await _db.authUserEntity.insertOnConflictUpdate(
|
|
AuthUserEntityCompanion(
|
|
id: Value(user.id),
|
|
name: Value(user.name),
|
|
email: Value(user.email),
|
|
hasProfileImage: Value(user.hasProfileImage),
|
|
profileChangedAt: Value(user.profileChangedAt),
|
|
isAdmin: Value(user.isAdmin),
|
|
quotaSizeInBytes: Value(user.quotaSizeInBytes),
|
|
quotaUsageInBytes: Value(user.quotaUsageInBytes),
|
|
avatarColor: Value(user.avatarColor),
|
|
),
|
|
);
|
|
return user;
|
|
}
|
|
}
|
|
|
|
extension on AuthUserEntityData {
|
|
UserDto toDto([List<UserMetadata>? metadata]) {
|
|
bool memoryEnabled = true;
|
|
|
|
if (metadata != null) {
|
|
for (final meta in metadata) {
|
|
if (meta.key == UserMetadataKey.preferences && meta.preferences != null) {
|
|
memoryEnabled = meta.preferences?.memoriesEnabled ?? true;
|
|
}
|
|
}
|
|
}
|
|
|
|
return UserDto(
|
|
id: id,
|
|
email: email,
|
|
name: name,
|
|
updatedAt: profileChangedAt,
|
|
profileChangedAt: profileChangedAt,
|
|
hasProfileImage: hasProfileImage,
|
|
avatarColor: avatarColor,
|
|
memoryEnabled: memoryEnabled,
|
|
isAdmin: isAdmin,
|
|
quotaSizeInBytes: quotaSizeInBytes,
|
|
quotaUsageInBytes: quotaUsageInBytes,
|
|
);
|
|
}
|
|
}
|