Add Path#resolve? macro method

Closes crystal-lang#4370
RX14 · May 12, 2017 · db2841a · db2841a
1 parent 08c6e8f
commit db2841a
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/spec/compiler/macro/macro_methods_spec.cr b/spec/compiler/macro/macro_methods_spec.cr
@@ -1566,6 +1566,21 @@ describe "macro methods" do
     end
   end
 
+  describe "path methods" do
+    it "executes resolve" do
+      assert_macro "x", %({{x.resolve}}), [Path.new("String")] of ASTNode, %(String)
+
+      expect_raises(Crystal::TypeException, "undefined constant Foo") do
+        assert_macro "x", %({{x.resolve}}), [Path.new("Foo")] of ASTNode, %(Foo)
+      end
+    end
+
+    it "executes resolve?" do
+      assert_macro "x", %({{x.resolve?}}), [Path.new("String")] of ASTNode, %(String)
+      assert_macro "x", %({{x.resolve?}}), [Path.new("Foo")] of ASTNode, %(nil)
+    end
+  end
+
   describe "env" do
     it "has key" do
       ENV["FOO"] = "foo"

diff --git a/src/compiler/crystal/macros/methods.cr b/src/compiler/crystal/macros/methods.cr
@@ -1739,6 +1739,15 @@ module Crystal
         interpret_argless_method(method, args) { BoolLiteral.new(@global) }
       when "resolve"
         interpret_argless_method(method, args) { interpreter.resolve(self) }
+      when "resolve?"
+        interpret_argless_method(method, args) do
+          begin
+            interpreter.resolve(self)
+          rescue ex : Crystal::TypeException
+            ::raise ex unless ex.message.try &.includes? "undefined constant"
+            NilLiteral.new
+          end
+        end
       else
         super
       end