[Writer] Unquote keyword as method name

Except `self` and `self?` because they need quote.
ruby · Dec 20, 2020 · 2b03212 · 2b03212
1 parent a1e6261
commit 2b03212
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/lib/rbs/writer.rb b/lib/rbs/writer.rb
@@ -233,7 +233,7 @@ def method_name(name)
       s = name.to_s
 
       if [:tOPERATOR, :kAMP, :kHAT, :kSTAR, :kLT, :kEXCLAMATION, :kSTAR2, :kBAR].include?(Parser::PUNCTS[s]) ||
-          (/\A[a-zA-Z_]\w*[?!=]?\z/.match?(s) && !/\A#{Parser::KEYWORDS_RE}\z/.match?(s))
+          (/\A[a-zA-Z_]\w*[?!=]?\z/.match?(s) && !/\Aself\??\z/.match?(s))
         s
       else
         "`#{s}`"

diff --git a/test/rbs/writer_test.rb b/test/rbs/writer_test.rb
@@ -111,7 +111,11 @@ def !: () -> void
 
   def __id__: () -> Integer
 
-  def `def`: () -> Symbol
+  def def: () -> Symbol
+
+  def `self`: () -> void
+
+  def `self?`: () -> void
 
   def timeout: () -> Integer