ofx_ruby 0.0.13 → 0.0.14

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c64f0acb70d7928cdf68a2a56f927f16930797e3
-  data.tar.gz: 9589928dfcaf5df283504eabd5d5a2b6112030ee
+  metadata.gz: b41ef0ed969fef00bb94b4685bc08edf1c1f301b
+  data.tar.gz: bfbe06e5523d12018f521bd39e9053a18dd35074
 SHA512:
-  metadata.gz: 2056a66be490fadf0072ec12117c563fabf9c13984ae5d46b580ffe5c292870d7697b640c0a3ff4bb54068a1446fb7731d8532467f5ee202c2dfdf236ea71004
-  data.tar.gz: 7d0108b6805f314665e78bddc5284fc759ce82f291088f3bd8ba787002665792d7b1b01031521bcc4b17f847e2ec4020535b1a3f2787e958c0afed5dd34ad231
+  metadata.gz: d2eef49cd742e94af56db6e2e4e1437609fed4d8933cad65e0e6f7ab0732a584a03e8de1dff1669589c8da293c5b0c6e32cdf593d92a529a42f580d4a8c97d3f
+  data.tar.gz: f17dc8ad3b4f30cb9ae0f29e2f324abffe0647b4b73bc3c3e18b2c652e66b3124f38859a1fd6db20167fe99caf2e0b597a7cc5ba51203d3d3c59e3a52a2efae2

data/lib/ofx/handler.rb

@@ -2,7 +2,8 @@
 module OFX
   class Handler < Ox::Sax
     TRANSACTION_ATTRS = [:TRNTYPE, :DTPOSTED, :TRNAMT, :FITID, :NAME]
-    ATTRS_MAP = { TRNTYPE: :as_s, DTPOSTED: :as_time,
+    BALANCE_ATTRS = [:BALAMT]
+    ATTRS_MAP = { TRNTYPE: :as_s, DTPOSTED: :as_time, BALAMT: :as_f,
                   TRNAMT: :as_f, FITID: :as_s, NAME: :as_s }
 
     def initialize(parser)
@@ -10,25 +11,40 @@ module OFX
     end                  
 
     def start_element(name)
-      @transaction = {} if name == :STMTTRN
-      @current_node = name
+      case name
+      when :STMTTRN then @transaction = {}
+      when :LEDGERBAL || :AVAILBAL then @balance = {}
+      end
+      @current = name
     end
 
     def value(value)
-      return unless TRANSACTION_ATTRS.include?(@current_node)
-      @transaction[@current_node] = value.send(ATTRS_MAP[@current_node])
+      case
+      when TRANSACTION_ATTRS.include?(@current)
+        @transaction[@current] = value.send(ATTRS_MAP[@current])
+      when BALANCE_ATTRS.include?(@current)
+        @balance[@current] = value.send(ATTRS_MAP[@current])
+      end
     end
 
     def end_element(name)
-      return unless name == :STMTTRN
-      @parser.output[:transactions] = [] if @parser.output[:transactions].nil?
-      @parser.output[:transactions].push(
-        type: @transaction[:TRNTYPE],
-        posted: @transaction[:DTPOSTED],
-        amount: @transaction[:TRNAMT],
-        fitid: @transaction[:FITID],
-        name: @transaction[:NAME]
-      )
+      case name
+      when :STMTTRN
+        @parser.output[:transactions] = [] unless @parser.output[:transactions]
+        @parser.output[:transactions].push(transaction(@transaction))
+      when :LEDGERBAL
+        @parser.output[:balance] = @balance[:BALAMT]
+      when :AVAILBAL
+        @parser.output[:pending] = @balance[:BALAMT] - @parser.output[:balance]
+      end
+    end
+    
+    protected
+    
+    def transaction(hash)
+      { type: @transaction[:TRNTYPE], posted: @transaction[:DTPOSTED],
+        amount: @transaction[:TRNAMT], fitid: @transaction[:FITID],
+        name: @transaction[:NAME] }
     end
   end
 end

data/lib/ofx/parser.rb

@@ -18,6 +18,14 @@ module OFX
       @output[:transactions]
     end
 
+    def balance
+      @output[:balance].round(2)
+    end
+
+    def pending
+      @output[:pending].round(2)
+    end
+
     protected
 
     def dump

data/lib/ofx/version.rb

@@ -1,3 +1,3 @@
 module OFX
-  VERSION = "0.0.13"
+  VERSION = "0.0.14"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ofx_ruby
 version: !ruby/object:Gem::Version
-  version: 0.0.13
+  version: 0.0.14
 platform: ruby
 authors:
 - rob
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-10-29 00:00:00.000000000 Z
+date: 2016-10-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler