Manage license file

This commit allows you to manage the license key file
associated with PE.  This file is managed in the same
way the r10k private key file is currently managed in
this tool per the request of @reidmv

Author: abrader

plans/action/install.pp

@@ -9,6 +9,11 @@
 #   over to the masters at /etc/puppetlabs/puppetserver/ssh/id-control_repo.rsa
 #   If the file does not exist the value will simply be supplied to the masters
 #
+# @param license_key
+#   The license key to use with Puppet Enterprise.  If this is a local file it
+#   will be copied over to the MoM at /etc/puppetlabs/license.key
+#   If the file does not exist the value will simply be supplied to the masters
+#
 # @param pe_conf_data
 #   Config data to plane into pe.conf when generated on all hosts, this can be
 #   used for tuning data etc.
@@ -36,6 +41,10 @@
   Optional[String]     $r10k_private_key_file    = undef,
   Optional[Peadm::Pem] $r10k_private_key_content = undef,
 
+  # License key
+  Optional[String]     $license_key_file    = undef,
+  Optional[String]     $license_key_content = undef,
+
   # Other
   String               $stagingdir   = '/tmp',
 ) {
@@ -106,6 +115,18 @@
     },
   }
 
+  $license_key = [
+    $license_key_file,
+    $license_key_content,
+  ].peadm::flatten_compact.size ? {
+    0 => undef, # no key data supplied
+    2 => fail('Must specify either one or neither of license_key_file and license_key_content; not both'),
+    1 => $license_key_file ? {
+      String  => file($license_key_file), # key file path supplied, read data from file
+      undef   => $license_key_content,    # key content supplied directly, use as-is
+    },
+  }
+
   $precheck_results = run_task('peadm::precheck', $all_targets)
   $platform = $precheck_results.first['platform'] # Assume the platform of the first result correct
 
@@ -221,6 +242,16 @@
     )
   }
 
+  if $license_key {
+    run_task('peadm::mkdir_p_file', [$master_target, $master_replica_target],
+      path    => '/etc/puppetlabs/license.key',
+      owner   => 'pe-puppet',
+      group   => 'pe-puppet',
+      mode    => '0400',
+      content => $license_key,
+    )
+  }
+
   # Configure autosigning for the puppetdb database hosts 'cause they need it
   $autosign_conf = $database_targets.reduce('') |$memo,$target| { "${target.name}\n${memo}" }
   run_task('peadm::mkdir_p_file', $master_target,

plans/provision.pp

@@ -27,6 +27,10 @@
   Optional[Peadm::Pem]              $r10k_private_key_content = undef,
   Optional[String]                  $deploy_environment       = undef,
 
+  # License Key
+  Optional[String]                  $license_key_file    = undef,
+  Optional[String]                  $license_key_content = undef,
+
   # Other
   Optional[String]                  $stagingdir = undef,
 ) {
@@ -54,6 +58,10 @@
     r10k_private_key_file          => $r10k_private_key_file,
     r10k_private_key_content       => $r10k_private_key_content,
 
+    # License Key
+    license_key_file               => $license_key_file,
+    license_key_content            => $license_key_content,
+
     # Other
     stagingdir                     => $stagingdir,
   )
@@ -81,3 +89,4 @@
   # Return a string banner reporting on what was done
   return([$install_result, $configure_result])
 }
+