50 lines
1.6 KiB
Ruby
Executable File
50 lines
1.6 KiB
Ruby
Executable File
#!/opt/puppetlabs/puppet/bin/ruby
|
|
|
|
require 'json'
|
|
|
|
param_environment_path = ARGV[0]
|
|
param_environment = ARGV[1]
|
|
|
|
if (param_environment_path && param_environment)
|
|
# Use the parameters, if passed.
|
|
environment_path = File.join(param_environment_path, param_environment)
|
|
else
|
|
# Or use the parent directory.
|
|
environment_path = File.dirname(File.expand_path(File.dirname(__FILE__)))
|
|
end
|
|
|
|
environment_r10k_deploy_file = File.join(environment_path, '.r10k-deploy.json')
|
|
environment_dot_git = File.join(environment_path, '.git')
|
|
pe_version_file = '/opt/puppetlabs/server/pe_version'
|
|
environment_version = nil
|
|
|
|
if (File.file?(environment_r10k_deploy_file))
|
|
# Use the r10k deploy file, if using Code Manager.
|
|
environment_version = JSON.parse(File.read(environment_r10k_deploy_file))['signature']
|
|
|
|
elsif (File.file?(pe_version_file))
|
|
# Or use rugged, if using Puppet Enterprise.
|
|
begin
|
|
require 'rugged'
|
|
rescue LoadError => e
|
|
# It's okay.
|
|
else
|
|
environment_version = Rugged::Repository.discover(environment_path).head.target_id
|
|
end
|
|
|
|
elsif (File.exist?(environment_dot_git))
|
|
# Or use git, just use git.
|
|
if (File.file?(environment_dot_git))
|
|
# It's a file, created via --separate-git-dir.
|
|
git_dir = File.read(environment_dot_git).split(': ')[1]
|
|
else
|
|
git_dir = environment_dot_git
|
|
end
|
|
git_rev_parse_head = %x( /usr/bin/git --git-dir #{git_dir} rev-parse HEAD 2>/dev/null )
|
|
environment_version = git_rev_parse_head if ($?.exitstatus == 0)
|
|
|
|
end
|
|
|
|
# Return the version, if found, or the current timestamp.
|
|
puts (environment_version || Time.now.to_i)
|