shipit-engine 0.31.0 → 0.35.0

data/app/models/shipit/webhooks/handlers/pull_request/labeled_handler.rb

@@ -0,0 +1,106 @@
+# frozen_string_literal: true
+
+module Shipit
+  module Webhooks
+    module Handlers
+      module PullRequest
+        class LabeledHandler < Shipit::Webhooks::Handlers::Handler
+          params do
+            requires :action, String
+            requires :number, Integer
+            requires :pull_request do
+              requires :id, Integer
+              requires :number, Integer
+              requires :url, String
+              requires :title, String
+              requires :state, String
+              requires :additions, Integer
+              requires :deletions, Integer
+              requires :head do
+                requires :sha, String
+                requires :ref, String
+              end
+              requires :user do
+                requires :login, String
+              end
+              requires :assignees, Array do
+                requires :login, String
+              end
+              requires :labels, Array do
+                requires :name, String
+              end
+            end
+            requires :repository do
+              requires :full_name, String
+            end
+            requires :sender do
+              requires :login, String
+            end
+          end
+
+          def process
+            return unless respond_to_label_change?
+
+            handle
+          end
+
+          private
+
+          def handle
+            if archive?
+              stack.archive!
+            elsif unarchive?
+              stack.unarchive!
+            end
+
+            stack
+          end
+
+          def stack
+            @stack ||=
+              Shipit::Webhooks::Handlers::PullRequest::ReviewStackAdapter
+                .new(params, scope: repository.review_stacks)
+          end
+
+          def repository
+            @repository ||= Shipit::Repository.from_github_repo_name(params.repository.full_name) ||
+              Shipit::NullRepository.new
+          end
+
+          def pull_request
+            params.pull_request
+          end
+
+          def pull_request_state
+            pull_request.state
+          end
+
+          def respond_to_label_change?
+            params.action == "labeled" &&
+              pull_request_state == "open" &&
+              repository.review_stacks_enabled &&
+              (archive? || unarchive?)
+          end
+
+          def archive?
+            (repository.provisioning_behavior_allow_with_label? && !pull_request_has_provisioning_label?) ||
+              (repository.provisioning_behavior_prevent_with_label? && pull_request_has_provisioning_label?)
+          end
+
+          def unarchive?
+            (repository.provisioning_behavior_allow_with_label? && pull_request_has_provisioning_label?) ||
+              (repository.provisioning_behavior_prevent_with_label? && !pull_request_has_provisioning_label?)
+          end
+
+          def pull_request_has_provisioning_label?
+            pull_request_label_names.include?(repository.provisioning_label_name)
+          end
+
+          def pull_request_label_names
+            Array.new(pull_request.labels).map(&:name)
+          end
+        end
+      end
+    end
+  end
+end

data/app/models/shipit/webhooks/handlers/pull_request/opened_handler.rb

@@ -0,0 +1,83 @@
+# frozen_string_literal: true
+
+module Shipit
+  module Webhooks
+    module Handlers
+      module PullRequest
+        class OpenedHandler < Shipit::Webhooks::Handlers::Handler
+          params do
+            requires :action, String
+            requires :number, Integer
+            requires :pull_request do
+              requires :id, Integer
+              requires :number, Integer
+              requires :url, String
+              requires :title, String
+              requires :state, String
+              requires :additions, Integer
+              requires :deletions, Integer
+              requires :head do
+                requires :sha, String
+                requires :ref, String
+              end
+              requires :user do
+                requires :login, String
+              end
+              requires :assignees, Array do
+                requires :login, String
+              end
+              requires :labels, Array do
+                requires :name, String
+              end
+            end
+            requires :repository do
+              requires :full_name, String
+            end
+            requires :sender do
+              requires :login, String
+            end
+          end
+
+          def process
+            return unless respond_to_pull_request_opened?
+
+            Shipit::Webhooks::Handlers::PullRequest::ReviewStackAdapter
+              .new(params, scope: repository.review_stacks).find_or_create!
+          end
+
+          private
+
+          def repository
+            @repository ||=
+              Shipit::Repository.from_github_repo_name(params.repository.full_name) ||
+              Shipit::NullRepository.new
+          end
+
+          def pull_request
+            params.pull_request
+          end
+
+          def respond_to_pull_request_opened?
+            params.action == "opened" &&
+              provision?
+          end
+
+          def provision?
+            repository.review_stacks_enabled &&
+              repository.provisioning_behavior_allow_all? ||
+              (repository.provisioning_behavior_allow_with_label? && pull_request_has_provisioning_label?) ||
+              (repository.provisioning_behavior_prevent_with_label? && !pull_request_has_provisioning_label?)
+          end
+
+          def pull_request_has_provisioning_label?
+            pull_request_label_names.include?(repository.provisioning_label_name)
+          end
+
+          def pull_request_label_names
+            Array.new(pull_request["labels"]).map { |label| label["name"] }
+          end
+        end
+      end
+    end
+  end
+end

data/app/models/shipit/webhooks/handlers/pull_request/reopened_handler.rb

@@ -0,0 +1,88 @@
+# frozen_string_literal: true
+
+module Shipit
+  module Webhooks
+    module Handlers
+      module PullRequest
+        class ReopenedHandler < Shipit::Webhooks::Handlers::Handler
+          params do
+            requires :action, String
+            requires :number, Integer
+            requires :pull_request do
+              requires :id, Integer
+              requires :number, Integer
+              requires :url, String
+              requires :title, String
+              requires :state, String
+              requires :additions, Integer
+              requires :deletions, Integer
+              requires :head do
+                requires :sha, String
+                requires :ref, String
+              end
+              requires :user do
+                requires :login, String
+              end
+              requires :assignees, Array do
+                requires :login, String
+              end
+              requires :labels, Array do
+                requires :name, String
+              end
+            end
+            requires :repository do
+              requires :full_name, String
+            end
+            requires :sender do
+              requires :login, String
+            end
+          end
+
+          def process
+            return unless respond_to_pull_request_reopened?
+
+            stack.unarchive!
+          end
+
+          private
+
+          def repository
+            @repository ||=
+              Shipit::Repository.from_github_repo_name(params.repository.full_name) ||
+              Shipit::NullRepository.new
+          end
+
+          def stack
+            @stack ||=
+              Shipit::Webhooks::Handlers::PullRequest::ReviewStackAdapter
+                .new(params, scope: repository.review_stacks)
+          end
+
+          def pull_request
+            params.pull_request
+          end
+
+          def respond_to_pull_request_reopened?
+            params.action == "reopened" &&
+              unarchive?
+          end
+
+          def unarchive?
+            repository.review_stacks_enabled &&
+              repository.provisioning_behavior_allow_all? ||
+              (repository.provisioning_behavior_allow_with_label? && pull_request_has_provisioning_label?) ||
+              (repository.provisioning_behavior_prevent_with_label? && !pull_request_has_provisioning_label?)
+          end
+
+          def pull_request_has_provisioning_label?
+            pull_request_label_names.include?(repository.provisioning_label_name)
+          end
+
+          def pull_request_label_names
+            Array.new(pull_request["labels"]).map { |label| label["name"] }
+          end
+        end
+      end
+    end
+  end
+end

data/app/models/shipit/webhooks/handlers/pull_request/review_stack_adapter.rb

@@ -0,0 +1,103 @@
+# frozen_string_literal: true
+
+module Shipit
+  module Webhooks
+    module Handlers
+      module PullRequest
+        class ReviewStackAdapter
+          delegate :archived?, to: :stack
+
+          def initialize(params, scope: Shipit::ReviewStack)
+            @params = params
+            @scope = scope
+          end
+
+          def stack
+            @stack ||= scope.find_by(environment: environment)
+          end
+
+          def find_or_create!
+            stack || create!
+          end
+
+          def archive!(*args, &block)
+            if stack.blank?
+              Rails.logger.info(
+                "Processing #{action} event for #{repo_name} PR #{pr_number} but no Stack exists. Ignoring."
+              )
+              return true
+            end
+            return if stack.archived?
+
+            stack.remove_from_provisioning_queue
+            stack.deprovision
+            stack.archive!(user, *args, &block)
+          end
+
+          def unarchive!(*args, &block)
+            if stack.blank?
+              Rails.logger.info(
+                "Processing #{action} event for #{repo_name} PR #{pr_number} but no ReviewStack exists. Creating."
+              )
+              return create!
+            end
+            return unless stack.archived?
+
+            stack.transaction do
+              Shipit::ReviewStackProvisioningQueue.add(stack)
+              stack.unarchive!(*args, &block)
+            end
+          end
+
+          def user
+            @user ||= Shipit::User.find_or_create_by_login!(params.sender["login"])
+          end
+
+          private
+
+          attr_reader :params, :scope
+
+          def action
+            params.action
+          end
+
+          def repo_name
+            params.repository["full_name"]
+          end
+
+          def pr_number
+            params.number
+          end
+
+          def create!
+            ReviewStack.transaction do
+              stack = scope.create!(stack_attributes)
+              stack
+                .build_pull_request
+                .update!(
+                  github_pull_request: params.pull_request
+                )
+            end
+
+            Shipit::ReviewStackProvisioningQueue.add(stack)
+
+            @stack = stack
+          end
+
+          def stack_attributes
+            {
+              branch: params.pull_request.head.ref,
+              environment: environment,
+              ignore_ci: false,
+              continuous_deployment: false,
+            }
+          end
+
+          def environment
+            "pr#{params.number}"
+          end
+        end
+      end
+    end
+  end
+end

data/app/models/shipit/webhooks/handlers/pull_request/unlabeled_handler.rb

@@ -0,0 +1,107 @@
+# frozen_string_literal: true
+
+module Shipit
+  module Webhooks
+    module Handlers
+      module PullRequest
+        class UnlabeledHandler < Shipit::Webhooks::Handlers::Handler
+          params do
+            requires :action, String
+            requires :number, Integer
+            requires :pull_request do
+              requires :id, Integer
+              requires :number, Integer
+              requires :url, String
+              requires :title, String
+              requires :state, String
+              requires :additions, Integer
+              requires :deletions, Integer
+              requires :head do
+                requires :sha, String
+                requires :ref, String
+              end
+              requires :user do
+                requires :login, String
+              end
+              requires :assignees, Array do
+                requires :login, String
+              end
+              requires :labels, Array do
+                requires :name, String
+              end
+            end
+            requires :repository do
+              requires :full_name, String
+            end
+            requires :sender do
+              requires :login, String
+            end
+          end
+
+          def process
+            return unless respond_to_label_change?
+
+            handle
+          end
+
+          private
+
+          def handle
+            if archive?
+              stack.archive!
+            elsif unarchive?
+              stack.unarchive!
+            end
+
+            stack
+          end
+
+          def repository
+            @repository ||=
+              Shipit::Repository.from_github_repo_name(params.repository.full_name) ||
+              Shipit::NullRepository.new
+          end
+
+          def stack
+            @stack ||=
+              Shipit::Webhooks::Handlers::PullRequest::ReviewStackAdapter
+                .new(params, scope: repository.review_stacks)
+          end
+
+          def pull_request
+            params.pull_request
+          end
+
+          def pull_request_state
+            pull_request["state"]
+          end
+
+          def respond_to_label_change?
+            params.action == "unlabeled" &&
+              pull_request_state == "open" &&
+              repository.review_stacks_enabled &&
+              (archive? || unarchive?)
+          end
+
+          def archive?
+            (repository.provisioning_behavior_allow_with_label? && !pull_request_has_provisioning_label?) ||
+              (repository.provisioning_behavior_prevent_with_label? && pull_request_has_provisioning_label?)
+          end
+
+          def unarchive?
+            (repository.provisioning_behavior_allow_with_label? && pull_request_has_provisioning_label?) ||
+              (repository.provisioning_behavior_prevent_with_label? && !pull_request_has_provisioning_label?)
+          end
+
+          def pull_request_has_provisioning_label?
+            pull_request_label_names.include?(repository.provisioning_label_name)
+          end
+
+          def pull_request_label_names
+            Array.new(pull_request.labels).map(&:name)
+          end
+        end
+      end
+    end
+  end
+end