-require 'chef/mixin/command'
+require "chef/mixin/command"
class Chef
class PostgreSQL
# Extract the record data
lines.collect do |line|
record = {}
- fields.zip(line.split("|")) { |name,value| record[name.to_sym] = value }
+ fields.zip(line.split("|")) { |name, value| record[name.to_sym] = value }
record
end
end
def users
- @users ||= query("SELECT * FROM pg_user").inject({}) do |users,user|
+ @users ||= query("SELECT * FROM pg_user").each_with_object({}) do |user, users|
users[user[:usename]] = {
:superuser => user[:usesuper] == "t",
:createdb => user[:usercreatedb] == "t",
:createrole => user[:usecatupd] == "t",
:replication => user[:userepl] == "t"
}
- users
end
end
def databases
- @databases ||= query("SELECT d.datname, u.usename, d.encoding, d.datcollate, d.datctype FROM pg_database AS d INNER JOIN pg_user AS u ON d.datdba = u.usesysid").inject({}) do |databases,database|
+ @databases ||= query("SELECT d.datname, u.usename, d.encoding, d.datcollate, d.datctype FROM pg_database AS d INNER JOIN pg_user AS u ON d.datdba = u.usesysid").each_with_object({}) do |database, databases|
databases[database[:datname]] = {
:owner => database[:usename],
:encoding => database[:encoding],
:collate => database[:datcollate],
:ctype => database[:datctype]
}
- databases
end
end
def extensions(database)
@extensions ||= {}
- @extensions[database] ||= query("SELECT extname, extversion FROM pg_extension", :database => database).inject({}) do |extensions,extension|
+ @extensions[database] ||= query("SELECT extname, extversion FROM pg_extension", :database => database).each_with_object({}) do |extension, extensions|
extensions[extension[:extname]] = {
:version => extension[:extversion]
}
- databases
end
end
def tables(database)
@tables ||= {}
- @tables[database] ||= query("SELECT n.nspname, c.relname, u.usename, c.relacl FROM pg_class AS c INNER JOIN pg_user AS u ON c.relowner = u.usesysid INNER JOIN pg_namespace AS n ON c.relnamespace = n.oid", :database => database).inject({}) do |tables,table|
+ @tables[database] ||= query("SELECT n.nspname, c.relname, u.usename, c.relacl FROM pg_class AS c INNER JOIN pg_user AS u ON c.relowner = u.usesysid INNER JOIN pg_namespace AS n ON c.relnamespace = n.oid", :database => database).each_with_object({}) do |table, tables|
name = "#{table[:nspname]}.#{table[:relname]}"
tables[name] = {
:owner => table[:usename],
:permissions => parse_acl(table[:relacl] || "{}")
}
-
- tables
end
end
- private
+ private
def parse_acl(acl)
- acl.sub(/^\{(.*)\}$/, "\\1").split(",").inject({}) do |permissions, entry|
- entry = entry.sub(/^"(.*)"$/) { $1.gsub(/\\"/, '"') }.sub(/\/.*$/, "")
+ acl.sub(/^\{(.*)\}$/, "\\1").split(",").each_with_object({}) do |entry, permissions|
+ entry = entry.sub(/^"(.*)"$/) { Regexp.last_match[1].gsub(/\\"/, '"') }.sub(/\/.*$/, "")
user, privileges = entry.split("=")
user = user.sub(/^"(.*)"$/, "\\1")
"a" => :insert, "r" => :select, "w" => :update, "d" => :delete,
"D" => :truncate, "x" => :references, "t" => :trigger
}.values_at(*(privileges.chars)).compact
-
- permissions
end
end
end