mercury_banking 0.5.34

data/lib/mercury_banking/reports/balance_sheet.rb

@@ -0,0 +1,586 @@
+module MercuryBanking
+  module Reports
+    # Module for generating balance sheet reports
+    module BalanceSheet
+      # Generate a ledger balance sheet
+      def generate_ledger_balance_sheet(file_path, end_date = nil)
+        # Check if ledger command exists
+        unless command_exists?('ledger')
+          puts "Error: 'ledger' command not found. Please install ledger (https://www.ledger-cli.org/)."
+          return nil
+        end
+        
+        # Verify the file exists
+        unless File.exist?(file_path)
+          puts "Error: Ledger file not found at #{file_path}"
+          return nil
+        end
+        
+        # Verify and potentially fix the ledger file
+        verified_file_path = verify_ledger_file_syntax(file_path)
+        return nil unless verified_file_path
+        
+        # Generate the balance sheet
+        generate_ledger_balance_report(verified_file_path, end_date)
+      end
+      
+      # Verify the syntax of a ledger file and fix common issues if needed
+      def verify_ledger_file_syntax(file_path)
+        check_cmd = "ledger -f #{file_path} --verify"
+        check_output = `#{check_cmd}`
+        
+        if $?.success?
+          return file_path
+        else
+          puts "Warning: Ledger file verification failed. The file may contain syntax errors."
+          puts "Error output: #{check_output}"
+          
+          # Try to identify problematic lines
+          problematic_lines = `grep -n '^[0-9]' #{file_path} | head -10`
+          puts "First few transaction lines for inspection:"
+          puts problematic_lines
+          
+          # Try to fix common issues
+          puts "Attempting to fix common issues in the ledger file..."
+          fix_ledger_file_syntax(file_path)
+        end
+      end
+      
+      # Fix common syntax issues in a ledger file
+      def fix_ledger_file_syntax(file_path)
+        fixed_file_path = "#{file_path}.fixed"
+        
+        # Create a fixed version of the file with safer formatting
+        File.open(fixed_file_path, 'w') do |fixed_file|
+          File.foreach(file_path) do |line|
+            if line.match?(/^\d{4}[\/\-\.]\d{2}[\/\-\.]\d{2}/)
+              # This is a transaction line, ensure description is safe
+              date, description = line.strip.split(' ', 2)
+              if description && (description.match?(/^\d/) || 
+                                 description.match?(/^[v#]/) || 
+                                 description.match?(/^\s*\d{4}/) || 
+                                 description.match?(/^\s*\d{2}\/\d{2}/))
+                fixed_file.puts "#{date} - #{description}"
+              else
+                fixed_file.puts line
+              end
+            else
+              fixed_file.puts line
+            end
+          end
+        end
+        
+        puts "Created fixed ledger file at #{fixed_file_path}"
+        fixed_file_path
+      end
+      
+      # Generate a balance sheet report using ledger
+      def generate_ledger_balance_report(file_path, end_date = nil)
+        # Construct the ledger command
+        cmd = "ledger -f #{file_path}"
+        cmd += " --end #{end_date}" if end_date
+        cmd += " balance --flat Assets Liabilities Equity"
+        
+        # Execute the command
+        output = `#{cmd}`
+        
+        # Check if the command was successful
+        if $?.success?
+          output
+        else
+          puts "Error executing ledger command: #{cmd}"
+          puts "Error output: #{output}"
+          
+          # Try with a more permissive approach
+          try_permissive_ledger_command(file_path, end_date)
+        end
+      end
+      
+      # Try a more permissive ledger command as a fallback
+      def try_permissive_ledger_command(file_path, end_date = nil)
+        puts "Trying with more permissive options..."
+        cmd = "ledger -f #{file_path} --permissive"
+        cmd += " --end #{end_date}" if end_date
+        cmd += " balance --flat Assets Liabilities Equity"
+        
+        output = `#{cmd}`
+        
+        if $?.success?
+          output
+        else
+          puts "Error executing permissive ledger command."
+          puts "Error output: #{output}"
+          nil
+        end
+      end
+      
+      # Generate a beancount balance sheet
+      def generate_beancount_balance_sheet(file_path, end_date = nil)
+        # Check if bean-report command exists
+        unless command_exists?('bean-report')
+          puts "Error: 'bean-report' command not found. Please install beancount (https://beancount.github.io/)."
+          return nil
+        end
+        
+        # Construct the beancount command
+        cmd = "bean-report #{file_path} balances"
+        cmd += " --end #{end_date}" if end_date
+        
+        # Execute the command
+        output = `#{cmd}`
+        
+        # Check if the command was successful
+        if $?.success?
+          output
+        else
+          puts "Error executing bean-report command: #{cmd}"
+          puts output
+          nil
+        end
+      end
+      
+      # Parse balance sheet output to extract account balances
+      def parse_balance_sheet_output(output, format, debug = false)
+        balances = {}
+        
+        if debug
+          puts "\n=== Balance Sheet Parsing Debug ==="
+          puts "Raw output to parse:"
+          puts output
+        end
+        
+        case format
+        when 'ledger'
+          balances = parse_ledger_balance_output(output, debug)
+        when 'beancount'
+          balances = parse_beancount_balance_output(output, debug)
+        else
+          puts "Unsupported format: #{format}"
+        end
+        
+        balances
+      end
+      
+      def parse_ledger_balance_output(output, debug = false)
+        balances = {}
+        
+        output.each_line do |line|
+          puts "Parsing line: #{line.inspect}" if debug
+          line = line.strip
+          
+          # Skip empty lines and summary lines
+          if line.empty? || line.include?('----')
+            puts "  Skipping line (empty or summary)" if debug
+            next
+          end
+          
+          # Extract account name and balance using a more flexible regex
+          # The ledger output format is typically something like:
+          #              $18.10  Assets:Mercury Checking ••1090
+          if match = line.match(/\$\s*([\d,.-]+)\s+(.+)/)
+            balance = match[1].gsub(',', '').to_f
+            account = match[2].strip
+            
+            puts "  Extracted: Account='#{account}', Balance=$#{balance}" if debug
+            
+            # Add to balances hash
+            balances[account] = balance
+            puts "  Added to balances hash" if debug
+          else
+            puts "  No match for balance pattern" if debug
+          end
+        end
+        
+        balances
+      end
+      
+      def parse_beancount_balance_output(output, debug = false)
+        balances = {}
+        
+        output.each_line do |line|
+          # Skip header lines and summary lines
+          next if line.strip.empty? || line.include?('---') || line.include?('Assets:') || line.include?('Liabilities:')
+          
+          # Extract account name and balance
+          if line =~ /(\S.*?)\s+([\d,.-]+)\s+USD/
+            account = $1.strip
+            balance = $2.gsub(',', '').to_f
+            
+            # Add to balances hash
+            balances[account] = balance
+          end
+        end
+        
+        balances
+      end
+      
+      # Generate ledger reports (P&L, balance sheet, etc.)
+      def generate_ledger_reports(file_path, report_type, category = nil, end_date = nil)
+        # Check if ledger is installed
+        unless command_exists?('ledger')
+          puts "Error: 'ledger' command not found. Please install Ledger CLI to generate reports."
+          return
+        end
+        
+        # Build date filter
+        date_filter = end_date ? " --end #{end_date}" : ""
+        
+        # Build category filter
+        category_filter = category ? " #{category}" : ""
+        
+        # Prepare to capture output
+        report_output = ""
+        report_output << "=== Mercury Banking Ledger Reports ===\n\n"
+        
+        puts "\n=== Mercury Banking Ledger Reports ===\n"
+        
+        # Generate requested reports
+        if ['all', 'balance'].include?(report_type)
+          report_output << generate_ledger_balance_report_section(file_path, date_filter, category_filter)
+        end
+        
+        if ['all', 'pl'].include?(report_type)
+          report_output << generate_ledger_pl_report_section(file_path, date_filter, category_filter)
+        end
+        
+        if ['all', 'monthly'].include?(report_type)
+          report_output << generate_ledger_monthly_report_section(file_path, date_filter, category)
+        end
+        
+        if report_type == 'register'
+          report_output << generate_ledger_register_report_section(file_path, date_filter, category)
+        end
+        
+        return report_output
+      end
+      
+      def generate_ledger_balance_report_section(file_path, date_filter, category_filter)
+        section_output = ""
+        
+        balance_cmd = "ledger -f #{file_path}#{date_filter} balance --flat Assets Liabilities Equity#{category_filter}"
+        balance_output = `#{balance_cmd}`
+        
+        puts "\n=== Balance Sheet ===\n"
+        puts balance_output
+        
+        section_output << "=== Balance Sheet ===\n"
+        section_output << balance_output
+        section_output << "\n"
+        
+        section_output
+      end
+      
+      def generate_ledger_pl_report_section(file_path, date_filter, category_filter)
+        section_output = ""
+        
+        pl_cmd = "ledger -f #{file_path}#{date_filter} balance --flat Income Expenses#{category_filter}"
+        pl_output = `#{pl_cmd}`
+        
+        puts "\n=== Income Statement ===\n"
+        puts pl_output
+        
+        section_output << "=== Income Statement ===\n"
+        section_output << pl_output
+        section_output << "\n"
+        
+        section_output
+      end
+      
+      def generate_ledger_monthly_report_section(file_path, date_filter, category)
+        section_output = ""
+        
+        if category
+          monthly_cmd = "ledger -f #{file_path}#{date_filter} --monthly --collapse register #{category}"
+          monthly_output = `#{monthly_cmd}`
+          
+          puts "\n=== Monthly #{category} ===\n"
+          puts monthly_output
+          
+          section_output << "=== Monthly #{category} ===\n"
+          section_output << monthly_output
+          section_output << "\n"
+        else
+          # Generate monthly expenses report
+          section_output << generate_ledger_monthly_category_report(file_path, date_filter, "Expenses")
+          
+          # Generate monthly income report
+          section_output << generate_ledger_monthly_category_report(file_path, date_filter, "Income")
+        end
+        
+        section_output
+      end
+      
+      def generate_ledger_monthly_category_report(file_path, date_filter, category)
+        section_output = ""
+        
+        monthly_cmd = "ledger -f #{file_path}#{date_filter} --monthly --collapse register #{category}"
+        monthly_output = `#{monthly_cmd}`
+        
+        puts "\n=== Monthly #{category} ===\n"
+        puts monthly_output
+        
+        section_output << "=== Monthly #{category} ===\n"
+        section_output << monthly_output
+        section_output << "\n"
+        
+        section_output
+      end
+      
+      def generate_ledger_register_report_section(file_path, date_filter, category)
+        section_output = ""
+        
+        register_filter = category || "Assets Liabilities Equity Income Expenses"
+        register_cmd = "ledger -f #{file_path}#{date_filter} register #{register_filter}"
+        register_output = `#{register_cmd}`
+        
+        puts "\n=== Transaction Register ===\n"
+        puts register_output
+        
+        section_output << "=== Transaction Register ===\n"
+        section_output << register_output
+        section_output << "\n"
+        
+        section_output
+      end
+      
+      # Generate beancount reports
+      def generate_beancount_reports(file_path, report_type, category = nil, end_date = nil)
+        # Check if bean-report is installed
+        unless command_exists?('bean-report')
+          puts "Error: 'bean-report' command not found. Please install Beancount to generate reports."
+          return
+        end
+        
+        # Prepare to capture output
+        report_output = ""
+        report_output << "=== Mercury Banking Beancount Reports ===\n\n"
+        
+        puts "\n=== Mercury Banking Beancount Reports ===\n"
+        
+        # Build date filter (beancount uses different syntax)
+        date_filter = end_date ? " -e #{end_date}" : ""
+        
+        # Generate requested reports
+        if ['all', 'balance'].include?(report_type)
+          report_output << generate_beancount_balance_report_section(file_path, date_filter)
+        end
+        
+        if ['all', 'pl'].include?(report_type)
+          report_output << generate_beancount_pl_report_section(file_path, date_filter)
+        end
+        
+        if ['all', 'monthly'].include?(report_type)
+          report_output << generate_beancount_monthly_report_section(file_path, date_filter)
+        end
+        
+        if report_type == 'register'
+          report_output << generate_beancount_register_report_section(file_path, date_filter, category)
+        end
+        
+        return report_output
+      end
+      
+      def generate_beancount_balance_report_section(file_path, date_filter)
+        section_output = ""
+        
+        balance_cmd = "bean-report #{file_path}#{date_filter} balsheet"
+        balance_output = `#{balance_cmd}`
+        
+        puts "\n=== Balance Sheet ===\n"
+        puts balance_output
+        
+        section_output << "=== Balance Sheet ===\n"
+        section_output << balance_output
+        section_output << "\n"
+        
+        section_output
+      end
+      
+      def generate_beancount_pl_report_section(file_path, date_filter)
+        section_output = ""
+        
+        pl_cmd = "bean-report #{file_path}#{date_filter} income"
+        pl_output = `#{pl_cmd}`
+        
+        puts "\n=== Income Statement ===\n"
+        puts pl_output
+        
+        section_output << "=== Income Statement ===\n"
+        section_output << pl_output
+        section_output << "\n"
+        
+        section_output
+      end
+      
+      def generate_beancount_monthly_report_section(file_path, date_filter)
+        section_output = ""
+        
+        monthly_cmd = "bean-report #{file_path}#{date_filter} monthly"
+        monthly_output = `#{monthly_cmd}`
+        
+        puts "\n=== Monthly Activity ===\n"
+        puts monthly_output
+        
+        section_output << "=== Monthly Activity ===\n"
+        section_output << monthly_output
+        section_output << "\n"
+        
+        section_output
+      end
+      
+      def generate_beancount_register_report_section(file_path, date_filter, category)
+        section_output = ""
+        
+        register_filter = category ? " --account #{category}" : ""
+        register_cmd = "bean-report #{file_path}#{date_filter}#{register_filter} register"
+        register_output = `#{register_cmd}`
+        
+        puts "\n=== Transaction Register ===\n"
+        puts register_output
+        
+        section_output << "=== Transaction Register ===\n"
+        section_output << register_output
+        section_output << "\n"
+        
+        section_output
+      end
+      
+      def export_transactions_to_ledger(transactions, file_path)
+        # Create a temporary file for ledger export
+        File.open(file_path, 'w') do |f|
+          write_ledger_header(f)
+          write_ledger_account_declarations(f, transactions)
+          write_ledger_transactions(f, transactions)
+        end
+      end
+      
+      def write_ledger_header(file)
+        file.puts "; Mercury Bank Transactions Export"
+        file.puts "; Generated on #{Time.now.strftime('%Y-%m-%d %H:%M:%S')}"
+        file.puts
+      end
+      
+      def write_ledger_account_declarations(file, transactions)
+        # Get unique account names from transactions
+        account_names = transactions.map { |t| t["accountName"] || "Unknown Account" }.uniq
+        
+        # Define accounts dynamically based on transaction data
+        account_names.each do |account_name|
+          file.puts "account Assets:#{account_name}"
+        end
+        file.puts "account Expenses:Unknown"
+        file.puts "account Income:Unknown"
+        file.puts
+      end
+      
+      def write_ledger_transactions(file, transactions)
+        # Filter out failed transactions
+        valid_transactions = transactions.reject { |t| t["status"] == "failed" }
+        
+        # Sort transactions by date
+        valid_transactions.sort_by! { |t| t["postedAt"] || t["createdAt"] }.each do |t|
+          write_ledger_transaction(file, t)
+        end
+      end
+      
+      def write_ledger_transaction(file, transaction)
+        date = transaction["postedAt"] ? 
+               Time.parse(transaction["postedAt"]).strftime("%Y/%m/%d") : 
+               Time.parse(transaction["createdAt"]).strftime("%Y/%m/%d")
+        description = transaction["bankDescription"] || transaction["externalMemo"] || "Unknown transaction"
+        amount = transaction["amount"]
+        account_name = transaction["accountName"] || "Unknown Account"
+        
+        file.puts "#{date} #{description}"
+        file.puts "    ; Transaction ID: #{transaction["id"]}"
+        file.puts "    ; Status: #{transaction["status"]}"
+        file.puts "    ; Reconciled: No"
+        
+        write_ledger_postings(file, amount, account_name)
+        file.puts
+      end
+      
+      def write_ledger_postings(file, amount, account_name)
+        if amount > 0
+          file.puts "    Income:Unknown          $-#{format("%.2f", amount)}"
+          file.puts "    Assets:#{account_name}    $#{format("%.2f", amount)}"
+        else
+          file.puts "    Expenses:Unknown          $#{format("%.2f", amount.abs)}"
+          file.puts "    Assets:#{account_name}    $-#{format("%.2f", amount.abs)}"
+        end
+      end
+      
+      def export_transactions_to_beancount(transactions, file_path)
+        # Create a temporary file for beancount export
+        File.open(file_path, 'w') do |f|
+          write_beancount_header(f)
+          write_beancount_account_declarations(f, transactions)
+          write_beancount_transactions(f, transactions)
+        end
+      end
+      
+      def write_beancount_header(file)
+        file.puts "; Mercury Bank Transactions Export"
+        file.puts "; Generated on #{Time.now.strftime('%Y-%m-%d %H:%M:%S')}"
+        file.puts
+        file.puts "option \"title\" \"Mercury Bank Transactions\""
+        file.puts "option \"operating_currency\" \"USD\""
+        file.puts
+      end
+      
+      def write_beancount_account_declarations(file, transactions)
+        # Get unique account names from transactions
+        account_names = transactions.map { |t| t["accountName"] || "Unknown Account" }.uniq
+        
+        # Define accounts dynamically based on transaction data
+        account_names.each do |account_name|
+          # Sanitize account name for beancount format
+          safe_account_name = account_name.gsub(/[^a-zA-Z0-9]/, '-')
+          file.puts "1970-01-01 open Assets:#{safe_account_name}"
+        end
+        file.puts "1970-01-01 open Expenses:Unknown"
+        file.puts "1970-01-01 open Income:Unknown"
+        file.puts
+      end
+      
+      def write_beancount_transactions(file, transactions)
+        # Filter out failed transactions
+        valid_transactions = transactions.reject { |t| t["status"] == "failed" }
+        
+        # Sort transactions by date
+        valid_transactions.sort_by! { |t| t["postedAt"] || t["createdAt"] }.each do |t|
+          write_beancount_transaction(file, t)
+        end
+      end
+      
+      def write_beancount_transaction(file, transaction)
+        date = transaction["postedAt"] ? 
+               Time.parse(transaction["postedAt"]).strftime("%Y-%m-%d") : 
+               Time.parse(transaction["createdAt"]).strftime("%Y-%m-%d")
+        description = transaction["bankDescription"] || transaction["externalMemo"] || "Unknown transaction"
+        amount = transaction["amount"]
+        account_name = transaction["accountName"] || "Unknown Account"
+        # Sanitize account name for beancount format
+        safe_account_name = account_name.gsub(/[^a-zA-Z0-9]/, '-')
+        
+        file.puts "#{date} * \"#{description}\""
+        file.puts "  ; Transaction ID: #{transaction["id"]}"
+        file.puts "  ; Status: #{transaction["status"]}"
+        file.puts "  ; Reconciled: No"
+        
+        write_beancount_postings(file, amount, safe_account_name)
+        file.puts
+      end
+      
+      def write_beancount_postings(file, amount, account_name)
+        if amount > 0
+          file.puts "  Income:Unknown          -#{format("%.2f", amount)} USD"
+          file.puts "  Assets:#{account_name}    #{format("%.2f", amount)} USD"
+        else
+          file.puts "  Expenses:Unknown          #{format("%.2f", amount.abs)} USD"
+          file.puts "  Assets:#{account_name}    -#{format("%.2f", amount.abs)} USD"
+        end
+      end
+    end
+  end
+end