BigW Consortium Gitlab

subscriptions.rb 1.91 KB
Newer Older
1 2 3 4 5 6 7 8 9
module API
  module V3
    class Subscriptions < Grape::API
      before { authenticate! }

      subscribable_types = {
        'merge_request' => proc { |id| find_merge_request_with_access(id, :update_merge_request) },
        'merge_requests' => proc { |id| find_merge_request_with_access(id, :update_merge_request) },
        'issues' => proc { |id| find_project_issue(id) },
10
        'labels' => proc { |id| find_project_label(id) }
11 12 13 14 15 16
      }

      params do
        requires :id, type: String, desc: 'The ID of a project'
        requires :subscribable_id, type: String, desc: 'The ID of a resource'
      end
17
      resource :projects, requirements: { id: %r{[^/]+} } do
18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
        subscribable_types.each do |type, finder|
          type_singularized = type.singularize
          entity_class = ::API::Entities.const_get(type_singularized.camelcase)

          desc 'Subscribe to a resource' do
            success entity_class
          end
          post ":id/#{type}/:subscribable_id/subscription" do
            resource = instance_exec(params[:subscribable_id], &finder)

            if resource.subscribed?(current_user, user_project)
              not_modified!
            else
              resource.subscribe(current_user, user_project)
              present resource, with: entity_class, current_user: current_user, project: user_project
            end
          end

          desc 'Unsubscribe from a resource' do
            success entity_class
          end
          delete ":id/#{type}/:subscribable_id/subscription" do
            resource = instance_exec(params[:subscribable_id], &finder)

            if !resource.subscribed?(current_user, user_project)
              not_modified!
            else
              resource.unsubscribe(current_user, user_project)
              present resource, with: entity_class, current_user: current_user, project: user_project
            end
          end
        end
      end
    end
  end
end