lunchpick/lib/presentation/pages/calendar/calendar_screen.dart

import 'package:flutter/material.dart';
import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:table_calendar/table_calendar.dart';
import '../../../core/constants/app_colors.dart';
import '../../../core/constants/app_typography.dart';
import '../../../domain/entities/visit_record.dart';
import '../../providers/visit_provider.dart';
import 'widgets/visit_record_card.dart';
import 'widgets/visit_statistics.dart';

class CalendarScreen extends ConsumerStatefulWidget {
  const CalendarScreen({super.key});

  @override
  ConsumerState<CalendarScreen> createState() => _CalendarScreenState();
}

class _CalendarScreenState extends ConsumerState<CalendarScreen>
    with SingleTickerProviderStateMixin {
  late DateTime _selectedDay;
  late DateTime _focusedDay;
  CalendarFormat _calendarFormat = CalendarFormat.month;
  late TabController _tabController;
  Map<DateTime, List<VisitRecord>> _visitRecordEvents = {};

  @override
  void initState() {
    super.initState();
    _selectedDay = DateTime.now();
    _focusedDay = DateTime.now();
    _tabController = TabController(length: 2, vsync: this);
  }

  @override
  void dispose() {
    _tabController.dispose();
    super.dispose();
  }

  List<VisitRecord> _getEventsForDay(DateTime day) {
    final normalizedDay = DateTime(day.year, day.month, day.day);
    return _visitRecordEvents[normalizedDay] ?? [];
  }

  @override
  Widget build(BuildContext context) {
    final isDark = Theme.of(context).brightness == Brightness.dark;

    return Scaffold(
      backgroundColor: isDark
          ? AppColors.darkBackground
          : AppColors.lightBackground,
      appBar: AppBar(
        title: const Text('방문 기록'),
        backgroundColor: isDark
            ? AppColors.darkPrimary
            : AppColors.lightPrimary,
        foregroundColor: Colors.white,
        elevation: 0,
        bottom: TabBar(
          controller: _tabController,
          indicatorColor: Colors.white,
          indicatorWeight: 3,
          tabs: const [
            Tab(text: '캘린더'),
            Tab(text: '통계'),
          ],
        ),
      ),
      body: TabBarView(
        controller: _tabController,
        children: [
          // 캘린더 탭
          _buildCalendarTab(isDark),
          // 통계 탭
          VisitStatistics(selectedMonth: _focusedDay),
        ],
      ),
    );
  }

  Widget _buildCalendarTab(bool isDark) {
    return Consumer(
      builder: (context, ref, child) {
        final visitRecordsAsync = ref.watch(visitRecordsProvider);

        // 방문 기록을 날짜별로 그룹화
        visitRecordsAsync.whenData((records) {
          _visitRecordEvents = {};
          for (final record in records) {
            final normalizedDate = DateTime(
              record.visitDate.year,
              record.visitDate.month,
              record.visitDate.day,
            );
            _visitRecordEvents[normalizedDate] = [
              ...(_visitRecordEvents[normalizedDate] ?? []),
              record,
            ];
          }
        });

        return Column(
          children: [
            // 캘린더
            Card(
              margin: const EdgeInsets.all(16),
              color: isDark ? AppColors.darkSurface : AppColors.lightSurface,
              elevation: 2,
              shape: RoundedRectangleBorder(
                borderRadius: BorderRadius.circular(12),
              ),
              child: TableCalendar(
                firstDay: DateTime.utc(2025, 1, 1),
                lastDay: DateTime.utc(2030, 12, 31),
                focusedDay: _focusedDay,
                calendarFormat: _calendarFormat,
                selectedDayPredicate: (day) => isSameDay(_selectedDay, day),
                onDaySelected: (selectedDay, focusedDay) {
                  setState(() {
                    _selectedDay = selectedDay;
                    _focusedDay = focusedDay;
                  });
                },
                onFormatChanged: (format) {
                  setState(() {
                    _calendarFormat = format;
                  });
                },
                eventLoader: _getEventsForDay,
                calendarBuilders: CalendarBuilders(
                  markerBuilder: (context, day, events) {
                    if (events.isEmpty) return null;

                    final visitRecords = events.cast<VisitRecord>();
                    final confirmedCount = visitRecords
                        .where((r) => r.isConfirmed)
                        .length;
                    final unconfirmedCount =
                        visitRecords.length - confirmedCount;

                    return Row(
                      mainAxisAlignment: MainAxisAlignment.center,
                      children: [
                        if (confirmedCount > 0)
                          Container(
                            width: 6,
                            height: 6,
                            margin: const EdgeInsets.symmetric(horizontal: 1),
                            decoration: const BoxDecoration(
                              color: AppColors.lightPrimary,
                              shape: BoxShape.circle,
                            ),
                          ),
                        if (unconfirmedCount > 0)
                          Container(
                            width: 6,
                            height: 6,
                            margin: const EdgeInsets.symmetric(horizontal: 1),
                            decoration: const BoxDecoration(
                              color: Colors.orange,
                              shape: BoxShape.circle,
                            ),
                          ),
                      ],
                    );
                  },
                ),
                calendarStyle: CalendarStyle(
                  outsideDaysVisible: false,
                  selectedDecoration: const BoxDecoration(
                    color: AppColors.lightPrimary,
                    shape: BoxShape.circle,
                  ),
                  todayDecoration: BoxDecoration(
                    color: AppColors.lightPrimary.withOpacity(0.5),
                    shape: BoxShape.circle,
                  ),
                  markersMaxCount: 2,
                  markerDecoration: const BoxDecoration(
                    color: AppColors.lightSecondary,
                    shape: BoxShape.circle,
                  ),
                  weekendTextStyle: const TextStyle(
                    color: AppColors.lightError,
                  ),
                ),
                headerStyle: HeaderStyle(
                  formatButtonVisible: true,
                  titleCentered: true,
                  formatButtonShowsNext: false,
                  formatButtonDecoration: BoxDecoration(
                    color: AppColors.lightPrimary.withOpacity(0.1),
                    borderRadius: BorderRadius.circular(12),
                  ),
                  formatButtonTextStyle: const TextStyle(
                    color: AppColors.lightPrimary,
                  ),
                ),
              ),
            ),

            // 범례
            Padding(
              padding: const EdgeInsets.symmetric(horizontal: 16),
              child: Row(
                mainAxisAlignment: MainAxisAlignment.center,
                children: [
                  _buildLegend('추천받음', Colors.orange, isDark),
                  const SizedBox(width: 24),
                  _buildLegend('방문완료', Colors.green, isDark),
                ],
              ),
            ),

            const SizedBox(height: 16),

            // 선택된 날짜의 기록
            Expanded(child: _buildDayRecords(_selectedDay, isDark)),
          ],
        );
      },
    );
  }

  Widget _buildLegend(String label, Color color, bool isDark) {
    return Row(
      children: [
        Container(
          width: 14,
          height: 14,
          decoration: BoxDecoration(color: color, shape: BoxShape.circle),
        ),
        const SizedBox(width: 6),
        Text(label, style: AppTypography.body2(isDark)),
      ],
    );
  }

  Widget _buildDayRecords(DateTime day, bool isDark) {
    final events = _getEventsForDay(day);

    if (events.isEmpty) {
      return Center(
        child: Column(
          mainAxisAlignment: MainAxisAlignment.center,
          children: [
            Icon(
              Icons.event_available,
              size: 48,
              color: isDark
                  ? AppColors.darkTextSecondary
                  : AppColors.lightTextSecondary,
            ),
            const SizedBox(height: 16),
            Text('이날의 기록이 없습니다', style: AppTypography.body2(isDark)),
          ],
        ),
      );
    }

    return Column(
      children: [
        Padding(
          padding: const EdgeInsets.all(16),
          child: Row(
            children: [
              Icon(
                Icons.calendar_today,
                size: 20,
                color: isDark
                    ? AppColors.darkTextSecondary
                    : AppColors.lightTextSecondary,
              ),
              const SizedBox(width: 8),
              Text(
                '${day.month}월 ${day.day}일 방문 기록',
                style: AppTypography.body1(
                  isDark,
                ).copyWith(fontWeight: FontWeight.bold),
              ),
              const Spacer(),
              Text(
                '${events.length}건',
                style: AppTypography.body2(isDark).copyWith(
                  color: AppColors.lightPrimary,
                  fontWeight: FontWeight.bold,
                ),
              ),
            ],
          ),
        ),
        Expanded(
          child: ListView.builder(
            itemCount: events.length,
            itemBuilder: (context, index) {
              final sortedEvents = events
                ..sort((a, b) => b.visitDate.compareTo(a.visitDate));
              return VisitRecordCard(
                visitRecord: sortedEvents[index],
                onTap: () {
                  // TODO: 맛집 상세 페이지로 이동
                },
              );
            },
          ),
        ),
      ],
    );
  }
}