Add enqueue service for canceling jobs and builds

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

@@ -0,0 +1,84 @@
+module Travis
+  module Enqueue
+    module Services
+
+      class CancelModel
+
+        attr_reader :current_user, :source, :target
+
+        def initialize(current_user, params)
+          @current_user = current_user
+          @params = params
+          @source = params.delete(:source) || 'unknown'
+          target
+        end
+
+        def run
+          cancel
+        end
+        instrument :run
+
+        def messages
+          messages = []
+          messages << { :notice => "The #{type} was successfully cancelled." } if can_cancel?
+          messages << { :error  => "You are not authorized to cancel this #{type}." } unless authorized?
+          messages << { :error  => "The #{type} could not be cancelled." } unless build.cancelable?
+          messages
+        end
+
+        def cancel(payload)
+          # target may have been retrieved with a :join query, so we need to reset the readonly status
+          if can_cancel?
+            target.send(:instance_variable_set, :@readonly, false)
+            target.cancel!
+
+            if type == :build
+              push_matrix(payload)
+            else
+              push(payload, target)
+            end
+          end
+        end
+
+        def push_matrix(payload)
+
+          target.matrix.each do |job|
+            push(payload, job)
+          end
+
+        end
+
+        def push(payload, job)
+          Travis.logger.info("Publishing cancel_job message to worker.commands queue for Job##{job.id}")
+          ::Sidekiq::Client.push(
+                'queue'   => 'hub',
+                'class'   => 'Travis::Hub::Sidekiq::Worker',
+                'args'    => ["#{type}:cancel", payload]
+              )
+        end
+
+        def type
+          @type ||= @params[:build_id] ? :build : :job
+        end
+
+        def target
+          if type == :build
+            @target = Build.find(@params[:build_id])
+          else
+            @target = Job.find(@params[:job_id])
+          end
+        end
+
+        def can_cancel?
+          authorized? && target.cancelable?
+        end
+
+        def authorized?
+          current_user.permission?(:pull, :repository_id => target.repository_id)
+        end
+
+      end
+
+    end
+  end
+end