Add support for importing CSV

david-swift · Mar 30, 2024 · a49c0ff · a49c0ff
1 parent afac0cd
commit a49c0ff
Show file tree

Hide file tree

Showing 4 changed files with 76 additions and 37 deletions.
diff --git a/Sources/Model/FlashcardsApp.swift b/Sources/Model/FlashcardsApp.swift
@@ -9,6 +9,7 @@ enum FlashcardsApp: String, CaseIterable, Identifiable {
 
     case quizlet
     case anki
+    case csv
 
     var id: String { rawValue }
 
@@ -18,6 +19,8 @@ enum FlashcardsApp: String, CaseIterable, Identifiable {
             Loc.quizlet
         case .anki:
             Loc.anki
+        case .csv:
+            Loc.csv
         }
     }
 

diff --git a/Sources/Model/Localized.yml b/Sources/Model/Localized.yml
@@ -733,3 +733,15 @@ searchFlashcards:
     en: Search Flashcards
     de: Karteikarten suchen
     it: Cerca Flashcard
+
+csv:
+    en: CSV
+    de: CSV
+
+csvDescription:
+    en: Paste the text into the text field in the next view. Flashcards are separated by newlines, the front and back by commas.
+    de: Füge den Text in das Textfeld der nächsten Ansicht ein. Karteikarten werden mit Zeilenumbrüchen, die Vorder- und Rückseite mit Kommas getrennt.
+
+importCSV:
+    en: Import CSV
+    de: CSV importieren
diff --git a/Sources/View/EditView.swift b/Sources/View/EditView.swift
@@ -143,7 +143,7 @@ struct EditView: View {
         } secondary: {
             importFlashcards = true
         }
-        .dialog(visible: $importFlashcards, width: 400, height: 500) {
+        .dialog(visible: $importFlashcards, width: 400, height: 450) {
             ImportView(set: $set) { importFlashcards = false }
         }
     }

diff --git a/Sources/View/ImportView.swift b/Sources/View/ImportView.swift
@@ -9,6 +9,10 @@ import RegexBuilder
 
 struct ImportView: View {
 
+    // swiftlint:disable large_tuple
+    typealias FlashcardsRegex = Regex<(Substring, Substring, Substring)>
+    // swiftlint:enable large_tuple
+
     @Binding var set: FlashcardsSet
     @State private var text = ""
     @State private var switchSides = false
@@ -28,10 +32,10 @@ struct ImportView: View {
                         .topToolbar {
                             toolbar(destination: destination)
                         }
-                case .paste:
+                case let .paste(app):
                     VStack {
                         entry
-                        preview
+                        preview(app: app)
                     }
                     .topToolbar {
                         toolbar(destination: destination)
@@ -75,37 +79,6 @@ struct ImportView: View {
         .padding(20)
     }
 
-    var preview: View {
-        ScrollView {
-            CarouselView(set: .constant(previewSet))
-        }
-        .vexpand()
-    }
-
-    var previewSet: FlashcardsSet {
-        var previewSet = set
-        previewSet.flashcards = []
-        let pattern = Regex {
-            Capture {
-                ZeroOrMore(.anyNonNewline)
-            }
-            "\t"
-            Capture {
-                ZeroOrMore(.anyNonNewline)
-            }
-        }
-        for (index, match) in text.matches(of: pattern).enumerated() {
-            let (front, back): (Substring, Substring)
-            if switchSides {
-                (_, back, front) = match.output
-            } else {
-                (_, front, back) = match.output
-            }
-            previewSet.flashcards.append(.init(id: "\(index)", front: .init(front), back: .init(back)))
-        }
-        return previewSet
-    }
-
     var quizletTutorial: View {
         StatusPage(
             Loc.exportQuizletSet,
@@ -125,21 +98,72 @@ struct ImportView: View {
         }
     }
 
-    @ViewBuilder var ankiTutorial: View {
+    var ankiTutorial: View {
         StatusPage(
             Loc.exportAnkiDeck,
             icon: .custom(name: "io.github.david_swift.Flashcards.set-symbolic"),
             description: Loc.ankiDescription
         )
     }
 
+    var csvTutorial: View {
+        StatusPage(
+            Loc.importCSV,
+            icon: .default(icon: .emblemDocuments),
+            description: Loc.csvDescription
+        )
+    }
+
+    func preview(app: FlashcardsApp) -> View {
+        ScrollView {
+            CarouselView(set: .constant(previewSet(app: app)))
+        }
+        .vexpand()
+    }
+
+    func previewSet(app: FlashcardsApp) -> FlashcardsSet {
+        var previewSet = set
+        previewSet.flashcards = []
+        let pattern: FlashcardsRegex
+        switch app {
+        case .csv:
+            pattern = newlineRegex(separator: ",")
+        default:
+            pattern = newlineRegex(separator: "\t")
+        }
+        for (index, match) in text.matches(of: pattern).enumerated() {
+            let (front, back): (Substring, Substring)
+            if switchSides {
+                (_, back, front) = match.output
+            } else {
+                (_, front, back) = match.output
+            }
+            previewSet.flashcards.append(.init(id: "\(index)", front: .init(front), back: .init(back)))
+        }
+        return previewSet
+    }
+
+    func newlineRegex(separator: String) -> FlashcardsRegex {
+        .init {
+            Capture {
+                ZeroOrMore(.anyNonNewline)
+            }
+            separator
+            Capture {
+                ZeroOrMore(.anyNonNewline)
+            }
+        }
+    }
+
     @ViewBuilder
     func tutorial(app: FlashcardsApp) -> View {
         switch app {
         case .quizlet:
             quizletTutorial
         case .anki:
             ankiTutorial
+        case .csv:
+            csvTutorial
         }
     }
 
@@ -160,8 +184,8 @@ struct ImportView: View {
                 }
             }()
             Button(label) {
-                if case .paste = destination {
-                    set.flashcards += previewSet.flashcards
+                if case let .paste(app) = destination {
+                    set.flashcards += previewSet(app: app).flashcards
                     close()
                 } else {
                     if case let .tutorial(app) = destination {