fluffychat/lib/pangea/analytics_misc/construct_use_model.dart

import 'dart:math';

import 'package:collection/collection.dart';

import 'package:fluffychat/pangea/analytics_misc/analytics_constants.dart';
import 'package:fluffychat/pangea/analytics_misc/construct_type_enum.dart';
import 'package:fluffychat/pangea/analytics_misc/construct_use_type_enum.dart';
import 'package:fluffychat/pangea/analytics_misc/constructs_model.dart';
import 'package:fluffychat/pangea/analytics_misc/practice_tier_enum.dart';
import 'package:fluffychat/pangea/constructs/construct_identifier.dart';
import 'package:fluffychat/pangea/constructs/construct_level_enum.dart';
import 'package:fluffychat/pangea/practice_activities/activity_type_enum.dart';

/// One lemma and a list of construct uses for that lemma
class ConstructUses {
  final List<OneConstructUse> _uses;
  final ConstructTypeEnum constructType;
  final String lemma;
  String? _category;

  ConstructUses({
    required List<OneConstructUse> uses,
    required this.constructType,
    required this.lemma,
    required category,
  }) : _category = category,
       _uses = List<OneConstructUse>.from(uses) {
    _sortUses();
  }

  // Total points for all uses of this lemma
  int get points {
    return min(
      _uses.fold<int>(0, (total, use) => total + use.xp),
      AnalyticsConstants.xpForFlower,
    );
  }

  DateTime? get lastUsed => _uses.lastOrNull?.timeStamp;
  DateTime? get cappedLastUse => cappedUses.lastOrNull?.timeStamp;

  String get category {
    if (_category == null || _category!.isEmpty) return "other";
    return _category!.toLowerCase();
  }

  bool get hasCorrectUse => _uses.any((use) => use.xp > 0);
  bool get hasIncorrectUse => _uses.any((use) => use.xp < 0);

  int get numTotalUses => _uses.length;

  ConstructIdentifier get id => ConstructIdentifier(
    lemma: lemma,
    type: constructType,
    category: category,
  );

  /// Get the lemma category, based on points
  ConstructLevelEnum get lemmaCategory {
    if (points < AnalyticsConstants.xpForGreens) {
      return ConstructLevelEnum.seeds;
    } else if (points >= AnalyticsConstants.xpForFlower) {
      return ConstructLevelEnum.flowers;
    }
    return ConstructLevelEnum.greens;
  }

  String get xpEmoji {
    if (points < 30) {
      // bean emoji
      return AnalyticsConstants.emojiForSeed;
    } else if (points < 100) {
      // sprout emoji
      return AnalyticsConstants.emojiForGreen;
    } else {
      // flower emoji
      return AnalyticsConstants.emojiForFlower;
    }
  }

  ConstructLevelEnum get constructLevel => switch (points) {
    < AnalyticsConstants.xpForGreens => ConstructLevelEnum.seeds,
    < AnalyticsConstants.xpForFlower => ConstructLevelEnum.greens,
    _ => ConstructLevelEnum.flowers,
  };

  List<OneConstructUse> get cappedUses {
    final result = <OneConstructUse>[];
    var totalXp = 0;

    for (final use in _uses) {
      if (totalXp >= AnalyticsConstants.xpForFlower) break;
      totalXp += use.xp;
      result.add(use);
    }

    return result;
  }

  /// Read-only view of all uses, sorted chronologically (oldest first).
  List<OneConstructUse> get uses => List.unmodifiable(_uses);

  /// Classify this construct into a [PracticeTier] based on use-type history.
  ///
  /// Walks uses in reverse chronological order to find the most recent
  /// chat use and any incorrect practice answers after it.
  PracticeTier get practiceTier {
    // Walk reverse chronologically. Everything seen before finding the
    // last chat use is more recent than that chat use.
    bool hasIncorrectAfterLastChatUse = false;

    for (int i = _uses.length - 1; i >= 0; i--) {
      final use = _uses[i];

      if (use.useType.isChatUse) {
        // Found the most recent chat use.
        if (use.useType == ConstructUseTypeEnum.wa &&
            !hasIncorrectAfterLastChatUse) {
          return PracticeTier.suppressed;
        }
        if (use.useType.isAssistedChatUse) {
          return PracticeTier.active;
        }
        // wa with incorrect after → active
        if (hasIncorrectAfterLastChatUse) {
          return PracticeTier.active;
        }
        return PracticeTier.maintenance;
      }

      if (use.useType.isIncorrectPractice) {
        hasIncorrectAfterLastChatUse = true;
      }
    }

    // No chat use found (only practice history).
    if (hasIncorrectAfterLastChatUse) return PracticeTier.active;
    return PracticeTier.maintenance;
  }

  DateTime? lastUseByTypes(List<ConstructUseTypeEnum> types) =>
      _uses.lastWhereOrNull((u) => types.contains(u.useType))?.timeStamp;

  /// Compute priority score for this construct.
  ///
  /// Higher score = higher priority (should be practiced sooner).
  /// Suppressed-tier constructs return 0.
  ///
  /// When [activityType] is provided, recency is checked against that
  /// activity's specific use types (e.g., corPA/incPA for wordMeaning).
  /// Otherwise, aggregate recency across all use types is used.
  int practiceScore({ActivityTypeEnum? activityType}) {
    final tier = practiceTier;
    if (tier == PracticeTier.suppressed) return 0;

    // Per-activity-type recency when available, otherwise aggregate.
    final DateTime? lastUsedDate = activityType != null
        ? lastUseByTypes(activityType.associatedUseTypes)
        : lastUsed;

    final daysSince = lastUsedDate == null
        ? AnalyticsConstants.defaultDaysSinceLastUsed
        : DateTime.now().difference(lastUsedDate).inDays;

    final wordMultiplier = id.isContentWord
        ? AnalyticsConstants.contentWordMultiplier
        : AnalyticsConstants.functionWordMultiplier;

    var score = daysSince * wordMultiplier;

    if (tier == PracticeTier.active) {
      score *= AnalyticsConstants.activeTierMultiplier;
    }

    return score;
  }

  Map<String, dynamic> toJson() {
    final json = {
      'construct_id': id.toJson(),
      'xp': points,
      'last_used': lastUsed?.toIso8601String(),
      'uses': _uses.map((e) => e.toJson()).toList(),
    };
    return json;
  }

  factory ConstructUses.fromJson(Map<String, dynamic> json) {
    final constructId = ConstructIdentifier.fromJson(
      Map<String, dynamic>.from(json['construct_id']),
    );

    List<dynamic> usesJson = [];
    if (json['uses'] is List) {
      usesJson = List<dynamic>.from(json['uses']);
    }

    final uses = usesJson
        .map((e) => OneConstructUse.fromJson(Map<String, dynamic>.from(e)))
        .toList();

    return ConstructUses(
      uses: uses,
      constructType: constructId.type,
      lemma: constructId.lemma,
      category: constructId.category,
    );
  }

  void _sortUses() {
    _uses.sort((a, b) => a.timeStamp.compareTo(b.timeStamp));
  }

  void addUses(List<OneConstructUse> uses) {
    _uses.addAll(uses);
    _sortUses();
  }

  void merge(ConstructUses other) {
    if (other.lemma.toLowerCase() != lemma.toLowerCase() ||
        other.constructType != constructType) {
      throw ArgumentError(
        'Cannot merge ConstructUses with different lemmas or types',
      );
    }

    _uses.addAll(other._uses);
    _sortUses();

    if (category == 'other' && other.category != 'other') {
      _category = other.category;
    }
  }

  ConstructUses copyWith({
    List<OneConstructUse>? uses,
    ConstructTypeEnum? constructType,
    String? lemma,
    String? category,
  }) {
    return ConstructUses(
      uses: uses ?? _uses,
      constructType: constructType ?? this.constructType,
      lemma: lemma ?? this.lemma,
      category: category ?? _category,
    );
  }
}