hledger-forecast 0.4.0 → 1.1.0

data/lib/hledger_forecast/transactions/modifiers.rb

@@ -0,0 +1,90 @@
+module HledgerForecast
+  module Transactions
+    # Generate auto-posting hledger transactions
+    # Example output:
+    # = Expenses:Groceries date:2024-01-01..2025-12-31
+    #    Expenses:Groceries    *0.1 ;  Groceries
+    #    Assets:Checking       *-0.1
+    class Modifiers
+      def self.generate(data, options)
+        new(data, options).generate
+      end
+
+      def generate
+        return nil unless modifiers?
+
+        process_modifier
+
+        output
+      end
+
+      def self.get_modifiers(transaction, block)
+        modifiers = []
+
+        transaction['modifiers'].each do |modifier|
+          description = transaction['description']
+          description += " - #{modifier['description']}" unless modifier['description'].empty?
+
+          modifiers << {
+            account: block['account'],
+            amount: modifier['amount'],
+            category: transaction['category'],
+            description: description,
+            from: Date.parse(modifier['from'] || block['from']),
+            to: modifier['to'] ? Date.parse(modifier['to']) : nil
+          }
+        end
+
+        modifiers
+      end
+
+      private
+
+      attr_reader :data, :options, :output
+
+      def initialize(data, options)
+        @data = data
+        @options = options
+        @output = []
+      end
+
+      def modifiers?
+        @data.any? do |_, blocks|
+          blocks.any? do |block|
+            block[:transactions].any? do |_, transactions|
+              transactions.any? { |t| !t[:modifiers].empty? }
+            end
+          end
+        end
+      end
+
+      def process_modifier
+        get_transactions.each do |modifier|
+          account = modifier[:account].ljust(@options[:max_category])
+          category = modifier[:category].ljust(@options[:max_category])
+          # Fix the ljust by counting strings in amount
+          amount = modifier[:amount].to_s.ljust(@options[:max_amount] - 1)
+          to = modifier[:to] ? "..#{modifier[:to]}" : nil
+
+          header = "= #{modifier[:category]} date:#{modifier[:from]}#{to}\n"
+          transactions = "    #{category}    *#{amount};  #{modifier[:description]}\n"
+          footer = "    #{account}    *#{modifier[:amount] * -1}\n\n"
+
+          output << { header: header, transactions: [transactions], footer: footer }
+        end
+      end
+
+      def get_transactions
+        @data.each_with_object([]) do |(_key, blocks), result|
+          blocks.each do |block|
+            block[:transactions].each_value do |transactions|
+              transactions.each do |t|
+                result.concat(t[:modifiers]) if t[:modifiers]
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/hledger_forecast/transactions/trackers.rb

@@ -0,0 +1,87 @@
+module HledgerForecast
+  module Transactions
+    # Generate hledger transactions based on the non-existance of a transaction
+    # in your ledger. This is useful for ensuring that certain expenses are
+    # accounted for, even if you forget to enter them.
+    #
+    # Example output:
+    # ~ 2023-05-1  * [TRACKED] Food expenses
+    #    Expenses:Groceries    $250.00 ;  Food expenses
+    #    Assets:Checking
+    class Trackers
+      def self.generate(data, options)
+        new(data, options).generate
+      end
+
+      def generate
+        return nil unless tracked?(data)
+
+        data.each_value do |blocks|
+          blocks.each do |block|
+            process_tracked(block)
+          end
+        end
+
+        output
+      end
+
+      def self.track?(transaction, data, options)
+        now = Date.today
+        transaction['track'] && Date.parse(data['from']) <= now && !exists?(transaction, data['account'],
+                                                                            data['from'], now, options)
+      end
+
+      def self.exists?(transaction, account, from, to, options)
+        # Format the money
+        amount = Formatter.format_money(transaction['amount'], options)
+        inverse_amount = Formatter.format_money(transaction['amount'] * -1, options)
+
+        category = transaction['category'].gsub('[', '\\[').gsub(']', '\\]').gsub('(', '\\(').gsub(')', '\\)')
+
+        # We run two commands and check to see if category +/- amount or account +/- amount exists
+        command1 = %(hledger print -f #{options[:transaction_file]} "date:#{from}..#{to}" | tr -s '[:space:]' ' ' | grep -q -Eo "#{category} (#{amount}|#{inverse_amount})")
+        command2 = %(hledger print -f #{options[:transaction_file]} "date:#{from}..#{to}" | tr -s '[:space:]' ' ' | grep -q -Eo "#{account} (#{amount}|#{inverse_amount})")
+
+        system(command1) || system(command2)
+      end
+
+      private
+
+      attr_reader :data, :options, :output
+
+      def initialize(data, options)
+        @data = data
+        @options = options
+        @output = []
+      end
+
+      def tracked?(data)
+        data.any? do |_, blocks|
+          blocks.any? do |block|
+            block[:transactions].any? do |_, transactions|
+              transactions.any? { |t| t[:track] }
+            end
+          end
+        end
+      end
+
+      def process_tracked(block)
+        block[:transactions].each do |_to, transactions|
+          transactions.each do |t|
+            next unless t[:track]
+
+            category = t[:category].ljust(options[:max_category])
+            amount = t[:amount].to_s.ljust(options[:max_amount])
+
+            header = "~ #{Date.new(Date.today.year, Date.today.month,
+                                   1).next_month}  * [TRACKED] #{t[:description]}\n"
+            transactions = "    #{category}    #{amount};  #{t[:description]}\n"
+            footer = "    #{block[:account]}\n\n"
+
+            output << { header: header, transactions: [transactions], footer: footer }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/hledger_forecast/version.rb

@@ -1,3 +1,3 @@
 module HledgerForecast
-  VERSION = "0.4.0"
+  VERSION = "1.1.0"
 end

data/lib/hledger_forecast.rb

@@ -12,8 +12,16 @@ require 'yaml'
 Money.locale_backend = nil
 Money.rounding_mode = BigDecimal::ROUND_HALF_UP
 
-require_relative 'hledger_forecast/version'
-require_relative 'hledger_forecast/generator'
-require_relative 'hledger_forecast/summarize'
-require_relative 'hledger_forecast/tracker'
+require_relative 'hledger_forecast/calculator'
 require_relative 'hledger_forecast/cli'
+require_relative 'hledger_forecast/formatter'
+require_relative 'hledger_forecast/generator'
+require_relative 'hledger_forecast/settings'
+require_relative 'hledger_forecast/summarizer'
+require_relative 'hledger_forecast/summarizer_formatter'
+require_relative 'hledger_forecast/version'
+
+require_relative 'hledger_forecast/transactions/default'
+require_relative 'hledger_forecast/transactions/modifiers'
+require_relative 'hledger_forecast/transactions/trackers'
+

data/spec/custom_spec.rb

@@ -1,6 +1,6 @@
 require_relative '../lib/hledger_forecast'
 
-config = <<~YAML
+base_config = <<~YAML
   custom:
     - frequency: "every 2 weeks"
       from: "2023-05-01"
@@ -21,7 +21,7 @@ config = <<~YAML
     currency: GBP
 YAML
 
-output = <<~JOURNAL
+base_output = <<~JOURNAL
   ~ every 2 weeks from 2023-05-01  * Hair and beauty
       [Expenses:Personal Care]    £80.00;  Hair and beauty
       [Assets:Bank]
@@ -32,9 +32,36 @@ output = <<~JOURNAL
 
 JOURNAL
 
+calculated_config = <<~YAML
+  custom:
+    - frequency: "every 2 weeks"
+      from: "2023-05-01"
+      account: "[Assets:Bank]"
+      transactions:
+        - amount: 80
+          category: "[Expenses:Personal Care]"
+          description: Hair and beauty
+          to: "=6"
+
+  settings:
+    currency: GBP
+YAML
+
+calculated_output = <<~JOURNAL
+  ~ every 2 weeks from 2023-05-01 to 2023-10-31  * Hair and beauty
+      [Expenses:Personal Care]    £80.00;  Hair and beauty
+      [Assets:Bank]
+
+JOURNAL
+
 RSpec.describe 'generate' do
   it 'generates a forecast with correct CUSTOM transactions' do
-    generated_journal = HledgerForecast::Generator.generate(config)
-    expect(generated_journal).to eq(output)
+    generated_journal = HledgerForecast::Generator.generate(base_config)
+    expect(generated_journal).to eq(base_output)
+  end
+
+  it 'generates a forecast with correct CUSTOM transactions and CALCULATED to dates' do
+    generated_journal = HledgerForecast::Generator.generate(calculated_config)
+    expect(generated_journal).to eq(calculated_output)
   end
 end

data/spec/modifier_spec.rb

@@ -5,7 +5,7 @@ base_config = <<~YAML
     - account: "Assets:Bank"
       from: "2023-01-01"
       transactions:
-        - amount: 500
+        - amount: 300
           category: "Expenses:Groceries"
           description: Food shopping
           modifiers:
@@ -17,6 +17,17 @@ base_config = <<~YAML
               description: "Y2 inflation"
               from: "2025-01-01"
               to: "2025-12-31"
+    - account: "Assets:Savings"
+      from: "2023-05-01"
+      transactions:
+        - amount: 500
+          category: "Assets:Bank"
+          description: Savings
+          modifiers:
+            - amount: 0.1
+              description: "Savings uplift"
+              from: "2024-05-01"
+              to: "2025-04-30"
 
   settings:
     currency: USD
@@ -24,9 +35,13 @@ YAML
 
 base_journal = <<~JOURNAL
   ~ monthly from 2023-01-01  * Food shopping
-      Expenses:Groceries    $500.00;  Food shopping
+      Expenses:Groceries    $300.00;  Food shopping
       Assets:Bank
 
+  ~ monthly from 2023-05-01  * Savings
+      Assets:Bank           $500.00;  Savings
+      Assets:Savings
+
   = Expenses:Groceries date:2024-01-01..2024-12-31
       Expenses:Groceries    *0.02 ;  Food shopping - Y1 inflation
       Assets:Bank           *-0.02
@@ -35,6 +50,10 @@ base_journal = <<~JOURNAL
       Expenses:Groceries    *0.05 ;  Food shopping - Y2 inflation
       Assets:Bank           *-0.05
 
+  = Assets:Bank date:2024-05-01..2025-04-30
+      Assets:Bank           *0.1  ;  Savings - Savings uplift
+      Assets:Savings        *-0.1
+
 JOURNAL
 
 no_date_config = <<~YAML
@@ -67,20 +86,16 @@ JOURNAL
 RSpec.describe 'Applying modifiers to transactions -' do
   it 'Auto-postings should be created correctly' do
     generated = HledgerForecast::Generator
-    generated.modified = {} # Clear modified transactions
 
     generated_journal = generated.generate(base_config)
-    generated.modified = {} # Clear modified transactions
 
     expect(generated_journal).to eq(base_journal)
   end
 
   it 'Auto-postings should be created correctly if no dates are set' do
     generated = HledgerForecast::Generator
-    generated.modified = {} # Clear modified transactions
 
     generated_journal = generated.generate(no_date_config)
-    generated.modified = {} # Clear modified transactions
 
     expect(generated_journal).to eq(no_date_journal)
   end

data/spec/monthly_end_date_spec.rb

@@ -1,47 +1,32 @@
 require_relative '../lib/hledger_forecast'
 
 config = <<~YAML
-settings:
-  currency: GBP
+  settings:
+    currency: GBP
 
-monthly:
-  - from: "2023-03-01"
-    account: "Assets:Bank"
-    transactions:
-      - description: Mortgage
-        to: "2023-06-01"
-        category: "Expenses:Mortgage"
-        amount: 2000.00
-      - description: Mortgage top up
-        to: "2023-06-01"
-        category: "Expenses:Mortgage Top Up"
-        amount: 200.00
-      - description: Food
-        category: "Expenses:Food"
-        amount: 100.00
-      - description: Party time
-        category: "Expenses:Going Out"
-        amount: 50.00
+  monthly:
+    - from: "2023-03-01"
+      to: "2023-06-01"
+      account: "Assets:Bank"
+      transactions:
+        - description: Mortgage
+          category: "Expenses:Mortgage"
+          amount: 2000.00
+        - description: Food
+          category: "Expenses:Food"
+          amount: 100.00
 YAML
 
 output = <<~JOURNAL
-~ monthly from 2023-03-01  * Food, Party time
-    Expenses:Food               £100.00  ;  Food
-    Expenses:Going Out          £50.00   ;  Party time
-    Assets:Bank
-
-~ monthly from 2023-03-01 to 2023-06-01  * Mortgage
-    Expenses:Mortgage           £2,000.00;  Mortgage
-    Assets:Bank
-
-~ monthly from 2023-03-01 to 2023-06-01  * Mortgage top up
-    Expenses:Mortgage Top Up    £200.00  ;  Mortgage top up
-    Assets:Bank
+  ~ monthly from 2023-03-01 to 2023-06-01  * Mortgage, Food
+      Expenses:Mortgage    £2,000.00;  Mortgage
+      Expenses:Food        £100.00  ;  Food
+      Assets:Bank
 
 JOURNAL
 
 RSpec.describe 'generate' do
-  it 'generates a forecast with correct MONTHLY transactions that have an end date' do
+  it 'generates a forecast with correct MONTHLY transactions that have an end date, at the top level' do
     expect(HledgerForecast::Generator.generate(config)).to eq(output)
   end
 end

data/spec/monthly_end_date_transaction_spec.rb

@@ -1,32 +1,69 @@
 require_relative '../lib/hledger_forecast'
 
-config = <<~YAML
+base_config = <<~YAML
   settings:
     currency: GBP
 
   monthly:
     - from: "2023-03-01"
-      to: "2023-06-01"
       account: "Assets:Bank"
       transactions:
         - description: Mortgage
+          to: "2023-06-01"
           category: "Expenses:Mortgage"
           amount: 2000.00
+        - description: Mortgage top up
+          to: "2023-06-01"
+          category: "Expenses:Mortgage Top Up"
+          amount: 200.00
         - description: Food
           category: "Expenses:Food"
           amount: 100.00
+        - description: Party time
+          category: "Expenses:Going Out"
+          amount: 50.00
 YAML
 
-output = <<~JOURNAL
-  ~ monthly from 2023-03-01 to 2023-06-01  * Mortgage, Food
+base_output = <<~JOURNAL
+  ~ monthly from 2023-03-01 to 2023-06-01  * Mortgage, Mortgage top up
+      Expenses:Mortgage           £2,000.00;  Mortgage
+      Expenses:Mortgage Top Up    £200.00  ;  Mortgage top up
+      Assets:Bank
+
+  ~ monthly from 2023-03-01  * Food, Party time
+      Expenses:Food               £100.00  ;  Food
+      Expenses:Going Out          £50.00   ;  Party time
+      Assets:Bank
+
+JOURNAL
+
+computed_config = <<~YAML
+  settings:
+    currency: GBP
+
+  monthly:
+    - from: "2023-03-01"
+      account: "Assets:Bank"
+      transactions:
+        - description: Mortgage
+          category: "Expenses:Mortgage"
+          to: "=12"
+          amount: 2000.00
+YAML
+
+computed_output = <<~JOURNAL
+  ~ monthly from 2023-03-01 to 2024-02-29  * Mortgage
       Expenses:Mortgage    £2,000.00;  Mortgage
-      Expenses:Food        £100.00  ;  Food
       Assets:Bank
 
 JOURNAL
 
 RSpec.describe 'generate' do
-  it 'generates a forecast with correct MONTHLY transactions that have an end date, at the top level' do
-    expect(HledgerForecast::Generator.generate(config)).to eq(output)
+  it 'generates a forecast with correct MONTHLY transactions that have an end date' do
+    expect(HledgerForecast::Generator.generate(base_config)).to eq(base_output)
+  end
+
+  it 'generates a forecast with correct MONTHLY transactions that have a COMPUTED end date' do
+    expect(HledgerForecast::Generator.generate(computed_config)).to eq(computed_output)
   end
 end

data/spec/monthly_spec.rb

@@ -23,14 +23,14 @@ config = <<~YAML
 YAML
 
 output = <<~JOURNAL
-~ monthly from 2023-03-01  * Mortgage, Food
-    Expenses:Mortgage    £2,000.55;  Mortgage
-    Expenses:Food        £100.00  ;  Food
-    Assets:Bank
+  ~ monthly from 2023-03-01  * Mortgage, Food
+      Expenses:Mortgage    £2,000.55;  Mortgage
+      Expenses:Food        £100.00  ;  Food
+      Assets:Bank
 
-~ monthly from 2023-03-01  * Savings
-    Assets:Bank          £-1,000.00;  Savings
-    Assets:Savings
+  ~ monthly from 2023-03-01  * Savings
+      Assets:Bank          £-1,000.00;  Savings
+      Assets:Savings
 
 JOURNAL
 

data/spec/summarizer_spec.rb

@@ -0,0 +1,86 @@
+require_relative '../lib/hledger_forecast'
+
+config = <<~YAML
+  monthly:
+    - account: "Assets:Bank"
+      from: "2023-03-01"
+      transactions:
+        - amount: 2000.55
+          category: "Expenses:Mortgage"
+          description: Mortgage
+          to: "=24"
+        - amount: 100
+          category: "Expenses:Food"
+          description: Food
+    - account: "Assets:Savings"
+      from: "2023-03-01"
+      transactions:
+        - amount: -1000
+          category: "Assets:Bank"
+          description: Savings
+
+  custom:
+    - frequency: "every 2 weeks"
+      from: "2023-05-01"
+      account: "[Assets:Bank]"
+      roll-up: 26
+      transactions:
+        - amount: 80
+          category: "[Expenses:Personal Care]"
+          description: Hair and beauty
+    - frequency: "every 5 days"
+      from: "2023-05-01"
+      account: "[Assets:Checking]"
+      roll-up: 73
+      transactions:
+        - amount: 50
+          category: "[Expenses:Groceries]"
+          description: Gotta feed that stomach
+
+  settings:
+    currency: GBP
+YAML
+
+RSpec.describe HledgerForecast::Summarizer do
+  let(:summarizer) { described_class.new }
+
+  describe '#generate with roll_up' do
+    let(:forecast) { YAML.safe_load(config) }
+    let(:cli_options) { { roll_up: 'monthly' } }
+
+    before do
+      summarizer.summarize(config, cli_options)
+    end
+
+    it 'generates the correct output' do
+      output = summarizer.send(:generate, forecast)
+
+      expect(output.first).to include(:account, :from, :to, :type, :frequency)
+      expect(output.first[:amount]).to eq(2000.55)
+      expect(output.last[:rolled_up_amount]).to eq(304) # ((50 * 73) / 12)
+      expect(output.length).to eq(5)
+    end
+
+    it 'transaction TO date take precedence over block TO date' do
+      output = summarizer.send(:generate, forecast)
+
+      expect(output.first[:to]).to eq(Date.parse("2025-02-28"))
+    end
+  end
+
+  describe '#generate' do
+    let(:forecast) { YAML.safe_load(config) }
+    let(:cli_options) { nil }
+
+    before do
+      summarizer.summarize(config, cli_options)
+    end
+
+    it 'generates the correct output' do
+      output = summarizer.send(:generate, forecast)
+
+      # expect(output.first).to include(:account, :from, :to, :type, :frequency)
+      expect(output.length).to eq(5)
+    end
+  end
+end

data/spec/track_spec.rb

@@ -42,70 +42,13 @@ base_output = <<~JOURNAL
 JOURNAL
 
 RSpec.describe 'Tracking transactions -' do
-  it 'Determines which transactions should be tracked' do
-    generated = HledgerForecast::Generator
-    generated.generate(base_config)
-    tracked = generated.tracked
-
-    expect(tracked[0]['transaction']).to eq(
-      { "amount" => "£3,000.00", "category" => "Expenses:Tax", "description" => "Tax owed",
-        "inverse_amount" => "£-3,000.00", "track" => true }
-    )
-    expect(tracked[0]['account']).to eq("Assets:Bank")
-
-    expect(tracked[1]['transaction']).to eq(
-      { "amount" => "£-1,500.00", "category" => "Income:Salary", "description" => "Salary", "to" => "2023-08-01",
-        "inverse_amount" => "£1,500.00", "track" => true }
-    )
-    expect(tracked[1]['account']).to eq("Assets:Bank")
-  end
-
-  it 'marks a transaction as NOT FOUND if it doesnt exist' do
-    generated = HledgerForecast::Generator
-    generated.tracked = {} # Clear tracked transactions
-    generated.generate(base_config)
-    transactions_to_track = generated.tracked
-
-    track = HledgerForecast::Tracker.track(transactions_to_track, 'spec/stubs/transactions_not_found.journal')
-
-    expect(track[0]['found']).to eq(false)
-    expect(track[1]['found']).to eq(false)
-  end
-
-  it 'marks a transaction as FOUND if it exists' do
-    generated = HledgerForecast::Generator
-    generated.tracked = {} # Clear tracked transactions
-    generated.generate(base_config)
-    transactions_to_track = generated.tracked
-
-    track = HledgerForecast::Tracker.track(transactions_to_track, 'spec/stubs/transactions_found.journal')
-
-    expect(track[0]['found']).to eq(true)
-    expect(track[1]['found']).to eq(true)
-  end
-
-  it 'marks a transaction as FOUND if it exists, even if the category/amount are inversed' do
-    generated = HledgerForecast::Generator
-    generated.tracked = {} # Clear tracked transactions
-    generated.generate(base_config)
-    transactions_to_track = generated.tracked
-
-    track = HledgerForecast::Tracker.track(transactions_to_track, 'spec/stubs/transactions_found_inverse.journal')
-
-    expect(track[0]['found']).to eq(true)
-  end
-
   it 'writes a NON-FOUND entry into a journal' do
     options = {}
     options[:transaction_file] = 'spec/stubs/transactions_not_found.journal'
 
-    generated = HledgerForecast::Generator
-    generated.tracked = {} # Clear tracked transactions
-
-    generated_journal = generated.generate(base_config, options)
+    generated_journal = HledgerForecast::Generator.generate(base_config, options)
 
-    expected_output = base_output
-    expect(generated_journal).to eq(expected_output)
+    expect(generated_journal).to eq(base_output)
   end
 
   it 'writes a NON-FOUND entry for dates that are close to the current period' do
@@ -147,12 +90,10 @@ RSpec.describe 'Tracking transactions -' do
     options = {}
     options[:transaction_file] = temp_file.path
 
-    generated = HledgerForecast::Generator
-    generated.tracked = {} # Clear tracked transactions
-
-    generated_journal = generated.generate(forecast_config, options)
+    generated_journal = HledgerForecast::Generator.generate(forecast_config, options)
 
     expected_output = <<~JOURNAL
+
       ~ #{next_month}  * [TRACKED] New kitchen
           Expenses:House    £5,000.00;  New kitchen
           Assets:Bank
@@ -180,10 +121,7 @@ RSpec.describe 'Tracking transactions -' do
     options = {}
     options[:transaction_file] = 'spec/stubs/transactions_not_found.journal'
 
-    generated = HledgerForecast::Generator
-    generated.tracked = {} # Clear tracked transactions
-
-    generated_journal = generated.generate(forecast_config, options)
+    generated_journal = HledgerForecast::Generator.generate(forecast_config, options)
 
     output = <<~JOURNAL
       ~ monthly from #{next_month}  * Food expenses