freighthop 0.0.6 → 0.1.0

data/modules/postgresql/files/validate_postgresql_connection.sh

@@ -0,0 +1,31 @@
+#!/usr/bin/env bash
+
+# usage is: validate_db_connection 2 50 psql
+
+SLEEP=$1
+TRIES=$2
+PSQL=$3
+
+STATE=1
+
+for (( c=1; c<=$TRIES; c++ ))
+do
+  echo $c
+  if [ $c -gt 1 ]
+  then
+    echo 'sleeping'
+    sleep $SLEEP
+  fi
+
+  /bin/echo "SELECT 1" | $PSQL
+  STATE=$?
+
+  if [ $STATE -eq 0 ]
+  then
+    exit 0
+  fi
+done
+
+echo 'Unable to connect to postgresql'
+
+exit 1

data/modules/postgresql/lib/puppet/provider/postgresql_conf/parsed.rb

@@ -0,0 +1,37 @@
+require 'puppet/provider/parsedfile'
+
+Puppet::Type.type(:postgresql_conf).provide(
+  :parsed,
+  :parent => Puppet::Provider::ParsedFile,
+  :default_target => '/etc/postgresql.conf',
+  :filetype => :flat
+) do
+  desc "Set key/values in postgresql.conf."
+
+  text_line :comment, :match => /^\s*#/
+  text_line :blank, :match => /^\s*$/
+
+  record_line :parsed,
+    :fields   => %w{name value comment},
+    :optional => %w{comment},
+    :match    => /^\s*([\w\.]+)\s*=?\s*(.*?)(?:\s*#\s*(.*))?\s*$/,
+    :to_line  => proc { |h|
+
+      # simple string and numeric values don't need to be enclosed in quotes
+      dontneedquote = h[:value].match(/^(\w+|[0-9.-]+)$/)
+      dontneedequal = h[:name].match(/^(include|include_if_exists)$/i)
+
+      str =  h[:name].downcase # normalize case
+      str += dontneedequal ? ' ' : ' = '
+      str += "'" unless dontneedquote && !dontneedequal
+      str += h[:value]
+      str += "'" unless dontneedquote && !dontneedequal
+      str += " # #{h[:comment]}" unless (h[:comment].nil? or h[:comment] == :absent)
+      str
+    },
+    :post_parse => proc { |h|
+      h[:name].downcase! # normalize case
+      h[:value].gsub!(/(^'|'$)/, '') # strip out quotes
+    }
+
+end

data/modules/postgresql/lib/puppet/provider/postgresql_psql/ruby.rb

@@ -51,6 +51,10 @@ Puppet::Type.type(:postgresql_psql).provide(:ruby) do
   end
 
   def run_sql_command(sql)
+    if resource[:search_path]
+      sql = "set search_path to #{Array(resource[:search_path]).join(',')}; #{sql}"
+    end
+
     command = [resource[:psql_path]]
     command.push("-d", resource[:db]) if resource[:db]
     command.push("-t", "-c", sql)

data/modules/postgresql/lib/puppet/type/postgresql_conf.rb

@@ -0,0 +1,31 @@
+module Puppet
+  newtype(:postgresql_conf) do
+
+    @doc = "This type allows puppet to manage postgresql.conf parameters."
+
+    ensurable
+
+    newparam(:name) do
+      desc "The postgresql parameter name to manage."
+      isnamevar
+
+      newvalues(/^[\w\.]+$/)
+    end
+
+    newproperty(:value) do
+      desc "The value to set for this parameter."
+    end
+
+    newproperty(:target) do
+      desc "The path to postgresql.conf"
+      defaultto {
+        if @resource.class.defaultprovider.ancestors.include?(Puppet::Provider::ParsedFile)
+          @resource.class.defaultprovider.default_target
+        else
+          nil
+        end
+      }
+    end
+
+  end
+end

data/modules/postgresql/lib/puppet/type/postgresql_psql.rb

@@ -49,6 +49,10 @@ Puppet::Type.newtype(:postgresql_psql) do
     desc "The name of the database to execute the SQL command against."
   end
 
+  newparam(:search_path) do
+    desc "The schema search path to use when executing the SQL command"
+  end
+
   newparam(:psql_path) do
     desc "The path to psql executable."
     defaultto("psql")

data/modules/postgresql/manifests/client.pp

@@ -1,23 +1,9 @@
-# Class: postgresql::client
-#
-#   This class installs postgresql client software.
-#
-#   *Note* don't forget to make sure to add any necessary yum or apt
-#   repositories if specifying a custom version.
-#
-# Parameters:
-#   [*package_name*]  - The name of the postgresql client package.
-#   [*ensure*] - the ensure parameter passed to the postgresql client package resource
-# Actions:
-#
-# Requires:
-#
-# Sample Usage:
-#
+# Install client cli tool. See README.md for more details.
 class postgresql::client (
   $package_name   = $postgresql::params::client_package_name,
   $package_ensure = 'present'
 ) inherits postgresql::params {
+  validate_string($package_name)
 
   package { 'postgresql-client':
     ensure  => $package_ensure,
@@ -25,4 +11,19 @@ class postgresql::client (
     tag     => 'postgresql',
   }
 
+  $file_ensure = $package_ensure ? {
+    'present' => 'file',
+    true      => 'file',
+    'absent'  => 'absent',
+    false     => 'absent',
+    default   => 'file',
+  }
+  file { "/usr/local/bin/validate_postgresql_connection.sh":
+    ensure => $file_ensure,
+    source => "puppet:///modules/postgresql/validate_postgresql_connection.sh",
+    owner  => 0,
+    group  => 0,
+    mode   => 0755,
+  }
+
 }

data/modules/postgresql/manifests/globals.pp

@@ -0,0 +1,95 @@
+# Class for setting cross-class global overrides. See README.md for more
+# details.
+class postgresql::globals (
+  $ensure               = undef,
+
+  $client_package_name  = undef,
+  $server_package_name  = undef,
+  $contrib_package_name = undef,
+  $devel_package_name   = undef,
+  $java_package_name    = undef,
+  $plperl_package_name  = undef,
+  $python_package_name  = undef,
+
+  $service_name         = undef,
+  $service_provider     = undef,
+  $service_status       = undef,
+  $default_database     = undef,
+
+  $initdb_path          = undef,
+  $createdb_path        = undef,
+  $psql_path            = undef,
+  $pg_hba_conf_path     = undef,
+  $postgresql_conf_path = undef,
+
+  $pg_hba_conf_defaults = undef,
+
+  $datadir              = undef,
+  $confdir              = undef,
+  $bindir               = undef,
+
+  $user                 = undef,
+  $group                = undef,
+
+  $version              = undef,
+
+  $needs_initdb         = undef,
+
+  $encoding             = undef,
+  $locale               = undef,
+
+  $manage_firewall      = undef,
+  $manage_pg_hba_conf   = undef,
+  $firewall_supported   = undef,
+
+  $manage_package_repo  = undef
+) {
+  # We are determining this here, because it is needed by the package repo
+  # class.
+  $default_version = $::osfamily ? {
+    /^(RedHat|Linux)/ => $::operatingsystem ? {
+      'Fedora' => $::operatingsystemrelease ? {
+        /^(18|19|20)$/ => '9.2',
+        /^(17)$/ => '9.1',
+        default => undef,
+      },
+      'Amazon' => '9.2',
+      default => $::operatingsystemrelease ? {
+        /^6\./ => '8.4',
+        /^5\./ => '8.1',
+        default => undef,
+      },
+    },
+    'Debian' => $::operatingsystem ? {
+      'Debian' => $::operatingsystemrelease ? {
+        /^6\./ => '8.4',
+        /^(wheezy|7\.)/ => '9.1',
+        default => undef,
+      },
+      'Ubuntu' => $::operatingsystemrelease ? {
+        /^(14.04)$/ => '9.3',
+        /^(11.10|12.04|12.10|13.04|13.10)$/ => '9.1',
+        /^(10.04|10.10|11.04)$/ => '8.4',
+        default => undef,
+      },
+      default => undef,
+    },
+    'Archlinux' => $::operatingsystem ? {
+      /Archlinux/ => '9.2',
+      default => '9.2',
+    },
+    default => undef,
+  }
+  $globals_version = pick($version, $default_version, 'unknown')
+  if($globals_version == 'unknown') {
+    fail('No preferred version defined or automatically detected.')
+  }
+
+  # Setup of the repo only makes sense globally, so we are doing this here.
+  if($manage_package_repo) {
+    class { 'postgresql::repo':
+      ensure  => $ensure,
+      version => $globals_version
+    }
+  }
+}

data/modules/postgresql/manifests/lib/devel.pp

@@ -0,0 +1,15 @@
+# This class installs postgresql development libraries. See README.md for more
+# details.
+class postgresql::lib::devel(
+  $package_name   = $postgresql::params::devel_package_name,
+  $package_ensure = 'present'
+) inherits postgresql::params {
+
+  validate_string($package_name)
+
+  package { 'postgresql-devel':
+    ensure => $package_ensure,
+    name   => $package_name,
+    tag    => 'postgresql',
+  }
+}

data/modules/postgresql/manifests/lib/java.pp

@@ -0,0 +1,15 @@
+# This class installs the postgresql jdbc connector. See README.md for more
+# details.
+class postgresql::lib::java (
+  $package_name   = $postgresql::params::java_package_name,
+  $package_ensure = 'present'
+) inherits postgresql::params {
+
+  validate_string($package_name)
+
+  package { 'postgresql-jdbc':
+    ensure => $package_ensure,
+    name   => $package_name,
+  }
+
+}

data/modules/postgresql/manifests/lib/python.pp

@@ -0,0 +1,13 @@
+# This class installs the python libs for postgresql. See README.md for more
+# details.
+class postgresql::lib::python(
+  $package_name   = $postgresql::params::python_package_name,
+  $package_ensure = 'present'
+) inherits postgresql::params {
+
+  package { 'python-psycopg2':
+    ensure => $package_ensure,
+    name   => $package_name,
+  }
+
+}