Merge pull request #57 from natemccurdy/config_version

Config Version Improvements

2 files changed, 5 insertions, 3 deletions

diff --git a/scripts/code_manager_config_version.rb b/scripts/code_manager_config_version.rb
index 1ef62af..beacabc 100755
--- a/scripts/code_manager_config_version.rb
+++ b/scripts/code_manager_config_version.rb
@@ -6,7 +6,8 @@ environmentpath = ARGV[0]
 environment     = ARGV[1]
 
 # Get the hostname of the Puppet master compiling the catalog.
-compiling_master = Socket.gethostname
+# Sometimes the hostname is the fqdn, so we'll take the first segment.
+compiling_master = Socket.gethostname.split('.').first
 
 # Get the path to the Code Manager deployment info file.
 r10k_deploy_file_path = File.join(environmentpath, environment, '.r10k-deploy.json')
diff --git a/scripts/config_version.rb b/scripts/config_version.rb
index 5aade34..3de7104 100755
--- a/scripts/config_version.rb
+++ b/scripts/config_version.rb
@@ -1,4 +1,4 @@
-#!/usr/bin/env ruby
+#!/opt/puppetlabs/puppet/bin/ruby
 begin
   require 'rugged'
   require 'socket'
@@ -10,7 +10,8 @@ else
   environment     = ARGV[1]
 
   # Get the hostname of the Puppet master compiling the catalog.
-  compiling_master = Socket.gethostname
+  # Sometimes the hostname is the fqdn, so we'll take the first segment.
+  compiling_master = Socket.gethostname.split('.').first
 
   # Get the path to the environment being compiled.
   repo = Rugged::Repository.discover(File.join(environmentpath, environment))