immich/mobile/lib/presentation/pages/drift_user_selection.page.dart

import 'package:auto_route/auto_route.dart';
import 'package:easy_localization/easy_localization.dart';
import 'package:flutter/material.dart';
import 'package:flutter_hooks/flutter_hooks.dart';
import 'package:hooks_riverpod/hooks_riverpod.dart';
import 'package:immich_mobile/domain/models/album/album.model.dart';
import 'package:immich_mobile/domain/models/user.model.dart';
import 'package:immich_mobile/extensions/asyncvalue_extensions.dart';
import 'package:immich_mobile/extensions/build_context_extensions.dart';
import 'package:immich_mobile/providers/infrastructure/db.provider.dart';
import 'package:immich_mobile/providers/infrastructure/remote_album.provider.dart';
import 'package:immich_mobile/providers/user.provider.dart';
import 'package:immich_mobile/widgets/common/user_circle_avatar.dart';

// TODO: Refactor this provider when we have user provider/service/repository pattern in place
final driftUsersProvider = FutureProvider.autoDispose<List<UserDto>>((ref) async {
  final drift = ref.watch(driftProvider);
  final currentUser = ref.watch(currentUserProvider);

  final userEntities = await drift.managers.userEntity.get();

  final users = userEntities
      .map(
        (entity) => UserDto(
          id: entity.id,
          name: entity.name,
          email: entity.email,
          isPartnerSharedBy: false,
          isPartnerSharedWith: false,
          avatarColor: entity.avatarColor,
          memoryEnabled: true,
          inTimeline: true,
          profileChangedAt: entity.profileChangedAt,
          hasProfileImage: entity.hasProfileImage,
        ),
      )
      .toList();

  users.removeWhere((u) => currentUser?.id == u.id);

  return users;
});

@RoutePage()
class DriftUserSelectionPage extends HookConsumerWidget {
  final RemoteAlbum album;

  const DriftUserSelectionPage({super.key, required this.album});

  @override
  Widget build(BuildContext context, WidgetRef ref) {
    final AsyncValue<List<UserDto>> suggestedShareUsers = ref.watch(driftUsersProvider);
    final sharedUsersList = useState<Set<UserDto>>({});

    addNewUsersHandler() {
      context.maybePop(sharedUsersList.value.map((e) => e.id).toList());
    }

    buildTileIcon(UserDto user) {
      if (sharedUsersList.value.contains(user)) {
        return CircleAvatar(backgroundColor: context.primaryColor, child: const Icon(Icons.check_rounded, size: 25));
      } else {
        return UserCircleAvatar(user: user);
      }
    }

    buildUserList(List<UserDto> users) {
      List<Widget> usersChip = [];

      for (var user in sharedUsersList.value) {
        usersChip.add(
          Padding(
            padding: const EdgeInsets.symmetric(horizontal: 8.0),
            child: Chip(
              backgroundColor: context.primaryColor.withValues(alpha: 0.15),
              label: Text(user.name, style: const TextStyle(fontSize: 12, fontWeight: FontWeight.bold)),
            ),
          ),
        );
      }
      return ListView(
        children: [
          Wrap(children: [...usersChip]),
          Padding(
            padding: const EdgeInsets.all(16.0),
            child: Text(
              'suggestions'.tr(),
              style: const TextStyle(fontSize: 14, color: Colors.grey, fontWeight: FontWeight.bold),
            ),
          ),
          ListView.builder(
            primary: false,
            shrinkWrap: true,
            itemBuilder: ((context, index) {
              return ListTile(
                leading: buildTileIcon(users[index]),
                dense: true,
                title: Text(users[index].name, style: const TextStyle(fontSize: 14, fontWeight: FontWeight.bold)),
                subtitle: Text(users[index].email, style: const TextStyle(fontSize: 12)),
                onTap: () {
                  if (sharedUsersList.value.contains(users[index])) {
                    sharedUsersList.value = sharedUsersList.value
                        .where((selectedUser) => selectedUser.id != users[index].id)
                        .toSet();
                  } else {
                    sharedUsersList.value = {...sharedUsersList.value, users[index]};
                  }
                },
              );
            }),
            itemCount: users.length,
          ),
        ],
      );
    }

    return Scaffold(
      appBar: AppBar(
        title: const Text('invite_to_album').tr(),
        elevation: 0,
        centerTitle: false,
        leading: IconButton(
          icon: const Icon(Icons.close_rounded),
          onPressed: () {
            context.maybePop(null);
          },
        ),
        actions: [
          TextButton(
            onPressed: sharedUsersList.value.isEmpty ? null : addNewUsersHandler,
            child: const Text("add", style: TextStyle(fontSize: 14, fontWeight: FontWeight.bold)).tr(),
          ),
        ],
      ),
      body: suggestedShareUsers.widgetWhen(
        onData: (users) {
          // Get shared users for this album from the database
          final sharedUsers = ref.watch(remoteAlbumSharedUsersProvider(album.id));

          return sharedUsers.when(
            data: (albumSharedUsers) {
              // Filter out users that are already shared with this album and the owner
              final filteredUsers = users.where((user) {
                return !albumSharedUsers.any((sharedUser) => sharedUser.id == user.id) && user.id != album.ownerId;
              }).toList();

              return buildUserList(filteredUsers);
            },
            loading: () => const Center(child: CircularProgressIndicator()),
            error: (error, stack) {
              // If we can't load shared users, just filter out the owner
              final filteredUsers = users.where((user) => user.id != album.ownerId).toList();
              return buildUserList(filteredUsers);
            },
          );
        },
      ),
    );
  }
}