cookbooks/postgresql/resources/user.rb

   1 #
   2 # Cookbook:: postgresql
   3 # Resource:: postgresql_user
   4 #
   5 # Copyright:: 2012, OpenStreetMap Foundation
   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 # https://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
  20 require "shellwords"
  21
  22 unified_mode true
  23
  24 default_action :create
  25
  26 property :user, :kind_of => String, :name_property => true
  27 property :cluster, :kind_of => String, :required => true
  28 property :password, :kind_of => String
  29 property :superuser, :kind_of => [TrueClass, FalseClass], :default => false
  30 property :createdb, :kind_of => [TrueClass, FalseClass], :default => false
  31 property :createrole, :kind_of => [TrueClass, FalseClass], :default => false
  32 property :replication, :kind_of => [TrueClass, FalseClass], :default => false
  33 property :roles, :kind_of => [String, Array]
  34
  35 action :create do
  36   password = new_resource.password ? "ENCRYPTED PASSWORD '#{new_resource.password.shellescape}'" : ""
  37   superuser = new_resource.superuser ? "SUPERUSER" : "NOSUPERUSER"
  38   createdb = new_resource.createdb ? "CREATEDB" : "NOCREATEDB"
  39   createrole = new_resource.createrole ? "CREATEROLE" : "NOCREATEROLE"
  40   replication = new_resource.replication ? "REPLICATION" : "NOREPLICATION"
  41
  42   if !cluster.users.include?(new_resource.user)
  43     converge_by "create role #{new_resource.user}" do
  44       cluster.execute(:command => "CREATE ROLE \"#{new_resource.user}\" LOGIN #{password} #{superuser} #{createdb} #{createrole}")
  45     end
  46   else
  47     current_user = cluster.users[new_resource.user]
  48
  49     if new_resource.superuser != current_user[:superuser]
  50       converge_by "alter role #{new_resource.user}" do
  51         cluster.execute(:command => "ALTER ROLE \"#{new_resource.user}\" #{superuser}")
  52       end
  53     end
  54
  55     unless new_resource.superuser
  56       if new_resource.createdb != current_user[:createdb]
  57         converge_by "alter role #{new_resource.user}" do
  58           cluster.execute(:command => "ALTER ROLE \"#{new_resource.user}\" #{createdb}")
  59         end
  60       end
  61
  62       if new_resource.createrole != current_user[:createrole]
  63         converge_by "alter role #{new_resource.user}" do
  64           cluster.execute(:command => "ALTER ROLE \"#{new_resource.user}\" #{createrole}")
  65         end
  66       end
  67
  68       if new_resource.replication != current_user[:replication]
  69         converge_by "alter role #{new_resource.user}" do
  70           cluster.execute(:command => "ALTER ROLE \"#{new_resource.user}\" #{replication}")
  71         end
  72       end
  73     end
  74
  75     roles = Array(new_resource.roles)
  76
  77     roles.each do |role|
  78       next if current_user[:roles].include?(role)
  79
  80       converge_by "grant #{role} to #{new_resource.user}" do
  81         cluster.execute(:command => "GRANT \"#{role}\" TO \"#{new_resource.user}\"")
  82       end
  83     end
  84
  85     current_user[:roles].each do |role|
  86       next if roles.include?(role)
  87
  88       converge_by "revoke #{role} from #{new_resource.user}" do
  89         cluster.execute(:command => "REVOKE \"#{role}\" FROM \"#{new_resource.user}\"")
  90       end
  91     end
  92   end
  93 end
  94
  95 action :drop do
  96   if cluster.users.include?(new_resource.user)
  97     converge_by "drop role #{new_resource.user}" do
  98       cluster.execute(:command => "DROP ROLE \"#{new_resource.user}\"")
  99     end
 100   end
 101 end
 102
 103 action_class do
 104   def cluster
 105     @cluster ||= OpenStreetMap::PostgreSQL.new(new_resource.cluster)
 106   end
 107 end