/
files.rb
94 lines (81 loc) · 2.98 KB
/
files.rb
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
require 'puppet/module'
#require 'puppet/parser/parser'
# This is a silly central module for finding
# different kinds of files while parsing. This code
# doesn't really belong in the Puppet::Module class,
# but it doesn't really belong anywhere else, either.
module Puppet; module Parser; module Files
module_function
# Return a list of manifests as absolute filenames matching the given
# pattern.
#
# @param pattern [String] A reference for a file in a module. It is the format "<modulename>/<file glob>"
# @param environment [Puppet::Node::Environment] the environment of modules
#
# @return [Array(String, Array<String>)] the module name and the list of files found
# @api private
def find_manifests_in_modules(pattern, environment)
module_name, file_pattern = split_file_path(pattern)
begin
if mod = Puppet::Module.find(module_name, environment)
return [mod.name, mod.match_manifests(file_pattern)]
end
rescue Puppet::Module::InvalidName
# one of the modules being loaded might have an invalid name and so
# looking for one might blow up since we load them lazily.
end
[nil, []]
end
# Find the concrete file denoted by +file+. If +file+ is absolute,
# return it directly. Otherwise try to find relative to the +templatedir+
# config param. If that fails try to find it as a template in a
# module.
# In all cases, an absolute path is returned, which does not
# necessarily refer to an existing file
def find_template(template, environment = nil)
if template == File.expand_path(template)
return template
end
if template_paths = templatepath(environment)
# If we can find the template in :templatedir, we return that.
template_paths.collect { |path|
File::join(path, template)
}.each do |f|
return f if FileTest.exist?(f)
end
end
# check in the default template dir, if there is one
if td_file = find_template_in_module(template, environment)
return td_file
end
nil
end
def find_template_in_module(template, environment = nil)
path, file = split_file_path(template)
# Because templates don't have an assumed template name, like manifests do,
# we treat templates with no name as being templates in the main template
# directory.
return nil unless file
if mod = Puppet::Module.find(path, environment) and t = mod.template(file)
return t
end
nil
end
# Return an array of paths by splitting the +templatedir+ config
# parameter.
def templatepath(environment = nil)
dirs = Puppet.settings.value(:templatedir, environment).split(File::PATH_SEPARATOR)
dirs.select do |p|
File::directory?(p)
end
end
# Split the path into the module and the rest of the path, or return
# nil if the path is empty or absolute (starts with a /).
def split_file_path(path)
if path == "" or Puppet::Util.absolute_path?(path)
nil
else
path.split(File::SEPARATOR, 2)
end
end
end; end; end