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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
# frozen_string_literal: true
require 'yaml'
require_relative '../puppet'
require_relative '../puppet/node/environment'
require_relative '../puppet/file_system'
require_relative '../puppet/indirector'
module Puppet
module ApplicationSupport
# Pushes a Puppet Context configured with a remote environment for an agent
# (one that exists at the master end), and a regular environment for other
# modes. The configuration is overridden with options from the command line
# before being set in a pushed Puppet Context.
#
# @param run_mode [Puppet::Util::RunMode] Puppet's current Run Mode.
# @param environment_mode [Symbol] optional, Puppet's
# current Environment Mode. Defaults to :local
# @return [void]
# @api private
def self.push_application_context(run_mode, environment_mode = :local)
Puppet.push_context_global(Puppet.base_context(Puppet.settings), "Update for application settings (#{run_mode})")
# This use of configured environment is correct, this is used to establish
# the defaults for an application that does not override, or where an override
# has not been made from the command line.
#
configured_environment_name = Puppet[:environment]
if run_mode.name == :agent
configured_environment = Puppet::Node::Environment.remote(configured_environment_name)
elsif environment_mode == :not_required
configured_environment =
Puppet.lookup(:environments).get(configured_environment_name) || Puppet::Node::Environment.remote(configured_environment_name)
else
configured_environment = Puppet.lookup(:environments).get!(configured_environment_name)
end
configured_environment = configured_environment.override_from_commandline(Puppet.settings)
# Setup a new context using the app's configuration
Puppet.push_context({ :current_environment => configured_environment },
"Update current environment from application's configuration")
end
# Reads the routes YAML settings from the file specified by Puppet[:route_file]
# and resets indirector termini for the current application class if listed.
#
# For instance, PE uses this to set the master facts terminus
# to 'puppetdb' and its cache terminus to 'yaml'.
#
# @param application_name [String] The name of the current application.
# @return [void]
# @api private
def self.configure_indirector_routes(application_name)
route_file = Puppet[:route_file]
if Puppet::FileSystem.exist?(route_file)
routes = Puppet::Util::Yaml.safe_load_file(route_file, [Symbol])
if routes["server"] && routes["master"]
Puppet.warning("Route file #{route_file} contains both server and master route settings.")
elsif routes["server"] && !routes["master"]
routes["master"] = routes["server"]
elsif routes["master"] && !routes["server"]
routes["server"] = routes["master"]
end
application_routes = routes[application_name]
Puppet::Indirector.configure_routes(application_routes) if application_routes
end
end
end
end
|
