Sha256: ea4f5c52833eb2b7f962d50fe1af780f2453576a5edcadbdb9d39a994a533434
Contents?: true
Size: 1.46 KB
Versions: 99
Compression:
Stored size: 1.46 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
register ConstantObject.new(namespace, name) {|o| o.source = statement; o.value = value.strip }
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
99 entries across 80 versions & 9 rubygems