🚀 3단계 - 로또(2등)

src/main/kotlin/lotto/application/LottoApplication.kt

@@ -1,8 +1,8 @@
 package lotto.application
 
-import lotto.Lotto
-import lotto.LottoResult
-import lotto.LottoVendingMachine
+import lotto.domain.Lotto
+import lotto.domain.LottoNumber
+import lotto.domain.Lottos
 import lotto.view.InputView
 import lotto.view.OutputView
 
@@ -12,18 +12,17 @@ class LottoApplication(
 ) {
 
     fun run() {
-        outputView.showInputMoneyMessage()
         val money = inputView.inputMoney()
-        val lotto = LottoVendingMachine.buyLotto(money)
+        val lotto = Lottos.buyLotto(money)
         outputView.showLottoCount(lotto)
         outputView.showLotto(lotto)
 
-        outputView.showInputWinningNumbersMessage()
-        val winningNumbers: List<Int> = inputView.inputWinningNumbers()
+        val winningLotto: Lotto = inputView.inputWinningNumbers()
+        val bonusNumber: LottoNumber = inputView.inputBonusNumber()
 
-        val lottoResult= LottoResult.makeLottoResult(
-            winningLotto = Lotto(*winningNumbers.toIntArray()),
-            lottos = lotto,
+        val lottoResult = lotto.getResult(
+            winningLotto = winningLotto,
+            bonusLottoNumber = bonusNumber
         )
         outputView.showResult(lottoResult, money)
     }

src/main/kotlin/lotto/domain/Lotto.kt

@@ -0,0 +1,35 @@
+package lotto.domain
+
+class Lotto private constructor(val lottoNumbers: Set<LottoNumber>) {
+
+    init {
+        val lottoNumbersSize = lottoNumbers.size
+        require(lottoNumbersSize == LOTTO_SIZE)
+    }
+
+    constructor() : this(
+        LottoPreset.shuffled()
+            .take(6)
+            .sortedBy { it.number }
+            .toSet()
+    )
+
+    constructor(vararg number: Int) : this(
+        number.map(::LottoNumber)
+            .toSet()
+    )
+
+    fun countMatch(winningLotto: Lotto): Int {
+        return lottoNumbers.intersect(winningLotto.lottoNumbers)
+            .count()
+    }
+
+    operator fun contains(lottoNumber: LottoNumber): Boolean {
+        return lottoNumber in lottoNumbers
+    }
+
+    companion object {
+        private val LottoPreset = List(LottoNumber.END_LOTTO_NUMBER) { LottoNumber(it + 1) }
+        private const val LOTTO_SIZE = 6
+    }
+}

src/main/kotlin/lotto/domain/LottoNumber.kt

@@ -0,0 +1,14 @@
+package lotto.domain
+
+@JvmInline
+value class LottoNumber(val number: Int) {
+    init {
+        require(number in lottoNumberRange)
+    }
+
+    companion object {
+        const val START_LOTTO_NUMBER = 1
+        const val END_LOTTO_NUMBER = 45
+        val lottoNumberRange = START_LOTTO_NUMBER..END_LOTTO_NUMBER
+    }
+}

src/main/kotlin/lotto/domain/LottoResult.kt

@@ -0,0 +1,18 @@
+package lotto.domain
+
+class LottoResult(private val result: Map<Prize, Int>) : Map<Prize, Int> by result {
+    private val totalPrizeMoney: Long
+        get() {
+            return result.map {
+                it.key.money.toLong() * it.value
+            }.sum()
+        }
+
+    override fun get(key: Prize): Int {
+        return result[key] ?: 0
+    }
+
+    fun getRateOfReturn(money: Int): Float {
+        return totalPrizeMoney.toFloat() / money
+    }
+}

src/main/kotlin/lotto/domain/Lottos.kt

@@ -0,0 +1,31 @@
+package lotto.domain
+
+class Lottos(lottos: List<Lotto>) {
+    val lottos: List<Lotto> = lottos.toList()
+
+    fun getResult(
+        winningLotto: Lotto,
+        bonusLottoNumber: LottoNumber,
+    ): LottoResult {
+        val result = lottos.groupingBy { lotto ->
+            val matchCount = lotto.countMatch(winningLotto)
+            Prize.of(matchCount = matchCount, bonusLottoNumber = bonusLottoNumber, lotto = lotto)
+        }.eachCount()
+        return LottoResult(result)
+    }
+
+    companion object {
+        private const val LOTTO_PRICE = 1000
+
+        fun buyLotto(money: Int?): Lottos {
+            requireNotNull(money) {
+                "구매 금액은 null이 아니어야 함"
+            }
+            require(money >= LOTTO_PRICE) {
+                "구매 금액은 1000 보다 커야 함"
+            }
+
+            return Lottos(List(money / LOTTO_PRICE) { Lotto() })
+        }
+    }
+}

src/main/kotlin/lotto/domain/Prize.kt

@@ -0,0 +1,21 @@
+package lotto.domain
+
+enum class Prize(val money: Int, val count: Int) {
+    FIRST(2_000_000_000, 6),
+    SECOND(30_000_000, 5),
+    THIRD(1_500_000, 5),
+    FOURTH(50_000, 4),
+    FIFTH(5_000, 3),
+    NONE(0, 0),
+    ;
+
+    companion object {
+        fun of(matchCount: Int, bonusLottoNumber: LottoNumber, lotto: Lotto) = when (matchCount) {
+            6 -> FIRST
+            5 -> if (bonusLottoNumber in lotto) SECOND else THIRD
+            4 -> FOURTH
+            3 -> FIFTH
+            else -> NONE
+        }
+    }
+}

src/main/kotlin/lotto/view/InputView.kt

@@ -1,13 +1,28 @@
 package lotto.view
 
+import lotto.domain.Lotto
+import lotto.domain.LottoNumber
+
 class InputView {
     fun inputMoney(): Int {
+        println("구입금액을 입력해 주세요.")
         val input = readlnOrNull()?.toIntOrNull()
         requireNotNull(input)
         return input
     }
 
-    fun inputWinningNumbers(): List<Int> {
-        return readlnOrNull()?.split(",")?.map { it.trim().toInt() } ?: emptyList()
+    fun inputWinningNumbers(): Lotto {
+        println("지난 주 당첨 번호를 입력해 주세요.")
+        val winningNumbers: List<Int> = readlnOrNull()?.split(",")
+            ?.map {
+                it.trim()
+                    .toInt()
+            } ?: emptyList()
+        return Lotto(*winningNumbers.toIntArray())
+    }
+
+    fun inputBonusNumber(): LottoNumber {
+        println("보너스 볼을 입력해 주세요.")
+        return LottoNumber(readlnOrNull()?.toIntOrNull() ?: 0)
     }
 }

src/main/kotlin/lotto/view/OutputView.kt

@@ -1,36 +1,40 @@
 package lotto.view
 
-import lotto.Lotto
-import lotto.LottoResult
-import lotto.Prize
+import lotto.domain.LottoResult
+import lotto.domain.Lottos
+import lotto.domain.Prize
 
 class OutputView {
-    fun showInputMoneyMessage() {
-        println("구입금액을 입력해 주세요.")
+    fun showLottoCount(lottos: Lottos) {
+        val size = lottos.lottos
+            .size
+        println("${size}개를 구매했습니다.")
     }
 
-    fun showLottoCount(lotto: List<Lotto>) {
-        println("${lotto.size}개를 구매했습니다.")
-    }
-
-    fun showLotto(lotto: List<Lotto>) {
-        lotto.forEach {
-            println(it)
-        }
-    }
-
-    fun showInputWinningNumbersMessage() {
-        println("지난 주 당첨 번호를 입력해 주세요.")
+    fun showLotto(lottos: Lottos) {
+        lottos.lottos
+            .forEach {
+                println(
+                    "[${
+                        it.lottoNumbers.map { lottoNumber -> lottoNumber.number }
+                            .joinToString(", ")
+                    }]")
+            }
     }
 
     fun showResult(lottoResult: LottoResult, money: Int) {
         println("당첨 통계")
         println("---------")
         Prize.entries
             .filter { it != Prize.NONE }
-            .sortedByDescending { it.count }
+            .reversed()
             .forEach { prize ->
-                println("${prize.count}개 일치 (${prize.money}원) - ${lottoResult[prize]}개")
+                val title = if (prize == Prize.SECOND) {
+                    "${prize.count}개 일치, 보너스 볼 일치 (${prize.money}원)"
+                } else {
+                    "${prize.count}개 일치 (${prize.money}원)"
+                }
+                println("$title - ${lottoResult[prize]}개")
             }
 
         println(String.format("총 수익률은 %.2f입니다.", lottoResult.getRateOfReturn(money)))

src/test/kotlin/lotto/LottoNumberTest.kt

@@ -2,6 +2,7 @@ package lotto
 
 import io.kotest.assertions.throwables.shouldNotThrow
 import io.kotest.assertions.throwables.shouldThrow
+import lotto.domain.LottoNumber
 import org.junit.jupiter.api.Test
 
 class LottoNumberTest {