Add accepts_block? macromethod to Def

crystal-lang · Aug 25, 2018 · 1b68a4c · 1b68a4c
1 parent a56dcbe
commit 1b68a4c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/spec/compiler/macro/macro_methods_spec.cr b/spec/compiler/macro/macro_methods_spec.cr
@@ -1304,6 +1304,11 @@ module Crystal
         assert_macro "x", %({{x.block_arg}}), [Def.new("some_def")] of ASTNode, ""
       end
 
+      it "executes accepts_block?" do
+        assert_macro "x", %({{x.accepts_block?}}), [Def.new("some_def", ["x".arg, "y".arg], yields: 1)] of ASTNode, "true"
+        assert_macro "x", %({{x.accepts_block?}}), [Def.new("some_def")] of ASTNode, "false"
+      end
+
       it "executes return_type" do
         assert_macro "x", %({{x.return_type}}), [Def.new("some_def", ["x".arg, "y".arg], return_type: "b".arg)] of ASTNode, "b"
         assert_macro "x", %({{x.return_type}}), [Def.new("some_def")] of ASTNode, ""

diff --git a/src/compiler/crystal/macros/methods.cr b/src/compiler/crystal/macros/methods.cr
@@ -1242,6 +1242,8 @@ module Crystal
         interpret_argless_method(method, args) { @double_splat || Nop.new }
       when "block_arg"
         interpret_argless_method(method, args) { @block_arg || Nop.new }
+      when "accepts_block?"
+        interpret_argless_method(method, args) { BoolLiteral.new(@yields != nil) }
       when "return_type"
         interpret_argless_method(method, args) { @return_type || Nop.new }
       when "body"