mirror of
https://gitea.invidious.io/iv-org/shard-ameba.git
synced 2024-08-15 00:53:29 +00:00
Make Scope#*_def?
methods accept check_outer_scopes
parameter
This commit is contained in:
parent
b6bd74e02f
commit
d23ad7f0ab
2 changed files with 31 additions and 14 deletions
|
@ -180,6 +180,28 @@ module Ameba::AST
|
|||
end
|
||||
end
|
||||
|
||||
describe "#def?" do
|
||||
context "when check_outer_scopes: true" do
|
||||
it "returns true if outer scope is Crystal::Def" do
|
||||
nodes = as_nodes("def foo; 3.times {}; end")
|
||||
outer_scope = Scope.new nodes.def_nodes.first
|
||||
scope = Scope.new nodes.block_nodes.first, outer_scope
|
||||
scope.def?(check_outer_scopes: true).should be_true
|
||||
end
|
||||
end
|
||||
|
||||
it "returns true if Crystal::Def" do
|
||||
nodes = as_nodes("def foo; end")
|
||||
scope = Scope.new nodes.def_nodes.first
|
||||
scope.def?.should be_true
|
||||
end
|
||||
|
||||
it "returns false otherwise" do
|
||||
scope = Scope.new as_node("a = 1")
|
||||
scope.def?.should be_false
|
||||
end
|
||||
end
|
||||
|
||||
describe "#in_macro?" do
|
||||
it "returns true if Crystal::Macro" do
|
||||
nodes = as_nodes <<-CRYSTAL
|
||||
|
|
|
@ -180,20 +180,15 @@ module Ameba::AST
|
|||
@visibility || outer_scope.try(&.visibility)
|
||||
end
|
||||
|
||||
# Returns `true` if current scope is a def, `false` otherwise.
|
||||
def def?
|
||||
node.is_a?(Crystal::Def)
|
||||
end
|
||||
|
||||
# Returns `true` if current scope is a class, `false` otherwise.
|
||||
def class_def?
|
||||
node.is_a?(Crystal::ClassDef)
|
||||
end
|
||||
|
||||
# Returns `true` if current scope is a module, `false` otherwise.
|
||||
def module_def?
|
||||
node.is_a?(Crystal::ModuleDef)
|
||||
end
|
||||
{% for type in %w[Def ClassDef ModuleDef EnumDef LibDef FunDef].map(&.id) %}
|
||||
{% method_name = type.underscore %}
|
||||
# Returns `true` if current scope is a {{ method_name[0..-5] }} def, `false` otherwise.
|
||||
def {{ method_name }}?(*, check_outer_scopes = false)
|
||||
node.is_a?(Crystal::{{ type }}) ||
|
||||
!!(check_outer_scopes &&
|
||||
outer_scope.try(&.{{ method_name }}?(check_outer_scopes: true)))
|
||||
end
|
||||
{% end %}
|
||||
|
||||
# Returns `true` if this scope is a top level scope, `false` otherwise.
|
||||
def top_level?
|
||||
|
|
Loading…
Reference in a new issue