File: custom_headers.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (43 lines) | stat: -rw-r--r-- 1,384 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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
# frozen_string_literal: true

module API
  module Hooks
    # rubocop: disable API/Base -- It is important that this re-usable module is not a Grape Instance, since it will be re-mounted.
    class CustomHeaders < ::Grape::API
      params do
        requires :hook_id, type: Integer, desc: 'The ID of the hook'
        requires :key, type: String, desc: 'The key of the custom header'
      end
      namespace ':hook_id/custom_headers' do
        desc 'Set a custom header'
        params do
          requires :value, type: String, desc: 'The value of the custom header'
        end
        put ":key" do
          hook = find_hook
          key = params.delete(:key)
          value = params.delete(:value)
          custom_headers = hook.custom_headers.merge(key => value)

          error!('Illegal key or value', 422) unless hook.update(custom_headers: custom_headers)

          status :no_content
        end

        desc 'Un-Set a custom header'
        delete ":key" do
          hook = find_hook
          key = params.delete(:key)
          not_found!('Custom header') unless hook.custom_headers.key?(key)

          vars = hook.custom_headers.reject { |k, _| k == key }

          error!('Could not unset custom header', 422) unless hook.update(custom_headers: vars)

          status :no_content
        end
      end
    end
    # rubocop: enable API/Base
  end
end