Implement parser

Author: soutaro

src/parser.c

@@ -3994,6 +3994,64 @@ static bool parse_inline_leading_annotation(rbs_parser_t *parser, rbs_ast_ruby_a
             );
             return true;
         }
+        case pAMP: {
+            rbs_parser_advance(parser);
+            rbs_location_range ampersand_loc = rbs_location_range_current_token(parser);
+
+            rbs_location_range name_loc = RBS_LOCATION_NULL_RANGE;
+            if (parser->next_token.type == tLIDENT) {
+                rbs_parser_advance(parser);
+                name_loc = rbs_location_range_current_token(parser);
+            }
+
+            ADVANCE_ASSERT(parser, pCOLON);
+            rbs_location_range colon_loc = rbs_location_range_current_token(parser);
+
+            rbs_location_range question_loc = RBS_LOCATION_NULL_RANGE;
+            if (parser->next_token.type == pQUESTION) {
+                rbs_parser_advance(parser);
+                question_loc = rbs_location_range_current_token(parser);
+            }
+
+            rbs_range_t type_range;
+            type_range.start = parser->next_token.range.start;
+
+            parse_function_result *result = rbs_allocator_alloc(ALLOCATOR(), parse_function_result);
+            if (!parse_function(parser, true, &result, true, true)) {
+                return false;
+            }
+
+            if (result->block != NULL) {
+                rbs_parser_set_error(parser, parser->current_token, true, "block type annotation cannot have a nested block");
+                return false;
+            }
+
+            type_range.end = parser->current_token.range.end;
+
+            rbs_location_range comment_loc = RBS_LOCATION_NULL_RANGE;
+            if (!parse_inline_comment(parser, &comment_loc)) {
+                return false;
+            }
+
+            rbs_range_t full_range = {
+                .start = rbs_range.start,
+                .end = parser->current_token.range.end
+            };
+
+            *annotation = (rbs_ast_ruby_annotations_t *) rbs_ast_ruby_annotations_block_param_type_annotation_new(
+                ALLOCATOR(),
+                RBS_RANGE_LEX2AST(full_range),
+                RBS_RANGE_LEX2AST(rbs_range),
+                ampersand_loc,
+                name_loc,
+                colon_loc,
+                question_loc,
+                RBS_RANGE_LEX2AST(type_range),
+                result->function,
+                comment_loc
+            );
+            return true;
+        }
         default: {
             rbs_parser_set_error(parser, parser->next_token, true, "unexpected token for @rbs annotation");
             return false;

test/rbs/inline_annotation_parsing_test.rb

@@ -455,4 +455,38 @@ def test_parse__param_type__skip
       assert_nil annot.comment_location
     end
   end
+
+  def test_parse__block_type_annotation
+    Parser.parse_inline_leading_annotation("@rbs &block: () [self: String] -> void", 0...).tap do |annot|
+      assert_instance_of AST::Ruby::Annotations::BlockParamTypeAnnotation, annot
+      assert_equal "@rbs &block: () [self: String] -> void", annot.location.source
+      assert_equal "@rbs", annot.prefix_location.source
+      assert_equal "&", annot.ampersand_location.source
+      assert_equal "block", annot.name_location.source
+      assert_equal ":", annot.colon_location.source
+      assert_nil annot.question_location
+      assert_equal "() [self: String] -> void", annot.type_location.source
+      assert_instance_of Types::Function, annot.type
+      assert_nil annot.comment_location
+      assert_equal :block, annot.name
+      assert_equal false, annot.optional?
+      assert_equal true, annot.required?
+    end
+
+      Parser.parse_inline_leading_annotation("@rbs &: ? (?) -> void", 0...).tap do |annot|
+      assert_instance_of AST::Ruby::Annotations::BlockParamTypeAnnotation, annot
+      assert_equal "@rbs &: ? (?) -> void", annot.location.source
+      assert_equal "@rbs", annot.prefix_location.source
+      assert_equal "&", annot.ampersand_location.source
+      assert_nil annot.name_location
+      assert_equal ":", annot.colon_location.source
+      assert_equal "?", annot.question_location.source
+      assert_equal "(?) -> void", annot.type_location.source
+      assert_instance_of Types::UntypedFunction, annot.type
+      assert_nil annot.comment_location
+      assert_nil annot.name
+      assert_equal true, annot.optional?
+      assert_equal false, annot.required?
+    end
+  end
 end