iv-org-mirror/shard-ameba
1
0
Fork 0
mirror of https://gitea.invidious.io/iv-org/shard-ameba.git synced 2024-08-15 00:53:29 +00:00
Code Issues Projects Releases Packages Wiki Activity

Use utility method to check if node is a loop

Browse source
This commit is contained in:
Vitalii Elenhaupt 2018-11-22 11:52:00 +02:00
parent 0fd5890738
commit ad2c6bad0e
No known key found for this signature in database
GPG key ID: 7558EF3A4056C706
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/ameba/ast/branchable.cr
View file

@ -1,3 +1,5 @@
require "./util"
module Ameba::AST module Ameba::AST
# A generic entity to represent a branchable Crystal node. # A generic entity to represent a branchable Crystal node.
# For example, `Crystal::If`, `Crystal::Unless`, `Crystal::While` # For example, `Crystal::If`, `Crystal::Unless`, `Crystal::While`
@ -11,6 +13,8 @@ module Ameba::AST
# end # end
# ``` # ```
class Branchable class Branchable
include Util
getter branches = [] of Crystal::ASTNode getter branches = [] of Crystal::ASTNode
# The actual Crystal node. # The actual Crystal node.
@ -32,10 +36,7 @@ module Ameba::AST
# Returns true if this node or one of the parent branchables is a loop, false otherwise. # Returns true if this node or one of the parent branchables is a loop, false otherwise.
def loop? def loop?
return true if node.is_a? Crystal::While || return true if loop?(node)
node.is_a? Crystal::Until ||
((n = node) && n.is_a?(Crystal::Call) && n.name == "loop")
parent.try(&.loop?) || false parent.try(&.loop?) || false
end end
end end