Add ability to specify windows install dir.

voxpupuli · Mar 1, 2016 · 53bda85 · 53bda85
1 parent 1ff6844
commit 53bda85
Show file tree

Hide file tree

Showing 7 changed files with 96 additions and 44 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,8 @@
+.*.sw?
+.bundle
+.pkg
+.ruby-version
+.vagrant
+Gemfile.lock
+modules/
+spec/fixtures
diff --git a/lib/puppet/provider/splunkforwarder_input/ini_setting.rb b/lib/puppet/provider/splunkforwarder_input/ini_setting.rb
@@ -3,13 +3,15 @@
   # set ini_setting as the parent provider
   :parent => Puppet::Type.type(:ini_setting).provider(:ruby)
 ) do
-  # hard code the file path (this allows purging)
+
+  def self.prefetch(resources)
+    catalog = resources[resources.keys.first].catalog
+    splunk_config = catalog.resources.find{|s| s.type == :splunk_config}
+    confdir = splunk_config['forwarder_confdir'] || raise(Puppet::Error, 'Unknown splunk forwarder confdir')
+    @file_path = File.join(confdir, 'inputs.conf')
+  end
+
   def self.file_path
-    case Facter.value(:osfamily)
-    when 'windows'
-      'C:\Program Files\SplunkUniversalForwarder\etc\system\local\inputs.conf'
-    else
-      '/opt/splunkforwarder/etc/system/local/inputs.conf'
-    end
+    @file_path
   end
 end
diff --git a/lib/puppet/provider/splunkforwarder_output/ini_setting.rb b/lib/puppet/provider/splunkforwarder_output/ini_setting.rb
@@ -3,13 +3,15 @@
   # set ini_setting as the parent provider
   :parent => Puppet::Type.type(:ini_setting).provider(:ruby)
 ) do
-  # hard code the file path (this allows purging)
+
+  def self.prefetch(resources)
+    catalog = resources[resources.keys.first].catalog
+    splunk_config = catalog.resources.find{|s| s.type == :splunk_config}
+    confdir = splunk_config['forwarder_confdir'] || raise(Puppet::Error, 'Unknown splunk forwarder confdir')
+    @file_path = File.join(confdir, 'outputs.conf')
+  end
+
   def self.file_path
-    case Facter.value(:osfamily)
-    when 'windows'
-      'C:\Program Files\SplunkUniversalForwarder\etc\system\local\outputs.conf'
-    else
-      '/opt/splunkforwarder/etc/system/local/outputs.conf'
-    end
+    @file_path
   end
 end
diff --git a/lib/puppet/provider/splunkforwarder_props/ini_setting.rb b/lib/puppet/provider/splunkforwarder_props/ini_setting.rb
@@ -3,13 +3,15 @@
   # set ini_setting as the parent provider
   :parent => Puppet::Type.type(:ini_setting).provider(:ruby)
 ) do
-  # hard code the file path (this allows purging)
+
+  def self.prefetch(resources)
+    catalog = resources[resources.keys.first].catalog
+    splunk_config = catalog.resources.find{|s| s.type == :splunk_config}
+    confdir = splunk_config['forwarder_confdir'] || raise(Puppet::Error, 'Unknown splunk forwarder confdir')
+    @file_path = File.join(confdir, 'props.conf')
+  end
+
   def self.file_path
-    case Facter.value(:osfamily)
-    when 'windows'
-      'C:\Program Files\SplunkUniversalForwarder\etc\system\local\props.conf'
-    else
-      '/opt/splunkforwarder/etc/system/local/props.conf'
-    end
+    @file_path
   end
 end
diff --git a/lib/puppet/provider/splunkforwarder_transforms/ini_setting.rb b/lib/puppet/provider/splunkforwarder_transforms/ini_setting.rb
@@ -3,13 +3,15 @@
   # set ini_setting as the parent provider
   :parent => Puppet::Type.type(:ini_setting).provider(:ruby)
 ) do
-  # hard code the file path (this allows purging)
+
+  def self.prefetch(resources)
+    catalog = resources[resources.keys.first].catalog
+    splunk_config = catalog.resources.find{|s| s.type == :splunk_config}
+    confdir = splunk_config['forwarder_confdir'] || raise(Puppet::Error, 'Unknown splunk forwarder confdir')
+    @file_path = File.join(confdir, 'transforms.conf')
+  end
+
   def self.file_path
-    case Facter.value(:osfamily)
-    when 'windows'
-      'C:\Program Files\SplunkUniversalForwarder\etc\system\local\transforms.conf'
-    else
-      '/opt/splunkforwarder/etc/system/local/transforms.conf'
-    end
+    @file_path
   end
 end
diff --git a/lib/puppet/type/splunk_config.rb b/lib/puppet/type/splunk_config.rb
@@ -0,0 +1,17 @@
+Puppet::Type.newtype(:splunk_config) do
+  newparam(:name, :namevar => true) do
+    desc 'splunk config'
+  end
+
+  newparam(:forwarder_installdir) do
+  end
+
+  newparam(:forwarder_confdir) do
+  end
+
+  newparam(:server_installdir) do
+  end
+
+  newparam(:server_confdir) do
+  end
+end
diff --git a/manifests/params.pp b/manifests/params.pp
@@ -67,12 +67,14 @@
 # Requires: nothing
 #
 class splunk::params (
-  $version      = '5.0.5',
-  $build        = '179365',
-  $src_root     = 'puppet:///modules/splunk',
-  $splunkd_port = '8089',
-  $logging_port = '9997',
-  $server       = 'splunk'
+  $version              = '5.0.5',
+  $build                = '179365',
+  $src_root             = 'puppet:///modules/splunk',
+  $splunkd_port         = '8089',
+  $logging_port         = '9997',
+  $server               = 'splunk'
+  $forwarder_installdir = undef,
+  $server_installdir    = undef,
 ) {
 
   # Based on the small number of inputs above, we can construct sane defaults
@@ -85,40 +87,48 @@
   $password_content = ':admin:$6$pIE/xAyP9mvBaewv$4GYFxC0SqonT6/x8qGcZXVCRLUVKODj9drDjdu/JJQ/Iw0Gg.aTkFzCjNAbaK4zcCHbphFz1g1HK18Z2bI92M0::Administrator:admin:changeme@example.com::'
 
 
+  if $::osfamily == 'Windows' {
+    $forwarder_dir = pick($forwarder_installdir, 'C:/Program Files/SplunkUniversalForwarder')
+    $server_dir    = pick($server_installdir, 'C:/Program Files/Splunk')
+  } else {
+    $forwarder_dir = pick($forwarder_installdir, '/opt/splunkforwarder')
+    $server_dir    = pick($server_installdir, '/opt/splunk')
+  }
+
   # Settings common to a kernel
   case $::kernel {
     'Linux': {
       $path_delimiter       = '/'
       $forwarder_src_subdir = 'universalforwarder/linux'
       $forwarder_service    = [ 'splunk' ]
-      $password_config_file = '/opt/splunkforwarder/etc/passwd'
-      $forwarder_confdir    = '/opt/splunkforwarder/etc/system/local'
-      $secret_file          =  '/opt/splunkforwarder/etc/splunk.secret'
+      $password_config_file = "${forwarder_dir}/etc/passwd"
+      $secret_file          = "${forwarder_dir}/etc/splunk.secret"
+      $forwarder_confdir    = "${forwarder_dir}/etc/system/local"
       $server_src_subdir    = 'splunk/linux'
       $server_service       = [ 'splunk', 'splunkd', 'splunkweb' ]
-      $server_confdir       = '/opt/splunk/etc/system/local'
+      $server_confdir       = "${server_dir}/etc/system/local"
     }
     'SunOS': {
       $path_delimiter       = '/'
       $forwarder_src_subdir = 'universalforwarder/solaris'
       $forwarder_service    = [ 'splunk' ]
-      $password_config_file = '/opt/splunkforwarder/etc/passwd'
-      $forwarder_confdir    = '/opt/splunkforwarder/etc/system/local'
-      $secret_file          =  '/opt/splunkforwarder/etc/splunk.secret'
+      $password_config_file = "${forwarder_dir}/etc/passwd"
+      $secret_file          = "${forwarder_dir}/etc/splunk.secret"
+      $forwarder_confdir    = "${forwarder_dir}/etc/system/local"
       $server_src_subdir    = 'splunk/solaris'
       $server_service       = [ 'splunk', 'splunkd', 'splunkweb' ]
-      $server_confdir       = '/opt/splunk/etc/system/local'
+      $server_confdir       = "${server_dir}/etc/system/local"
     }
     'Windows': {
       $path_delimiter       = '\\'
       $forwarder_src_subdir = 'universalforwarder/windows'
       $password_config_file = 'C:/Program Files/SplunkUniversalForwarder/etc/passwd'
       $secret_file          =  'C:/Program Files/SplunkUniversalForwarder/etc/splunk.secret'
       $forwarder_service    = [ 'SplunkForwarder' ] # UNKNOWN
-      $forwarder_confdir    = 'C:/Program Files/SplunkUniversalForwarder/etc/system/local'
+      $forwarder_confdir    = "${forwarder_dir}/etc/system/local"
       $server_src_subdir    = 'splunk/windows'
       $server_service       = [ 'Splunkd', 'Splunkweb' ] # UNKNOWN
-      $server_confdir       = 'C:/Program Files/Splunk/etc/system/local' # UNKNOWN
+      $server_confdir       = "${server_dir}/etc/system/local"
       $forwarder_install_options = [
         'AGREETOLICENSE=Yes',
         'LAUNCHSPLUNK=0',
@@ -129,6 +139,7 @@
         'WINEVENTLOG_FWD_ENABLE=1',
         'WINEVENTLOG_SET_ENABLE=1',
         'ENABLEADMON=1',
+        "INSTALLDIR=\"${forwarder_dir}\"",
       ]
       $server_install_options = [
         'LAUNCHSPLUNK=1',
@@ -227,4 +238,12 @@
   $create_password   = true
 
   $forwarder_pkg_ensure = 'installed'
+
+  # A meta resource so providers know where splunk is installed:
+  splunk_config { 'splunk':
+    forwarder_installdir => $forwarder_dir,
+    forwarder_confdir    => $forwarder_confdir,
+    server_installdir    => $server_dir,
+    server_confdir       => $server_confdir,
+  }
 }