cuke_profiles 0.0.2

Sign up to get free protection for your applications and to get access to all the features.
data/.gitignore ADDED
@@ -0,0 +1,5 @@
1
+ *.gem
2
+ .bundle
3
+ pkg/*
4
+ .idea
5
+ features/run.txt
data/.travis.yml ADDED
@@ -0,0 +1 @@
1
+ rvm: 1.9.2
data/Gemfile ADDED
@@ -0,0 +1,9 @@
1
+ source "http://rubygems.org"
2
+
3
+ group :development do
4
+ gem 'rake', '>= 0.9.2'
5
+ gem 'rspec'
6
+ gem 'rspec-expectations', :git => "git://github.com/rspec/rspec-expectations.git"
7
+ end
8
+
9
+ gemspec
data/Gemfile.lock ADDED
@@ -0,0 +1,36 @@
1
+ GIT
2
+ remote: git://github.com/rspec/rspec-expectations.git
3
+ revision: 0b63bf446ce84a613875417121980492571bfa90
4
+ specs:
5
+ rspec-expectations (2.6.0)
6
+ diff-lcs (~> 1.1.2)
7
+
8
+ PATH
9
+ remote: .
10
+ specs:
11
+ cuke_profiles (0.0.1)
12
+ gherkin
13
+
14
+ GEM
15
+ remote: http://rubygems.org/
16
+ specs:
17
+ diff-lcs (1.1.2)
18
+ gherkin (2.4.11)
19
+ json (>= 1.4.6)
20
+ json (1.5.3)
21
+ rake (0.9.2)
22
+ rspec (2.6.0)
23
+ rspec-core (~> 2.6.0)
24
+ rspec-expectations (~> 2.6.0)
25
+ rspec-mocks (~> 2.6.0)
26
+ rspec-core (2.6.4)
27
+ rspec-mocks (2.6.0)
28
+
29
+ PLATFORMS
30
+ ruby
31
+
32
+ DEPENDENCIES
33
+ cuke_profiles!
34
+ rake (>= 0.9.2)
35
+ rspec
36
+ rspec-expectations!
data/README.md ADDED
@@ -0,0 +1,103 @@
1
+ # Cuke Profiles [![Build Status](https://secure.travis-ci.org/jmerrifield/cuke_profiles.png)](http://travis-ci.org/jmerrifield/cuke_profiles)
2
+
3
+ Multiple invocations of your scenarios, without running Cucumber multiple times.
4
+
5
+ ## Why would I want that?
6
+
7
+ In order to run different variations of your tests e.g.
8
+
9
+ * Different browsers/devices (run all my tests in Chrome, Firefox, iPhone simulator etc)
10
+ * Different platforms (run all my tests against a Windows installation, Linux, OS X etc)
11
+ * Some types of multi-tenant scenarios (run all my tests against the Initrode deployment, the Acme deployment etc)
12
+ * Systems which have different, configurable, modes of operation (but behave the same externally) - got a system that can optionally use MySQL, Sqlite or Oracle as the data store?
13
+
14
+ ## Surely there's some way of doing that already?
15
+
16
+ The only way of 'iterating' over a scenario in Cucumber is to provide it with example rows. You certainly don't want a table of 'devices' under every single scenario.
17
+
18
+ ## Damn right. I can just call Cucumber multiple times from a Rake script though!
19
+
20
+ Absolutely. It's not without disadvantages though:
21
+
22
+ * Rake will naturally stop after the first failed command, so you have to code around that fact to have Rake ignore the cucumber failures, and fail at the end if any of the Cucumber runs had failures.
23
+ * No aggregation of results - you have to write your own result aggregation code to get an overall view of how all the variations of your tests did.
24
+ * When your Cucumber suite gets quite large, it can take a significant chunk of time for Cucumber to run your `env.rb` code, and load up all your support and feature files. That overhead, multiplied by the number of variations, can really add up.
25
+
26
+ ## So how does this gem work?
27
+
28
+ It's very simple. Cucumber can accept a file specifying a list of exact scenarios to run, in the format `filename:line_number`. This is used to re-run failed scenarios, usually in conjunction with the 'rerun' formatter. This gem takes advantage of that, to send Cucumber a long list of all your scenarios, repeated for each profile.
29
+
30
+ ## How do I use it?
31
+
32
+ First, install the gem: `gem install cuke_profiles`
33
+
34
+ ### Configuration - profiles.yml
35
+
36
+ In your `features` folder, create a YAML file to contain your profiles. Each one must have a set of included and excluded tags, e.g:
37
+
38
+ ```yaml
39
+ iphone:
40
+ :include_tags:
41
+ - ! '@mobile'
42
+ - ! '@iphone_only'
43
+ :exclude_tags: []
44
+ android:
45
+ :include_tags:
46
+ - ! '@mobile'
47
+ :exclude_tags:
48
+ - ! '@iphone_only'
49
+ ```
50
+
51
+ At present, you *must* specify all tags that are to be included. Excluded tags have priority over included ones (e.g. an `@iphone_only` scenario would not be included in the `android` profile, even if it had the `@mobile` tag.
52
+
53
+ ### Running it
54
+
55
+ `cd features`
56
+
57
+ With defaults:
58
+ `cuke_profiles`
59
+
60
+ To see other options:
61
+ `cuke_profiles -h`
62
+
63
+ ### What did it do?
64
+
65
+ First, it created a set of empty directories under the `profiles` folder - one for each profile that you defined. You should add this folder to your source-control ignore list (unless you're using Git which doesn't track empty folders).
66
+
67
+ Next, it created a run file for Cucumber. If you didn't specify a filename it will be called `run.txt`. Definitely add this to your source-control ignore list.
68
+
69
+ ### Run Cucumber with the run file instead of a features folder
70
+
71
+ `cucumber @run.txt`
72
+
73
+ Note we haven't told Cucumber about our `features` folder like we normally would. This also means it doesn't know about your `support` or `step_definitions` folders, so make sure to add those to the command-line as well.
74
+
75
+ ### Use the profiles inside your scenarios
76
+
77
+ In `env.rb`:
78
+
79
+ ```ruby
80
+ require 'cuke_profiles'
81
+ World(CukeProfiles::CucumberHelper)
82
+ ```
83
+
84
+ Create a 'Before' hook:
85
+
86
+ ```ruby
87
+ Before do |scenario|
88
+ profile = current_profile(scenario)
89
+ puts "I'm running under the '#{profile}' profile!"
90
+ end
91
+ ```
92
+
93
+ ### What should I do now I know which profile the scenario is running under?
94
+
95
+ You should `puts` the profile name to distinguish between instances of the scenario in the Cucumber output.
96
+
97
+ Depending upon the nature of your profiles, you might like to:
98
+
99
+ * Switch the browser driver to a device-specific one
100
+ * Change the URL you're running tests against
101
+ * Change the configuration of the system under test
102
+
103
+ Happy cuking!
data/Rakefile ADDED
@@ -0,0 +1,7 @@
1
+ require 'bundler/gem_tasks'
2
+ require 'rspec/core/rake_task'
3
+
4
+ desc "Run all specs"
5
+ RSpec::Core::RakeTask.new(:spec)
6
+
7
+ task :default => :spec
data/bin/cuke_profiles ADDED
@@ -0,0 +1,27 @@
1
+ #!/usr/bin/env ruby
2
+
3
+ $:.unshift File.dirname(__FILE__) + "/../lib"
4
+ require 'cuke_profiles'
5
+ require 'trollop'
6
+ require 'fileutils'
7
+
8
+ opts = Trollop::options do
9
+ opt :profiles, "Profiles file to use", :default => 'cuke_profiles.yml'
10
+ opt :out, "Name of runfile to write", :default => 'run.txt'
11
+ end
12
+
13
+ Trollop::die "Couldn't find profiles file '#{opts[:profiles]}'" unless File.exist?(opts[:profiles])
14
+ profiles = YAML.load(File.read(opts[:profiles]))
15
+
16
+ all_scenarios = []
17
+ features = CukeProfiles::CukeParser.parse_features(Dir.getwd)
18
+ profiles.each do |profile_name, profile|
19
+ profile_dir = File.join('profiles', profile_name, '..', '..')
20
+ FileUtils.mkdir_p profile_dir
21
+
22
+ CukeProfiles::ScenarioListGenerator.for_profile(features, profile).each do |scenario|
23
+ all_scenarios << File.join(profile_dir, scenario[:filename] + ":#{scenario[:line]}")
24
+ end
25
+ end
26
+
27
+ File.open(opts[:out], 'w') {|f| f.puts all_scenarios}
@@ -0,0 +1,21 @@
1
+ # -*- encoding: utf-8 -*-
2
+ $:.push File.expand_path("../lib", __FILE__)
3
+ require "cuke_profiles/version"
4
+
5
+ Gem::Specification.new do |s|
6
+ s.name = "cuke_profiles"
7
+ s.version = CukeProfiles::VERSION
8
+ s.authors = ["Jon Merrifield"]
9
+ s.email = ["jon@jmerrifield.com"]
10
+ s.homepage = "https://github.com/jmerrifield/cuke_profiles"
11
+ s.summary = %q{Multiple profiles for Cucumber features}
12
+ s.description = %q{Run your Cucumber features multiple times within one Cucumber invocation}
13
+
14
+ s.rubyforge_project = "cuke_profiles"
15
+
16
+ s.files = `git ls-files`.split("\n")
17
+ s.test_files = `git ls-files -- {test,spec,features}/*`.split("\n")
18
+ s.executables = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
19
+ s.require_paths = ["lib"]
20
+ s.add_dependency 'gherkin'
21
+ end
@@ -0,0 +1,10 @@
1
+ iphone:
2
+ :include_tags:
3
+ - ! '@tag2'
4
+ - ! '@tag8'
5
+ :exclude_tags: []
6
+ android:
7
+ :include_tags:
8
+ - ! '@tag2'
9
+ :exclude_tags:
10
+ - ! '@tag8'
@@ -0,0 +1,22 @@
1
+ @feature_tag1 @feature_tag2
2
+ Feature: My test feature with tags
3
+ In order to be awesome
4
+ As a cuker
5
+ I need to know how many cukes I have, and what colour they are
6
+
7
+ Background:
8
+ Given I have "10" cukes
9
+
10
+ Scenario: My scenario
11
+ When I check my cukes
12
+ Then I should have "10" cukes
13
+
14
+ Scenario Outline: My scenario outline
15
+ When I check my cukes
16
+ Then I should have '<cukes>' cukes
17
+ And I they should be '<colour>' in colour
18
+
19
+ Examples:
20
+ | cukes | colour |
21
+ | 10 | green |
22
+ | 11 | red |
@@ -0,0 +1,25 @@
1
+ Feature: I love cukes
2
+
3
+ @tag1 @tag2
4
+ Scenario: Scenario with tags
5
+ Given I have "10" cukes
6
+ When I check my cukes
7
+ Then I should have "10" cukes
8
+
9
+ @tag3 @tag4
10
+ Scenario Outline: Scenario outline with tags
11
+ Given I have "<cukes>" cukes
12
+
13
+ Examples:
14
+ | cukes |
15
+ | 10 |
16
+ | 11 |
17
+
18
+ @tag5 @tag6
19
+ Scenario Outline: Scenario outline with tags and tagged example sections
20
+ Given I have "<cukes>" cukes
21
+
22
+ @tag7 @tag8
23
+ Examples:
24
+ | cukes |
25
+ | 10 |
@@ -0,0 +1,3 @@
1
+ require File.dirname(__FILE__) + '/../../../lib/cuke_profiles'
2
+
3
+ World(CukeProfiles::CucumberHelper)
@@ -0,0 +1,4 @@
1
+ Before do |scenario|
2
+ profile = current_profile(scenario)
3
+ puts "This step is running under the '#{profile}' profile!"
4
+ end
@@ -0,0 +1,7 @@
1
+ $:.push File.expand_path("../cuke_profiles", __FILE__)
2
+
3
+ require 'version'
4
+ require 'cuke_parser'
5
+ require 'scenario_extracting_formatter'
6
+ require 'scenario_list_generator'
7
+ require 'cucumber_helper'
@@ -0,0 +1,13 @@
1
+ module CukeProfiles
2
+ module CucumberHelper
3
+ def current_profile(scenario)
4
+ if scenario.is_a? Cucumber::Ast::Scenario
5
+ scenario.feature.file
6
+ elsif scenario.is_a? Cucumber::Ast::OutlineTable::ExampleRow
7
+ scenario.scenario_outline.feature.file
8
+ else
9
+ raise "Can't deal with #{scenario.class.name}"
10
+ end[/profiles\/(.+?)\//, 1]
11
+ end
12
+ end
13
+ end
@@ -0,0 +1,21 @@
1
+ require 'gherkin'
2
+
3
+ module CukeProfiles
4
+ class CukeParser
5
+ class << self
6
+ def parse_features(dir)
7
+ scenarios = []
8
+ Dir.glob(File.join(dir, '**/*.feature')).each do |feature_file|
9
+ formatter = ScenarioExtractingFormatter.new
10
+ parser = Gherkin::Parser::Parser.new (formatter)
11
+ text = File.open(feature_file, 'r') { |f| f.read }
12
+ parser.parse(text, __FILE__, __LINE__-1)
13
+
14
+ scenarios << formatter.discovered_scenarios.each {|s| s[:filename] = File.basename(feature_file)}
15
+ end
16
+
17
+ scenarios.flatten
18
+ end
19
+ end
20
+ end
21
+ end
@@ -0,0 +1,40 @@
1
+ module CukeProfiles
2
+ class ScenarioExtractingFormatter
3
+ attr_reader :discovered_scenarios
4
+
5
+ def initialize
6
+ @discovered_scenarios = []
7
+ @feature_tags = []
8
+ end
9
+
10
+ def feature(feature)
11
+ @feature_tags = feature.tags.map { |t| t.name }
12
+ end
13
+
14
+ def scenario(scenario)
15
+ @discovered_scenarios << {
16
+ line: scenario.line,
17
+ tags: @feature_tags + scenario.tags.map { |t| t.name }
18
+ }
19
+ end
20
+
21
+ def scenario_outline(outline)
22
+ @outline_tags = outline.tags.map { |t| t.name }
23
+ end
24
+
25
+ def examples(examples)
26
+ examples.rows.each do |row|
27
+ next if row == examples.rows.first
28
+
29
+ @discovered_scenarios << {
30
+ line: row.line,
31
+ tags: @feature_tags + @outline_tags + examples.tags.map { |t| t.name }
32
+ }
33
+ end
34
+ end
35
+
36
+ def method_missing(sym, *args, &block)
37
+ # We don't care
38
+ end
39
+ end
40
+ end
@@ -0,0 +1,12 @@
1
+ module CukeProfiles
2
+ class ScenarioListGenerator
3
+ class << self
4
+ def for_profile(parsed_features, profile)
5
+ parsed_features.select do |f|
6
+ next if (f[:tags] & profile[:exclude_tags]).any?
7
+ (f[:tags] & profile[:include_tags]).any?
8
+ end
9
+ end
10
+ end
11
+ end
12
+ end
@@ -0,0 +1,3 @@
1
+ module CukeProfiles
2
+ VERSION = "0.0.2"
3
+ end
data/lib/trollop.rb ADDED
@@ -0,0 +1,782 @@
1
+ ## lib/trollop.rb -- trollop command-line processing library
2
+ ## Author:: William Morgan (mailto: wmorgan-trollop@masanjin.net)
3
+ ## Copyright:: Copyright 2007 William Morgan
4
+ ## License:: the same terms as ruby itself
5
+
6
+ require 'date'
7
+
8
+ module Trollop
9
+
10
+ VERSION = "1.16.2"
11
+
12
+ ## Thrown by Parser in the event of a commandline error. Not needed if
13
+ ## you're using the Trollop::options entry.
14
+ class CommandlineError < StandardError; end
15
+
16
+ ## Thrown by Parser if the user passes in '-h' or '--help'. Handled
17
+ ## automatically by Trollop#options.
18
+ class HelpNeeded < StandardError; end
19
+
20
+ ## Thrown by Parser if the user passes in '-h' or '--version'. Handled
21
+ ## automatically by Trollop#options.
22
+ class VersionNeeded < StandardError; end
23
+
24
+ ## Regex for floating point numbers
25
+ FLOAT_RE = /^-?((\d+(\.\d+)?)|(\.\d+))([eE][-+]?[\d]+)?$/
26
+
27
+ ## Regex for parameters
28
+ PARAM_RE = /^-(-|\.$|[^\d\.])/
29
+
30
+ ## The commandline parser. In typical usage, the methods in this class
31
+ ## will be handled internally by Trollop::options. In this case, only the
32
+ ## #opt, #banner and #version, #depends, and #conflicts methods will
33
+ ## typically be called.
34
+ ##
35
+ ## If you want to instantiate this class yourself (for more complicated
36
+ ## argument-parsing logic), call #parse to actually produce the output hash,
37
+ ## and consider calling it from within
38
+ ## Trollop::with_standard_exception_handling.
39
+ class Parser
40
+
41
+ ## The set of values that indicate a flag option when passed as the
42
+ ## +:type+ parameter of #opt.
43
+ FLAG_TYPES = [:flag, :bool, :boolean]
44
+
45
+ ## The set of values that indicate a single-parameter (normal) option when
46
+ ## passed as the +:type+ parameter of #opt.
47
+ ##
48
+ ## A value of +io+ corresponds to a readable IO resource, including
49
+ ## a filename, URI, or the strings 'stdin' or '-'.
50
+ SINGLE_ARG_TYPES = [:int, :integer, :string, :double, :float, :io, :date]
51
+
52
+ ## The set of values that indicate a multiple-parameter option (i.e., that
53
+ ## takes multiple space-separated values on the commandline) when passed as
54
+ ## the +:type+ parameter of #opt.
55
+ MULTI_ARG_TYPES = [:ints, :integers, :strings, :doubles, :floats, :ios, :dates]
56
+
57
+ ## The complete set of legal values for the +:type+ parameter of #opt.
58
+ TYPES = FLAG_TYPES + SINGLE_ARG_TYPES + MULTI_ARG_TYPES
59
+
60
+ INVALID_SHORT_ARG_REGEX = /[\d-]/ #:nodoc:
61
+
62
+ ## The values from the commandline that were not interpreted by #parse.
63
+ attr_reader :leftovers
64
+
65
+ ## The complete configuration hashes for each option. (Mainly useful
66
+ ## for testing.)
67
+ attr_reader :specs
68
+
69
+ ## Initializes the parser, and instance-evaluates any block given.
70
+ def initialize *a, &b
71
+ @version = nil
72
+ @leftovers = []
73
+ @specs = {}
74
+ @long = {}
75
+ @short = {}
76
+ @order = []
77
+ @constraints = []
78
+ @stop_words = []
79
+ @stop_on_unknown = false
80
+
81
+ #instance_eval(&b) if b # can't take arguments
82
+ cloaker(&b).bind(self).call(*a) if b
83
+ end
84
+
85
+ ## Define an option. +name+ is the option name, a unique identifier
86
+ ## for the option that you will use internally, which should be a
87
+ ## symbol or a string. +desc+ is a string description which will be
88
+ ## displayed in help messages.
89
+ ##
90
+ ## Takes the following optional arguments:
91
+ ##
92
+ ## [+:long+] Specify the long form of the argument, i.e. the form with two dashes. If unspecified, will be automatically derived based on the argument name by turning the +name+ option into a string, and replacing any _'s by -'s.
93
+ ## [+:short+] Specify the short form of the argument, i.e. the form with one dash. If unspecified, will be automatically derived from +name+.
94
+ ## [+:type+] Require that the argument take a parameter or parameters of type +type+. For a single parameter, the value can be a member of +SINGLE_ARG_TYPES+, or a corresponding Ruby class (e.g. +Integer+ for +:int+). For multiple-argument parameters, the value can be any member of +MULTI_ARG_TYPES+ constant. If unset, the default argument type is +:flag+, meaning that the argument does not take a parameter. The specification of +:type+ is not necessary if a +:default+ is given.
95
+ ## [+:default+] Set the default value for an argument. Without a default value, the hash returned by #parse (and thus Trollop::options) will have a +nil+ value for this key unless the argument is given on the commandline. The argument type is derived automatically from the class of the default value given, so specifying a +:type+ is not necessary if a +:default+ is given. (But see below for an important caveat when +:multi+: is specified too.) If the argument is a flag, and the default is set to +true+, then if it is specified on the the commandline the value will be +false+.
96
+ ## [+:required+] If set to +true+, the argument must be provided on the commandline.
97
+ ## [+:multi+] If set to +true+, allows multiple occurrences of the option on the commandline. Otherwise, only a single instance of the option is allowed. (Note that this is different from taking multiple parameters. See below.)
98
+ ##
99
+ ## Note that there are two types of argument multiplicity: an argument
100
+ ## can take multiple values, e.g. "--arg 1 2 3". An argument can also
101
+ ## be allowed to occur multiple times, e.g. "--arg 1 --arg 2".
102
+ ##
103
+ ## Arguments that take multiple values should have a +:type+ parameter
104
+ ## drawn from +MULTI_ARG_TYPES+ (e.g. +:strings+), or a +:default:+
105
+ ## value of an array of the correct type (e.g. [String]). The
106
+ ## value of this argument will be an array of the parameters on the
107
+ ## commandline.
108
+ ##
109
+ ## Arguments that can occur multiple times should be marked with
110
+ ## +:multi+ => +true+. The value of this argument will also be an array.
111
+ ## In contrast with regular non-multi options, if not specified on
112
+ ## the commandline, the default value will be [], not nil.
113
+ ##
114
+ ## These two attributes can be combined (e.g. +:type+ => +:strings+,
115
+ ## +:multi+ => +true+), in which case the value of the argument will be
116
+ ## an array of arrays.
117
+ ##
118
+ ## There's one ambiguous case to be aware of: when +:multi+: is true and a
119
+ ## +:default+ is set to an array (of something), it's ambiguous whether this
120
+ ## is a multi-value argument as well as a multi-occurrence argument.
121
+ ## In thise case, Trollop assumes that it's not a multi-value argument.
122
+ ## If you want a multi-value, multi-occurrence argument with a default
123
+ ## value, you must specify +:type+ as well.
124
+
125
+ def opt name, desc="", opts={}
126
+ raise ArgumentError, "you already have an argument named '#{name}'" if @specs.member? name
127
+
128
+ ## fill in :type
129
+ opts[:type] = # normalize
130
+ case opts[:type]
131
+ when :boolean, :bool; :flag
132
+ when :integer; :int
133
+ when :integers; :ints
134
+ when :double; :float
135
+ when :doubles; :floats
136
+ when Class
137
+ case opts[:type].name
138
+ when 'TrueClass', 'FalseClass'; :flag
139
+ when 'String'; :string
140
+ when 'Integer'; :int
141
+ when 'Float'; :float
142
+ when 'IO'; :io
143
+ when 'Date'; :date
144
+ else
145
+ raise ArgumentError, "unsupported argument type '#{opts[:type].class.name}'"
146
+ end
147
+ when nil; nil
148
+ else
149
+ raise ArgumentError, "unsupported argument type '#{opts[:type]}'" unless TYPES.include?(opts[:type])
150
+ opts[:type]
151
+ end
152
+
153
+ ## for options with :multi => true, an array default doesn't imply
154
+ ## a multi-valued argument. for that you have to specify a :type
155
+ ## as well. (this is how we disambiguate an ambiguous situation;
156
+ ## see the docs for Parser#opt for details.)
157
+ disambiguated_default =
158
+ if opts[:multi] && opts[:default].is_a?(Array) && !opts[:type]
159
+ opts[:default].first
160
+ else
161
+ opts[:default]
162
+ end
163
+
164
+ type_from_default =
165
+ case disambiguated_default
166
+ when Integer; :int
167
+ when Numeric; :float
168
+ when TrueClass, FalseClass; :flag
169
+ when String; :string
170
+ when IO; :io
171
+ when Date; :date
172
+ when Array
173
+ if opts[:default].empty?
174
+ raise ArgumentError, "multiple argument type cannot be deduced from an empty array for '#{opts[:default][0].class.name}'"
175
+ end
176
+ case opts[:default][0] # the first element determines the types
177
+ when Integer; :ints
178
+ when Numeric; :floats
179
+ when String; :strings
180
+ when IO; :ios
181
+ when Date; :dates
182
+ else
183
+ raise ArgumentError, "unsupported multiple argument type '#{opts[:default][0].class.name}'"
184
+ end
185
+ when nil; nil
186
+ else
187
+ raise ArgumentError, "unsupported argument type '#{opts[:default].class.name}'"
188
+ end
189
+
190
+ raise ArgumentError, ":type specification and default type don't match (default type is #{type_from_default})" if opts[:type] && type_from_default && opts[:type] != type_from_default
191
+
192
+ opts[:type] = opts[:type] || type_from_default || :flag
193
+
194
+ ## fill in :long
195
+ opts[:long] = opts[:long] ? opts[:long].to_s : name.to_s.gsub("_", "-")
196
+ opts[:long] =
197
+ case opts[:long]
198
+ when /^--([^-].*)$/
199
+ $1
200
+ when /^[^-]/
201
+ opts[:long]
202
+ else
203
+ raise ArgumentError, "invalid long option name #{opts[:long].inspect}"
204
+ end
205
+ raise ArgumentError, "long option name #{opts[:long].inspect} is already taken; please specify a (different) :long" if @long[opts[:long]]
206
+
207
+ ## fill in :short
208
+ opts[:short] = opts[:short].to_s if opts[:short] unless opts[:short] == :none
209
+ opts[:short] = case opts[:short]
210
+ when /^-(.)$/; $1
211
+ when nil, :none, /^.$/; opts[:short]
212
+ else raise ArgumentError, "invalid short option name '#{opts[:short].inspect}'"
213
+ end
214
+
215
+ if opts[:short]
216
+ raise ArgumentError, "short option name #{opts[:short].inspect} is already taken; please specify a (different) :short" if @short[opts[:short]]
217
+ raise ArgumentError, "a short option name can't be a number or a dash" if opts[:short] =~ INVALID_SHORT_ARG_REGEX
218
+ end
219
+
220
+ ## fill in :default for flags
221
+ opts[:default] = false if opts[:type] == :flag && opts[:default].nil?
222
+
223
+ ## autobox :default for :multi (multi-occurrence) arguments
224
+ opts[:default] = [opts[:default]] if opts[:default] && opts[:multi] && !opts[:default].is_a?(Array)
225
+
226
+ ## fill in :multi
227
+ opts[:multi] ||= false
228
+
229
+ opts[:desc] ||= desc
230
+ @long[opts[:long]] = name
231
+ @short[opts[:short]] = name if opts[:short] && opts[:short] != :none
232
+ @specs[name] = opts
233
+ @order << [:opt, name]
234
+ end
235
+
236
+ ## Sets the version string. If set, the user can request the version
237
+ ## on the commandline. Should probably be of the form "<program name>
238
+ ## <version number>".
239
+ def version s=nil; @version = s if s; @version end
240
+
241
+ ## Adds text to the help display. Can be interspersed with calls to
242
+ ## #opt to build a multi-section help page.
243
+ def banner s; @order << [:text, s] end
244
+ alias :text :banner
245
+
246
+ ## Marks two (or more!) options as requiring each other. Only handles
247
+ ## undirected (i.e., mutual) dependencies. Directed dependencies are
248
+ ## better modeled with Trollop::die.
249
+ def depends *syms
250
+ syms.each { |sym| raise ArgumentError, "unknown option '#{sym}'" unless @specs[sym] }
251
+ @constraints << [:depends, syms]
252
+ end
253
+
254
+ ## Marks two (or more!) options as conflicting.
255
+ def conflicts *syms
256
+ syms.each { |sym| raise ArgumentError, "unknown option '#{sym}'" unless @specs[sym] }
257
+ @constraints << [:conflicts, syms]
258
+ end
259
+
260
+ ## Defines a set of words which cause parsing to terminate when
261
+ ## encountered, such that any options to the left of the word are
262
+ ## parsed as usual, and options to the right of the word are left
263
+ ## intact.
264
+ ##
265
+ ## A typical use case would be for subcommand support, where these
266
+ ## would be set to the list of subcommands. A subsequent Trollop
267
+ ## invocation would then be used to parse subcommand options, after
268
+ ## shifting the subcommand off of ARGV.
269
+ def stop_on *words
270
+ @stop_words = [*words].flatten
271
+ end
272
+
273
+ ## Similar to #stop_on, but stops on any unknown word when encountered
274
+ ## (unless it is a parameter for an argument). This is useful for
275
+ ## cases where you don't know the set of subcommands ahead of time,
276
+ ## i.e., without first parsing the global options.
277
+ def stop_on_unknown
278
+ @stop_on_unknown = true
279
+ end
280
+
281
+ ## Parses the commandline. Typically called by Trollop::options,
282
+ ## but you can call it directly if you need more control.
283
+ ##
284
+ ## throws CommandlineError, HelpNeeded, and VersionNeeded exceptions.
285
+ def parse cmdline=ARGV
286
+ vals = {}
287
+ required = {}
288
+
289
+ opt :version, "Print version and exit" if @version unless @specs[:version] || @long["version"]
290
+ opt :help, "Show this message" unless @specs[:help] || @long["help"]
291
+
292
+ @specs.each do |sym, opts|
293
+ required[sym] = true if opts[:required]
294
+ vals[sym] = opts[:default]
295
+ vals[sym] = [] if opts[:multi] && !opts[:default] # multi arguments default to [], not nil
296
+ end
297
+
298
+ resolve_default_short_options
299
+
300
+ ## resolve symbols
301
+ given_args = {}
302
+ @leftovers = each_arg cmdline do |arg, params|
303
+ sym = case arg
304
+ when /^-([^-])$/
305
+ @short[$1]
306
+ when /^--([^-]\S*)$/
307
+ @long[$1]
308
+ else
309
+ raise CommandlineError, "invalid argument syntax: '#{arg}'"
310
+ end
311
+ raise CommandlineError, "unknown argument '#{arg}'" unless sym
312
+
313
+ if given_args.include?(sym) && !@specs[sym][:multi]
314
+ raise CommandlineError, "option '#{arg}' specified multiple times"
315
+ end
316
+
317
+ given_args[sym] ||= {}
318
+
319
+ given_args[sym][:arg] = arg
320
+ given_args[sym][:params] ||= []
321
+
322
+ # The block returns the number of parameters taken.
323
+ num_params_taken = 0
324
+
325
+ unless params.nil?
326
+ if SINGLE_ARG_TYPES.include?(@specs[sym][:type])
327
+ given_args[sym][:params] << params[0, 1] # take the first parameter
328
+ num_params_taken = 1
329
+ elsif MULTI_ARG_TYPES.include?(@specs[sym][:type])
330
+ given_args[sym][:params] << params # take all the parameters
331
+ num_params_taken = params.size
332
+ end
333
+ end
334
+
335
+ num_params_taken
336
+ end
337
+
338
+ ## check for version and help args
339
+ raise VersionNeeded if given_args.include? :version
340
+ raise HelpNeeded if given_args.include? :help
341
+
342
+ ## check constraint satisfaction
343
+ @constraints.each do |type, syms|
344
+ constraint_sym = syms.find { |sym| given_args[sym] }
345
+ next unless constraint_sym
346
+
347
+ case type
348
+ when :depends
349
+ syms.each { |sym| raise CommandlineError, "--#{@specs[constraint_sym][:long]} requires --#{@specs[sym][:long]}" unless given_args.include? sym }
350
+ when :conflicts
351
+ syms.each { |sym| raise CommandlineError, "--#{@specs[constraint_sym][:long]} conflicts with --#{@specs[sym][:long]}" if given_args.include?(sym) && (sym != constraint_sym) }
352
+ end
353
+ end
354
+
355
+ required.each do |sym, val|
356
+ raise CommandlineError, "option --#{@specs[sym][:long]} must be specified" unless given_args.include? sym
357
+ end
358
+
359
+ ## parse parameters
360
+ given_args.each do |sym, given_data|
361
+ arg = given_data[:arg]
362
+ params = given_data[:params]
363
+
364
+ opts = @specs[sym]
365
+ raise CommandlineError, "option '#{arg}' needs a parameter" if params.empty? && opts[:type] != :flag
366
+
367
+ vals["#{sym}_given".intern] = true # mark argument as specified on the commandline
368
+
369
+ case opts[:type]
370
+ when :flag
371
+ vals[sym] = !opts[:default]
372
+ when :int, :ints
373
+ vals[sym] = params.map { |pg| pg.map { |p| parse_integer_parameter p, arg } }
374
+ when :float, :floats
375
+ vals[sym] = params.map { |pg| pg.map { |p| parse_float_parameter p, arg } }
376
+ when :string, :strings
377
+ vals[sym] = params.map { |pg| pg.map { |p| p.to_s } }
378
+ when :io, :ios
379
+ vals[sym] = params.map { |pg| pg.map { |p| parse_io_parameter p, arg } }
380
+ when :date, :dates
381
+ vals[sym] = params.map { |pg| pg.map { |p| parse_date_parameter p, arg } }
382
+ end
383
+
384
+ if SINGLE_ARG_TYPES.include?(opts[:type])
385
+ unless opts[:multi] # single parameter
386
+ vals[sym] = vals[sym][0][0]
387
+ else # multiple options, each with a single parameter
388
+ vals[sym] = vals[sym].map { |p| p[0] }
389
+ end
390
+ elsif MULTI_ARG_TYPES.include?(opts[:type]) && !opts[:multi]
391
+ vals[sym] = vals[sym][0] # single option, with multiple parameters
392
+ end
393
+ # else: multiple options, with multiple parameters
394
+ end
395
+
396
+ ## modify input in place with only those
397
+ ## arguments we didn't process
398
+ cmdline.clear
399
+ @leftovers.each { |l| cmdline << l }
400
+
401
+ ## allow openstruct-style accessors
402
+ class << vals
403
+ def method_missing(m, *args)
404
+ self[m] || self[m.to_s]
405
+ end
406
+ end
407
+ vals
408
+ end
409
+
410
+ def parse_date_parameter param, arg #:nodoc:
411
+ begin
412
+ begin
413
+ time = Chronic.parse(param)
414
+ rescue NameError
415
+ # chronic is not available
416
+ end
417
+ time ? Date.new(time.year, time.month, time.day) : Date.parse(param)
418
+ rescue ArgumentError => e
419
+ raise CommandlineError, "option '#{arg}' needs a date"
420
+ end
421
+ end
422
+
423
+ ## Print the help message to +stream+.
424
+ def educate stream=$stdout
425
+ width # just calculate it now; otherwise we have to be careful not to
426
+ # call this unless the cursor's at the beginning of a line.
427
+
428
+ left = {}
429
+ @specs.each do |name, spec|
430
+ left[name] = "--#{spec[:long]}" +
431
+ (spec[:short] && spec[:short] != :none ? ", -#{spec[:short]}" : "") +
432
+ case spec[:type]
433
+ when :flag; ""
434
+ when :int; " <i>"
435
+ when :ints; " <i+>"
436
+ when :string; " <s>"
437
+ when :strings; " <s+>"
438
+ when :float; " <f>"
439
+ when :floats; " <f+>"
440
+ when :io; " <filename/uri>"
441
+ when :ios; " <filename/uri+>"
442
+ when :date; " <date>"
443
+ when :dates; " <date+>"
444
+ end
445
+ end
446
+
447
+ leftcol_width = left.values.map { |s| s.length }.max || 0
448
+ rightcol_start = leftcol_width + 6 # spaces
449
+
450
+ unless @order.size > 0 && @order.first.first == :text
451
+ stream.puts "#@version\n" if @version
452
+ stream.puts "Options:"
453
+ end
454
+
455
+ @order.each do |what, opt|
456
+ if what == :text
457
+ stream.puts wrap(opt)
458
+ next
459
+ end
460
+
461
+ spec = @specs[opt]
462
+ stream.printf " %#{leftcol_width}s: ", left[opt]
463
+ desc = spec[:desc] + begin
464
+ default_s = case spec[:default]
465
+ when $stdout; "<stdout>"
466
+ when $stdin; "<stdin>"
467
+ when $stderr; "<stderr>"
468
+ when Array
469
+ spec[:default].join(", ")
470
+ else
471
+ spec[:default].to_s
472
+ end
473
+
474
+ if spec[:default]
475
+ if spec[:desc] =~ /\.$/
476
+ " (Default: #{default_s})"
477
+ else
478
+ " (default: #{default_s})"
479
+ end
480
+ else
481
+ ""
482
+ end
483
+ end
484
+ stream.puts wrap(desc, :width => width - rightcol_start - 1, :prefix => rightcol_start)
485
+ end
486
+ end
487
+
488
+ def width #:nodoc:
489
+ @width ||= if $stdout.tty?
490
+ begin
491
+ require 'curses'
492
+ Curses::init_screen
493
+ x = Curses::cols
494
+ Curses::close_screen
495
+ x
496
+ rescue Exception
497
+ 80
498
+ end
499
+ else
500
+ 80
501
+ end
502
+ end
503
+
504
+ def wrap str, opts={} # :nodoc:
505
+ if str == ""
506
+ [""]
507
+ else
508
+ str.split("\n").map { |s| wrap_line s, opts }.flatten
509
+ end
510
+ end
511
+
512
+ ## The per-parser version of Trollop::die (see that for documentation).
513
+ def die arg, msg
514
+ if msg
515
+ $stderr.puts "Error: argument --#{@specs[arg][:long]} #{msg}."
516
+ else
517
+ $stderr.puts "Error: #{arg}."
518
+ end
519
+ $stderr.puts "Try --help for help."
520
+ exit(-1)
521
+ end
522
+
523
+ private
524
+
525
+ ## yield successive arg, parameter pairs
526
+ def each_arg args
527
+ remains = []
528
+ i = 0
529
+
530
+ until i >= args.length
531
+ if @stop_words.member? args[i]
532
+ remains += args[i .. -1]
533
+ return remains
534
+ end
535
+ case args[i]
536
+ when /^--$/ # arg terminator
537
+ remains += args[(i + 1) .. -1]
538
+ return remains
539
+ when /^--(\S+?)=(.*)$/ # long argument with equals
540
+ yield "--#{$1}", [$2]
541
+ i += 1
542
+ when /^--(\S+)$/ # long argument
543
+ params = collect_argument_parameters(args, i + 1)
544
+ unless params.empty?
545
+ num_params_taken = yield args[i], params
546
+ unless num_params_taken
547
+ if @stop_on_unknown
548
+ remains += args[i + 1 .. -1]
549
+ return remains
550
+ else
551
+ remains += params
552
+ end
553
+ end
554
+ i += 1 + num_params_taken
555
+ else # long argument no parameter
556
+ yield args[i], nil
557
+ i += 1
558
+ end
559
+ when /^-(\S+)$/ # one or more short arguments
560
+ shortargs = $1.split(//)
561
+ shortargs.each_with_index do |a, j|
562
+ if j == (shortargs.length - 1)
563
+ params = collect_argument_parameters(args, i + 1)
564
+ unless params.empty?
565
+ num_params_taken = yield "-#{a}", params
566
+ unless num_params_taken
567
+ if @stop_on_unknown
568
+ remains += args[i + 1 .. -1]
569
+ return remains
570
+ else
571
+ remains += params
572
+ end
573
+ end
574
+ i += 1 + num_params_taken
575
+ else # argument no parameter
576
+ yield "-#{a}", nil
577
+ i += 1
578
+ end
579
+ else
580
+ yield "-#{a}", nil
581
+ end
582
+ end
583
+ else
584
+ if @stop_on_unknown
585
+ remains += args[i .. -1]
586
+ return remains
587
+ else
588
+ remains << args[i]
589
+ i += 1
590
+ end
591
+ end
592
+ end
593
+
594
+ remains
595
+ end
596
+
597
+ def parse_integer_parameter param, arg
598
+ raise CommandlineError, "option '#{arg}' needs an integer" unless param =~ /^\d+$/
599
+ param.to_i
600
+ end
601
+
602
+ def parse_float_parameter param, arg
603
+ raise CommandlineError, "option '#{arg}' needs a floating-point number" unless param =~ FLOAT_RE
604
+ param.to_f
605
+ end
606
+
607
+ def parse_io_parameter param, arg
608
+ case param
609
+ when /^(stdin|-)$/i; $stdin
610
+ else
611
+ require 'open-uri'
612
+ begin
613
+ open param
614
+ rescue SystemCallError => e
615
+ raise CommandlineError, "file or url for option '#{arg}' cannot be opened: #{e.message}"
616
+ end
617
+ end
618
+ end
619
+
620
+ def collect_argument_parameters args, start_at
621
+ params = []
622
+ pos = start_at
623
+ while args[pos] && args[pos] !~ PARAM_RE && !@stop_words.member?(args[pos]) do
624
+ params << args[pos]
625
+ pos += 1
626
+ end
627
+ params
628
+ end
629
+
630
+ def resolve_default_short_options
631
+ @order.each do |type, name|
632
+ next unless type == :opt
633
+ opts = @specs[name]
634
+ next if opts[:short]
635
+
636
+ c = opts[:long].split(//).find { |d| d !~ INVALID_SHORT_ARG_REGEX && !@short.member?(d) }
637
+ if c # found a character to use
638
+ opts[:short] = c
639
+ @short[c] = name
640
+ end
641
+ end
642
+ end
643
+
644
+ def wrap_line str, opts={}
645
+ prefix = opts[:prefix] || 0
646
+ width = opts[:width] || (self.width - 1)
647
+ start = 0
648
+ ret = []
649
+ until start > str.length
650
+ nextt =
651
+ if start + width >= str.length
652
+ str.length
653
+ else
654
+ x = str.rindex(/\s/, start + width)
655
+ x = str.index(/\s/, start) if x && x < start
656
+ x || str.length
657
+ end
658
+ ret << (ret.empty? ? "" : " " * prefix) + str[start ... nextt]
659
+ start = nextt + 1
660
+ end
661
+ ret
662
+ end
663
+
664
+ ## instance_eval but with ability to handle block arguments
665
+ ## thanks to why: http://redhanded.hobix.com/inspect/aBlockCostume.html
666
+ def cloaker &b
667
+ (class << self; self; end).class_eval do
668
+ define_method :cloaker_, &b
669
+ meth = instance_method :cloaker_
670
+ remove_method :cloaker_
671
+ meth
672
+ end
673
+ end
674
+ end
675
+
676
+ ## The easy, syntactic-sugary entry method into Trollop. Creates a Parser,
677
+ ## passes the block to it, then parses +args+ with it, handling any errors or
678
+ ## requests for help or version information appropriately (and then exiting).
679
+ ## Modifies +args+ in place. Returns a hash of option values.
680
+ ##
681
+ ## The block passed in should contain zero or more calls to +opt+
682
+ ## (Parser#opt), zero or more calls to +text+ (Parser#text), and
683
+ ## probably a call to +version+ (Parser#version).
684
+ ##
685
+ ## The returned block contains a value for every option specified with
686
+ ## +opt+. The value will be the value given on the commandline, or the
687
+ ## default value if the option was not specified on the commandline. For
688
+ ## every option specified on the commandline, a key "<option
689
+ ## name>_given" will also be set in the hash.
690
+ ##
691
+ ## Example:
692
+ ##
693
+ ## require 'trollop'
694
+ ## opts = Trollop::options do
695
+ ## opt :monkey, "Use monkey mode" # a flag --monkey, defaulting to false
696
+ ## opt :goat, "Use goat mode", :default => true # a flag --goat, defaulting to true
697
+ ## opt :num_limbs, "Number of limbs", :default => 4 # an integer --num-limbs <i>, defaulting to 4
698
+ ## opt :num_thumbs, "Number of thumbs", :type => :int # an integer --num-thumbs <i>, defaulting to nil
699
+ ## end
700
+ ##
701
+ ## ## if called with no arguments
702
+ ## p opts # => { :monkey => false, :goat => true, :num_limbs => 4, :num_thumbs => nil }
703
+ ##
704
+ ## ## if called with --monkey
705
+ ## p opts # => {:monkey_given=>true, :monkey=>true, :goat=>true, :num_limbs=>4, :help=>false, :num_thumbs=>nil}
706
+ ##
707
+ ## See more examples at http://trollop.rubyforge.org.
708
+ def options args=ARGV, *a, &b
709
+ @last_parser = Parser.new(*a, &b)
710
+ with_standard_exception_handling(@last_parser) { @last_parser.parse args }
711
+ end
712
+
713
+ ## If Trollop::options doesn't do quite what you want, you can create a Parser
714
+ ## object and call Parser#parse on it. That method will throw CommandlineError,
715
+ ## HelpNeeded and VersionNeeded exceptions when necessary; if you want to
716
+ ## have these handled for you in the standard manner (e.g. show the help
717
+ ## and then exit upon an HelpNeeded exception), call your code from within
718
+ ## a block passed to this method.
719
+ ##
720
+ ## Note that this method will call System#exit after handling an exception!
721
+ ##
722
+ ## Usage example:
723
+ ##
724
+ ## require 'trollop'
725
+ ## p = Trollop::Parser.new do
726
+ ## opt :monkey, "Use monkey mode" # a flag --monkey, defaulting to false
727
+ ## opt :goat, "Use goat mode", :default => true # a flag --goat, defaulting to true
728
+ ## end
729
+ ##
730
+ ## opts = Trollop::with_standard_exception_handling p do
731
+ ## o = p.parse ARGV
732
+ ## raise Trollop::HelpNeeded if ARGV.empty? # show help screen
733
+ ## o
734
+ ## end
735
+ ##
736
+ ## Requires passing in the parser object.
737
+
738
+ def with_standard_exception_handling parser
739
+ begin
740
+ yield
741
+ rescue CommandlineError => e
742
+ $stderr.puts "Error: #{e.message}."
743
+ $stderr.puts "Try --help for help."
744
+ exit(-1)
745
+ rescue HelpNeeded
746
+ parser.educate
747
+ exit
748
+ rescue VersionNeeded
749
+ puts parser.version
750
+ exit
751
+ end
752
+ end
753
+
754
+ ## Informs the user that their usage of 'arg' was wrong, as detailed by
755
+ ## 'msg', and dies. Example:
756
+ ##
757
+ ## options do
758
+ ## opt :volume, :default => 0.0
759
+ ## end
760
+ ##
761
+ ## die :volume, "too loud" if opts[:volume] > 10.0
762
+ ## die :volume, "too soft" if opts[:volume] < 0.1
763
+ ##
764
+ ## In the one-argument case, simply print that message, a notice
765
+ ## about -h, and die. Example:
766
+ ##
767
+ ## options do
768
+ ## opt :whatever # ...
769
+ ## end
770
+ ##
771
+ ## Trollop::die "need at least one filename" if ARGV.empty?
772
+ def die arg, msg=nil
773
+ if @last_parser
774
+ @last_parser.die arg, msg
775
+ else
776
+ raise ArgumentError, "Trollop::die can only be called after Trollop::options"
777
+ end
778
+ end
779
+
780
+ module_function :options, :die, :with_standard_exception_handling
781
+
782
+ end # module