Add CSV input support

Author: hannotify

.gitignore

@@ -5,3 +5,6 @@ build/
 #IntelliJ IDEA
 .idea
 *.iml
+
+#Test bank statements
+src/test/resources/*.csv

build.gradle.kts

@@ -16,19 +16,16 @@ repositories {
     // Use jcenter for resolving dependencies.
     // You can declare any Maven/Ivy/file repository here.
     jcenter()
+    maven("https://jitpack.io")
 }
 
 dependencies {
-    // Align versions of all Kotlin components
+    // Kotlin
     implementation(platform("org.jetbrains.kotlin:kotlin-bom"))
-
-    // Use the Kotlin JDK 8 standard library.
     implementation("org.jetbrains.kotlin:kotlin-stdlib-jdk8")
+    implementation("com.github.ptnplanet:Java-Naive-Bayes-Classifier:1.0.7")
 
-    // Use the Kotlin test library.
     testImplementation("org.jetbrains.kotlin:kotlin-test")
-
-    // Use the Kotlin JUnit integration.
     testImplementation("org.jetbrains.kotlin:kotlin-test-junit")
 }
 

…main/kotlin/com/github/hannotify/Main.kt …/com/github/hannotify/classyfire/Main.ktsrc/main/kotlin/com/github/hannotify/Main.kt renamed to src/main/kotlin/com/github/hannotify/classyfire/Main.kt src/main/kotlin/com/github/hannotify/Main.kt renamed to src/main/kotlin/com/github/hannotify/classyfire/Main.kt

@@ -1,4 +1,4 @@
-package com.github.hannotify
+package com.github.hannotify.classyfire
 
 fun main() {
     println("Hello from the Classy Fire!")

src/main/kotlin/com/github/hannotify/classyfire/data/Transaction.kt

@@ -0,0 +1,7 @@
+package com.github.hannotify.classyfire.data
+
+import java.math.BigDecimal
+import java.time.LocalDate
+
+data class Transaction(val date: LocalDate, val description: String, val beneficiaryIban: String,
+                       val amount: BigDecimal, val transactionType: String, val remarks: String)

src/main/kotlin/com/github/hannotify/classyfire/input/InputReader.kt

@@ -0,0 +1,7 @@
+package com.github.hannotify.classyfire.input
+
+import com.github.hannotify.classyfire.data.Transaction
+
+interface InputReader {
+    fun read(filePath: String): Collection<Transaction>
+}

src/main/kotlin/com/github/hannotify/classyfire/input/impl/IngCsvInputReader.kt

@@ -0,0 +1,45 @@
+package com.github.hannotify.classyfire.input.impl
+
+import com.github.hannotify.classyfire.data.Transaction
+import com.github.hannotify.classyfire.input.InputReader
+import java.io.File
+import java.math.BigDecimal
+import java.time.LocalDate
+import java.time.format.DateTimeFormatter
+
+class IngCsvInputReader : InputReader {
+    override fun read(filePath: String): Collection<Transaction> {
+        val transactions = mutableListOf<Transaction>()
+
+        File(filePath).readLines()
+                .filter { !isHeaderLine(it) }
+                .forEach { transactions.add(toTransaction(it)) }
+
+        return transactions
+    }
+
+    private fun isHeaderLine(line: String): Boolean {
+        return line.startsWith("\"Datum")
+    }
+
+    private fun toTransaction(line: String): Transaction {
+        val fields = line.split("\",\"")
+                .map { it.removePrefix("\"") }
+                .map { it.removeSuffix("\"") }
+
+        val date = LocalDate.parse(fields[0], DateTimeFormatter.ofPattern("yyyyMMdd"))
+        val description = fields[1]
+        val beneficiaryIban = fields[3]
+        val amount = toAmount(fields[6], fields[5])
+        val transactionType = fields[7]
+        val remarks = fields[8]
+
+        return Transaction(date, description, beneficiaryIban, amount, transactionType, remarks);
+    }
+
+    private fun toAmount(amountString: String, bijAf: String): BigDecimal {
+        val amount = BigDecimal(amountString.replace(',', '.'))
+
+        return if ("Bij" == bijAf) amount else amount.negate()
+    }
+}

src/test/kotlin/com/github/hannotify/AppTest.kt

@@ -0,0 +1,20 @@
+package com.github.hannotify.classyfire.input.impl
+
+import kotlin.test.Test
+import kotlin.test.assertEquals
+
+internal class IngCsvInputReaderTest {
+
+    @Test
+    fun read() {
+        // Arrange
+        val inputReader = IngCsvInputReader()
+
+        // Act
+        val transactions = inputReader.read("src/test/resources/test.csv")
+
+        // Assert
+        assertEquals(96, transactions.size);
+        println(transactions)
+    }
+}