Class: Ditto::Options

Inherits:
OpenStruct
  • Object
show all
Defined in:
lib/ditto/options.rb

Constant Summary collapse

DEFAULTS = 
{
  :connstring => ENV['DATABASE_URL'] || 'nodb://a/b',
  :entitydir  => 'ditto',
  :dittomart  => ENV['DITTO_MART'],
  :debug     => false,
  :check     => false,
  :droptables => false,
  :verbose    => 0,
  :loadfiles  => nil
}

Instance Method Summary collapse

Constructor Details

#initialize(argv) ⇒ Options

Returns a new instance of Options.



18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
 
# File 'lib/ditto/options.rb', line 18

def initialize argv
  super(DEFAULTS)

  Ditto::Options::DEFAULTS.each do |k,v|
  self.send "#{k}=".to_s, v
  end

  OptionParser.new do |opts|
  opts.banner = "Usage: ditto [ options ] files..."
  opts.separator "Files specified (in yaml format) will be loaded into database"
  opts.on("-c STRING", "--connection", String, "Database connection string",
    "Defaults to environment variable DATABASE_URL",
    "Current default is #{DEFAULTS[:connstring]}") do |s|
    self.connstring = s
  end
  opts.on("-e DIR", "--entitydir", String, "Directory to load Ditto entities from (default #{DEFAULTS[:entitydir]})") do |m|
    raise "Entity dir #{m} does not exist!" unless Dir.exist?(m)
    self.entitydir = m
  end
  opts.on("-m FILE", "--dittomart", String, "File containing Datamapper models for target Datamart (default #{DEFAULTS[:dittomart]})") do |m|
    self.dittomart = m
  end
  opts.on("-d", "--debug", "Debug") do |s|
    self.debug = true
  end
  opts.on("-v", "--verbose", "Verbose (repeat for more)") do
    self.verbose += 1
  end
  opts.on("--check", "Check given files, don't do anything!") do
    self.check = true
  end
  opts.on("--droptables", "Delete existing database tables!") do
    self.droptables = true
  end
  opts.on("-?", "-h", "--help", "Show this message") do
    puts opts
    exit
  end
  begin
    opts.parse!(argv)
  rescue OptionParser::ParseError => e
    STDERR.puts e.message, "\n", opts
    exit(-1)
  end
  end
  self.martfiles = atfile self.dittomart
  self.loadfiles = argv.map{ |m| atfile m}.flatten
  return self
end

Instance Method Details

#atfile(fn, dir = '.') ⇒ Object

Take a filename and expand it if it’s an @file relative paths are always relative to the including file return an array of absolute filenames

Raises:

  • (Errno::ENOENT) 


72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
 
# File 'lib/ditto/options.rb', line 72

def atfile fn, dir = '.'
  return [] unless fn
  af = fn.split('@',2)
  fn = af[1] if af.size == 2
  fn = File.expand_path(fn,dir)
  raise Errno::ENOENT, fn unless File.exist? fn
  return Array[fn] if af.size == 1
  dir = File.dirname File.absolute_path(fn)
  fh = File.open(fn)
  files = []
  fh.each_with_index do |line,ix|
  line.strip!
  begin
    files.push *atfile(line,dir) unless line =~ /(^\s*$|^\s*#)/
  rescue
    $!.message << " at '#{fn}:#{ix}'"
    raise
  end
  end
  return files
end