vault_client 0.0.3 → 0.1.0

data/lib/helpers/http_helpers.rb

@@ -0,0 +1,45 @@
+module VC
+  module HttpHelpers
+    extend self
+
+    attr_accessor :url
+    def url
+      @url || ENV["VAULT_URL"]
+    end
+
+    def headers
+      {:content_type => :json, :accept => :json}
+    end
+
+    def enc_json(hash)
+      Yajl::Encoder.encode(hash)
+    end
+
+    def dec_json(json)
+      Yajl::Parser.parse(json)
+    end
+
+    def handle_req(&blk)
+      begin
+        resp = yield
+        dec_json(resp.body)
+      rescue RestClient::Exception => e
+        body = dec_json(e.http_body)
+        case e.http_code
+        when 404
+        when 401
+          raise(AuthorizationError, "Response: #{body.inspect}")
+        when 403
+        when 409
+        when 500
+          raise(UnexpectedError, "Response: #{body.inspect}")
+        when 503
+          raise(ServiceDownError, "#{VC.url} is down for maintenance.")
+        else
+          raise(UnexpectedError, "Response: #{body.inspect}")
+        end
+      end
+    end
+
+  end
+end

data/lib/models/account.rb

@@ -0,0 +1,13 @@
+module VC
+  module Account
+    extend self
+
+    def create
+      VC.handle_req {RestClient.post(accounts_url, {}, VC.headers)}
+    end
+
+    def accounts_url
+      [VC.url, "/accounts"].join
+    end
+  end
+end

data/lib/models/acct_own.rb

@@ -0,0 +1,31 @@
+module VC
+  module AcctOwn
+    extend self
+
+    def act(args)
+      pmid, eid = args.delete(:payment_method_id), args.delete(:entity_id)
+      VC.handle_req {RestClient.post(acct_own_url(pmid, eid), args, VC.headers)}
+    end
+
+    def xfr(args)
+      prev_pmid, prev_eid = args.delete(:prev_payment_method_id), args.delete(:prev_entity_id)
+      VC.handle_req {RestClient.put(acct_own_url(prev_pmid, prev_eid), args, VC.headers)}
+    end
+
+    def deact(args)
+      pmid = args.delete(:payment_method_id)
+      eid = args.delete(:entity_id)
+      account_id = args[:account_id]
+      time = args[:time]
+      VC.handle_req {RestClient.delete([acct_own_url(pmid, eid), CGI.encode("?account_id=#{account_id}&time=#{time}")].join)}
+    end
+
+    def query(args)
+      VC.handle_req {RestClient.get([VC.url, "/accounts/#{args[:account_id]}/resource_ownerships"].join)}
+    end
+
+    def acct_own_url(payment_method_id, entity_id)
+      [VC.url, "/payment_methods/#{payment_method_id}/account_ownerships/#{entity_id}"].join
+    end
+  end
+end

data/lib/models/b_event.rb

@@ -0,0 +1,27 @@
+module VC
+  module BEvent
+    extend self
+
+    OPEN = "open"
+    CLOSE = "close"
+
+    def open(args)
+      hid = args.delete(:hid)
+      entity_id = args.delete(:entity_id)
+      args[:state] = OPEN
+      VC.handle_req {RestClient.put(events_url(hid, entity_id), args, VC.headers)}
+    end
+
+    def close(args)
+      hid = args.delete(:hid)
+      entity_id = args.delete(:entity_id)
+      args[:state] = CLOSE
+      VC.handle_req {RestClient.put(events_url(hid, entity_id), args, VC.headers)}
+    end
+
+    def events_url(hid, entity_id)
+      [VC.url, "/resources/#{hid}/billable_events/#{entity_id}"].join
+    end
+
+  end
+end

data/lib/models/billable_unit.rb

@@ -1,11 +1,10 @@
-module VaultClient
+module VC
   class BillableUnit
     attr_accessor(
       :hid,
       :from,
       :to,
       :qty,
-      :total,
       :rate,
       :rate_period,
       :product_group,
@@ -28,5 +27,9 @@ module VaultClient
       Time.now.to_i
     end
 
+    def total
+      @rate * @qty
+    end
+
   end
 end

data/lib/models/line_item.rb

@@ -1,4 +1,4 @@
-module VaultClient
+module VC
   class LineItem
 
     attr_reader :unit_groups
@@ -12,10 +12,8 @@ module VaultClient
     end
 
     def total
-      @unit_groups.inject(0) {|memo, unit_group| unit_group.total}
+      @unit_groups.map(&:total).reduce(:+)
     end
 
   end
 end
-
-

data/lib/models/report.rb

@@ -0,0 +1,51 @@
+module VC
+  class Report
+    attr_accessor(
+      :from,
+      :to,
+      :billable_units
+    )
+
+    def fetch
+      VC.handle_req do
+        RestClient.get([VC.url, resource].join, {:params => {:from => from.utc.to_s, :to => to.utc.to_s}})
+      end
+    end
+
+    def billable_units
+      @billable_units ||= report["billable_units"].map {|buh| BillableUnit.new(buh)}
+    end
+
+    def total
+      report["total"]
+    end
+
+    def report
+      @report ||= fetch
+    end
+  end
+
+  class UsageReport < Report
+    attr_accessor :account_id
+
+    def initialize(account_id, from, to)
+      @account_id, @from, @to = account_id, from, to
+    end
+
+    def resource
+      "/accounts/#{@account_id}/usage_reports"
+    end
+  end
+
+  class Invoice < Report
+    attr_accessor :payment_method_id
+
+    def initialize(payment_method_id, from, to)
+      @payment_method_id, @from, @to = payment_method_id, from, to
+    end
+
+    def resource
+      "/payment_methods/#{payment_method_id}/invoices"
+    end
+  end
+end

data/lib/models/res_own.rb

@@ -0,0 +1,31 @@
+module VC
+  module ResOwn
+    extend self
+
+    def act(args)
+      aid, eid = args.delete(:account_id), args.delete(:entity_id)
+      VC.handle_req {RestClient.post(res_own_url(aid, eid), args, VC.headers)}
+    end
+
+    def xfr(args)
+      prev_aid, prev_eid = args.delete(:prev_account_id), args.delete(:prev_entity_id)
+      VC.handle_req {RestClient.put(res_own_url(prev_aid, prev_eid), args, VC.headers)}
+    end
+
+    def deact(args)
+      aid = args.delete(:account_id)
+      eid = args.delete(:entity_id)
+      hid = args[:hid]
+      time = args[:time]
+      VC.handle_req {RestClient.delete([res_own_url(aid, eid), CGI.encode("?hid=#{hid}&time=#{time}")].join)}
+    end
+
+    def query(args)
+      VC.handle_req {RestClient.get([VC.url, "/accounts/#{args[:account_id]}/resource_ownerships"].join)}
+    end
+
+    def res_own_url(account_id, entity_id)
+      [VC.url, "/accounts/#{account_id}/resource_ownerships/#{entity_id}"].join
+    end
+  end
+end

data/lib/models/unit_group.rb

@@ -1,6 +1,6 @@
-module VaultClient
+module VC
   class UnitGroup
-    InvalidUnitGroup = Class.new(Exception)
+    InvalidUnitGroup = Class.new(VCException)
 
     attr_reader :units
 
@@ -15,45 +15,44 @@ module VaultClient
       end
     end
 
-    def name
-      @name ||= units.first.product_name
+    def total
+      rate * qty
     end
 
-    def hid
-      units.first.hid
+    def qty
+      @units.map(&:qty).reduce(:+)
     end
 
-    def total
-      units.inject(0) {|memo, unit| unit.rate * unit.qty}
+    def rate
+      sample_unit.rate
     end
 
-    def dyno_type?
-      sample_unit.product_group == "dyno"
+    def rate_period
+      sample_unit.rate_period
     end
 
-    def add_on_type?
-      sample_unit.product_group == "addon"
+    def name
+      sample_unit.product_name
     end
 
-    def run_type?
-      return false
-      sample_unit.product_name == "run"
+    def hid
+      sample_unit.hid
     end
 
-    def description
+    def product_name
       sample_unit.product_name
     end
 
-    def qty
-      @units.inject(0) {|memo, unit| unit.qty}
+    def product_group
+      sample_unit.product_group
     end
 
-    def rate
-      sample_unit.rate
+    def description
+      sample_unit.product_name
     end
 
     def sample_unit
-      @sample_unit ||= @units.first
+      @sample_unit ||= @units.sample
     end
 
   end

data/lib/presenters/base_presenter.rb

@@ -1,9 +1,12 @@
-module VaultClient
+module VC
   class BasePresenter
 
     def money(cents)
-      return if cents.nil?
-      pennies_to_dollar(cents.to_i)
+      add_commas(pennies_to_dollar(cents.to_i))
+    end
+
+    def add_commas(str)
+      str.reverse.scan(/(?:\d*\.)?\d{1,3}-?/).join(',').reverse
     end
 
     def pennies_to_dollar(qty)

data/lib/presenters/line_item_presenter.rb

@@ -1,67 +1,49 @@
-module VaultClient
+module VC
   class LineItemPresenter < BasePresenter
 
     def initialize(line_item)
       @line_item = line_item
+      @unit_groups = @line_item.unit_groups
+      @unit_group_presenters = @unit_groups.map {|ug| UnitGroupPresenter.new(ug)}
     end
 
-    def add_on_unit_group_total
-      result = add_on_unit_group_presenters.map do |ugp|
-        ugp.unit_group
-      end.inject(0) do |memo, ug|
-        ug.total
-      end
-      money(result)
-    end
-
-    def dyno_unit_group_total
-      result = dyno_unit_group_presenters.map do |ugp|
-        ugp.unit_group
-      end.inject(0) do |memo, ug|
-        ug.total
-      end
-      money(result)
+    def total
+      money(@line_item.total)
     end
 
-    def dyno_unit_group_rate
-      0.05
+    def name
+      @line_item.app_name
     end
 
-    def dyno_unit_group_qty
-      dyno_unit_group_presenters.map do |upg|
-        upg.unit_group
-      end.inject(0) do |memo, ug|
-        ug.qty
-      end
+    def unit_group_rate(product_group)
+      unit_group_presenters(product_group).sample.rate
     end
 
-    def dyno_unit_group_presenters
-      unit_group_presenters.select {|ugp| ugp.unit_group.dyno_type?}
+    def unit_group_total(product_group)
+      money(
+        unit_group_presenters(product_group).
+        map(&:unit_group).
+        map(&:total).
+        reduce(:+)
+      )
     end
 
-    def add_on_unit_group_presenters
-      unit_group_presenters.select {|ugp| ugp.unit_group.add_on_type?}
+    def unit_group_qty(product_group)
+      trunc_hours(
+        unit_group_presenters(product_group).
+        map(&:unit_group).
+        map(&:qty).
+        reduce(:+)
+      )
     end
 
-    def unit_group_presenters
-      if defined?(@unit_group_presenters)
-        @unit_group_presenters
+    def unit_group_presenters(product_group=nil)
+      if product_group
+        @unit_group_presenters.select {|ugp| ugp.product_group == product_group}
       else
-        @unit_group_presenters = unit_groups.map {|ug| UnitGroupPresenter.new(ug)}
+        @unit_group_presenters
       end
     end
 
-    def total
-      money(@line_item.total)
-    end
-
-    def name
-      @line_item.app_name
-    end
-
-    def unit_groups
-      @unit_groups ||= @line_item.unit_groups
-    end
-
   end
 end

data/lib/presenters/report_presenter.rb

@@ -1,13 +1,19 @@
-module VaultClient
+module VC
   class ReportPresenter < BasePresenter
 
-    def initialize(units)
-      @units = units
+    def initialize(report, line_item_builder=LineItemBuilder)
+      @report = report
+      @units = @report.billable_units
+      @builder = line_item_builder
+    end
+
+    def total
+      money(@report.total)
     end
 
     def line_item_presenters
       @line_item_presenters ||= begin
-        LineItemBuilder.build(@units).map do |li|
+        @builder.build(@units).map do |li|
           LineItemPresenter.new(li)
         end
       end