# encoding: utf-8

require 'csv'

# Ruby wrapper module for calling ledger
module Ledger
  module_function

  @binary = 'ledger'
  @file = ENV[ 'LEDGER_FILE' ]

  def run( options, command = '', command_parameters = '' )
    `#{@binary} -f #{@file} #{options} #{command} #{command_parameters}`
  end

  def version
    run '--version'
  end

  def accounts( depth = 9999 )
    run( '', 'accounts' )
      .split( "\n" )
      .map do |a|
      a.split( ':' )
       .each_slice( depth )
       .to_a.first
    end.uniq
  end

  def dates_salaries( category = 'salaire' )
    CSV.parse( run( '', 'csv', category ) )
       .map do
      |row|
      Date.parse row[ 0 ]
    end
       .uniq
  end

  def register( period = nil, categories = '' )
    period = period.nil? ? '' : "-p '#{period}'"

    CSV.parse( run( "--exchange '#{CURRENCY}' #{period}", 'csv', categories ) )
       .map do
      |row|
      { date: row[ 0 ],
        payee: row[ 2 ],
        account: row[ 3 ],
        amount: row[ 5 ],
        currency: row[ 4 ] }
    end
  end

  def cleared
    run( "--flat --no-total --exchange '#{CURRENCY}'", 'cleared', 'Assets Equity' )
      .split( "\n" )
      .map do |row|
      fields = row.match( /\s*(\S+ €)\s*(\S+ €)\s*(\S+)\s*(\S+)/ )
      { account: fields[ 4 ],
        amount: { cleared: fields[ 2 ],
                  all: fields[ 1 ] } } unless fields.nil?
    end
  end
end