mercury_banking 0.5.34

data/lib/mercury_banking/reconciliation.rb

@@ -0,0 +1,139 @@
+require 'json'
+require 'fileutils'
+
+module MercuryBanking
+  # Class to manage reconciliation status for transactions
+  class Reconciliation
+    RECONCILIATION_DIR = File.join(Dir.home, '.mercury-banking', 'reconciliation')
+    
+    def initialize
+      # Ensure reconciliation directory exists
+      FileUtils.mkdir_p(RECONCILIATION_DIR)
+    end
+    
+    # Get the reconciliation file path for an account
+    def reconciliation_file(account_id)
+      File.join(RECONCILIATION_DIR, "#{account_id}.json")
+    end
+    
+    # Load reconciled transactions for an account
+    def load_reconciled_transactions(account_id)
+      file_path = reconciliation_file(account_id)
+      
+      if File.exist?(file_path)
+        begin
+          data = JSON.parse(File.read(file_path))
+          # Handle both old format (array of transaction IDs) and new format (hash with dates)
+          if data.is_a?(Array)
+            # Convert old format to new format
+            new_data = {}
+            data.each do |transaction_id|
+              new_data[transaction_id] = nil
+            end
+            return new_data
+          else
+            return data
+          end
+        rescue JSON::ParserError
+          # If the file is corrupted, return an empty hash
+          return {}
+        end
+      else
+        # If the file doesn't exist, return an empty hash
+        return {}
+      end
+    end
+    
+    # Save reconciled transactions for an account
+    def save_reconciled_transactions(account_id, transactions)
+      file_path = reconciliation_file(account_id)
+      File.write(file_path, JSON.pretty_generate(transactions))
+    end
+    
+    # Mark a transaction as reconciled
+    def mark_reconciled(account_id, transaction_id)
+      transactions = load_reconciled_transactions(account_id)
+      
+      # Add the transaction ID if it's not already in the list
+      if transactions.key?(transaction_id)
+        return false # Already reconciled
+      else
+        transactions[transaction_id] = Time.now.strftime('%Y-%m-%d')
+        save_reconciled_transactions(account_id, transactions)
+        return true # Successfully reconciled
+      end
+    end
+    
+    # Mark a transaction as unreconciled
+    def mark_unreconciled(account_id, transaction_id)
+      transactions = load_reconciled_transactions(account_id)
+      
+      # Remove the transaction ID if it's in the list
+      if transactions.key?(transaction_id)
+        transactions.delete(transaction_id)
+        save_reconciled_transactions(account_id, transactions)
+        return true # Successfully unreconciled
+      else
+        return false # Not reconciled to begin with
+      end
+    end
+    
+    # Check if a transaction is reconciled
+    def reconciled?(account_id, transaction_id)
+      transactions = load_reconciled_transactions(account_id)
+      transactions.key?(transaction_id)
+    end
+    
+    # Get all reconciled transaction IDs for an account
+    def get_reconciled_transactions(account_id)
+      load_reconciled_transactions(account_id).keys
+    end
+    
+    # Get the reconciliation date for a transaction
+    def get_reconciliation_date(account_id, transaction_id)
+      transactions = load_reconciled_transactions(account_id)
+      transactions[transaction_id]
+    end
+    
+    # Get reconciliation status for all transactions
+    def get_reconciliation_status(account_id, transactions)
+      reconciled_data = load_reconciled_transactions(account_id)
+      
+      transactions.map do |t|
+        {
+          transaction_id: t["id"],
+          date: t["postedAt"] || t["createdAt"],
+          description: t["bankDescription"] || t["externalMemo"] || "Unknown transaction",
+          amount: t["amount"],
+          reconciled: reconciled_data.key?(t["id"]),
+          reconciled_at: reconciled_data[t["id"]]
+        }
+      end
+    end
+    
+    # Get reconciliation summary
+    def get_reconciliation_summary(account_id, transactions)
+      reconciled_data = load_reconciled_transactions(account_id)
+      
+      total_transactions = transactions.size
+      reconciled_count = transactions.count { |t| reconciled_data.key?(t["id"]) }
+      unreconciled_count = total_transactions - reconciled_count
+      
+      reconciled_amount = transactions
+        .select { |t| reconciled_data.key?(t["id"]) }
+        .sum { |t| t["amount"].to_f }
+      
+      unreconciled_amount = transactions
+        .reject { |t| reconciled_data.key?(t["id"]) }
+        .sum { |t| t["amount"].to_f }
+      
+      {
+        total_transactions: total_transactions,
+        reconciled_count: reconciled_count,
+        unreconciled_count: unreconciled_count,
+        reconciled_amount: reconciled_amount,
+        unreconciled_amount: unreconciled_amount
+      }
+    end
+  end
+end