forked from at-grandpa/clim
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
8879f5f
commit 595a23b
Showing
2 changed files
with
157 additions
and
148 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,155 @@ | ||
| class Clim | ||
| abstract class Command | ||
| module Macros | ||
| macro desc(description) | ||
| def desc : String | ||
| {{ description }} | ||
| end | ||
| end | ||
|
|
||
| macro usage(usage) | ||
| def usage : String | ||
| {{ usage }} | ||
| end | ||
| end | ||
|
|
||
| macro alias_name(*names) | ||
| {% raise "'alias_name' is not supported on main command." if @type == Command_Main_command_of_clim_library %} | ||
| def alias_name : Array(String) | ||
| {{ names }}.to_a | ||
| end | ||
| end | ||
|
|
||
| macro version(version_str, short = nil) | ||
| {% if short == nil %} | ||
| option "--version", type: Bool, desc: "Show version.", default: false | ||
| {% else %} | ||
| option {{short.id.stringify}}, "--version", type: Bool, desc: "Show version.", default: false | ||
| {% end %} | ||
|
|
||
| def version_str : String | ||
| {{ version_str }} | ||
| end | ||
| end | ||
|
|
||
| macro main | ||
| main_command | ||
| end | ||
|
|
||
| macro main_command | ||
| {% raise "Can not be declared 'main_command' or 'main' as sub command." if @type.superclass.id.stringify == "Clim::Command" %} | ||
| end | ||
|
|
||
| macro sub(name, &block) | ||
| sub_command({{name}}) do | ||
| {{ yield }} | ||
| end | ||
| end | ||
|
|
||
| macro sub_command(name, &block) | ||
| command({{name}}) do | ||
| {{ yield }} | ||
| end | ||
| end | ||
|
|
||
| macro run(&block) | ||
| def run(io : IO) | ||
| return RunProc.new { io.puts help_template_def }.call(@options, @arguments) if @options.help == true | ||
| options = @options | ||
| if options.responds_to?(:version) | ||
| return RunProc.new { io.puts version_str }.call(options, @arguments) if options.version == true | ||
| end | ||
| RunProc.new {{ block.id }} .call(@options, @arguments) | ||
| end | ||
| end | ||
|
|
||
| macro option_base(short, long, type, desc, default, required) | ||
| {% raise "Empty option name." if short.empty? %} | ||
| {% raise "Type [#{type}] is not supported on option." unless SUPPORT_TYPES.keys.includes?(type) %} | ||
|
|
||
| {% base_option_name = long == nil ? short : long %} | ||
| {% option_name = base_option_name.id.stringify.gsub(/\=/, " ").split(" ").first.id.stringify.gsub(/^-+/, "").gsub(/-/, "_").id %} | ||
| class OptionsForEachCommand | ||
| class Option_{{option_name}} < Option | ||
| define_option_macro({{option_name}}, {{type}}, {{default}}, {{required}}) | ||
| end | ||
|
|
||
| {% default = false if type.id.stringify == "Bool" %} | ||
| {% raise "You can not specify 'required: true' for Bool option." if type.id.stringify == "Bool" && required == true %} | ||
|
|
||
| {% if default == nil %} | ||
| {% default_value = SUPPORT_TYPES[type][:nilable] ? default : SUPPORT_TYPES[type][:default] %} | ||
| {% else %} | ||
| {% default_value = default %} | ||
| {% end %} | ||
|
|
||
| property {{ option_name }}_instance : Option_{{option_name}} = Option_{{option_name}}.new({{ short }}, {% unless long == nil %} {{ long }}, {% end %} {{ desc }}, {{ default_value }}, {{ required }}) | ||
| def {{ option_name }} | ||
| {{ option_name }}_instance.@value | ||
| end | ||
| end | ||
| end | ||
|
|
||
| macro option(short, long, type = String, desc = "Option description.", default = nil, required = false) | ||
| option_base({{short}}, {{long}}, {{type}}, {{desc}}, {{default}}, {{required}}) | ||
| end | ||
|
|
||
| macro option(short, type = String, desc = "Option description.", default = nil, required = false) | ||
| option_base({{short}}, nil, {{type}}, {{desc}}, {{default}}, {{required}}) | ||
| end | ||
|
|
||
| macro command(name, &block) | ||
| {% if @type.constants.map(&.id.stringify).includes?("Command_" + name.id.capitalize.stringify) %} | ||
| {% raise "Command \"" + name.id.stringify + "\" is already defined." %} | ||
| {% end %} | ||
|
|
||
| class Command_{{ name.id.capitalize }} < Command | ||
| property name : String = {{name.id.stringify}} | ||
|
|
||
| class Options_{{ name.id.capitalize }} < Options | ||
| end | ||
|
|
||
| alias OptionsForEachCommand = Options_{{ name.id.capitalize }} | ||
|
|
||
| def initialize | ||
| @display_version_flag = false | ||
| @parser = OptionParser.new | ||
| @options = OptionsForEachCommand.new | ||
| @options.setup_parser(@parser) | ||
| \{% for command_class in @type.constants.select { |c| @type.constant(c).superclass.id.stringify == "Clim::Command" } %} | ||
| @sub_commands << \{{ command_class.id }}.new | ||
| \{% end %} | ||
| end | ||
|
|
||
| def options_info | ||
| @options.info | ||
| end | ||
|
|
||
| private def parse_by_parser(argv) | ||
| @parser.invalid_option { |opt_name| raise ClimInvalidOptionException.new "Undefined option. \"#{opt_name}\"" } | ||
| @parser.missing_option { |opt_name| raise ClimInvalidOptionException.new "Option that requires an argument. \"#{opt_name}\"" } | ||
| @parser.unknown_args { |unknown_args| @arguments = unknown_args } | ||
| @parser.parse(argv.dup) | ||
| required_validate! if @options.help == false | ||
| @options.help_str = help_template_def | ||
| self | ||
| end | ||
|
|
||
| private def required_validate! | ||
| raise "Required options. \"#{@options.invalid_required_names.join("\", \"")}\"" unless @options.invalid_required_names.empty? | ||
| end | ||
|
|
||
| {{ yield }} | ||
|
|
||
| option "--help", type: Bool, desc: "Show this help.", default: false | ||
|
|
||
| alias RunProc = Proc(OptionsForEachCommand, Array(String), Nil) | ||
| property options : OptionsForEachCommand = OptionsForEachCommand.new | ||
|
|
||
| class OptionsForEachCommand | ||
| end | ||
| end | ||
| end | ||
| end | ||
| end | ||
| end |