mirror of
https://github.com/gwenhael-le-moine/credger.git
synced 2024-12-26 09:58:36 +01:00
73 lines
2.3 KiB
Crystal
73 lines
2.3 KiB
Crystal
# encoding: utf-8
|
|
|
|
require "csv"
|
|
|
|
# Crystal wrapper module for calling ledger
|
|
class Ledger
|
|
def initialize( binary : String = "ledger", ledger_file : String = ENV[ "LEDGER_FILE" ] )
|
|
@binary = binary
|
|
@file = ledger_file
|
|
end
|
|
|
|
def run( options : String, command : String = "", command_parameters : String = "" ) : String
|
|
STDERR.puts "#{@binary} -f #{@file} #{options} #{command} #{command_parameters}"
|
|
`#{@binary} -f #{@file} #{options} #{command} #{command_parameters}`
|
|
end
|
|
|
|
def version : String
|
|
run "--version"
|
|
end
|
|
|
|
def accounts( depth : Number = 9999 ) : Array( Array( String ) )
|
|
accounts = run( "", "accounts" )
|
|
.split( "\n" )
|
|
.map do |a|
|
|
a.split( ":" )
|
|
.each_slice( depth )
|
|
.to_a.first
|
|
end.uniq
|
|
|
|
accounts.map{|a| a.size}.max.times do |i|
|
|
accounts += accounts.map { |acc| acc.first( i ) }
|
|
end
|
|
|
|
accounts
|
|
.uniq
|
|
.sort
|
|
.reject { |a| a.empty? || a.first.empty? }
|
|
.sort_by { |a| a.size }
|
|
end
|
|
|
|
def balance( cleared : Bool = false, depth : Int32 = 9999, period : String = nil, categories : String = "" ) : Array( NamedTuple( account: String, amount: Float64 ) )
|
|
period = period.nil? ? "" : "-p '#{period}'"
|
|
depth = depth.nil? ? "" : "--depth #{depth}"
|
|
operation = cleared ? "cleared" : "balance"
|
|
|
|
run( "--flat --no-total --exchange '#{CURRENCY}' #{period} #{depth}", operation, categories )
|
|
.split( "\n" )
|
|
.reject {|line| line.empty?}
|
|
.map do |line|
|
|
line_array = line.split( "#{CURRENCY}" )
|
|
|
|
{ account: line_array[ 1 ].strip,
|
|
amount: line_array[ 0 ].tr( SEPARATOR, "." ).to_f }
|
|
end
|
|
end
|
|
|
|
def graph_values( period : String = "", categories : Array(String) = ["Expenses"] ) : Hash( String, Array( NamedTuple( date: String, amount: String, currency: String ) ) )
|
|
period = period == "" ? "" : "-p '#{period}'"
|
|
|
|
result = {} of String => Array(NamedTuple(date: String, amount: String, currency: String))
|
|
categories.map do |category|
|
|
result[category] = CSV
|
|
.parse( run( "-MAn --exchange '#{CURRENCY}' #{period}", "csv --no-revalued", category ) )
|
|
.map do |row|
|
|
{ date: row[ 0 ],
|
|
amount: row[ 5 ],
|
|
currency: row[ 4 ] }
|
|
end
|
|
end
|
|
|
|
result
|
|
end
|
|
end
|