Api WIP for v2 plan change

Author: AndriiMysko

lib/travis/api/v3/billing_client.rb

@@ -104,6 +104,18 @@ def update_v2_subscription(subscription_id, plan_data)
       handle_v2_subscription_response(response)
     end
 
+    def v2_plans_for_organization(organization_id)
+      connection.get("/v2/plans_for/organization/#{organization_id}").body.map do |plan_data|
+        Travis::API::V3::Models::V2PlanConfig.new(plan_data)
+      end
+    end
+
+    def v2_plans_for_user
+      connection.get("/v2/plans_for/user").body.map do |plan_data|
+        Travis::API::V3::Models::V2PlanConfig.new(plan_data)
+      end
+    end
+
     def cancel_subscription(id, reason_data)
       response = connection.post("/subscriptions/#{id}/cancel", reason_data)
       handle_subscription_response(response)

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

@@ -0,0 +1,14 @@
+module Travis::API::V3
+  class Models::V2PlanConfig
+    attr_reader :id, :name, :private_repos, :default_addons, :starting_price, :starting_users
+
+    def initialize(attrs)
+      @id = attrs.fetch('id')
+      @name = attrs.fetch('name')
+      @private_repos = attrs.fetch('private_repos')
+      @default_addons = attrs.fetch('default_addons')
+      @starting_price = attrs.fetch('starting_price')
+      @starting_users = attrs.fetch('starting_users')
+    end
+  end
+end

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

@@ -0,0 +1,12 @@
+module Travis::API::V3
+  class Queries::V2Plans < Query
+    def all(user_id)
+      client = BillingClient.new(user_id)
+      if params['organization.id']
+        client.v2_plans_for_organization(params['organization.id'])
+      else
+        client.v2_plans_for_user
+      end
+    end
+  end
+end

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

@@ -0,0 +1,46 @@
+module Travis::API::V3
+  class Queries::V2Subscription < Query
+    def update_address(user_id)
+      address_data = params.dup.tap { |h| h.delete('subscription.id') }
+      client = BillingClient.new(user_id)
+      client.update_v2_address(params['subscription.id'], address_data)
+    end
+
+    def cancel(user_id)
+      reason_data = params.dup.tap { |h| h.delete('subscription.id') }
+      client = BillingClient.new(user_id)
+      client.cancel_subscription(params['subscription.id'], reason_data)
+    end
+
+    def update_creditcard(user_id)
+      client = BillingClient.new(user_id)
+      client.update_v2_creditcard(params['subscription.id'], params['token'])
+    end
+
+    def update_plan(user_id)
+      plan_data = params.dup.tap { |h| h.delete('subscription.id') }
+      client = BillingClient.new(user_id)
+      client.update_v2_subscription(params['subscription.id'], plan_data)
+    end
+
+    def resubscribe(user_id)
+      client = BillingClient.new(user_id)
+      client.resubscribe(params['subscription.id'])
+    end
+
+    def invoices(user_id)
+      client = BillingClient.new(user_id)
+      client.get_invoices_for_subscription(params['subscription.id'])
+    end
+
+    def trial(user_id)
+      client = BillingClient.new(user_id)
+      client.get_trial_info_for_subscription(params['subscription.id'])
+    end
+
+    def pay(user_id)
+      client = BillingClient.new(user_id)
+      client.pay(params['subscription.id'])
+    end
+  end
+end

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

@@ -0,0 +1,24 @@
+module Travis::API::V3
+  class Queries::V2Subscriptions < Query
+    params :plan, :coupon, :organization_id, :client_secret
+    params :first_name, :last_name, :company, :address, :address2, :city, :country, :state, :vat_id, :zip_code, :billing_email, prefix: :billing_info
+    params :token, prefix: :credit_card_info
+
+    def all(user_id)
+      client = BillingClient.new(user_id)
+      client.all_v2
+    end
+
+    def create(user_id)
+      client = BillingClient.new(user_id)
+      client.create_v2_subscription(
+        :plan => params['plan'],
+        :client_secret => params['client_secret'],
+        :coupon => params['coupon'],
+        :organization_id => params['organization_id'],
+        :billing_info => billing_info_params,
+        :credit_card_info => credit_card_info_params
+        )
+    end
+  end
+end

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

@@ -0,0 +1,6 @@
+module Travis::API::V3
+  class Renderer::V2PlanConfig < ModelRenderer
+    representation(:standard, :id, :name, :private_repos, :default_addons, :starting_price, :starting_users)
+    representation(:minimal, :id, :name, :private_repos, :default_addons, :starting_price, :starting_users)
+  end
+end

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

@@ -0,0 +1,6 @@
+module Travis::API::V3
+  class Renderer::V2Plans < CollectionRenderer
+    type           :v2_plans
+    collection_key :v2_plans
+  end
+end

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

@@ -0,0 +1,25 @@
+module Travis::API::V3
+  class Renderer::V2Subscription < ModelRenderer
+    representation(:standard, :id, :plan, :addons, :status, :source, :owner, :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?
+    end
+
+    def credit_card_info
+      Renderer.render_model(model.credit_card_info, mode: :standard) unless model.credit_card_info.nil?
+    end
+
+    def plan
+      Renderer.render_model(model.plan, mode: :standard) unless model.plan.nil?
+    end
+
+    def addons
+      Renderer.render_model(model.addons, mode: :standard) unless model.addons.nil?
+    end
+
+    def payment_intent
+      Renderer.render_model(model.payment_intent, mode: :standard) unless model.payment_intent.nil?
+    end
+  end
+end

lib/travis/api/v3/routes.rb

@@ -296,6 +296,12 @@ module Routes
       post :create
     end
 
+    hidden_resource :v2_subscriptions do
+      route '/v2_subscriptions'
+      get :all
+      post :create
+    end
+
     hidden_resource :subscription do
       route '/subscription/{subscription.id}'
       patch :update_address, '/address'
@@ -307,6 +313,17 @@ module Routes
       get :invoices, '/invoices'
     end
 
+    hidden_resource :v2_subscription do
+      route '/v2_subscription/{subscription.id}'
+      patch :update_address, '/address'
+      patch :update_creditcard, '/creditcard'
+      patch :update_plan, '/plan'
+      patch :resubscribe, '/resubscribe'
+      post :cancel, '/cancel'
+      post :pay, '/pay'
+      get :invoices, '/invoices'
+    end
+
     hidden_resource :trials do
       route '/trials'
       get :all
@@ -325,6 +342,13 @@ module Routes
 
     end
 
+    hidden_resource :v2_plans do
+      route '/v2_plans_for'
+      get :all, '/user'
+      get :all, '/organization/{organization.id}'
+
+    end
+
     if ENV['GDPR_ENABLED']
       hidden_resource :gdpr do
         route '/gdpr'

lib/travis/api/v3/services.rb

@@ -36,6 +36,7 @@ module Services
     Organizations         = Module.new { extend Services }
     Owner                 = Module.new { extend Services }
     Plans                 = Module.new { extend Services }
+    V2Plans               = Module.new { extend Services }
     Preferences           = Module.new { extend Services }
     Preference            = Module.new { extend Services }
     Queues                = Module.new { extend Services }
@@ -47,6 +48,8 @@ module Services
     Stages                = Module.new { extend Services }
     Subscription          = Module.new { extend Services }
     Subscriptions         = Module.new { extend Services }
+    V2Subscription        = Module.new { extend Services }
+    V2Subscriptions       = Module.new { extend Services }
     Trials                = Module.new { extend Services }
     User                  = Module.new { extend Services }
     UserSetting           = Module.new { extend Services }