workouttest_app/lib/repository/training_plan_repository.dart

import 'package:aitrainer_app/model/cache.dart';
import 'package:aitrainer_app/model/customer_training_plan.dart';
import 'package:aitrainer_app/model/customer_training_plan_details.dart';
import 'package:aitrainer_app/model/exercise.dart';
import 'package:aitrainer_app/model/exercise_plan_detail.dart';
import 'package:aitrainer_app/model/exercise_tree.dart';
import 'package:aitrainer_app/model/training_plan.dart';
import 'package:aitrainer_app/repository/training_plan_day_repository.dart';
import 'package:aitrainer_app/service/training_plan_service.dart';
import 'package:aitrainer_app/util/common.dart';

class TrainingPlanRepository {
  ExerciseTree? parentTree;
  List<TrainingPlan> getPlansByParent(String parent) {
    final List<TrainingPlan> resultList = [];
    final List<ExerciseTree>? exerciseTree = Cache().getExerciseTree();
    int? parentId;
    if (exerciseTree != null) {
      exerciseTree.forEach((element) {
        if (element.internalName == parent) {
          parentId = element.treeId;
          parentTree = element;
        }
      });
    }

    final List<TrainingPlan>? plans = Cache().getTrainingPlans();
    if (plans != null && parentId != null) {
      plans.forEach((element) {
        if (element.treeId == parentId) {
          resultList.add(element);
        }
      });
    }
    return resultList;
  }

  /// 1. deactivate old training plans - update all

  /// 2. calculate customer_training_plan_details weights / repleats
  /// 3. create new customer_training_plan

  Future<CustomerTrainingPlan?> activateTrainingPlan(int trainingPlanId) async {
    print(" **** Activate Plan: $trainingPlanId");
    // 1. deactivate
    if (Cache().getCustomerTrainingPlans() != null) {
      Cache().getCustomerTrainingPlans()!.forEach((plan) {
        plan.active = false;
        if (plan.customerTrainingPlanId != null) {
          TrainingPlanApi().updateCustomerTrainingPlan(plan, plan.customerTrainingPlanId!);
        }
      });
    }

    CustomerTrainingPlan plan = CustomerTrainingPlan();
    plan.customerId = Cache().userLoggedIn!.customerId;
    plan.trainingPlanId = trainingPlanId;
    plan.active = true;
    plan.status = "open";
    plan.dateAdd = DateTime.now();
    plan.name = getTrainingPlanById(trainingPlanId)!.nameTranslations["hu"];

    TrainingPlan? trainingPlan = this.getTrainingPlanById(trainingPlanId);
    if (trainingPlan == null || trainingPlan.details == null) {
      return null;
    }

    // 3 calculate weights
    int index = 0;

    trainingPlan.details!.forEach((elem) {
      CustomerTrainingPlanDetails detail = CustomerTrainingPlanDetails();
      detail.customerTrainingPlanDetailsId = ++index;
      detail.trainingPlanDetailsId = elem.trainingPlanDetailId;
      detail.exerciseTypeId = elem.exerciseTypeId;
      detail.repeats = elem.repeats;
      detail.set = elem.set;
      detail.dayId = elem.dayId;
      TrainingPlanDayRepository trainingPlanDayRepository = TrainingPlanDayRepository();
      detail.day = trainingPlanDayRepository.getNameById(elem.dayId);
      detail.parallel = elem.parallel;
      detail.restingTime = elem.restingTime;
      detail.exerciseType = Cache().getExerciseTypeById(detail.exerciseTypeId!);
      if (elem.weight == -1) {
        if (detail.exerciseType!.unitQuantityUnit != null) {
          detail = getCalculatedWeightRepeats(elem.exerciseTypeId, detail);
        } else {
          detail.weight = 0;
        }
      } else if (elem.weight == -2) {
        final CustomerTrainingPlanDetails calculated = this.isWeightCalculatedByExerciseType(elem.exerciseTypeId, detail, plan);
        if (calculated.weight != -1) {
          detail.weight = calculated.weight;
        } else {
          detail.weight = -2;
        }
      } else {
        detail.weight = elem.weight;
      }
      //print("Detail $detail exerciseType: ${detail.exerciseType}");

      detail.state = ExercisePlanDetailState.start;
      plan.details.add(detail);
    });

    Cache().myTrainingPlan = plan;

    //TrainingPlanApi().saveCustomerTrainingPlan(plan);
    return plan;
  }

  CustomerTrainingPlanDetails isWeightCalculatedByExerciseType(
      int exerciseTypeId, CustomerTrainingPlanDetails detail, CustomerTrainingPlan plan) {
    CustomerTrainingPlanDetails calculated = detail;
    for (var element in plan.details) {
      if (element.exerciseTypeId == exerciseTypeId) {
        calculated = element;
        break;
      }
    }

    return calculated;
  }

  TrainingPlan? getTrainingPlanById(int trainingPlanId) {
    TrainingPlan? plan;
    if (Cache().getTrainingPlans() == null) {
      return plan;
    }

    for (var trainingPlan in Cache().getTrainingPlans()!) {
      if (trainingPlan.trainingPlanId == trainingPlanId) {
        plan = trainingPlan;
        break;
      }
    }

    return plan;
  }

  CustomerTrainingPlanDetails getCalculatedWeightRepeats(int exerciseTypeId, CustomerTrainingPlanDetails detail) {
    double weight = -1;
    if (Cache().getExercises() == null) {
      detail.weight = weight;
      detail.isTest = true;
      return detail;
    }

    Exercise? lastExercise1RM;
    Cache().getExercises()!.forEach((exercise) {
      if (exercise.exercisePlanDetailId == 0 && exercise.exerciseTypeId == exerciseTypeId) {
        detail.weight = weight;
        lastExercise1RM = exercise;
      }
    });

    if (lastExercise1RM == null || lastExercise1RM!.unitQuantity == null) {
      detail.weight = weight;
      detail.isTest = true;
      return detail;
    }

    double oneRepMax = Common.calculate1RM(lastExercise1RM!.unitQuantity!, lastExercise1RM!.quantity!);
    //print("Exercise $exerciseTypeId - 1RM : $oneRepMax");
    weight = oneRepMax * Common.get1RMPercent(detail.repeats!);
    //print("Exercise $exerciseTypeId - weight : $weight");
    weight = Common.roundWeight(weight);

    detail.repeats = Common.calculateQuantityByChangedWeight(oneRepMax, weight, detail.repeats!.toDouble());

    detail.weight = weight;
    return detail;
  }

  int getOriginalRepeats(int trainingPlanId, CustomerTrainingPlanDetails detail) {
    TrainingPlan? plan = getTrainingPlanById(trainingPlanId);
    if (plan == null) {
      return 0;
    }
    int originalRepeats = 0;
    plan.details!.forEach((element) {
      if (element.trainingPlanDetailId == detail.trainingPlanDetailsId) {
        originalRepeats = element.repeats ?? 0;
      }
    });
    return originalRepeats;
  }

  double getOriginalWeight(int trainingPlanId, CustomerTrainingPlanDetails detail) {
    TrainingPlan? plan = getTrainingPlanById(trainingPlanId);
    if (plan == null) {
      return 0;
    }
    double originalWeight = 0;
    plan.details!.forEach((element) {
      if (element.trainingPlanDetailId == detail.trainingPlanDetailsId) {
        originalWeight = element.weight ?? 0;
      }
    });
    return originalWeight;
  }

  CustomerTrainingPlanDetails recalculateDetailFixRepeats(int trainingPlanId, CustomerTrainingPlanDetails detail) {
    TrainingPlan? plan = getTrainingPlanById(trainingPlanId);
    if (plan == null) {
      return detail;
    }
    int originalRepeats = getOriginalRepeats(trainingPlanId, detail);

    detail.weight = Common.calculateWeigthByChangedQuantity(detail.weight!, detail.repeats!.toDouble(), originalRepeats.toDouble());
    detail.weight = Common.roundWeight(detail.weight!);
    print("Recalculated weight: ${detail.weight}");
    detail.repeats = originalRepeats;
    return detail;
  }

  CustomerTrainingPlanDetails recalculateDetail(
      int trainingPlanId, CustomerTrainingPlanDetails detail, CustomerTrainingPlanDetails nextDetail) {
    CustomerTrainingPlanDetails recalculatedDetail = nextDetail;

    // 1. get original repeats

    // 1a get original plan
    TrainingPlan? plan = getTrainingPlanById(trainingPlanId);
    if (plan == null) {
      return recalculatedDetail;
    }

    // 1.b get the original detail's repeat
    int originalRepeats = detail.repeats!;
    plan.details!.forEach((element) {
      if (element.trainingPlanDetailId == detail.trainingPlanDetailsId) {
        originalRepeats = element.repeats ?? 0;
      }
    });

    // 2 get recalculated repeats
    recalculatedDetail.weight =
        Common.calculateWeigthByChangedQuantity(detail.weight!, detail.repeats!.toDouble(), originalRepeats.toDouble());
    recalculatedDetail.weight = Common.roundWeight(recalculatedDetail.weight!);
    print("recalculated repeats for $originalRepeats: ${recalculatedDetail.weight}");
    //recalculatedDetail.repeats = originalRepeats;

    return recalculatedDetail;
  }
}