class PuppetStrings::Yard::Handlers::Ruby::ProviderHandler

Implements the handler for Puppet providers written in Ruby.

Private Instance Methods

populate_provider_data(object) click to toggle source
# File lib/puppet-strings/yard/handlers/ruby/provider_handler.rb, line 77
def populate_provider_data(object)
  # Traverse the block looking for confines/defaults/commands
  block = statement.block
  return unless block && block.count >= 2

  block[1].children.each do |node|
    next unless node.is_a?(YARD::Parser::Ruby::MethodCallNode) && node.method_name

    method_name = node.method_name.source
    parameters = node.parameters(false)

    if method_name == 'confine'
      # Add a confine to the object
      next unless parameters.count >= 1

      parameters[0].each do |kvp|
        next unless kvp.count == 2

        object.add_confine(node_as_string(kvp[0]) || kvp[0].source, node_as_string(kvp[1]) || kvp[1].source)
      end
    elsif %w[has_feature has_features].include?(method_name)
      # Add the features to the object
      parameters.each do |parameter|
        object.add_feature(node_as_string(parameter) || parameter.source)
      end
    elsif method_name == 'defaultfor'
      # Add a default to the object
      next unless parameters.count >= 1

      # Some defaultfor statements contain multiple constraints.
      parameters.each do |kvps|
        next unless kvps.count >= 1

        defaultfor = kvps.map do |kvp|
          [node_as_string(kvp[0]) || kvp[0].source, node_as_string(kvp[1]) || kvp[1].source]
        end
        object.add_default(defaultfor)
      end
    elsif method_name == 'commands'
      # Add the commands to the object
      next unless parameters.count >= 1

      parameters[0].each do |kvp|
        next unless kvp.count == 2

        object.add_command(node_as_string(kvp[0]) || kvp[0].source, node_as_string(kvp[1]) || kvp[1].source)
      end
    end
  end
end
register_provider_docstring(object) click to toggle source
# File lib/puppet-strings/yard/handlers/ruby/provider_handler.rb, line 48
def register_provider_docstring(object)
  # Walk the tree searching for assignments or calls to desc/doc=
  statement.traverse do |child|
    if child.type == :assign
      ivar = child.jump(:ivar)
      next unless ivar != child && ivar.source == '@doc'

      docstring = node_as_string(child[1])
      log.error "Failed to parse docstring for Puppet provider '#{object.name}' (resource type '#{object.type_name}') near #{child.file}:#{child.line}." and return nil unless docstring

      register_docstring(object, PuppetStrings::Yard::Util.scrub_string(docstring), nil)
      return nil
    elsif child.is_a?(YARD::Parser::Ruby::MethodCallNode)
      # Look for a call to a dispatch method with a block
      next unless
        child.method_name &&
        (child.method_name.source == 'desc' || child.method_name.source == 'doc=') &&
        child.parameters(false).count == 1

      docstring = node_as_string(child.parameters[0])
      log.error "Failed to parse docstring for Puppet provider '#{object.name}' (resource type '#{object.type_name}') near #{child.file}:#{child.line}." and return nil unless docstring

      register_docstring(object, PuppetStrings::Yard::Util.scrub_string(docstring), nil)
      return nil
    end
  end
  log.warn "Missing a description for Puppet provider '#{object.name}' (resource type '#{object.type_name}') at #{statement.file}:#{statement.line}."
end