Add flag to ledger-cli file to change currency

darcys22 · Jan 9, 2022 · cf943e7 · cf943e7
1 parent 6984a11
commit cf943e7
Show file tree

Hide file tree

Showing 5 changed files with 31 additions and 7 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,7 +3,6 @@ bazel-*
 
 /tmp
 */**/*un~
-*/**/*.test
 *un~
 .DS_Store
 */**/.DS_Store

diff --git a/ledger-cli/file.go b/ledger-cli/file.go
@@ -38,7 +38,14 @@ var commandFile = &cli.Command{
 	Description: `
 	Loads a file in the ledger cli format
 `,
-	Flags: []cli.Flag{},
+	Flags: []cli.Flag{
+    &cli.StringFlag{
+      Name: "currency",
+			Aliases: []string{"c"},
+      Value: "USD",
+      Usage: "Specify the currency that the ledger file will be in, default to USD",
+    },
+	},
 	Action: func(ctx *cli.Context) error {
 		err, cfg := cmd.MakeConfig(ctx)
 		if err != nil {
@@ -60,7 +67,7 @@ var commandFile = &cli.Command{
 				return fmt.Errorf("Could not read file %s (%v)", ledgerFileName, err)
 			}
 
-			generalLedger, parseError := ParseLedger(ledgerFileReader)
+			generalLedger, parseError := ParseLedger(ledgerFileReader, ctx.String("currency"))
 			if parseError != nil {
 				return fmt.Errorf("Could not parse file (%v)", parseError)
 			}

diff --git a/ledger-cli/parse.go b/ledger-cli/parse.go
@@ -34,8 +34,8 @@ const (
 // ParseLedger parses a ledger file and returns a list of Transactions.
 //
 // Transactions are sorted by date.
-func ParseLedger(ledgerReader io.Reader) (generalLedger []*Transaction, err error) {
-	parseLedger(ledgerReader, func(t *Transaction, e error) (stop bool) {
+func ParseLedger(ledgerReader io.Reader, currency string) (generalLedger []*Transaction, err error) {
+	parseLedger(ledgerReader, currency, func(t *Transaction, e error) (stop bool) {
 		if e != nil {
 			err = e
 			stop = true
@@ -55,7 +55,7 @@ func ParseLedger(ledgerReader io.Reader) (generalLedger []*Transaction, err erro
 
 var accountToAmountSpace = regexp.MustCompile(" {2,}|\t+")
 
-func parseLedger(ledgerReader io.Reader, callback func(t *Transaction, err error) (stop bool)) {
+func parseLedger(ledgerReader io.Reader, currency string , callback func(t *Transaction, err error) (stop bool)) {
 	var trans *Transaction
 	scanner := bufio.NewScanner(ledgerReader)
 	var line string
@@ -122,7 +122,7 @@ func parseLedger(ledgerReader io.Reader, callback func(t *Transaction, err error
 			}
 			lastIndex := len(nonEmptyWords) - 1
 			balErr, rationalNum := getBalance(strings.Trim(nonEmptyWords[lastIndex], whitespace))
-			accChange.Currency = "USD"
+			accChange.Currency = currency
 			if !balErr {
 				// Assuming no balance and whole line is account name
 				accChange.Name = strings.Join(nonEmptyWords, " ")

diff --git a/utils/ledgercli-files/transaction-codes-1.test b/utils/ledgercli-files/transaction-codes-1.test
@@ -0,0 +1,3 @@
+2009/10/29 (XFER) Panera Bread
+    Expenses:Food               4.50
+    Assets:Checking             
diff --git a/utils/ledgercli-files/transaction-codes-2.test b/utils/ledgercli-files/transaction-codes-2.test
@@ -0,0 +1,15 @@
+2009/10/29 (XFER) Panera Bread
+    Expenses:Food               4.50
+    Assets:Checking             
+
+2009/10/30 (DEP) Pay day!
+    Assets:Checking            20.00
+    Income
+
+2009/10/30 (XFER) Panera Bread
+    Expenses:Food               4.50
+    Assets:Checking
+
+2009/10/31 (559385768438A8D7) Panera Bread
+    Expenses:Food               4.50
+    Liabilities:Credit Card