Contents

#!/usr/bin/env ruby

require 'optparse'
require 'ostruct'
require 'pg_export'

LOGS_TIMESTAMPED = ->(severity, datetime, progname, message) { "#{datetime} #{Process.pid} TID-#{Thread.current.object_id.to_s(36)}#{progname} #{severity}: #{message}\n" }
LOGS_NORMAL = ->(_, _, _, message) { "#{message}\n" }

options = OpenStruct.new
option_parser = OptionParser.new do |opts|
  opts.banner = 'Usage: pg_export [options]'

  opts.on('-d', '--database DATABASE', '[Required] Name of the database to export') do |database|
    options.database = database
  end

  opts.on('-k', '--keep [KEEP]', Integer, '[Optional] Number of dump files to keep locally and on FTP (default: 10)') do |keep|
    options.keep = keep
  end

  opts.on('-t', '--timestamped', '[Optional] Enables log messages with timestamps') do
    options.timestamped = true
  end

  opts.on('-h', '--help', 'Show this message') do
    puts opts
    exit
  end

  opts.separator "\nSetting can be verified by running following commands:"

  opts.on('-c', '--configuration', 'Prints the configuration') do
    puts PgExport::Configuration.new.to_s
    exit
  end

  opts.on('-f', '--ftp', 'Tries connecting to FTP to verify the connection') do
    PgExport::Logging.logger.formatter = LOGS_NORMAL
    PgExport::FtpService.new(PgExport::Configuration.new.ftp_params)
    exit
  end
end

begin
  option_parser.parse!

  PgExport::Logging.logger.formatter = options.timestamped ? LOGS_TIMESTAMPED : LOGS_NORMAL

  PgExport.new do |config|
    config.database = options.database if options.database
    config.keep_dumps = options.keep if options.keep
    config.keep_ftp_dumps = options.keep if options.keep
  end.call
rescue OptionParser::MissingArgument, PgExport::InvalidConfigurationError => e
  puts "Error: #{e}"
  puts option_parser
rescue PgExport::DatabaseDoesNotExistError, PgExport::DependencyRequiredError => e
  puts e
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
pg_export-0.2.0	bin/pg_export