3rdparty/modules/postgresql/manifests/psql.pp

   1 # puppet-postgresql
   2 # For all details and documentation:
   3 # http://github.com/inkling/puppet-postgresql
   4 #
   5 # Copyright 2012- Inkling Systems, Inc.
   6 #
   7 # Licensed under the Apache License, Version 2.0 (the "License");
   8 # you may not use this file except in compliance with the License.
   9 # You may obtain a copy of the License at
  10 #
  11 #     http://www.apache.org/licenses/LICENSE-2.0
  12 #
  13 # Unless required by applicable law or agreed to in writing, software
  14 # distributed under the License is distributed on an "AS IS" BASIS,
  15 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16 # See the License for the specific language governing permissions and
  17 # limitations under the License.
  18
  19 define postgresql::psql(
  20     $db,
  21     $unless,
  22     $command     = $title,
  23     $refreshonly = false,
  24     $user        = $postgresql::params::user
  25 ) {
  26
  27   include postgresql::params
  28
  29   # TODO: FIXME: shellquote does not work, and this regex works for trivial
  30   # things but not nested escaping.  Need a lexer, preferably a ruby SQL parser
  31   # to catch errors at catalog time.  Possibly https://github.com/omghax/sql ?
  32
  33   if ($postgresql::params::version != '8.1') {
  34     $no_password_option = '--no-password'
  35   }
  36
  37   $psql = "${postgresql::params::psql_path} ${no_password_option} --tuples-only --quiet --dbname ${db}"
  38
  39   $quoted_command = regsubst($command, '"', '\\"', 'G')
  40   $quoted_unless  = regsubst($unless,  '"', '\\"', 'G')
  41
  42   $final_cmd = "/bin/echo \"${quoted_command}\" | ${psql} |egrep -v -q '^$'"
  43
  44   notify { "deprecation warning: ${final_cmd}":
  45     message => 'postgresql::psql is deprecated ; please use postgresql_psql instead.',
  46   } ->
  47
  48   exec { $final_cmd:
  49     cwd         => '/tmp',
  50     user        => $user,
  51     returns     => 1,
  52     unless      => "/bin/echo \"${quoted_unless}\" | ${psql} | egrep -v -q '^$'",
  53     refreshonly => $refreshonly,
  54   }
  55 }
  56