Update report to receive flag for source type

Report previously tried to determine the type of input data from the data itself. However, before the report is generated, we already know the source type from the command line arguments. Report now gets passed a flag to specify the type of source data
stonish · Oct 13, 2021 · 816f9cd · 816f9cd
1 parent 7b9942e
commit 816f9cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/transactions/process.go b/transactions/process.go
@@ -13,8 +13,8 @@ func Process(revolutFile string, aibFile string, startDate time.Time) {
   revolutLines := utils.ReadInputFile(revolutFile)
   aibLines := utils.ReadInputFile(aibFile)
 
-  report = report.AddTransactions(revolutLines)
-  report = report.AddTransactions(aibLines)
+  report = report.AddTransactions(revolutLines, true)
+  report = report.AddTransactions(aibLines, false)
 
   report.Sort()
   report.Output()

diff --git a/transactions/report.go b/transactions/report.go
@@ -3,7 +3,6 @@ package transactions
 import (
   "fmt"
   "time"
-  "strings"
   "sort"
 )
 
@@ -20,8 +19,7 @@ func NewReport(startDate time.Time) *report {
   return r
 }
 
-func (r report) AddTransactions(rawData[]string) *report {
-  isRevolut := strings.HasPrefix(rawData[0], "Completed Date")
+func (r report) AddTransactions(rawData[]string, isRevolut bool) *report {
   if isRevolut {
     for i := len(rawData) - 1; i > 0; i-- {
       transaction := NewRevolutTransaction(rawData[i])