Sha256: 39bff21e8c77c3ce6206da135201ad7e6315d83adf573c699b874020b63e3519
Contents?: true
Size: 1.69 KB
Versions: 7
Compression:
Stored size: 1.69 KB
Contents
class Posgra::CLI::Grant < Thor
include Posgra::CLI::Helper
include Posgra::Logger::Helper
DEFAULT_FILENAME = 'pg_grants.rb'
class_option :'include-schema'
class_option :'exclude-schema'
class_option :'include-role'
class_option :'exclude-role'
class_option :'include-object'
class_option :'exclude-object'
desc 'apply FILE', 'Apply grants'
option :'dry-run', :type => :boolean, :default => false
def apply(file)
check_fileanem(file)
updated = client.apply_grants(file)
unless updated
Posgra::Logger.instance.info('No change'.intense_blue)
end
end
desc 'export [FILE]', 'Export grants'
option :split, :type => :boolean, :default => false
def export(file = nil)
check_fileanem(file)
dsl = client.export_grants
if options[:split]
file = DEFAULT_FILENAME unless file
log(:info, 'Export Grants')
requires = []
dsl.each do |user, content|
user = user.gsub(/\s+/, '_')
user = '_' if user.empty?
grant_file = "#{user}.rb"
requires << grant_file
log(:info, " write `#{grant_file}`")
open(grant_file, 'wb') do |f|
f.puts Posgra::CLI::MAGIC_COMMENT
f.puts content
end
end
log(:info, " write `#{file}`")
open(file, 'wb') do |f|
f.puts Posgra::CLI::MAGIC_COMMENT
requires.each do |grant_file|
f.puts "require '#{File.basename grant_file}'"
end
end
else
if file.nil? or file == '-'
puts dsl
else
log(:info, "Export Grants to `#{file}`")
open(file, 'wb') do |f|
f.puts Posgra::CLI::MAGIC_COMMENT
f.puts dsl
end
end
end
end
end
Version data entries
7 entries across 7 versions & 1 rubygems