bashly 1.3.8 → 1.4.0.rc1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 411dc6262c3232175e5a968a31ad90b56aa59be9d9b46203c3eb2db14f35d491
-  data.tar.gz: 2ce20d069c93e1e7081984ff1fb2dd9c7bca530d6fbf127e0bb1c996c0117cc1
+  metadata.gz: c4417e2acaa7177450ab206c368039c46db5e2e3839b015c7dac235d23d3899a
+  data.tar.gz: e62782cd28d229c36adcc4be6d09243b693f8c1536fadcfb98ceaddb203fbc49
 SHA512:
-  metadata.gz: 3bff0a363b5c52c0f54b2b1e413b909256579c0b29ef8a44e88cd73b1803c584c3de26af99fa737edac8cc103a218f418a7a5f463e38b27c416ebf26bbcfe8ef
-  data.tar.gz: 2691947ed69d6f693625be4ba66a1b9803d01e46f50768d266c5f0f27c60f74242977fb91131443f6bd29161a9810970b542ecef3d3be0e9b3a84641da8dc080
+  metadata.gz: f2827e99a65bbfc2cdaa77b2533400d6a85d7af72cbb35471654bb2ce840cef897f76e19c51504027559405a410cb46d150c2b2680d9b471526a94ee29dc4083
+  data.tar.gz: 1cbe9b74014155e5cbd920fa29249a4102afea7738334b811583d5bdc1ef9064a1595d87d238bc350dd46b4dbe32926546a341956b1f417b521f5d459947ba1c

data/lib/bashly/commands/doc.rb

@@ -75,7 +75,7 @@ module Bashly
         @raw_data ||= begin
           result = {}
           Dir["#{docs_dir}/*.yml"].each do |path|
-            result.merge! YAML.load_file(path)
+            result.merge! YAML.trusted_load_file(path)
           end
           result
         end

data/lib/bashly/completion_builder.rb

@@ -0,0 +1,188 @@
+module Bashly
+  class CompletionBuilder
+    BUILTIN_PATTERN = /\A<([^>]+)>\z/
+
+    def initialize(command, with_version: true)
+      @command = command
+      @with_version = with_version
+      @patterns = []
+      @options = {}
+      @tokens = {}
+      @token_sources = {}
+    end
+
+    def call
+      add_command @command, inherited_global_groups: []
+
+      result = { 'patterns' => @patterns }
+      result['options'] = @options if @options.any?
+      result['tokens'] = @tokens if @tokens.any?
+      result
+    end
+
+  private
+
+    def add_command(command, inherited_global_groups:)
+      local_group = add_local_options command
+      pattern_groups = inherited_global_groups.dup
+      pattern_groups << local_group if local_group
+
+      @patterns << pattern_for(command, pattern_groups)
+
+      child_global_groups = inherited_global_groups.dup
+      if command.global_flags?
+        global_group = add_global_options command
+        child_global_groups << global_group if global_group
+      end
+
+      command.visible_commands.each do |child|
+        add_command child, inherited_global_groups: child_global_groups
+      end
+    end
+
+    def pattern_for(command, option_groups)
+      parts = [command_path(command)]
+      parts.concat(option_groups.map { |group| "[#{group} options]" })
+      parts.concat positional_tokens(command)
+      parts.join ' '
+    end
+
+    def command_path(command)
+      command_chain(command).map.with_index do |item, index|
+        index.zero? ? item.name : item.aliases.join('|')
+      end.join ' '
+    end
+
+    def command_chain(command)
+      result = []
+      current = command
+      while current
+        result.unshift current
+        current = current.parent_command
+      end
+      result
+    end
+
+    def add_local_options(command)
+      entries = fixed_option_entries(command) + flag_option_entries(command.visible_flags, command)
+      return if entries.empty?
+
+      name = group_name command
+      @options[name] = entries
+      name
+    end
+
+    def add_global_options(command)
+      entries = flag_option_entries command.visible_flags, command
+      return if entries.empty?
+
+      name = "#{group_name(command)}_global"
+      @options[name] = entries
+      name
+    end
+
+    def fixed_option_entries(command)
+      return [] if !command.root_command? && command.catch_all.catch_help?
+
+      entries = %w[--help|-h]
+      entries << '--version|-v' if command.root_command? && @with_version
+      entries
+    end
+
+    def flag_option_entries(flags, command)
+      flags.map do |flag|
+        token_name = flag_token_name flag, command
+        flag.completion_option_entry token_name
+      end
+    end
+
+    def flag_token_name(flag, command)
+      return unless flag.arg || flag.allowed || flag.completions
+
+      register_token flag.arg || flag.name, command, flag_source(flag)
+    end
+
+    def positional_tokens(command)
+      command.args.map do |arg|
+        token_name = register_token arg.name, command, arg_source(arg, command)
+        suffix = arg.repeatable ? '...' : nil
+        "<#{token_name}>#{suffix}"
+      end
+    end
+
+    def flag_source(flag)
+      return static_source(flag.allowed) if flag.allowed
+      return completion_source(flag.completions) if flag.completions
+
+      nil
+    end
+
+    def arg_source(arg, command)
+      return static_source(arg.allowed) if arg.allowed
+      return completion_source(arg.completions) if arg.completions
+      return completion_source(command.completions) if command.completions
+
+      nil
+    end
+
+    def static_source(values)
+      Array(values).compact.map { |value| escape_static_source value }
+    end
+
+    def completion_source(values)
+      Array(values).compact.map do |value|
+        string = value.to_s
+        builtin = string[BUILTIN_PATTERN, 1]
+
+        if builtin
+          "+#{builtin}"
+        else
+          escape_static_source string
+        end
+      end
+    end
+
+    def escape_static_source(value)
+      string = value.to_s
+      string.start_with?('+') ? "+#{string}" : string
+    end
+
+    def register_token(base_name, command, source)
+      preferred = token_name base_name
+      return preferred if register_token_name preferred, source
+
+      scoped = token_name "#{group_name command}_#{base_name}"
+      return scoped if register_token_name scoped, source
+
+      suffix = 2
+      loop do
+        candidate = "#{scoped}_#{suffix}"
+        return candidate if register_token_name candidate, source
+
+        suffix += 1
+      end
+    end
+
+    def register_token_name(name, source)
+      if @token_sources.has_key? name
+        return false unless @token_sources[name] == source
+      else
+        @token_sources[name] = source
+        @tokens[name] = source
+      end
+
+      true
+    end
+
+    def group_name(command)
+      token_name command.root_command? ? 'root' : command.action_name
+    end
+
+    def token_name(value)
+      value.to_s
+        .gsub(/[^a-zA-Z0-9]+/, '_')
+        .gsub(/\A_+|_+\z/, '')
+        .downcase
+    end
+  end
+end