auto refill (#1195)

auto refill functionality

Author: gbarc80

.travis.yml

@@ -7,11 +7,6 @@ rvm: 2.6.5
 
 script: "bundle exec rake knapsack:rspec"
 
-addons:
-  - snaps:
-    - name: docker
-      channel: latest/beta
-
 env:
   global:
     - PATH=/snap/bin:$PATH

lib/travis/api/v3/billing_client.rb

@@ -205,6 +205,16 @@ def update_organization_billing_permission(organization_id, billing_admin_only)
       handle_subscription_response(response)
     end
 
+    def create_auto_refill(plan_id, is_enabled)
+      response = connection.post('/auto_refill', {plan: plan_id, enabled: is_enabled})
+      handle_errors_and_respond(response)
+    end
+
+    def get_auto_refill(plan_id)
+      response = connection.get("/auto_refill?plan_id=#{plan_id}")
+      handle_errors_and_respond(response) { |r| Travis::API::V3::Models::AutoRefill.new(r) }
+    end
+
     private
 
     def handle_subscription_response(response)

lib/travis/api/v3/models/auto_refill.rb

@@ -0,0 +1,12 @@
+module Travis::API::V3
+  class Models::AutoRefill
+
+    attr_reader :enabled, :threshold, :amount
+    def initialize(attributes = {})
+      @enabled = attributes.key?('enabled') ? attributes.fetch('enabled') : true
+      @threshold = attributes.key?('refill_threshold') ? attributes.fetch('refill_threshold') : 25000
+      @amount = attributes.key?('refill_amount') ? attributes.fetch('refill_amount'): 10000
+    end
+  end
+
+end

lib/travis/api/v3/models/v2_plan_config.rb

@@ -1,7 +1,7 @@
 module Travis::API::V3
   class Models::V2PlanConfig
     attr_reader :id, :name, :private_repos, :starting_price, :starting_users, :plan_type,
-                :private_credits, :public_credits, :addon_configs, :concurrency_limit, :available_standalone_addons
+                :private_credits, :public_credits, :addon_configs, :concurrency_limit, :available_standalone_addons, :auto_refill_enabled
 
     def initialize(attrs)
       @id = attrs.fetch('id')

lib/travis/api/v3/models/v2_subscription.rb

@@ -3,7 +3,7 @@ class Models::V2Subscription
     include Models::Owner
 
     attr_reader :id, :plan, :permissions, :source, :billing_info, :credit_card_info, :owner, :status, :valid_to, :canceled_at,
-                :client_secret, :payment_intent, :addons, :available_standalone_addons, :created_at
+                :client_secret, :payment_intent, :addons, :auto_refill, :available_standalone_addons, :created_at
 
     def initialize(attributes = {})
       @id = attributes.fetch('id')
@@ -15,7 +15,17 @@ def initialize(attributes = {})
       @payment_intent = attributes['payment_intent'] && Models::PaymentIntent.new(attributes['payment_intent'])
       @owner = fetch_owner(attributes.fetch('owner'))
       @client_secret = attributes.fetch('client_secret')
-      @addons = attributes['addons'].select { |addon| addon['current_usage']['status'] != 'expired' }.map { |addon| Models::V2Addon.new(addon) }
+      @addons = attributes['addons'].select { |addon| addon['current_usage']['status'] != 'expired' if addon['current_usage'] }.map { |addon| Models::V2Addon.new(addon) }
+      refill = attributes['addons'].detect { |addon| addon['addon_config_id'] === 'auto_refill' } || {"enabled" => false};
+      default_refill = @plan.respond_to?('available_standalone_addons') ?
+        @plan.available_standalone_addons.detect { |addon| addon['id'] === 'auto_refill' } : nil
+      
+      refill['enabled'] = attributes['auto_refill_enabled']
+      if default_refill
+        refill['refill_threshold'] = default_refill['refill_threshold'] unless refill.key?('refill_threshold')
+        refill['refill_amount'] = default_refill['refill_amount'] unless refill.key?('refill_amount')
+      end
+      @auto_refill = Models::AutoRefill.new(refill)
       @created_at = attributes.fetch('created_at')
       @status = attributes.fetch('status')
       @valid_to = attributes.fetch('valid_to')

lib/travis/api/v3/queries/v2_subscription.rb

@@ -1,5 +1,7 @@
 module Travis::API::V3
   class Queries::V2Subscription < Query
+    params :enabled
+
     def update_address(user_id)
       address_data = params.dup.tap { |h| h.delete('subscription.id') }
       client = BillingClient.new(user_id)
@@ -37,5 +39,15 @@ def pay(user_id)
       client = BillingClient.new(user_id)
       client.pay_v2(params['subscription.id'])
     end
+
+    def get_auto_refill(user_id, plan_id)
+      client = BillingClient.new(user_id)
+      client.get_auto_refill(plan_id)
+    end
+
+    def toggle_auto_refill(user_id, plan_id)
+      client = BillingClient.new(user_id)
+      client.create_auto_refill(plan_id, enabled)
+    end
   end
 end

lib/travis/api/v3/renderer/auto_refill.rb

@@ -0,0 +1,6 @@
+module Travis::API::V3
+  class Renderer::AutoRefill < ModelRenderer
+    representation(:standard, :enabled, :threshold, :amount)
+    representation(:minimal, :enabled, :threshold, :amount)
+  end
+end

lib/travis/api/v3/renderer/v2_subscription.rb

@@ -1,6 +1,6 @@
 module Travis::API::V3
   class Renderer::V2Subscription < ModelRenderer
-    representation(:standard, :id, :plan, :addons, :status, :valid_to, :canceled_at, :source, :owner, :client_secret, :billing_info, :credit_card_info, :payment_intent, :created_at)
+    representation(:standard, :id, :plan, :addons, :auto_refill, :status, :valid_to, :canceled_at, :source, :owner, :client_secret, :billing_info, :credit_card_info, :payment_intent, :created_at)
 
     def billing_info
       Renderer.render_model(model.billing_info, mode: :standard) unless model.billing_info.nil?

lib/travis/api/v3/routes.rb

@@ -367,6 +367,8 @@ module Routes
       post :buy_addon, '/addon/{addon.id}'
       get :user_usages, '/user_usages'
       get :invoices, '/invoices'
+      get :auto_refill, '/auto_refill'
+      patch :toggle_auto_refill,  '/auto_refill'
     end
 
     hidden_resource :trials do

lib/travis/api/v3/services/v2_subscription/auto_refill.rb

@@ -0,0 +1,10 @@
+module Travis::API::V3
+  class Services::V2Subscription::AutoRefill < Service
+    result_type :auto_refill
+
+    def run!
+      raise LoginRequired unless access_control.full_access_or_logged_in?
+      result query.get_auto_refill(access_control.user.id, params['subscription.id']), status: 200
+    end
+  end
+end