Merge pull request puppetlabs#138 from kbarber/ticket/master/128-provide_more_custom_parameters_for_custom_packaging

Ticket/master/128 provide more custom parameters for custom packaging

Author: cprice404

README.md

@@ -164,6 +164,39 @@ This will set the default database locale for all databases created with this mo
 ####`charset`
 This will set the default charset for all databases created with this module. On certain operating systems this will be used during the `template1` initialization as well so it becomes a default outside of the module as well. Defaults to `UTF8`.
 
+####`datadir`
+This setting can be used to override the default postgresql data directory for the target platform. If not specified, the module will use whatever directory is the default for your OS distro.
+
+####`confdir`
+This setting can be used to override the default postgresql configuration directory for the target platform. If not specified, the module will use whatever directory is the default for your OS distro.
+
+####`bindir`
+This setting can be used to override the default postgresql binaries directory for the target platform. If not specified, the module will use whatever directory is the default for your OS distro.
+
+####`client_package_name`
+This setting can be used to override the default postgresql client package name. If not specified, the module will use whatever package name is the default for your OS distro.
+
+####`server_package_name`
+This setting can be used to override the default postgresql server package name. If not specified, the module will use whatever package name is the default for your OS distro.
+
+####`devel_package_name`
+This setting can be used to override the default postgresql devel package name. If not specified, the module will use whatever package name is the default for your OS distro.
+
+####`java_package_name`
+This setting can be used to override the default postgresql java package name. If not specified, the module will use whatever package name is the default for your OS distro.
+
+####`service_name`
+This setting can be used to override the default postgresql service name. If not specified, the module will use whatever service name is the default for your OS distro.
+
+####`user`
+This setting can be used to override the default postgresql super user and owner of postgresql related files in the file system. If not specified, the module will use the user name 'postgres'.
+
+####`group`
+This setting can be used to override the default postgresql user group to be used for related files in the file system. If not specified, the module will use the group name 'postgres'.
+
+####`run_initdb`
+This setting can be used to explicitly call the initdb operation after server package is installed and before the postgresql service is started. If not specified, the module will decide whether to call initdb or not depending on your OS distro.
+
 ###Class: postgresql::server
 Here are the options that you can set in the `config_hash` parameter of `postgresql::server`:
 

lib/puppet/provider/postgresql_psql/ruby.rb

@@ -51,7 +51,8 @@ def run_unless_sql_command(sql)
   end
 
   def run_sql_command(sql)
-    command = %{psql #{"-d #{resource[:db]}" if resource[:db]} -t -c "#{sql.gsub('"', '\"')}"}
+    command = %{#{resource[:psql_path]} #{"-d #{resource[:db]}" if resource[:db]} -t -c "#{sql.gsub('"', '\"')}"}
+
     if resource[:cwd]
       Dir.chdir resource[:cwd] do
         Puppet::Util::SUIDManager.run_and_capture(command, resource[:psql_user], resource[:psql_group])

lib/puppet/type/postgresql_psql.rb

@@ -49,6 +49,11 @@ def sync(refreshing = false)
     desc "The name of the database to execute the SQL command against."
   end
 
+  newparam(:psql_path) do
+    desc "The path to psql executable."
+    defaultto("psql")
+  end
+
   newparam(:psql_user) do
     desc "The system user account under which the psql command should be executed."
     defaultto("postgres")

manifests/config/afterservice.pp

@@ -28,7 +28,7 @@
     #  for pg_hba.conf.
     exec { 'set_postgres_postgrespw':
         # This command works w/no password because we run it as postgres system user
-        command     => "psql -c \"ALTER ROLE postgres PASSWORD '${postgres_password}'\"",
+        command     => "psql -c \"ALTER ROLE ${postgresql::params::user} PASSWORD '${postgres_password}'\"",
         user        => $postgresql::params::user,
         group       => $postgresql::params::group,
         logoutput   => true,

manifests/database.pp

@@ -27,6 +27,13 @@
 ) {
   include postgresql::params
 
+  # Set the defaults for the postgresql_psql resource
+  Postgresql_psql {
+    psql_user    => $postgresql::params::user,
+    psql_group   => $postgresql::params::group,
+    psql_path    => $postgresql::params::psql_path,
+  }
+
   # Optionally set the locale switch. Older versions of createdb may not accept
   # --locale, so if the parameter is undefined its safer not to pass it.
   if ($postgresql::params::version != '8.1') {
@@ -52,23 +59,20 @@
   postgresql_psql { "Check for existence of db '${dbname}'":
     command => 'SELECT 1',
     unless  => "SELECT datname FROM pg_database WHERE datname='${dbname}'",
-    cwd     => $postgresql::params::datadir,
     require => Class['postgresql::server']
   } ~>
 
   exec { $createdb_command :
     refreshonly => true,
-    user        => 'postgres',
-    cwd         => $postgresql::params::datadir,
+    user        => $postgresql::params::user,
     logoutput   => on_failure,
   } ~>
 
   # This will prevent users from connecting to the database unless they've been
   #  granted privileges.
-  postgresql_psql {"REVOKE ${public_revoke_privilege} ON DATABASE ${dbname} FROM public":
-    db          => 'postgres',
+  postgresql_psql {"REVOKE ${public_revoke_privilege} ON DATABASE \"${dbname}\" FROM public":
+    db          => $postgresql::params::user,
     refreshonly => true,
-    cwd         => $postgresql::params::datadir,
   }
 
 }

manifests/database_grant.pp

@@ -25,16 +25,22 @@
 #  in the modules or not.
 
 define postgresql::database_grant(
-    # TODO: mysql supports an array of privileges here.  We should do that if we
-    #  port this to ruby.
-    $privilege,
-    $db,
-    $role,
-    $psql_db   = 'postgres',
-    $psql_user ='postgres'
+  # TODO: mysql supports an array of privileges here.  We should do that if we
+  #  port this to ruby.
+  $privilege,
+  $db,
+  $role,
+  $psql_db   = $postgresql::params::user,
+  $psql_user = $postgresql::params::user
 ) {
   include postgresql::params
 
+  Postgresql_psql {
+    psql_user    => $postgresql::params::user,
+    psql_group   => $postgresql::params::group,
+    psql_path    => $postgresql::params::psql_path,
+  }
+
   # TODO: FIXME: only works on databases, due to using has_database_privilege
 
   # TODO: this is a terrible hack; if they pass "ALL" as the desired privilege,
@@ -50,11 +56,9 @@
     default => $privilege,
   }
 
-  postgresql_psql {"GRANT ${privilege} ON database ${db} TO ${role}":
+  postgresql_psql {"GRANT ${privilege} ON database \"${db}\" TO \"${role}\"":
     db           => $psql_db,
     psql_user    => $psql_user,
     unless       => "SELECT 1 WHERE has_database_privilege('${role}', '${db}', '${unless_privilege}')",
-    cwd          => $postgresql::params::datadir,
   }
 }
-

manifests/database_user.pp

@@ -38,12 +38,12 @@
 #
 
 define postgresql::database_user(
-    $password_hash,
-    $createdb   = false,
-    $createrole = false,
-    $db         = 'postgres',
-    $superuser  = false,
-    $user       = $title
+  $password_hash,
+  $createdb   = false,
+  $createrole = false,
+  $db         = $postgresql::params::user,
+  $superuser  = false,
+  $user       = $title
 ) {
   postgresql::role { $user:
     db              => $db,

manifests/init.pp

@@ -10,6 +10,7 @@
 # For examples, see the files in the `tests` directory; in particular,
 # `/server-yum-postgresql-org.pp`.
 #
+# === Parameters
 #
 # [*version*]
 #    The postgresql version to install.  If not specified, the
@@ -26,15 +27,70 @@
 #    set to `true`.  It determines which package repository should
 #    be used to install the postgres packages.  Currently supported
 #    values include `yum.postgresql.org`.
-#
 # [*locale*]
 #    This setting defines the default locale for initdb and createdb
 #    commands. This default to 'undef' which is effectively 'C'.
 # [*charset*]
 #    Sets the default charset to be used for initdb and createdb.
 #    Defaults to 'UTF8'.
+# [*datadir*]
+#    This setting can be used to override the default postgresql
+#    data directory for the target platform. If not specified, the
+#    module will use whatever directory is the default for your
+#    OS distro.
+# [*confdir*]
+#    This setting can be used to override the default postgresql
+#    configuration directory for the target platform. If not
+#    specified, the module will use whatever directory is the
+#    default for your OS distro.
+# [*bindir*]
+#    This setting can be used to override the default postgresql
+#    binaries directory for the target platform. If not
+#    specified, the module will use whatever directory is the
+#    default for your OS distro.
+# [*client_package_name*]
+#    This setting can be used to override the default
+#    postgresql client package name. If not specified, the module
+#    will use whatever package name is the default for your
+#    OS distro.
+# [*server_package_name*]
+#    This setting can be used to override the default
+#    postgresql server package name. If not specified, the module
+#    will use whatever package name is the default for your
+#    OS distro.
+# [*devel_package_name*]
+#    This setting can be used to override the default
+#    postgresql devel package name. If not specified, the module
+#    will use whatever package name is the default for your
+#    OS distro.
+# [*java_package_name*]
+#    This setting can be used to override the default
+#    postgresql java package name. If not specified, the module
+#    will use whatever package name is the default for your
+#    OS distro.
+# [*service_name*]
+#    This setting can be used to override the default
+#    postgresql service name. If not specified, the module
+#    will use whatever service name is the default for your
+#    OS distro.
+# [*user*]
+#    This setting can be used to override the default
+#    postgresql super user and owner of postgresql related files
+#    in the file system. If not specified, the module will use
+#    the user name 'postgres'.
+# [*group*]
+#    This setting can be used to override the default
+#    postgresql user group to be used for related files
+#    in the file system. If not specified, the module will use
+#    the group name 'postgres'.
+# [*run_initdb*]
+#    This setting can be used to explicitly call the initdb
+#    operation after server package is installed and before
+#    the postgresql service is started. If not specified, the
+#    module will decide whether to call initdb or not depending
+#    on your OS distro.
 #
-# === Examples:
+# === Examples
 #
 #   class { 'postgresql':
 #     version               => '9.2',
@@ -47,13 +103,36 @@
   $manage_package_repo = false,
   $package_source      = undef,
   $locale              = undef,
-  $charset             = 'UTF8'
+  $charset             = 'UTF8',
+  $datadir             = undef,
+  $confdir             = undef,
+  $bindir              = undef,
+  $client_package_name = undef,
+  $server_package_name = undef,
+  $devel_package_name  = undef,
+  $java_package_name   = undef,
+  $service_name        = undef,
+  $user                = undef,
+  $group               = undef,
+  $run_initdb          = undef
 ) {
+
   class { 'postgresql::params':
-    version             => $version,
-    manage_package_repo => $manage_package_repo,
-    package_source      => $package_source,
-    locale              => $locale,
-    charset             => $charset,
+    version                    => $version,
+    manage_package_repo        => $manage_package_repo,
+    package_source             => $package_source,
+    locale                     => $locale,
+    charset                    => $charset,
+    custom_datadir             => $datadir,
+    custom_confdir             => $confdir,
+    custom_bindir              => $bindir,
+    custom_client_package_name => $client_package_name,
+    custom_server_package_name => $server_package_name,
+    custom_devel_package_name  => $devel_package_name,
+    custom_java_package_name   => $java_package_name,
+    custom_service_name        => $service_name,
+    custom_user                => $user,
+    custom_group               => $group,
+    run_initdb                 => $run_initdb,
   }
 }

manifests/initdb.pp

@@ -19,9 +19,9 @@
 class postgresql::initdb(
   $datadir     = $postgresql::params::datadir,
   $encoding    = $postgresql::params::charset,
-  $group       = 'postgres',
+  $group       = $postgresql::params::group,
   $initdb_path = $postgresql::params::initdb_path,
-  $user        = 'postgres'
+  $user        = $postgresql::params::user
 ) inherits postgresql::params {
   # Build up the initdb command.
   #