Refactor `HeaderNode#find_parent`

Now the instance variable assignment is all contained within
`#initialize`, and the `children.push` behavior is more explicit.
This commit is contained in:
Robert Speicher 2017-09-01 14:58:40 -04:00
parent b43aefbd9d
commit fc27ca62dd
1 changed files with 7 additions and 6 deletions

View File

@ -86,7 +86,8 @@ module Banzai
@href = href
@children = []
find_parent(previous_header)
@parent = find_parent(previous_header)
@parent.children.push(self) if @parent
end
def level
@ -107,15 +108,15 @@ module Banzai
return unless previous_header
if level == previous_header.level
@parent = previous_header.parent
parent = previous_header.parent
elsif level > previous_header.level
@parent = previous_header
parent = previous_header
else
@parent = previous_header
@parent = @parent.parent while @parent.level >= level
parent = previous_header
parent = parent.parent while parent.level >= level
end
@parent.children.push(self)
parent
end
end
end