From 216632d7672ed8db4b6e0752e2271b0c523c2570 Mon Sep 17 00:00:00 2001
From: oliverhnat <oliver.hnat@gmail.com>
Date: Sat, 13 Apr 2024 11:57:28 +0200
Subject: [PATCH] Don't offer one minute as a next milestone

---
 WordAX/Model/Flashcard+CoreDataClass.swift | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/WordAX/Model/Flashcard+CoreDataClass.swift b/WordAX/Model/Flashcard+CoreDataClass.swift
index 9dd5541..d6d9f56 100644
--- a/WordAX/Model/Flashcard+CoreDataClass.swift
+++ b/WordAX/Model/Flashcard+CoreDataClass.swift
@@ -34,8 +34,8 @@ public class Flashcard: NSManagedObject {
                 return .TenMinutes
             }
             let milestoneIndex = sorted.firstIndex(where: {$0.rawValue == milestone!.rawValue})!
-            if milestoneIndex < SpacedRepetitionMilestoneEnum.allCasesSorted.count {
-                return sorted[milestoneIndex + 1]
+            if milestoneIndex < sorted.count {
+                return sorted[milestoneIndex + 1] == .OneMinute ? .TenMinutes : sorted[milestoneIndex + 1]
             }
             return .OneYear
         }
@@ -47,6 +47,7 @@ public class Flashcard: NSManagedObject {
     }
     
     func getSpacedRepetitionMilestone() -> SpacedRepetitionMilestoneEnum {
-        return SpacedRepetitionMilestoneEnum.getMilestoneFromInt(value: self.nextSpacedRepetitionMilestone)
+        let milestone = SpacedRepetitionMilestoneEnum.getMilestoneFromInt(value: self.nextSpacedRepetitionMilestone)
+        return milestone == .OneMinute ? .TenMinutes : milestone
     }
 }