Sha256: fc27b8c49eb09617fd57c65c04df78337d52bed88c1abd7f8a6ef92767865ada
Contents?: true
Size: 1.63 KB
Versions: 7
Compression:
Stored size: 1.63 KB
Contents
# Handles any constant assignment
class YARD::Handlers::Ruby::ConstantHandler < YARD::Handlers::Ruby::Base
include YARD::Handlers::Ruby::StructHandlerMethods
handles :assign
namespace_only
process do
if statement[1].call? && statement[1][0][0] == s(:const, "Struct") &&
statement[1][2] == s(:ident, "new")
process_structclass(statement)
elsif statement[0].type == :var_field && statement[0][0].type == :const
process_constant(statement)
end
end
private
def process_constant(statement)
name = statement[0][0][0]
value = statement[1].source
obj = P(namespace, name)
if obj.is_a?(NamespaceObject)
raise YARD::Parser::UndocumentableError, "constant for existing #{obj.type} #{obj}"
else
register ConstantObject.new(namespace, name) {|o| o.source = statement; o.value = value.strip }
end
end
def process_structclass(statement)
lhs = statement[0][0]
if lhs.type == :const
klass = create_class(lhs[0], P(:Struct))
create_attributes(klass, extract_parameters(statement[1]))
else
raise YARD::Parser::UndocumentableError, "Struct assignment to #{statement[0].source}"
end
end
# Extract the parameters from the Struct.new AST node, returning them as a list
# of strings
#
# @param [MethodCallNode] superclass the AST node for the Struct.new call
# @return [Array<String>] the member names to generate methods for
def extract_parameters(superclass)
return [] unless superclass.parameters
members = superclass.parameters.select {|x| x && x.type == :symbol_literal}
members.map! {|x| x.source.strip[1..-1]}
members
end
end
Version data entries
7 entries across 6 versions & 2 rubygems