125 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			125 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Ruby
		
	
	
	
| # frozen_string_literal: true
 | |
| 
 | |
| class DatadogService < Service
 | |
|   DEFAULT_SITE = 'datadoghq.com'.freeze
 | |
|   URL_TEMPLATE = 'https://webhooks-http-intake.logs.%{datadog_site}/v1/input/'.freeze
 | |
|   URL_TEMPLATE_API_KEYS = 'https://app.%{datadog_site}/account/settings#api'.freeze
 | |
|   URL_API_KEYS_DOCS = "https://docs.#{DEFAULT_SITE}/account_management/api-app-keys/".freeze
 | |
| 
 | |
|   SUPPORTED_EVENTS = %w[
 | |
|     pipeline job
 | |
|   ].freeze
 | |
| 
 | |
|   prop_accessor :datadog_site, :api_url, :api_key, :datadog_service, :datadog_env
 | |
| 
 | |
|   with_options presence: true, if: :activated? do
 | |
|     validates :api_key, format: { with: /\A\w+\z/ }
 | |
|     validates :datadog_site, format: { with: /\A[\w\.]+\z/ }, unless: :api_url
 | |
|     validates :api_url, public_url: true, unless: :datadog_site
 | |
|   end
 | |
| 
 | |
|   after_save :compose_service_hook, if: :activated?
 | |
| 
 | |
|   def self.supported_events
 | |
|     SUPPORTED_EVENTS
 | |
|   end
 | |
| 
 | |
|   def self.default_test_event
 | |
|     'pipeline'
 | |
|   end
 | |
| 
 | |
|   def configurable_events
 | |
|     [] # do not allow to opt out of required hooks
 | |
|   end
 | |
| 
 | |
|   def title
 | |
|     'Datadog'
 | |
|   end
 | |
| 
 | |
|   def description
 | |
|     'Trace your GitLab pipelines with Datadog'
 | |
|   end
 | |
| 
 | |
|   def help
 | |
|     nil
 | |
|     # Maybe adding something in the future
 | |
|     # We could link to static help pages as well
 | |
|     # [More information](#{Gitlab::Routing.url_helpers.help_page_url('integration/datadog')})"
 | |
|   end
 | |
| 
 | |
|   def self.to_param
 | |
|     'datadog'
 | |
|   end
 | |
| 
 | |
|   def fields
 | |
|     [
 | |
|       {
 | |
|         type: 'text', name: 'datadog_site',
 | |
|         placeholder: DEFAULT_SITE, default: DEFAULT_SITE,
 | |
|         help: 'Choose the Datadog site to send data to. Set to "datadoghq.eu" to send data to the EU site',
 | |
|         required: false
 | |
|       },
 | |
|       {
 | |
|         type: 'text', name: 'api_url', title: 'Custom URL',
 | |
|         help: '(Advanced) Define the full URL for your Datadog site directly',
 | |
|         required: false
 | |
|       },
 | |
|       {
 | |
|         type: 'password', name: 'api_key', title: 'API key',
 | |
|         help: "<a href=\"#{api_keys_url}\" target=\"_blank\">API key</a> used for authentication with Datadog",
 | |
|         required: true
 | |
|       },
 | |
|       {
 | |
|         type: 'text', name: 'datadog_service', title: 'Service', placeholder: 'gitlab-ci',
 | |
|         help: 'Name of this GitLab instance that all data will be tagged with'
 | |
|       },
 | |
|       {
 | |
|         type: 'text', name: 'datadog_env', title: 'Env',
 | |
|         help: 'The environment tag that traces will be tagged with'
 | |
|       }
 | |
|     ]
 | |
|   end
 | |
| 
 | |
|   def compose_service_hook
 | |
|     hook = service_hook || build_service_hook
 | |
|     hook.url = hook_url
 | |
|     hook.save
 | |
|   end
 | |
| 
 | |
|   def hook_url
 | |
|     url = api_url.presence || sprintf(URL_TEMPLATE, datadog_site: datadog_site)
 | |
|     url = URI.parse(url)
 | |
|     url.path = File.join(url.path || '/', api_key)
 | |
|     query = { service: datadog_service, env: datadog_env }.compact
 | |
|     url.query = query.to_query unless query.empty?
 | |
|     url.to_s
 | |
|   end
 | |
| 
 | |
|   def api_keys_url
 | |
|     return URL_API_KEYS_DOCS unless datadog_site.presence
 | |
| 
 | |
|     sprintf(URL_TEMPLATE_API_KEYS, datadog_site: datadog_site)
 | |
|   end
 | |
| 
 | |
|   def execute(data)
 | |
|     return if project.disabled_services.include?(to_param)
 | |
| 
 | |
|     object_kind = data[:object_kind]
 | |
|     object_kind = 'job' if object_kind == 'build'
 | |
|     return unless supported_events.include?(object_kind)
 | |
| 
 | |
|     service_hook.execute(data, "#{object_kind} hook")
 | |
|   end
 | |
| 
 | |
|   def test(data)
 | |
|     begin
 | |
|       result = execute(data)
 | |
|       return { success: false, result: result[:message] } if result[:http_status] != 200
 | |
|     rescue StandardError => error
 | |
|       return { success: false, result: error }
 | |
|     end
 | |
| 
 | |
|     { success: true, result: result[:message] }
 | |
|   end
 | |
| end
 |