fluffychat/lib/pangea/choreographer/igc/igc_request_model.dart

import 'dart:convert';

import 'package:fluffychat/pangea/choreographer/igc/igc_response_model.dart';
import 'package:fluffychat/pangea/common/constants/model_keys.dart';
import 'package:fluffychat/pangea/common/models/base_request_model.dart';
import 'package:fluffychat/pangea/common/models/llm_feedback_model.dart';
import 'package:fluffychat/widgets/matrix.dart';

class IGCRequestModel with BaseRequestModel {
  final String fullText;
  final bool enableIT;
  final bool enableIGC;
  final String userId;
  final List<PreviousMessage> prevMessages;
  final List<LLMFeedbackModel<IGCResponseModel>> feedback;

  @override
  String get userCefr => MatrixState
      .pangeaController
      .userController
      .profile
      .userSettings
      .cefrLevel
      .string;

  @override
  String get userL1 => MatrixState.pangeaController.userController.userL1Code!;

  @override
  String get userL2 => MatrixState.pangeaController.userController.userL2Code!;

  const IGCRequestModel({
    required this.fullText,
    required this.enableIGC,
    required this.enableIT,
    required this.userId,
    required this.prevMessages,
    this.feedback = const [],
  });

  /// Creates a copy of this request with optional feedback.
  IGCRequestModel copyWithFeedback(
    List<LLMFeedbackModel<IGCResponseModel>> newFeedback,
  ) => IGCRequestModel(
    fullText: fullText,
    enableIGC: enableIGC,
    enableIT: enableIT,
    userId: userId,
    prevMessages: prevMessages,
    feedback: newFeedback,
  );

  Map<String, dynamic> toJson() {
    final json = {
      ModelKey.fullText: fullText,
      ModelKey.userL1: userL1,
      ModelKey.userL2: userL2,
      ModelKey.enableIT: enableIT,
      ModelKey.enableIGC: enableIGC,
      ModelKey.userId: userId,
      ModelKey.prevMessages: jsonEncode(
        prevMessages.map((x) => x.toJson()).toList(),
      ),
    };
    if (feedback.isNotEmpty) {
      json[ModelKey.feedback] = feedback.map((f) => f.toJson()).toList();
    }
    return json;
  }

  @override
  bool operator ==(Object other) {
    if (identical(this, other)) return true;

    if (other is! IGCRequestModel) return false;

    return fullText.trim() == other.fullText.trim() &&
        fullText == other.fullText &&
        userL1 == other.userL1 &&
        userL2 == other.userL2 &&
        enableIT == other.enableIT &&
        userId == other.userId &&
        _feedbackHash == other._feedbackHash;
  }

  /// Hash of feedback content for cache differentiation
  int get _feedbackHash =>
      feedback.isEmpty ? 0 : Object.hashAll(feedback.map((f) => f.feedback));

  @override
  int get hashCode => Object.hash(
    fullText.trim(),
    userL1,
    userL2,
    enableIT,
    enableIGC,
    userId,
    _feedbackHash,
  );
}

/// Previous text/audio message sent in chat
/// Contain message content, sender, and timestamp
class PreviousMessage {
  final String content;
  final String sender;
  final DateTime timestamp;

  const PreviousMessage({
    required this.content,
    required this.sender,
    required this.timestamp,
  });

  factory PreviousMessage.fromJson(Map<String, dynamic> json) =>
      PreviousMessage(
        content: json[ModelKey.prevContent] ?? "",
        sender: json[ModelKey.prevSender] ?? "",
        timestamp: json[ModelKey.prevTimestamp] == null
            ? DateTime.now()
            : DateTime.parse(json[ModelKey.prevTimestamp]),
      );

  Map<String, dynamic> toJson() => {
    ModelKey.prevContent: content,
    ModelKey.prevSender: sender,
    ModelKey.prevTimestamp: timestamp.toIso8601String(),
  };

  @override
  bool operator ==(Object other) {
    if (identical(this, other)) return true;

    if (other is! PreviousMessage) return false;

    return content == other.content &&
        sender == other.sender &&
        timestamp == other.timestamp;
  }

  @override
  int get hashCode {
    return Object.hash(content, sender, timestamp);
  }
}