Immich/mobile/lib/presentation/widgets/bottom_sheet/map_bottom_sheet.widget.dart

import 'dart:async';

import 'package:flutter/material.dart';
import 'package:hooks_riverpod/hooks_riverpod.dart';
import 'package:immich_mobile/domain/models/events.model.dart';
import 'package:immich_mobile/domain/models/map.model.dart';
import 'package:immich_mobile/domain/utils/event_stream.dart';
import 'package:immich_mobile/extensions/build_context_extensions.dart';
import 'package:immich_mobile/generated/translations.g.dart';
import 'package:immich_mobile/presentation/widgets/bottom_sheet/base_bottom_sheet.widget.dart';
import 'package:immich_mobile/presentation/widgets/bottom_sheet/general_bottom_sheet.widget.dart';
import 'package:immich_mobile/presentation/widgets/map/map.state.dart';
import 'package:immich_mobile/presentation/widgets/timeline/timeline.widget.dart';
import 'package:immich_mobile/providers/infrastructure/timeline.provider.dart';
import 'package:immich_mobile/providers/user.provider.dart';

class MapBottomSheet extends StatelessWidget {
  final Key? sheetKey;

  const MapBottomSheet({super.key, this.sheetKey});

  @override
  Widget build(BuildContext context) {
    return BaseBottomSheet(
      key: sheetKey,
      initialChildSize: 0.25,
      maxChildSize: 0.75,
      shouldCloseOnMinExtent: false,
      resizeOnScroll: false,
      actions: const [],
      backgroundColor: context.themeData.colorScheme.surface,
      slivers: const [
        SliverFillRemaining(hasScrollBody: false, child: SizedBox(height: 0, child: _ScopedMapTimeline())),
      ],
    );
  }
}

class _ScopedMapTimeline extends StatelessWidget {
  const _ScopedMapTimeline();

  @override
  Widget build(BuildContext context) {
    return ProviderScope(
      overrides: [
        timelineServiceProvider.overrideWith((ref) {
          final user = ref.watch(currentUserProvider);
          if (user == null) {
            throw Exception('User must be logged in to access archive');
          }

          final withPartners = ref.watch(mapStateProvider.select((s) => s.withPartners));
          final users = withPartners ? ref.watch(timelineUsersProvider).valueOrNull ?? [user.id] : [user.id];

          final optionsController = StreamController<TimelineMapOptions>.broadcast();
          ref.onDispose(optionsController.close);

          var currentOptions = ref.read(mapStateProvider).toOptions();

          ref.listen(mapStateProvider.select((state) => state.toOptions()), (_, newOptions) {
            currentOptions = newOptions;
            optionsController.add(newOptions);
          });

          final timelineService = ref
              .watch(timelineFactoryProvider)
              .geographicMap(users, () => currentOptions, optionsController.stream);
          ref.onDispose(timelineService.dispose);

          return timelineService;
        }),
      ],
      child: const Column(
        children: [
          _MapAssetCount(),
          Expanded(
            child: Timeline(appBar: null, bottomSheet: GeneralBottomSheet(minChildSize: 0.23), withScrubber: false),
          ),
        ],
      ),
    );
  }
}

class _MapAssetCount extends ConsumerStatefulWidget {
  const _MapAssetCount();

  @override
  ConsumerState<_MapAssetCount> createState() => _MapAssetCountState();
}

class _MapAssetCountState extends ConsumerState<_MapAssetCount> {
  StreamSubscription? _reloadSubscription;

  @override
  void initState() {
    super.initState();
    _reloadSubscription = EventStream.shared.listen<TimelineReloadEvent>((_) => setState(() {}));
  }

  @override
  void dispose() {
    unawaited(_reloadSubscription?.cancel());
    super.dispose();
  }

  @override
  Widget build(BuildContext context) {
    final count = ref.watch(timelineServiceProvider.select((s) => s.totalAssets));
    return Text(context.t.map_assets_in_bounds(count: count), style: context.themeData.textTheme.headlineSmall);
  }
}