Add buttons for light and dark mode, fix functions for setting/getting milestone

WordAX/AnkiView.swift

@@ -9,12 +9,26 @@ import SwiftUI
 
 struct AnkiView: View {
     @EnvironmentObject var model: WordAXModelView
+    @State var showDescription = true
     var word: WordAX.Word? {
         model.getWordToDisplay()
     }
     var body: some View {
         if word != nil {
-            WordView(word: word!)
+            VStack {
+                WordView(word: word!, showDescription: $showDescription)
+                if showDescription {
+                    Text("How did you do?")
+                        .font(.subheadline)
+                        .foregroundStyle(.gray)
+                    HStack {
+                        NextRepetitionButtonView(buttonText: "Poor", nextMilestone: word!.nextSpacedRepetitionMilestone, wordId: word!.id)
+                        NextRepetitionButtonView(buttonText: "Good", nextMilestone: WordAX.SpacedRepetitionMilestoneEnum.getNext(milestone: word!.nextSpacedRepetitionMilestone), wordId: word!.id)
+                        NextRepetitionButtonView(buttonText: "Excellent", nextMilestone: WordAX.SpacedRepetitionMilestoneEnum.getNext(milestone: WordAX.SpacedRepetitionMilestoneEnum.getNext(milestone: word!.nextSpacedRepetitionMilestone)), wordId: word!.id)
+                    }
+                    .padding(.bottom)
+                }
+            }
         } else {
             Text("There is no word to display, come back later")
         }

WordAX/NextRepetitionButtonView.swift

@@ -0,0 +1,39 @@
+//
+//  NextRepetitionButtonView.swift
+//  WordAX
+//
+//  Created by Oliver Hnát on 25.02.2024.
+//
+
+import SwiftUI
+
+struct NextRepetitionButtonView: View {
+    let buttonText: String
+    let nextMilestone: WordAX.SpacedRepetitionMilestoneEnum?
+    let wordId: Int
+    @EnvironmentObject var model: WordAXModelView
+    @Environment(\.colorScheme) var colorScheme
+    var body: some View {
+        Button(action: {
+            model.setSpacedRepetitionMilestone(wordId: wordId, milestone: nextMilestone)
+        }) {
+            Text(buttonText)
+                .padding()
+                .foregroundColor(colorScheme == .light ? .black : .white)
+                .background(colorScheme == .light ? .cyan : .darkCyan)
+                .clipShape(RoundedRectangle(cornerRadius: 50))
+        }
+    }
+}
+
+extension ShapeStyle where Self == Color {
+    // color darker version of cyan from hex palette
+    public static var darkCyan: Color {
+        Color(red: 34/255, green: 121/255, blue: 161/255)
+    }
+}
+
+#Preview {
+    NextRepetitionButtonView(buttonText: "Excellent", nextMilestone: WordAX.SpacedRepetitionMilestoneEnum.OneDay, wordId: 0)
+        .environmentObject(WordAXModelView())
+}

WordAX/WordAX.swift

@@ -36,9 +36,12 @@ struct WordAX {
             allCases.sorted {$0.rawValue < $1.rawValue }
         }
         
-        func getNext() -> SpacedRepetitionMilestoneEnum? {
+        static func getNext(milestone: SpacedRepetitionMilestoneEnum?) -> SpacedRepetitionMilestoneEnum? {
+            if milestone == nil {
+                return SpacedRepetitionMilestoneEnum.OneDay
+            }
             let sorted = WordAX.SpacedRepetitionMilestoneEnum.allCasesSorted
-            let milestoneIndex = sorted.firstIndex(where: {$0.rawValue == self.rawValue})!
+            let milestoneIndex = sorted.firstIndex(where: {$0.rawValue == milestone!.rawValue})!
             if milestoneIndex < WordAX.SpacedRepetitionMilestoneEnum.allCasesSorted.count {
                 return sorted[milestoneIndex + 1]
             }
@@ -52,7 +55,7 @@ struct WordAX {
         var dateFormatter: DateFormatter
     }
     
-    private mutating func setNextSpacedRepetitionMilestone(word: Word) {
+    public mutating func setNextSpacedRepetitionMilestone(word: Word) {
         if word.nextSpacedRepetitionMilestone != nil {
             let current = SpacedRepetitionMilestoneEnum.allCasesSorted.firstIndex(of: word.nextSpacedRepetitionMilestone!) ?? SpacedRepetitionMilestoneEnum.allCases.count
             let index = words.firstIndex(where:{$0.id == word.id}) ?? nil
@@ -64,6 +67,13 @@ struct WordAX {
         }
     }
     
+    public mutating func setSpacedRepetitionMilestone(wordId: Int, milestone: SpacedRepetitionMilestoneEnum?) {
+        let index = words.firstIndex(where:{$0.id == wordId}) ?? nil
+        if index != nil {
+            words[index!].nextSpacedRepetitionMilestone = milestone
+        }
+    }
+    
     var words: [Word] = []
     var settings: Settings
     

WordAX/WordAXModelView.swift

@@ -45,6 +45,10 @@ class WordAXModelView: ObservableObject {
         // otherwise show nothing
         return nil
     }
+    
+    public func setSpacedRepetitionMilestone(wordId: Int, milestone: WordAX.SpacedRepetitionMilestoneEnum?) {
+        model.setSpacedRepetitionMilestone(wordId: wordId, milestone: milestone)
+    }
 }
 
 

WordAX/WordView.swift

@@ -10,7 +10,7 @@ import UIKit
 
 struct WordView: View {
     var word: WordAX.Word
-    @State var showDescription: Bool = false
+    @Binding var showDescription: Bool
     @EnvironmentObject var model: WordAXModelView
     @Environment(\.colorScheme) var colorScheme
     
@@ -19,7 +19,6 @@ struct WordView: View {
         let wordText = Text(word.name)
             .font(.title)
             .bold()
-        
         VStack {
             if word.shown && word.lastSeenOn != nil {
                 Text(model.getDateFormatter().string(from: word.lastSeenOn!))
@@ -46,6 +45,7 @@ struct WordView: View {
 }
 
 #Preview {
-    WordView(word: WordAXModelView().getWordToDisplay()!)
+    @State var showDescription = false
+    return WordView(word: WordAXModelView().getWordToDisplay()!, showDescription: $showDescription)
         .environmentObject(WordAXModelView())
 }