-
Notifications
You must be signed in to change notification settings - Fork 13
/
pgsql.rb
77 lines (63 loc) · 2.46 KB
/
pgsql.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
# Copyright 2011 Red Hat, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
require 'puppet/provider/package'
Puppet::Type.type(:pgsql_database).provide(:pgsql,
:parent => Puppet::Provider::Package) do
desc "Use pgsql as database."
# retrieve the current set of pgsql users
def self.instances
dbs = []
output = execute(['psql', '-Aqtl'], :failonfail => true, :uid => "postgres")
output.each do |line|
dbs << new( query_line_to_hash(line) )
end
return dbs
end
def self.query_line_to_hash(line)
fields = line.chomp.split('|')
{
:name => fields[0],
:owner => fields[1],
:ensure => :present
}
end
def query
result = {
:name => @resource[:name],
:owner => @resource[:owner],
:ensure => :absent
}
output = execute(['psql', '-Aqtc', "SELECT pg_database.datname, pg_user.usename FROM pg_database, pg_user WHERE pg_database.datname='#{resource[:name]} AND pg_user.usesysid = ( SELECT datdba FROM pg_database WHERE pg_database.datname='#{@resource[:name]}')" ], :failonfail => true, :uid => "postgres")
output.each do |line|
result = query_line_to_hash(line)
end
result
end
def create
execute(['createdb', '-q', '-O', "#{@resource.should(:owner)}", "#{@resource[:name]}"], :failonfail => true, :uid => "postgres")
end
def destroy
execute(['dropdb', '-q', "#{@resource[:name]}"], :failonfail => true, :uid => "postgres")
end
def exists?
output = execute(['psql', '-Aqtc', "SELECT datname FROM pg_database WHERE datname='#{@resource[:name]}'"], :failonfail => true, :uid => "postgres")
output.chomp.match(/^#{@resource[:name]}$/)
end
def owner
@property_hash[:owner]
end
def owner=(string)
execute(['psql', '-Aqtc', "UPDATE pg_database SET datdba=(SELECT oid FROM pg_roles WHERE rolname='#{string}') where datname='#{@resource[:name]}'"], :failonfail => true, :uid => "postgres")
end
end