15 lines
421 B
Ruby
15 lines
421 B
Ruby
|
# Custom function to extract the index from a list.
|
||
|
|
# The list are a list of hostname, and the index is the n'th
|
||
|
|
# position of the host in list
|
||
|
|
module Puppet::Parser::Functions
|
||
|
|
newfunction(:extract_id, :type => :rvalue) do |argv|
|
||
|
|
hosts = argv[0]
|
||
|
|
if hosts.class != Array
|
||
|
|
hosts = [hosts]
|
||
|
|
end
|
||
|
|
hostname = argv[1]
|
||
|
|
hash = Hash[hosts.map.with_index.to_a]
|
||
|
|
return hash[hostname].to_i + 1
|
||
|
|
end
|
||
|
|
end
|