fluffychat/lib/pangea/widgets/download_analytics_dialog.dart

import 'package:flutter/material.dart';

import 'package:collection/collection.dart';
import 'package:csv/csv.dart';
import 'package:excel/excel.dart';
import 'package:flutter_gen/gen_l10n/l10n.dart';
import 'package:matrix/matrix.dart';

import 'package:fluffychat/config/app_config.dart';
import 'package:fluffychat/config/themes.dart';
import 'package:fluffychat/pangea/enum/analytics/analytics_summary_enum.dart';
import 'package:fluffychat/pangea/extensions/pangea_room_extension/pangea_room_extension.dart';
import 'package:fluffychat/pangea/models/analytics/analytics_summary_model.dart';
import 'package:fluffychat/pangea/models/analytics/construct_list_model.dart';
import 'package:fluffychat/pangea/models/analytics/construct_use_model.dart';
import 'package:fluffychat/pangea/models/analytics/constructs_model.dart';
import 'package:fluffychat/pangea/utils/bot_name.dart';
import 'package:fluffychat/pangea/utils/download_file.dart';
import 'package:fluffychat/pangea/utils/error_handler.dart';
import 'package:fluffychat/pangea/utils/grammar/get_grammar_copy.dart';
import 'package:fluffychat/widgets/matrix.dart';

class DownloadAnalyticsDialog extends StatefulWidget {
  final Room space;
  const DownloadAnalyticsDialog({
    required this.space,
    super.key,
  });

  @override
  DownloadAnalyticsDialogState createState() => DownloadAnalyticsDialogState();
}

class DownloadAnalyticsDialogState extends State<DownloadAnalyticsDialog> {
  bool _initialized = false;
  bool _downloaded = false;
  bool _joiningRooms = false;
  bool _downloading = false;

  bool get _loading => _joiningRooms || _downloading || !_initialized;

  String? _error;

  Map<String, int> _downloadStatuses = {};

  @override
  void initState() {
    super.initState();
    _initialize();
  }

  Future<void> _initialize() async {
    try {
      await widget.space.requestParticipants(
        [Membership.join],
        false,
        true,
      );
    } catch (e, s) {
      ErrorHandler.logError(
        e: e,
        s: s,
        data: {
          "spaceID": widget.space.id,
        },
      );
    } finally {
      if (mounted) setState(() => _initialized = true);
    }
  }

  DownloadType _downloadType = DownloadType.csv;

  void _setDownloadType(DownloadType type) {
    _clean();
    if (mounted) setState(() => _downloadType = type);
  }

  void _clean() {
    _error = null;
    _joiningRooms = false;
    _downloading = false;
    _downloaded = false;
    _downloadStatuses = Map.fromEntries(
      _usersToDownload.map((user) => MapEntry(user.id, 0)),
    );
  }

  List<User> get _usersToDownload => widget.space
      .getParticipants()
      .where((member) => member.id != BotName.byEnvironment)
      .toList();

  Color _downloadStatusColor(String userID) {
    final status = _downloadStatuses[userID];
    if (status == 1) return Colors.yellow;
    if (status == 2) return Colors.green;
    if ((status ?? 0) < 0) return Colors.red;
    return Colors.grey;
  }

  String? get _statusText {
    if (_joiningRooms) return L10n.of(context).accessingMemberAnalytics;
    if (_downloading) return L10n.of(context).downloading;
    if (_downloaded) return L10n.of(context).downloadComplete;
    return null;
  }

  Room? _userAnalyticsRoom(String userID) {
    final rooms = widget.space.client.rooms;
    final l2 = MatrixState.pangeaController.languageController.userL2?.langCode;
    if (l2 == null) return null;
    return rooms.firstWhereOrNull((room) {
      return room.isAnalyticsRoomOfUser(userID) && room.isMadeForLang(l2);
    });
  }

  Future<void> _runDownload() async {
    try {
      if (!mounted) return;
      setState(() {
        _error = null;
        _joiningRooms = true;
      });

      await widget.space.joinAnalyticsRooms();
      if (mounted) {
        setState(() {
          _joiningRooms = false;
          _downloading = true;
        });
      }

      await _downloadSpaceAnalytics();
      if (mounted) {
        setState(() {
          _downloading = false;
          _downloaded = true;
        });
      }
    } catch (e, s) {
      ErrorHandler.logError(
        e: e,
        s: s,
        data: {
          "spaceID": widget.space.id,
        },
      );

      _clean();
      _error = e.toString();
      if (mounted) setState(() {});
    }
  }

  Future<void> _downloadSpaceAnalytics() async {
    final l2 = MatrixState.pangeaController.languageController.userL2?.langCode;
    if (l2 == null) return;

    final List<AnalyticsSummaryModel?> summaries = await Future.wait(
      _usersToDownload.map((user) => _getUserAnalyticsModel(user.id)),
    );

    final allSummaries = summaries.whereType<AnalyticsSummaryModel>().toList();
    final content = _downloadType == DownloadType.xlsx
        ? _getExcelFileContent(allSummaries)
        : _getCSVFileContent(allSummaries);

    final fileName =
        "analytics_${widget.space.name}_${DateTime.now().toIso8601String()}.${_downloadType == DownloadType.xlsx ? 'xlsx' : 'csv'}";

    await downloadFile(
      content,
      fileName,
      DownloadType.csv,
    );
  }

  Future<AnalyticsSummaryModel?> _getUserAnalyticsModel(String userID) async {
    try {
      final userAnalyticsRoom = _userAnalyticsRoom(userID);
      _downloadStatuses[userID] = userAnalyticsRoom != null ? 1 : -1;
      if (mounted) setState(() {});

      final constructEvents = await userAnalyticsRoom?.getAnalyticsEvents(
        userId: userID,
      );

      if (constructEvents == null) {
        if (mounted) setState(() => _downloadStatuses[userID] = -1);
        return AnalyticsSummaryModel.emptyModel(userID);
      }

      final List<OneConstructUse> uses = [];
      for (final event in constructEvents) {
        uses.addAll(event.content.uses);
      }

      final constructs = ConstructListModel(uses: uses);
      final summary = AnalyticsSummaryModel.fromConstructListModel(
        userID,
        constructs,
        getCopy,
        context,
      );
      if (mounted) setState(() => _downloadStatuses[userID] = 2);
      return summary;
    } catch (e, s) {
      ErrorHandler.logError(
        e: e,
        s: s,
        data: {
          "spaceID": widget.space.id,
          "userID": userID,
        },
      );
      if (mounted) setState(() => _downloadStatuses[userID] = -2);
    }
    return null;
  }

  List<CellValue> _formatExcelRow(
    AnalyticsSummaryModel summary,
  ) {
    final List<CellValue> row = [];
    for (int i = 0; i < AnalyticsSummaryEnum.values.length; i++) {
      final key = AnalyticsSummaryEnum.values[i];
      final value = summary.getValue(key, context);
      if (value is int) {
        row.add(IntCellValue(value));
      } else if (value is String) {
        row.add(TextCellValue(value));
      } else if (value is List<String>) {
        row.add(TextCellValue(value.join(", ")));
      }
    }
    return row;
  }

  List<int> _getExcelFileContent(
    List<AnalyticsSummaryModel> summaries,
  ) {
    final excel = Excel.createExcel();
    final sheet = excel['Sheet1'];

    for (final key in AnalyticsSummaryEnum.values) {
      sheet
          .cell(
            CellIndex.indexByColumnRow(
              rowIndex: 0,
              columnIndex: key.index,
            ),
          )
          .value = TextCellValue(key.header(L10n.of(context)));
    }

    final rows = summaries.map((summary) => _formatExcelRow(summary)).toList();

    for (int i = 0; i < rows.length; i++) {
      final row = rows[i];
      for (int j = 0; j < row.length; j++) {
        final cell = row[j];
        sheet
            .cell(CellIndex.indexByColumnRow(rowIndex: i + 2, columnIndex: j))
            .value = cell;
      }
    }
    return excel.encode() ?? [];
  }

  String _getCSVFileContent(
    List<AnalyticsSummaryModel> summaries,
  ) {
    final List<List<dynamic>> rows = [];
    final headerRow = [];
    for (final key in AnalyticsSummaryEnum.values) {
      headerRow.add(key.header(L10n.of(context)));
    }
    rows.add(headerRow);

    for (final summary in summaries) {
      final row = [];
      for (int i = 0; i < AnalyticsSummaryEnum.values.length; i++) {
        final key = AnalyticsSummaryEnum.values[i];
        final value = summary.getValue(key, context);
        if (value == null) continue;
        value is List<String> ? row.add(value.join(", ")) : row.add(value);
      }
      rows.add(row);
    }

    final String fileString = const ListToCsvConverter().convert(rows);
    return fileString;
  }

  String getCopy(ConstructUses use) {
    return getGrammarCopy(
          category: use.category,
          lemma: use.lemma,
          context: context,
        ) ??
        use.lemma;
  }

  @override
  Widget build(BuildContext context) {
    return Dialog(
      child: Container(
        constraints: const BoxConstraints(
          maxWidth: 400,
        ),
        padding: const EdgeInsets.symmetric(vertical: 20),
        child: Column(
          mainAxisSize: MainAxisSize.min,
          children: [
            Text(
              L10n.of(context).fileType,
              style: TextStyle(
                fontSize: AppConfig.fontSizeFactor * AppConfig.messageFontSize,
              ),
            ),
            Padding(
              padding: const EdgeInsets.all(8.0),
              child: SegmentedButton<DownloadType>(
                selected: {_downloadType},
                onSelectionChanged: (c) => _setDownloadType(c.first),
                segments: [
                  ButtonSegment(
                    value: DownloadType.csv,
                    label: Text(L10n.of(context).commaSeparatedFile),
                  ),
                  ButtonSegment(
                    value: DownloadType.xlsx,
                    label: Text(L10n.of(context).excelFile),
                  ),
                ],
              ),
            ),
            Padding(
              padding: const EdgeInsets.all(8.0),
              child: ConstrainedBox(
                constraints: const BoxConstraints(
                  maxHeight: 300,
                  minHeight: 0,
                ),
                child: ListView.builder(
                  shrinkWrap: true,
                  itemCount: _usersToDownload.length,
                  itemBuilder: (context, index) {
                    final user = _usersToDownload[index];

                    String tooltip = "";
                    if (_downloadStatuses[user.id] == -1) {
                      tooltip = L10n.of(context).analyticsNotAvailable;
                    } else if (_downloadStatuses[user.id] == -2) {
                      tooltip = L10n.of(context).failedFetchUserAnalytics;
                    }

                    return Padding(
                      padding: const EdgeInsets.all(4.0),
                      child: AnimatedOpacity(
                        duration: FluffyThemes.animationDuration,
                        opacity:
                            (_downloadStatuses[user.id] ?? 0) > 0 ? 1 : 0.5,
                        child: Row(
                          children: [
                            SizedBox(
                              width: 40,
                              height: 30,
                              child: (_downloadStatuses[user.id] ?? 0) < 0
                                  ? const Icon(
                                      Icons.error_outline,
                                      size: 16,
                                    )
                                  : Center(
                                      child: AnimatedContainer(
                                        duration:
                                            FluffyThemes.animationDuration,
                                        height: 12,
                                        width: 12,
                                        decoration: BoxDecoration(
                                          color: _downloadStatusColor(user.id),
                                          borderRadius:
                                              BorderRadius.circular(100),
                                        ),
                                      ),
                                    ),
                            ),
                            Flexible(
                              child: Column(
                                crossAxisAlignment: CrossAxisAlignment.start,
                                children: [
                                  Text(user.displayName ?? user.id),
                                  if (tooltip.isNotEmpty)
                                    Text(
                                      tooltip,
                                      style: const TextStyle(fontSize: 10),
                                    ),
                                ],
                              ),
                            ),
                          ],
                        ),
                      ),
                    );
                  },
                ),
              ),
            ),
            Padding(
              padding: const EdgeInsets.fromLTRB(8.0, 16.0, 8.0, 8.0),
              child: OutlinedButton(
                onPressed: _loading || !_initialized ? null : _runDownload,
                child: _initialized && !_loading
                    ? Text(
                        _loading
                            ? L10n.of(context).downloading
                            : L10n.of(context).download,
                      )
                    : const SizedBox(
                        height: 10,
                        width: 100,
                        child: LinearProgressIndicator(),
                      ),
              ),
            ),
            AnimatedSize(
              duration: FluffyThemes.animationDuration,
              child: _statusText != null
                  ? Padding(
                      padding: const EdgeInsets.all(8.0),
                      child: Text(_statusText!),
                    )
                  : const SizedBox(),
            ),
            AnimatedSize(
              duration: FluffyThemes.animationDuration,
              child: _error != null
                  ? Padding(
                      padding: const EdgeInsets.all(8.0),
                      child: Text(L10n.of(context).oopsSomethingWentWrong),
                    )
                  : const SizedBox(),
            ),
          ],
        ),
      ),
    );
  }
}