From f1a1791e103befcf3425b3df79630372959c7e74 Mon Sep 17 00:00:00 2001
From: Takumi Muraishi <mrs1669.develop@gmail.com>
Date: Thu, 16 May 2024 01:20:11 +0900
Subject: [PATCH] [feat] add normal and hard data

---
 PrimePickApp/QuizData/PrimeData.swift | 26 ++++++++++++++++++++++++++
 PrimePickApp/QuizView.swift           | 10 +++++++++-
 2 files changed, 35 insertions(+), 1 deletion(-)

diff --git a/PrimePickApp/QuizData/PrimeData.swift b/PrimePickApp/QuizData/PrimeData.swift
index a5a21d5..7440282 100644
--- a/PrimePickApp/QuizData/PrimeData.swift
+++ b/PrimePickApp/QuizData/PrimeData.swift
@@ -23,6 +23,32 @@ final class PrimeData {
         return twoDigitPrimes
     }
     
+    public func generateThreeDigitPrimes() -> [Int] {
+        var twoDigitPrimes = [Int]()
+        outerLoop: for num in 100..<1000 {
+            for i in 2..<num {
+                if num % i == 0 {
+                    continue outerLoop
+                }
+            }
+            twoDigitPrimes.append(num)
+        }
+        return twoDigitPrimes
+    }
+    
+    public func generateFourDigitPrimes() -> [Int] {
+        var twoDigitPrimes = [Int]()
+        outerLoop: for num in 1000..<10000 {
+            for i in 2..<num {
+                if num % i == 0 {
+                    continue outerLoop
+                }
+            }
+            twoDigitPrimes.append(num)
+        }
+        return twoDigitPrimes
+    }
+    
     public func generateThreeOrFourDigitPrimes() -> [Int] {
         var twoDigitPrimes = [Int]()
         outerLoop: for num in 100..<10000 {
diff --git a/PrimePickApp/QuizView.swift b/PrimePickApp/QuizView.swift
index ab7b6e1..f181e78 100644
--- a/PrimePickApp/QuizView.swift
+++ b/PrimePickApp/QuizView.swift
@@ -14,7 +14,15 @@ struct QuizView: View {
     
     init(difficulty: String) {
         self.difficulty = difficulty
-        primes = primeData.generateOneOrTwoDigitPrimes()
+        if self.difficulty == "Easy" {
+            primes = primeData.generateOneOrTwoDigitPrimes()
+        } else if self.difficulty == "Normal" {
+            primes = primeData.generateThreeDigitPrimes()
+        } else if self.difficulty == "Hard" {
+            primes = primeData.generateFourDigitPrimes()
+        } else {
+            primes = []
+        }
     }
 
     var body: some View {