Add wallet / porfolio report

src/hodlerctl/main.cr

@@ -4,6 +4,7 @@ require "yaml"
 require "colorize"
 require "xdg_basedir"
 require "completion"
+require "log"
 
 require "../lib/version"
 require "../lib/types"

src/lib/actions/portfolio-get.cr

@@ -3,14 +3,10 @@ require "../actions"
 
 module Hodler
   class GetPortfolioAction < Action
-    def self.match(action)
-      # return (action == Action::Type::Report)
-    end
-
     # Display Wallet, Symbol, Held, Initial Value, Current Value, Evolution (%)
     def perform
       portfolio = PortfolioFactory.build(@global_options, @config)
-      puts portfolio.to_report
+      puts portfolio.symbol_report
     end
   end
 end

src/lib/actions/wallet-get.cr

@@ -2,14 +2,10 @@
 require "../actions"
 
 module Hodler
-  class GetPortfolioAction < Action
-    def self.match(action)
-      # return (action == Action::Type::Report)
-    end
-
+  class GetWalletAction < Action
     def perform
       portfolio = PortfolioFactory.build(@global_options, @config)
-      puts portfolio.to_report
+      puts portfolio.wallet_report
     end
   end
 end

src/lib/models/portfolio.cr

@@ -4,17 +4,24 @@ require "../models"
 module Hodler
   class PortfolioModel < Model
 
-    property wallets : Array(WalletModel)
-
+    getter wallets : Array(WalletModel)
 
     def initialize()
       @wallets = [] of WalletModel
+      @trades = [] of TradeModel
+      @deposits = [] of DepositModel
+    end
+
+    def import_deposits(deposits : Array(DepositModel))
+      @deposits = deposits
     end
 
     def import_wallets(wallets : Array(WalletModel))
-      wallets.each do |wallet|
-        @wallets << wallet
-      end
+      @wallets = wallets
+    end
+
+    def import_trades(trades : Array(TradeModel))
+      @trades = trades
     end
 
     def add_wallet
@@ -23,9 +30,49 @@ module Hodler
     def remove_wallet
     end
 
-    def to_report
-      data = @wallets.map{|row| [row.name, row.type.to_s, row.refs.map{|ref| ref.sym}.join(", ")] }
-      # pp data
+    def symbol_report
+      wallets = {} of Tuple(String,String) => Float32
+      @deposits.each do |deposit_wrapper|
+        deposit = deposit_wrapper.deposit
+        id = {deposit.wallet, deposit.sym}
+        wallets[id] ||= 0
+        wallets[id] += deposit.amount
+      end
+
+      @trades.each do |trade_wrapper|
+        trade = trade_wrapper.transaction
+        from_id = {trade.from.wallet, trade.from.sym}
+        to_id = {trade.to.wallet, trade.to.sym}
+        fee_id = {trade.from.wallet, trade.fee.sym}
+
+        wallets[from_id] ||= 0
+        wallets[fee_id] ||= 0
+        wallets[to_id] ||= 0
+        wallets[from_id] -= trade.from.amount
+        wallets[fee_id] -= trade.fee.amount
+        wallets[to_id] += trade.to.amount
+      end
+      data = [] of Array(String)
+      wallets
+        .each{|key, val| data << [ key[0], key[1].to_s, "%f" % val] }
+      data.sort! do |x,y| 
+        resA = x[0] <=> y[0]
+        resB = x[1] <=> y[1] 
+        ((resA == 0) ? resB : resA)
+      end
+
+      table = Tablo::Table.new(data, connectors: Tablo::CONNECTORS_SINGLE_ROUNDED) do |t|
+        t.add_column("Wallet") { |row| row[0] }
+        t.add_column("Symbol") { |row| row[1] }
+        t.add_column("Held", align_header: Tablo::Justify::Right, align_body: Tablo::Justify::Right) { |row| row[2] }
+        t.add_column("Invested") { |row| "FIXME" }
+        t.add_column("Cur. Price") { |row| "FIXME" }
+        t.add_column("Cur. Value") { |row| "FIXME" }
+      end
+    end
+
+    def wallet_report
+      data = @wallets.map{|row| [row.name, row.type.to_s, row.refs.map{|ref| ref.sym}.join(",")] }
       table = Tablo::Table.new(data, connectors: Tablo::CONNECTORS_SINGLE_ROUNDED) do |t|
         t.add_column("Name") { |row| row[0] }
         t.add_column("Type") { |row| row[1] }
@@ -39,6 +86,8 @@ module Hodler
       portfolio = PortfolioModel.new
 
       portfolio.import_wallets(config.wallets)
+      portfolio.import_trades(config.trades)
+      portfolio.import_deposits(config.deposits)
 
       return portfolio
     end

src/lib/types.cr

@@ -4,6 +4,7 @@ require "./actions/*"
 module Hodler
   alias AmountT = Float64 | Int64
 
+  TABLO_CONNECTORS_EMPTY="               "
   alias GlobalOptions = {
     action: Action.class,
     verbose_enable: Bool,