Merge pull request #1161 from travis-ci/epic-pricing-rc

The new billing system going live

Author: Shairyar Baig

.travis.yml

@@ -1,7 +1,7 @@
 language: ruby
 
 import:
-  - travis-ci/build-configs:db-setup.yml@bionic
+  - travis-ci/build-configs:db-setup.yml@epic-pricing
 
 rvm: 2.6.5
 

lib/travis/api/app/endpoint/authorization.rb

@@ -106,7 +106,6 @@ class Authorization < Endpoint
       get '/handshake/?:provider?' do
         method = org? ? :handshake : :vcs_handshake
         params[:provider] ||= 'github'
-
         send(method) do |user, token, redirect_uri|
           if target_ok? redirect_uri
             content_type :html
@@ -212,7 +211,9 @@ def vcs_handshake
               return
             end
 
-            yield serialize_user(User.find(vcs_data['user']['id'])), vcs_data['token'], payload(params[:provider])
+            user = User.find(vcs_data['user']['id'])
+            update_first_login(user)
+            yield serialize_user(user), vcs_data['token'], payload(params[:provider])
           else
             state = vcs_create_state(params[:origin] || params[:redirect_uri])
 

lib/travis/api/enqueue/services/restart_model.rb

@@ -4,7 +4,7 @@ module Services
 
       class RestartModel
         attr_reader :current_user, :target
-        ABUSE_DETECTED = "abuse_detected"
+        ABUSE_DETECTED = 'abuse_detected'
 
         def initialize(current_user, params)
           @current_user = current_user
@@ -15,23 +15,43 @@ def initialize(current_user, params)
         def push(event, payload)
           if current_user && target && accept?
             ::Sidekiq::Client.push(
-                  'queue'   => 'hub',
-                  'class'   => 'Travis::Hub::Sidekiq::Worker',
-                  'args'    => [event, payload]
-                )
+              'queue'   => 'hub',
+              'class'   => 'Travis::Hub::Sidekiq::Worker',
+              'args'    => [event, payload]
+            )
+
+            Result.new(value: payload)
           else
-            @cause_of_denial
+            Result.new(error: @cause_of_denial || 'restart failed')
           end
         end
 
         def accept?
-          current_user && permission? && resetable?
+          current_user && permission? && resetable? && billing?
+        end
+
+        def billing?
+          @_billing_ok ||= begin
+            jobs = target.is_a?(Job) ? [target] : target.matrix
+
+            jobs_attrs = jobs.map do |job|
+              job.config ? job.config.slice(:os) : {}
+            end
+
+            client = Travis::API::V3::BillingClient.new(current_user.id)
+            client.authorize_build(repository, current_user.id, jobs_attrs)
+            true
+          rescue Travis::API::V3::InsufficientAccess => e
+            @cause_of_denial = e.message
+            false
+          end
         end
 
         def messages
           messages = []
           messages << { notice: "The #{type} was successfully restarted." } if accept?
           messages << { error:  'You do not seem to have sufficient permissions.' } unless permission?
+          messages << { error:  'You do not have enough credits.' } unless billing?
           messages << { error:  "This #{type} currently can not be restarted." } unless resetable?
           messages
         end
@@ -54,23 +74,36 @@ def target
 
         private
 
-          def permission?
-            current_user && current_user.permission?(required_role, repository_id: target.repository_id) && !abusive?
-          end
+        def permission?
+          current_user && current_user.permission?(required_role, repository_id: target.repository_id) && !abusive?
+        end
 
-          def abusive?
-            abusive = Travis.redis.sismember("abuse:offenders", "#{@target.owner.class.name}:#{@target.owner_id}")
-            @cause_of_denial = ABUSE_DETECTED if abusive
-            abusive
-          end
+        def abusive?
+          abusive = Travis.redis.sismember("abuse:offenders", "#{@target.owner.class.name}:#{@target.owner_id}")
+          @cause_of_denial = ABUSE_DETECTED if abusive
+          abusive
+        end
+
+        def resetable?
+          target.resetable?
+        end
 
-          def resetable?
-            target.resetable?
+        def required_role
+          Travis.config.roles.reset_model
+        end
+
+        class Result
+          attr_reader :error, :value
+
+          def initialize(value: nil, error: nil)
+            @value = value
+            @error = error
           end
 
-          def required_role
-            Travis.config.roles.reset_model
+          def success?
+            !@error
           end
+        end
       end
     end
   end

lib/travis/api/v3/billing_client.rb

@@ -2,10 +2,46 @@ module Travis::API::V3
   class BillingClient
     class ConfigurationError < StandardError; end
 
+    ALLOWANCE_TIMEOUT = 1 # second
+
     def initialize(user_id)
       @user_id = user_id
     end
 
+    def allowance(owner_type, owner_id)
+      response = connection(timeout: ALLOWANCE_TIMEOUT).get("/usage/#{owner_type.downcase}s/#{owner_id}/allowance")
+      return BillingClient.default_allowance_response unless response.status == 200
+
+      Travis::API::V3::Models::Allowance.new(2, owner_id, response.body)
+    end
+
+    def authorize_build(repo, sender_id, jobs)
+      response = connection.post("/#{repo.owner.class.name.downcase.pluralize}/#{repo.owner.id}/authorize_build", { repository: { private: repo.private? }, sender_id: sender_id, jobs: jobs })
+      handle_errors_and_respond(response)
+    end
+
+    def self.default_allowance_response(id = 0)
+      Travis::API::V3::Models::Allowance.new(1, id, {
+        "public_repos" => true,
+        "private_repos" => false,
+        "concurrency_limit" => 1,
+        "user_usage" => false,
+        "pending_user_licenses" => false
+      }.freeze)
+    end
+
+    def self.minimal_allowance_response(id = 0)
+      Travis::API::V3::Models::Allowance.new(2, id, {})
+    end
+
+    def executions(owner_type, owner_id, page, per_page, from, to)
+      response = connection.get("/usage/#{owner_type.downcase}s/#{owner_id}/executions?page=#{page}&per_page=#{per_page}&from=#{from}&to=#{to}")
+      executions = response.body.map do |execution_data|
+        Travis::API::V3::Models::Execution.new(execution_data)
+      end
+      executions
+    end
+
     def all
       data = connection.get('/subscriptions').body
       subscriptions = data.fetch('subscriptions').map do |subscription_data|
@@ -16,17 +52,38 @@ def all
       Travis::API::V3::Models::SubscriptionsCollection.new(subscriptions, permissions)
     end
 
+    def all_v2
+      data = connection.get('/v2/subscriptions').body
+      subscriptions = data.fetch('plans').map do |subscription_data|
+        Travis::API::V3::Models::V2Subscription.new(subscription_data)
+      end
+      permissions = data.fetch('permissions')
+
+      Travis::API::V3::Models::SubscriptionsCollection.new(subscriptions, permissions)
+    end
+
     def get_subscription(id)
       response = connection.get("/subscriptions/#{id}")
       handle_subscription_response(response)
     end
 
+    def get_v2_subscription(id)
+      response = connection.get("/v2/subscriptions/#{id}")
+      handle_v2_subscription_response(response)
+    end
+
     def get_invoices_for_subscription(id)
       connection.get("/subscriptions/#{id}/invoices").body.map do |invoice_data|
         Travis::API::V3::Models::Invoice.new(invoice_data)
       end
     end
 
+    def get_invoices_for_v2_subscription(id)
+      connection.get("/v2/subscriptions/#{id}/invoices").body.map do |invoice_data|
+        Travis::API::V3::Models::Invoice.new(invoice_data)
+      end
+    end
+
     def trials
       connection.get('/trials').body.map do | trial_data |
         Travis::API::V3::Models::Trial.new(trial_data)
@@ -43,11 +100,21 @@ def update_address(subscription_id, address_data)
       handle_subscription_response(response)
     end
 
+    def update_v2_address(subscription_id, address_data)
+      response = connection.patch("/v2/subscriptions/#{subscription_id}/address", address_data)
+      handle_v2_subscription_response(response)
+    end
+
     def update_creditcard(subscription_id, creditcard_token)
       response = connection.patch("/subscriptions/#{subscription_id}/creditcard", token: creditcard_token)
       handle_subscription_response(response)
     end
 
+    def update_v2_creditcard(subscription_id, creditcard_token)
+      response = connection.patch("/v2/subscriptions/#{subscription_id}/creditcard", token: creditcard_token)
+      handle_v2_subscription_response(response)
+    end
+
     def update_plan(subscription_id, plan_data)
       response = connection.patch("/subscriptions/#{subscription_id}/plan", plan_data)
       handle_subscription_response(response)
@@ -58,6 +125,44 @@ def create_subscription(subscription_data)
       handle_subscription_response(response)
     end
 
+    def create_v2_subscription(subscription_data)
+      response = connection.post('/v2/subscriptions', subscription_data)
+      handle_v2_subscription_response(response)
+    end
+
+    def changetofree_v2_subscription(subscription_id, data)
+      response = connection.patch("/v2/subscriptions/#{subscription_id}/changetofree", data)
+      handle_v2_subscription_response(response)
+    end
+
+    def update_v2_subscription(subscription_id, plan_data)
+      response = connection.patch("/v2/subscriptions/#{subscription_id}/plan", plan_data)
+      handle_v2_subscription_response(response)
+    end
+
+    def purchase_addon(subscription_id, addon_config_id)
+      response = connection.patch("/v2/subscriptions/#{subscription_id}/addon", { addon: addon_config_id })
+      handle_v2_subscription_response(response)
+    end
+
+    def v2_subscription_user_usages(subscription_id)
+      connection.get("/v2/subscriptions/#{subscription_id}/user_usage").body.map do |usage_data|
+        Travis::API::V3::Models::V2AddonUsage.new(usage_data)
+      end
+    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)
@@ -70,7 +175,7 @@ def plans_for_organization(organization_id)
     end
 
     def plans_for_user
-      connection.get("/plans_for/user").body.map do |plan_data|
+      connection.get('/plans_for/user').body.map do |plan_data|
         Travis::API::V3::Models::Plan.new(plan_data)
       end
     end
@@ -85,6 +190,11 @@ def pay(id)
       handle_subscription_response(response)
     end
 
+    def pay_v2(id)
+      response = connection.post("/v2/subscriptions/#{id}/pay")
+      handle_v2_subscription_response(response)
+    end
+
     def get_coupon(code)
       response = connection.get("/coupons/#{code}")
       handle_coupon_response(response)
@@ -101,6 +211,10 @@ def handle_subscription_response(response)
       handle_errors_and_respond(response) { |r| Travis::API::V3::Models::Subscription.new(r) }
     end
 
+    def handle_v2_subscription_response(response)
+      handle_errors_and_respond(response) { |r| Travis::API::V3::Models::V2Subscription.new(r) }
+    end
+
     def handle_coupon_response(response)
       handle_errors_and_respond(response) { |r| Travis::API::V3::Models::Coupon.new(r) }
     end
@@ -113,24 +227,28 @@ def handle_errors_and_respond(response)
         true
       when 204
         true
-      when 404
-        raise Travis::API::V3::NotFound, response.body['error']
       when 400
         raise Travis::API::V3::ClientError, response.body['error']
+      when 403
+        raise Travis::API::V3::InsufficientAccess, response.body['rejection_code']
+      when 404
+        raise Travis::API::V3::NotFound, response.body['error']
       when 422
         raise Travis::API::V3::UnprocessableEntity, response.body['error']
       else
         raise Travis::API::V3::ServerError, 'Billing system failed'
       end
     end
 
-    def connection
+    def connection(timeout: 10)
       @connection ||= Faraday.new(url: billing_url, ssl: { ca_path: '/usr/lib/ssl/certs' }) do |conn|
         conn.basic_auth '_', billing_auth_key
         conn.headers['X-Travis-User-Id'] = @user_id.to_s
         conn.headers['Content-Type'] = 'application/json'
         conn.request :json
         conn.response :json
+        conn.options[:open_timeout] = timeout
+        conn.options[:timeout] = timeout
         conn.use OpenCensus::Trace::Integrations::FaradayMiddleware if Travis::Api::App::Middleware::OpenCensus.enabled?
         conn.adapter :net_http
       end

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

@@ -0,0 +1,15 @@
+module Travis::API::V3
+  class Models::Allowance
+    attr_reader :subscription_type, :public_repos, :private_repos, :concurrency_limit, :user_usage, :pending_user_licenses, :id
+
+    def initialize(subscription_type, owner_id, attributes = {})
+      @subscription_type = subscription_type
+      @id = owner_id
+      @public_repos = attributes.fetch('public_repos', nil)
+      @private_repos = attributes.fetch('private_repos', nil)
+      @concurrency_limit = attributes.fetch('concurrency_limit', nil)
+      @user_usage = attributes.fetch('user_usage', nil)
+      @pending_user_licenses = attributes.fetch('pending_user_licenses', nil)
+    end
+  end
+end

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

@@ -0,0 +1,11 @@
+module Travis::API::V3
+  class Models::BuildPermission
+    attr_accessor :user, :permission, :role
+
+    def initialize(attrs = {})
+      @user  = attrs.fetch(:user)
+      @role  = attrs.fetch(:role)
+      @permission = attrs.fetch(:permission)
+    end
+  end
+end