Merge branch 'epic-pricing-3' into epic-pricing-rc

Author: pavel-d

.travis.yml

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

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,19 +2,40 @@ 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.get("/usage/#{owner_type.downcase}s/#{owner_id}/allowance")
+      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 }.freeze)
+      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 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
@@ -202,24 +223,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

@@ -1,13 +1,15 @@
 module Travis::API::V3
   class Models::Allowance
-    attr_reader :subscription_type, :public_repos, :private_repos, :concurrency_limit, :id
+    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')
       @private_repos = attributes.fetch('private_repos')
       @concurrency_limit = attributes.fetch('concurrency_limit')
+      @user_usage = attributes.fetch('user_usage')
+      @pending_user_licenses = attributes.fetch('pending_user_licenses')
     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

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

@@ -0,0 +1,27 @@
+module Travis::API::V3
+  class Models::Execution
+    attr_reader :id, :os, :instance_size, :arch, :virtualization_type, :queue, :job_id, :repository_id, :owner_id,
+                :owner_type, :plan_id, :sender_id, :credits_consumed, :started_at, :finished_at, :created_at,
+                :updated_at
+
+    def initialize(attributes = {})
+      @id = attributes.fetch('id')
+      @os = attributes.fetch('os')
+      @instance_size = attributes.fetch('instance_size')
+      @arch = attributes.fetch('arch')
+      @virtualization_type = attributes.fetch('virtualization_type')
+      @queue = attributes.fetch('queue')
+      @job_id = attributes.fetch('job_id')
+      @repository_id = attributes.fetch('repository_id')
+      @owner_id = attributes.fetch('owner_id')
+      @owner_type = attributes.fetch('owner_type')
+      @plan_id = attributes.fetch('plan_id')
+      @sender_id = attributes.fetch('sender_id')
+      @credits_consumed = attributes.fetch('credits_consumed')
+      @started_at = attributes.fetch('started_at')
+      @finished_at = attributes.fetch('finished_at')
+      @created_at = attributes.fetch('created_at')
+      @updated_at = attributes.fetch('updated_at')
+    end
+  end
+end

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

@@ -1,10 +1,11 @@
 module Travis::API::V3
   class Models::Invoice
-    attr_reader :id, :created_at, :url, :amount_due
+    attr_reader :id, :created_at, :status, :url, :amount_due
 
     def initialize(attributes = {})
       @id = attributes.fetch('id')
       @created_at = attributes.fetch('created_at') && DateTime.parse(attributes.fetch('created_at'))
+      @status = attributes.fetch('status')
       @url = attributes.fetch('url')
       @amount_due = attributes.fetch('amount_due')
     end

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

@@ -4,6 +4,8 @@ module Travis::API::V3
   class Models::OrganizationPreferences < Models::JsonSlice
     child Models::Preference
 
+    attribute :consume_oss_credits, Boolean, default: true
+
     # whether to show insights about the organization's private repositories to
     # only admins, all members of the organization, or everybody (public) (note:
     # insights about public repositories are always public)

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

@@ -6,6 +6,8 @@ class Models::UserPreferences < Models::JsonSlice
 
     attribute :build_emails, Boolean, default: true
 
+    attribute :consume_oss_credits, Boolean, default: true
+
     # whether to show insights about the user's private repositories to
     # everybody or keep them only for the user (note: insights about public
     # repositories are always public)

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

@@ -1,6 +1,6 @@
 module Travis::API::V3
   class Models::V2AddonUsage
-    attr_reader :id, :addon_id, :addon_quantity, :addon_usage, :remaining, :purchase_date, :valid_to, :active
+    attr_reader :id, :addon_id, :addon_quantity, :addon_usage, :remaining, :purchase_date, :valid_to, :active, :status
 
     def initialize(attrs)
       @id = attrs.fetch('id')
@@ -11,6 +11,7 @@ def initialize(attrs)
       @purchase_date = attrs.fetch('purchase_date')
       @valid_to = attrs.fetch('valid_to')
       @active = attrs.fetch('active')
+      @status = attrs.fetch('status')
     end
   end
 end