125 lines
3.2 KiB
Ruby
125 lines
3.2 KiB
Ruby
class Mrsk::Configuration::Role
|
|
delegate :argumentize, :argumentize_env_with_secrets, :optionize, to: Mrsk::Utils
|
|
|
|
attr_accessor :name
|
|
|
|
def initialize(name, config:)
|
|
@name, @config = name.inquiry, config
|
|
end
|
|
|
|
def primary_host
|
|
hosts.first
|
|
end
|
|
|
|
def hosts
|
|
@hosts ||= extract_hosts_from_config
|
|
end
|
|
|
|
def labels
|
|
default_labels.merge(traefik_labels).merge(custom_labels)
|
|
end
|
|
|
|
def label_args
|
|
argumentize "--label", labels
|
|
end
|
|
|
|
def env
|
|
if config.env && config.env["secret"]
|
|
merged_env_with_secrets
|
|
else
|
|
merged_env
|
|
end
|
|
end
|
|
|
|
def env_args
|
|
argumentize_env_with_secrets env
|
|
end
|
|
|
|
def cmd
|
|
specializations["cmd"]
|
|
end
|
|
|
|
def option_args
|
|
if args = specializations["options"]
|
|
optionize args
|
|
else
|
|
[]
|
|
end
|
|
end
|
|
|
|
def running_traefik?
|
|
name.web? || specializations["traefik"]
|
|
end
|
|
|
|
private
|
|
attr_accessor :config
|
|
|
|
def extract_hosts_from_config
|
|
if config.servers.is_a?(Array)
|
|
config.servers
|
|
else
|
|
servers = config.servers[name]
|
|
servers.is_a?(Array) ? servers : Array(servers["hosts"])
|
|
end
|
|
end
|
|
|
|
def default_labels
|
|
if config.destination
|
|
{ "service" => config.service, "role" => name, "destination" => config.destination }
|
|
else
|
|
{ "service" => config.service, "role" => name }
|
|
end
|
|
end
|
|
|
|
def traefik_labels
|
|
if running_traefik?
|
|
{
|
|
"traefik.http.routers.#{config.service}.rule" => "PathPrefix(`/`)",
|
|
"traefik.http.services.#{config.service}.loadbalancer.healthcheck.path" => config.healthcheck["path"],
|
|
"traefik.http.services.#{config.service}.loadbalancer.healthcheck.interval" => "1s",
|
|
"traefik.http.middlewares.#{config.service}-retry.retry.attempts" => "5",
|
|
"traefik.http.middlewares.#{config.service}-retry.retry.initialinterval" => "500ms",
|
|
"traefik.http.routers.#{config.service}.middlewares" => "#{config.service}-retry@docker"
|
|
}
|
|
else
|
|
{}
|
|
end
|
|
end
|
|
|
|
def custom_labels
|
|
Hash.new.tap do |labels|
|
|
labels.merge!(config.labels) if config.labels.present?
|
|
labels.merge!(specializations["labels"]) if specializations["labels"].present?
|
|
end
|
|
end
|
|
|
|
def specializations
|
|
if config.servers.is_a?(Array) || config.servers[name].is_a?(Array)
|
|
{ }
|
|
else
|
|
config.servers[name].except("hosts")
|
|
end
|
|
end
|
|
|
|
def specialized_env
|
|
specializations["env"] || {}
|
|
end
|
|
|
|
def merged_env
|
|
config.env&.merge(specialized_env) || {}
|
|
end
|
|
|
|
# Secrets are stored in an array, which won't merge by default, so have to do it by hand.
|
|
def merged_env_with_secrets
|
|
merged_env.tap do |new_env|
|
|
new_env["secret"] = Array(config.env["secret"]) + Array(specialized_env["secret"])
|
|
|
|
# If there's no secret/clear split, everything is clear
|
|
clear_app_env = config.env["secret"] ? Array(config.env["clear"]) : Array(config.env["clear"] || config.env)
|
|
clear_role_env = specialized_env["secret"] ? Array(specialized_env["clear"]) : Array(specialized_env["clear"] || specialized_env)
|
|
|
|
new_env["clear"] = (clear_app_env + clear_role_env).uniq
|
|
end
|
|
end
|
|
end
|