debian-mirror-gitlab/lib/gitlab/mail_room.rb

106 lines
2.8 KiB
Ruby
Raw Normal View History

2018-12-13 13:39:08 +05:30
# frozen_string_literal: true
2016-09-13 17:45:13 +05:30
require 'yaml'
require 'json'
2020-03-13 15:44:24 +05:30
require 'pathname'
2017-09-10 17:25:29 +05:30
require_relative 'redis/queues' unless defined?(Gitlab::Redis::Queues)
2016-09-13 17:45:13 +05:30
2020-01-01 13:55:28 +05:30
# This service is run independently of the main Rails process,
# therefore the `Rails` class and its methods are unavailable.
2016-09-13 17:45:13 +05:30
module Gitlab
module MailRoom
2020-01-01 13:55:28 +05:30
RAILS_ROOT_DIR = Pathname.new('../..').expand_path(__dir__).freeze
2018-03-17 18:26:18 +05:30
DEFAULT_CONFIG = {
enabled: false,
port: 143,
ssl: false,
start_tls: false,
mailbox: 'inbox',
2020-01-01 13:55:28 +05:30
idle_timeout: 60,
2020-05-24 23:13:21 +05:30
log_path: RAILS_ROOT_DIR.join('log', 'mail_room_json.log'),
expunge_deleted: false
2018-03-17 18:26:18 +05:30
}.freeze
2020-03-13 15:44:24 +05:30
# Email specific configuration which is merged with configuration
# fetched from YML config file.
2022-03-02 08:16:31 +05:30
MAILBOX_SPECIFIC_CONFIGS = {
2020-03-13 15:44:24 +05:30
incoming_email: {
queue: 'email_receiver',
worker: 'EmailReceiverWorker'
},
service_desk_email: {
queue: 'service_desk_email_receiver',
worker: 'ServiceDeskEmailReceiverWorker'
}
}.freeze
2016-09-13 17:45:13 +05:30
class << self
2020-03-13 15:44:24 +05:30
def enabled_configs
2022-03-02 08:16:31 +05:30
@enabled_configs ||= configs.select { |_key, config| enabled?(config) }
end
def enabled_mailbox_types
enabled_configs.keys.map(&:to_s)
end
def worker_for(mailbox_type)
MAILBOX_SPECIFIC_CONFIGS.try(:[], mailbox_type.to_sym).try(:[], :worker).try(:safe_constantize)
2016-09-13 17:45:13 +05:30
end
2020-03-13 15:44:24 +05:30
private
2016-09-13 17:45:13 +05:30
2020-03-13 15:44:24 +05:30
def enabled?(config)
config[:enabled] && !config[:address].to_s.empty?
2016-09-13 17:45:13 +05:30
end
2020-03-13 15:44:24 +05:30
def configs
2022-03-02 08:16:31 +05:30
MAILBOX_SPECIFIC_CONFIGS.to_h { |key, _value| [key, fetch_config(key)] }
2020-03-13 15:44:24 +05:30
end
2016-09-13 17:45:13 +05:30
2020-03-13 15:44:24 +05:30
def fetch_config(config_key)
2016-09-13 17:45:13 +05:30
return {} unless File.exist?(config_file)
2020-03-13 15:44:24 +05:30
config = merged_configs(config_key)
config.merge!(redis_config) if enabled?(config)
config[:log_path] = File.expand_path(config[:log_path], RAILS_ROOT_DIR)
config
end
def merged_configs(config_key)
yml_config = load_yaml.fetch(config_key, {})
2022-03-02 08:16:31 +05:30
specific_config = MAILBOX_SPECIFIC_CONFIGS.fetch(config_key, {})
2020-03-13 15:44:24 +05:30
DEFAULT_CONFIG.merge(specific_config, yml_config) do |_key, oldval, newval|
2018-03-17 18:26:18 +05:30
newval.nil? ? oldval : newval
end
2020-03-13 15:44:24 +05:30
end
2016-09-13 17:45:13 +05:30
2020-03-13 15:44:24 +05:30
def redis_config
gitlab_redis_queues = Gitlab::Redis::Queues.new(rails_env)
2021-11-18 22:05:49 +05:30
config = { redis_url: gitlab_redis_queues.url, redis_db: gitlab_redis_queues.db }
2017-08-17 22:00:37 +05:30
2020-03-13 15:44:24 +05:30
if gitlab_redis_queues.sentinels?
config[:sentinels] = gitlab_redis_queues.sentinels
2016-09-13 17:45:13 +05:30
end
config
end
2018-03-17 18:26:18 +05:30
def rails_env
@rails_env ||= ENV['RAILS_ENV'] || ENV['RACK_ENV'] || 'development'
end
2016-09-13 17:45:13 +05:30
def config_file
2018-11-18 11:00:15 +05:30
ENV['MAIL_ROOM_GITLAB_CONFIG_FILE'] || File.expand_path('../../config/gitlab.yml', __dir__)
2016-09-13 17:45:13 +05:30
end
2020-01-01 13:55:28 +05:30
2020-03-13 15:44:24 +05:30
def load_yaml
@yaml ||= YAML.load_file(config_file)[rails_env].deep_symbolize_keys
2020-01-01 13:55:28 +05:30
end
2016-09-13 17:45:13 +05:30
end
end
end