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|
        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

4 entries across 4 versions & 1 rubygems

Version	Path
posgra-0.1.6	lib/posgra/cli/grant.rb
posgra-0.1.5	lib/posgra/cli/grant.rb
posgra-0.1.4	lib/posgra/cli/grant.rb
posgra-0.1.3	lib/posgra/cli/grant.rb