fluffychat/lib/pangea/token_info_feedback/token_info_feedback_dialog.dart

import 'package:flutter/material.dart';

import 'package:fluffychat/l10n/l10n.dart';
import 'package:fluffychat/pangea/common/constants/model_keys.dart';
import 'package:fluffychat/pangea/common/widgets/feedback_dialog.dart';
import 'package:fluffychat/pangea/events/event_wrappers/pangea_message_event.dart';
import 'package:fluffychat/pangea/events/models/language_detection_model.dart';
import 'package:fluffychat/pangea/events/models/pangea_token_model.dart';
import 'package:fluffychat/pangea/events/models/tokens_event_content_model.dart';
import 'package:fluffychat/pangea/extensions/pangea_room_extension.dart';
import 'package:fluffychat/pangea/languages/language_arc_model.dart';
import 'package:fluffychat/pangea/languages/p_language_store.dart';
import 'package:fluffychat/pangea/lemmas/lemma_info_repo.dart';
import 'package:fluffychat/pangea/lemmas/lemma_info_response.dart';
import 'package:fluffychat/pangea/phonetic_transcription/phonetic_transcription_repo.dart';
import 'package:fluffychat/pangea/phonetic_transcription/phonetic_transcription_request.dart';
import 'package:fluffychat/pangea/phonetic_transcription/phonetic_transcription_response.dart';
import 'package:fluffychat/pangea/token_info_feedback/token_info_feedback_repo.dart';
import 'package:fluffychat/pangea/token_info_feedback/token_info_feedback_request.dart';
import 'package:fluffychat/pangea/token_info_feedback/token_info_feedback_response.dart';
import 'package:fluffychat/pangea/toolbar/word_card/word_zoom_widget.dart';
import 'package:fluffychat/widgets/future_loading_dialog.dart';
import 'package:fluffychat/widgets/matrix.dart';

class TokenInfoFeedbackDialog extends StatelessWidget {
  final TokenInfoFeedbackRequestData requestData;
  final String langCode;
  final PangeaMessageEvent? event;

  const TokenInfoFeedbackDialog({
    super.key,
    required this.requestData,
    required this.langCode,
    this.event,
  });

  Future<String> _submitFeedback(String feedback) async {
    final request = TokenInfoFeedbackRequest(
      userFeedback: feedback,
      data: requestData,
    );

    final TokenInfoFeedbackResponse response =
        await TokenInfoFeedbackRepo.submitFeedback(request);

    final originalToken = requestData.tokens[requestData.selectedToken];
    final token = response.updatedToken ?? originalToken;

    // first, update lemma info if changed
    if (response.updatedLemmaInfo != null) {
      await _updateLemmaInfo(
        token,
        response.updatedLemmaInfo!,
      );
    }

    // second, update the phonetic info if changed
    if (response.updatedPhonetics != null) {
      await _updatePhoneticTranscription(
        response.updatedPhonetics!,
      );
    }

    final originalSent = event?.originalSent;

    // if no other changes, just return the message
    final hasTokenUpdate = response.updatedToken != null;
    final hasLangUpdate = originalSent != null &&
        response.updatedLanguage != null &&
        response.updatedLanguage != originalSent.langCode;

    if (!hasTokenUpdate && !hasLangUpdate) {
      return response.userFriendlyMessage;
    }

    // update the tokens to be sent in the message edit
    final tokens = List<PangeaToken>.from(requestData.tokens);
    if (hasTokenUpdate) {
      tokens[requestData.selectedToken] = response.updatedToken!;
    }

    final updatedLanguage =
        response.updatedLanguage ?? event?.originalSent?.langCode;

    final tokensSent = PangeaMessageTokens(
      tokens: tokens,
      detections: [
        if (updatedLanguage != null)
          LanguageDetectionModel(
            langCode: updatedLanguage,
            confidence: 1,
          ),
      ],
    );

    if (requestData.fullText != null && event != null) {
      await event!.room.pangeaSendTextEvent(
        requestData.fullText!,
        editEventId: event!.eventId,
        originalWritten: event!.originalWritten?.content,
        tokensSent: tokensSent,
        tokensWritten: event!.originalWritten?.tokens != null
            ? PangeaMessageTokens(
                tokens: event!.originalWritten!.tokens!,
                detections: event!.originalWritten?.detections,
              )
            : null,
        choreo: originalSent?.choreo,
        messageTag: ModelKey.tokenFeedbackEdit,
      );
    }

    return response.userFriendlyMessage;
  }

  Future<void> _submit(String feedback, BuildContext context) async {
    final resp = await showFutureLoadingDialog(
      context: context,
      future: () => _submitFeedback(feedback),
    );

    if (!resp.isError) {
      Navigator.of(context).pop(resp.result!);
    }
  }

  Future<void> _updateLemmaInfo(
    PangeaToken token,
    LemmaInfoResponse response,
  ) =>
      LemmaInfoRepo.set(
        token.vocabConstructID.lemmaInfoRequest(
          event?.event.content ?? {},
        ),
        response,
      );

  Future<void> _updatePhoneticTranscription(
    PhoneticTranscriptionResponse response,
  ) async {
    final req = PhoneticTranscriptionRequest(
      arc: LanguageArc(
        l1: PLanguageStore.byLangCode(requestData.wordCardL1) ??
            MatrixState.pangeaController.userController.userL1!,
        l2: PLanguageStore.byLangCode(langCode) ??
            MatrixState.pangeaController.userController.userL2!,
      ),
      content: response.content,
    );
    await PhoneticTranscriptionRepo.set(req, response);
  }

  @override
  Widget build(BuildContext context) {
    final selectedToken = requestData.tokens[requestData.selectedToken];
    return FeedbackDialog(
      title: L10n.of(context).tokenInfoFeedbackDialogTitle,
      onSubmit: (feedback) => _submit(feedback, context),
      extraContent: WordZoomWidget(
        token: selectedToken.text,
        construct: selectedToken.vocabConstructID,
        langCode: langCode,
        enableEmojiSelection: false,
      ),
    );
  }
}