rbs 4.0.0.dev.2 → 4.0.0.dev.3

data/ext/rbs_extension/legacy_location.c

@@ -10,307 +10,308 @@ rbs_loc_range RBS_LOC_NULL_RANGE = { -1, -1 };
 VALUE RBS_Location;
 
 rbs_position_t rbs_loc_position(int char_pos) {
-  return (rbs_position_t) { 0, char_pos, -1, -1 };
+    return (rbs_position_t) { 0, char_pos, -1, -1 };
 }
 
 rbs_position_t rbs_loc_position3(int char_pos, int line, int column) {
-  return (rbs_position_t) { 0, char_pos, line, column };
+    return (rbs_position_t) { 0, char_pos, line, column };
 }
 
 static rbs_loc_range rbs_new_loc_range(rbs_range_t rg) {
-  rbs_loc_range r = { rg.start.char_pos, rg.end.char_pos };
-  return r;
+    rbs_loc_range r = { rg.start.char_pos, rg.end.char_pos };
+    return r;
 }
 
 static void check_children_max(unsigned short n) {
-  size_t max = sizeof(rbs_loc_entry_bitmap) * 8;
-  if (n > max) {
-    rb_raise(rb_eRuntimeError, "Too many children added to location: %d", n);
-  }
+    size_t max = sizeof(rbs_loc_entry_bitmap) * 8;
+    if (n > max) {
+        rb_raise(rb_eRuntimeError, "Too many children added to location: %d", n);
+    }
 }
 
 void rbs_loc_legacy_alloc_children(rbs_loc *loc, unsigned short cap) {
-  check_children_max(cap);
+    check_children_max(cap);
 
-  size_t s = RBS_LOC_CHILDREN_SIZE(cap);
-  loc->children = malloc(s);
+    size_t s = RBS_LOC_CHILDREN_SIZE(cap);
+    loc->children = malloc(s);
 
-  *loc->children = (rbs_loc_children) {
-    .len = 0,
-    .required_p = 0,
-    .cap = cap,
-    .entries = {{ 0 }},
-  };
+    *loc->children = (rbs_loc_children) {
+        .len = 0,
+        .required_p = 0,
+        .cap = cap,
+        .entries = { { 0 } },
+    };
 }
 
 static void check_children_cap(rbs_loc *loc) {
-  if (loc->children == NULL) {
-    rbs_loc_legacy_alloc_children(loc, 1);
-  } else {
-    if (loc->children->len == loc->children->cap) {
-      check_children_max(loc->children->cap + 1);
-      size_t s = RBS_LOC_CHILDREN_SIZE(++loc->children->cap);
-      loc->children = realloc(loc->children, s);
+    if (loc->children == NULL) {
+        rbs_loc_legacy_alloc_children(loc, 1);
+    } else {
+        if (loc->children->len == loc->children->cap) {
+            check_children_max(loc->children->cap + 1);
+            size_t s = RBS_LOC_CHILDREN_SIZE(++loc->children->cap);
+            loc->children = realloc(loc->children, s);
+        }
     }
-  }
 }
 
 void rbs_loc_legacy_add_optional_child(rbs_loc *loc, rbs_constant_id_t name, rbs_range_t r) {
-  check_children_cap(loc);
+    check_children_cap(loc);
 
-  unsigned short i = loc->children->len++;
-  loc->children->entries[i] = (rbs_loc_entry) {
-    .name = name,
-    .rg = rbs_new_loc_range(r),
-  };
+    unsigned short i = loc->children->len++;
+    loc->children->entries[i] = (rbs_loc_entry) {
+        .name = name,
+        .rg = rbs_new_loc_range(r),
+    };
 }
 
 void rbs_loc_legacy_add_required_child(rbs_loc *loc, rbs_constant_id_t name, rbs_range_t r) {
-  rbs_loc_legacy_add_optional_child(loc, name, r);
+    rbs_loc_legacy_add_optional_child(loc, name, r);
 
-  unsigned short last_index = loc->children->len - 1;
-  loc->children->required_p |= 1 << last_index;
+    unsigned short last_index = loc->children->len - 1;
+    loc->children->required_p |= 1 << last_index;
 }
 
 void rbs_loc_init(rbs_loc *loc, VALUE buffer, rbs_loc_range rg) {
-  *loc = (rbs_loc) {
-    .buffer = buffer,
-    .rg = rg,
-    .children = NULL,
-  };
+    *loc = (rbs_loc) {
+        .buffer = buffer,
+        .rg = rg,
+        .children = NULL,
+    };
 }
 
 void rbs_loc_free(rbs_loc *loc) {
-  free(loc->children);
-  ruby_xfree(loc);
+    free(loc->children);
+    ruby_xfree(loc);
 }
 
-static void rbs_loc_mark(void *ptr)
-{
-  rbs_loc *loc = ptr;
-  rb_gc_mark(loc->buffer);
+static void rbs_loc_mark(void *ptr) {
+    rbs_loc *loc = ptr;
+    rb_gc_mark(loc->buffer);
 }
 
 static size_t rbs_loc_memsize(const void *ptr) {
-  const rbs_loc *loc = ptr;
-  if (loc->children == NULL) {
-    return sizeof(rbs_loc);
-  } else {
-    return sizeof(rbs_loc) + RBS_LOC_CHILDREN_SIZE(loc->children->cap);
-  }
+    const rbs_loc *loc = ptr;
+    if (loc->children == NULL) {
+        return sizeof(rbs_loc);
+    } else {
+        return sizeof(rbs_loc) + RBS_LOC_CHILDREN_SIZE(loc->children->cap);
+    }
 }
 
 static rb_data_type_t location_type = {
-  "RBS::Location",
-  {rbs_loc_mark, (RUBY_DATA_FUNC)rbs_loc_free, rbs_loc_memsize},
-  0, 0, RUBY_TYPED_FREE_IMMEDIATELY
+    "RBS::Location",
+    { rbs_loc_mark, (RUBY_DATA_FUNC) rbs_loc_free, rbs_loc_memsize },
+    0,
+    0,
+    RUBY_TYPED_FREE_IMMEDIATELY
 };
 
 static VALUE location_s_allocate(VALUE klass) {
-  rbs_loc *loc;
-  VALUE obj = TypedData_Make_Struct(klass, rbs_loc, &location_type, loc);
+    rbs_loc *loc;
+    VALUE obj = TypedData_Make_Struct(klass, rbs_loc, &location_type, loc);
 
-  rbs_loc_init(loc, Qnil, RBS_LOC_NULL_RANGE);
+    rbs_loc_init(loc, Qnil, RBS_LOC_NULL_RANGE);
 
-  return obj;
+    return obj;
 }
 
 rbs_loc *rbs_check_location(VALUE obj) {
-  return rb_check_typeddata(obj, &location_type);
+    return rb_check_typeddata(obj, &location_type);
 }
 
 static VALUE location_initialize(VALUE self, VALUE buffer, VALUE start_pos, VALUE end_pos) {
-  rbs_loc *loc = rbs_check_location(self);
+    rbs_loc *loc = rbs_check_location(self);
 
-  int start = FIX2INT(start_pos);
-  int end = FIX2INT(end_pos);
+    int start = FIX2INT(start_pos);
+    int end = FIX2INT(end_pos);
 
-  *loc = (rbs_loc) {
-    .buffer = buffer,
-    .rg = (rbs_loc_range) { start, end },
-    .children = NULL,
-  };
+    *loc = (rbs_loc) {
+        .buffer = buffer,
+        .rg = (rbs_loc_range) { start, end },
+        .children = NULL,
+    };
 
-  return Qnil;
+    return Qnil;
 }
 
 static VALUE location_initialize_copy(VALUE self, VALUE other) {
-  rbs_loc *self_loc = rbs_check_location(self);
-  rbs_loc *other_loc = rbs_check_location(other);
-
-  *self_loc = (rbs_loc) {
-    .buffer = other_loc->buffer,
-    .rg = other_loc->rg,
-    .children = NULL,
-  };
-
-  if (other_loc->children != NULL) {
-    rbs_loc_legacy_alloc_children(self_loc, other_loc->children->cap);
-    memcpy(self_loc->children, other_loc->children, RBS_LOC_CHILDREN_SIZE(other_loc->children->cap));
-  }
+    rbs_loc *self_loc = rbs_check_location(self);
+    rbs_loc *other_loc = rbs_check_location(other);
+
+    *self_loc = (rbs_loc) {
+        .buffer = other_loc->buffer,
+        .rg = other_loc->rg,
+        .children = NULL,
+    };
+
+    if (other_loc->children != NULL) {
+        rbs_loc_legacy_alloc_children(self_loc, other_loc->children->cap);
+        memcpy(self_loc->children, other_loc->children, RBS_LOC_CHILDREN_SIZE(other_loc->children->cap));
+    }
 
-  return Qnil;
+    return Qnil;
 }
 
 static VALUE location_buffer(VALUE self) {
-  rbs_loc *loc = rbs_check_location(self);
-  return loc->buffer;
+    rbs_loc *loc = rbs_check_location(self);
+    return loc->buffer;
 }
 
 static VALUE location_start_pos(VALUE self) {
-  rbs_loc *loc = rbs_check_location(self);
-  return INT2FIX(loc->rg.start);
+    rbs_loc *loc = rbs_check_location(self);
+    return INT2FIX(loc->rg.start);
 }
 
 static VALUE location_end_pos(VALUE self) {
-  rbs_loc *loc = rbs_check_location(self);
-  return INT2FIX(loc->rg.end);
+    rbs_loc *loc = rbs_check_location(self);
+    return INT2FIX(loc->rg.end);
 }
 
 static rbs_constant_id_t rbs_constant_pool_insert_ruby_symbol(VALUE symbol) {
-  VALUE name = rb_sym2str(symbol);
+    VALUE name = rb_sym2str(symbol);
 
-  // Constants inserted here will never be freed, but that's acceptable because:
-  // 1. Most symbols passed into here will be the ones already inserted into the constant pool by `parser.c`.
-  // 2. Methods like `add_required_child` and `add_optional_child` will usually only get called with a few different symbols.
-  return rbs_constant_pool_insert_constant(RBS_GLOBAL_CONSTANT_POOL, (const uint8_t *) RSTRING_PTR(name), RSTRING_LEN(name));
+    // Constants inserted here will never be freed, but that's acceptable because:
+    // 1. Most symbols passed into here will be the ones already inserted into the constant pool by `parser.c`.
+    // 2. Methods like `add_required_child` and `add_optional_child` will usually only get called with a few different symbols.
+    return rbs_constant_pool_insert_constant(RBS_GLOBAL_CONSTANT_POOL, (const uint8_t *) RSTRING_PTR(name), RSTRING_LEN(name));
 }
 
 static VALUE location_add_required_child(VALUE self, VALUE name, VALUE start, VALUE end) {
-  rbs_loc *loc = rbs_check_location(self);
+    rbs_loc *loc = rbs_check_location(self);
 
-  rbs_range_t rg;
-  rg.start = rbs_loc_position(FIX2INT(start));
-  rg.end = rbs_loc_position(FIX2INT(end));
+    rbs_range_t rg;
+    rg.start = rbs_loc_position(FIX2INT(start));
+    rg.end = rbs_loc_position(FIX2INT(end));
 
-  rbs_loc_legacy_add_required_child(loc, rbs_constant_pool_insert_ruby_symbol(name), rg);
+    rbs_loc_legacy_add_required_child(loc, rbs_constant_pool_insert_ruby_symbol(name), rg);
 
-  return Qnil;
+    return Qnil;
 }
 
 static VALUE location_add_optional_child(VALUE self, VALUE name, VALUE start, VALUE end) {
-  rbs_loc *loc = rbs_check_location(self);
+    rbs_loc *loc = rbs_check_location(self);
 
-  rbs_range_t rg;
-  rg.start = rbs_loc_position(FIX2INT(start));
-  rg.end = rbs_loc_position(FIX2INT(end));
+    rbs_range_t rg;
+    rg.start = rbs_loc_position(FIX2INT(start));
+    rg.end = rbs_loc_position(FIX2INT(end));
 
-  rbs_loc_legacy_add_optional_child(loc, rbs_constant_pool_insert_ruby_symbol(name), rg);
+    rbs_loc_legacy_add_optional_child(loc, rbs_constant_pool_insert_ruby_symbol(name), rg);
 
-  return Qnil;
+    return Qnil;
 }
 
 static VALUE location_add_optional_no_child(VALUE self, VALUE name) {
-  rbs_loc *loc = rbs_check_location(self);
+    rbs_loc *loc = rbs_check_location(self);
 
-  rbs_loc_legacy_add_optional_child(loc, rbs_constant_pool_insert_ruby_symbol(name), NULL_RANGE);
+    rbs_loc_legacy_add_optional_child(loc, rbs_constant_pool_insert_ruby_symbol(name), NULL_RANGE);
 
-  return Qnil;
+    return Qnil;
 }
 
 VALUE rbs_new_location(VALUE buffer, rbs_range_t rg) {
-  rbs_loc *loc;
-  VALUE obj = TypedData_Make_Struct(RBS_Location, rbs_loc, &location_type, loc);
+    rbs_loc *loc;
+    VALUE obj = TypedData_Make_Struct(RBS_Location, rbs_loc, &location_type, loc);
 
-  rbs_loc_init(loc, buffer, rbs_new_loc_range(rg));
+    rbs_loc_init(loc, buffer, rbs_new_loc_range(rg));
 
-  return obj;
+    return obj;
 }
 
 static VALUE rbs_new_location_from_loc_range(VALUE buffer, rbs_loc_range rg) {
-  rbs_loc *loc;
-  VALUE obj = TypedData_Make_Struct(RBS_Location, rbs_loc, &location_type, loc);
+    rbs_loc *loc;
+    VALUE obj = TypedData_Make_Struct(RBS_Location, rbs_loc, &location_type, loc);
 
-  rbs_loc_init(loc, buffer, rg);
+    rbs_loc_init(loc, buffer, rg);
 
-  return obj;
+    return obj;
 }
 
 static rbs_constant_id_t rbs_constant_pool_find_ruby_symbol(VALUE symbol) {
-  VALUE name = rb_sym2str(symbol);
+    VALUE name = rb_sym2str(symbol);
 
-  return rbs_constant_pool_find(RBS_GLOBAL_CONSTANT_POOL, (const uint8_t *) RSTRING_PTR(name), RSTRING_LEN(name));
+    return rbs_constant_pool_find(RBS_GLOBAL_CONSTANT_POOL, (const uint8_t *) RSTRING_PTR(name), RSTRING_LEN(name));
 }
 
 static VALUE location_aref(VALUE self, VALUE name) {
-  rbs_loc *loc = rbs_check_location(self);
+    rbs_loc *loc = rbs_check_location(self);
 
-  rbs_constant_id_t id = rbs_constant_pool_find_ruby_symbol(name);
+    rbs_constant_id_t id = rbs_constant_pool_find_ruby_symbol(name);
 
-  if (loc->children != NULL && id != RBS_CONSTANT_ID_UNSET) {
-    for (unsigned short i = 0; i < loc->children->len; i++) {
-      if (loc->children->entries[i].name == id) {
-        rbs_loc_range result = loc->children->entries[i].rg;
+    if (loc->children != NULL && id != RBS_CONSTANT_ID_UNSET) {
+        for (unsigned short i = 0; i < loc->children->len; i++) {
+            if (loc->children->entries[i].name == id) {
+                rbs_loc_range result = loc->children->entries[i].rg;
 
-        if (RBS_LOC_OPTIONAL_P(loc, i) && NULL_LOC_RANGE_P(result)) {
-          return Qnil;
-        } else {
-          return rbs_new_location_from_loc_range(loc->buffer, result);
+                if (RBS_LOC_OPTIONAL_P(loc, i) && NULL_LOC_RANGE_P(result)) {
+                    return Qnil;
+                } else {
+                    return rbs_new_location_from_loc_range(loc->buffer, result);
+                }
+            }
         }
-      }
     }
-  }
 
-  VALUE string = rb_funcall(name, rb_intern("to_s"), 0);
-  rb_raise(rb_eRuntimeError, "Unknown child name given: %s", RSTRING_PTR(string));
+    VALUE string = rb_funcall(name, rb_intern("to_s"), 0);
+    rb_raise(rb_eRuntimeError, "Unknown child name given: %s", RSTRING_PTR(string));
 }
 
 static VALUE rbs_constant_to_ruby_symbol(rbs_constant_t *constant) {
-  return ID2SYM(rb_intern2((const char *) constant->start, constant->length));
+    return ID2SYM(rb_intern2((const char *) constant->start, constant->length));
 }
 
 static VALUE location_optional_keys(VALUE self) {
-  VALUE keys = rb_ary_new();
+    VALUE keys = rb_ary_new();
 
-  rbs_loc *loc = rbs_check_location(self);
-  rbs_loc_children *children = loc->children;
-  if (children == NULL) {
-    return keys;
-  }
+    rbs_loc *loc = rbs_check_location(self);
+    rbs_loc_children *children = loc->children;
+    if (children == NULL) {
+        return keys;
+    }
 
-  for (unsigned short i = 0; i < children->len; i++) {
-    if (RBS_LOC_OPTIONAL_P(loc, i)) {
-      rbs_constant_t *key_id = rbs_constant_pool_id_to_constant(RBS_GLOBAL_CONSTANT_POOL, children->entries[i].name);
-      VALUE key_sym = rbs_constant_to_ruby_symbol(key_id);
-      rb_ary_push(keys, key_sym);
+    for (unsigned short i = 0; i < children->len; i++) {
+        if (RBS_LOC_OPTIONAL_P(loc, i)) {
+            rbs_constant_t *key_id = rbs_constant_pool_id_to_constant(RBS_GLOBAL_CONSTANT_POOL, children->entries[i].name);
+            VALUE key_sym = rbs_constant_to_ruby_symbol(key_id);
+            rb_ary_push(keys, key_sym);
+        }
     }
-  }
 
-  return keys;
+    return keys;
 }
 
 static VALUE location_required_keys(VALUE self) {
-  VALUE keys = rb_ary_new();
+    VALUE keys = rb_ary_new();
 
-  rbs_loc *loc = rbs_check_location(self);
-  rbs_loc_children *children = loc->children;
-  if (children == NULL) {
-    return keys;
-  }
+    rbs_loc *loc = rbs_check_location(self);
+    rbs_loc_children *children = loc->children;
+    if (children == NULL) {
+        return keys;
+    }
 
-  for (unsigned short i = 0; i < children->len; i++) {
-    if (RBS_LOC_REQUIRED_P(loc, i)) {
-      rbs_constant_t *key_id = rbs_constant_pool_id_to_constant(RBS_GLOBAL_CONSTANT_POOL, children->entries[i].name);
-      VALUE key_sym = rbs_constant_to_ruby_symbol(key_id);
-      rb_ary_push(keys, key_sym);
+    for (unsigned short i = 0; i < children->len; i++) {
+        if (RBS_LOC_REQUIRED_P(loc, i)) {
+            rbs_constant_t *key_id = rbs_constant_pool_id_to_constant(RBS_GLOBAL_CONSTANT_POOL, children->entries[i].name);
+            VALUE key_sym = rbs_constant_to_ruby_symbol(key_id);
+            rb_ary_push(keys, key_sym);
+        }
     }
-  }
 
-  return keys;
+    return keys;
 }
 
 void rbs__init_location(void) {
-  RBS_Location = rb_define_class_under(RBS, "Location", rb_cObject);
-  rb_define_alloc_func(RBS_Location, location_s_allocate);
-  rb_define_private_method(RBS_Location, "initialize", location_initialize, 3);
-  rb_define_private_method(RBS_Location, "initialize_copy", location_initialize_copy, 1);
-  rb_define_method(RBS_Location, "buffer", location_buffer, 0);
-  rb_define_method(RBS_Location, "_start_pos", location_start_pos, 0);
-  rb_define_method(RBS_Location, "_end_pos", location_end_pos, 0);
-  rb_define_method(RBS_Location, "_add_required_child", location_add_required_child, 3);
-  rb_define_method(RBS_Location, "_add_optional_child", location_add_optional_child, 3);
-  rb_define_method(RBS_Location, "_add_optional_no_child", location_add_optional_no_child, 1);
-  rb_define_method(RBS_Location, "_optional_keys", location_optional_keys, 0);
-  rb_define_method(RBS_Location, "_required_keys", location_required_keys, 0);
-  rb_define_method(RBS_Location, "[]", location_aref, 1);
+    RBS_Location = rb_define_class_under(RBS, "Location", rb_cObject);
+    rb_define_alloc_func(RBS_Location, location_s_allocate);
+    rb_define_private_method(RBS_Location, "initialize", location_initialize, 3);
+    rb_define_private_method(RBS_Location, "initialize_copy", location_initialize_copy, 1);
+    rb_define_method(RBS_Location, "buffer", location_buffer, 0);
+    rb_define_method(RBS_Location, "_start_pos", location_start_pos, 0);
+    rb_define_method(RBS_Location, "_end_pos", location_end_pos, 0);
+    rb_define_method(RBS_Location, "_add_required_child", location_add_required_child, 3);
+    rb_define_method(RBS_Location, "_add_optional_child", location_add_optional_child, 3);
+    rb_define_method(RBS_Location, "_add_optional_no_child", location_add_optional_no_child, 1);
+    rb_define_method(RBS_Location, "_optional_keys", location_optional_keys, 0);
+    rb_define_method(RBS_Location, "_required_keys", location_required_keys, 0);
+    rb_define_method(RBS_Location, "[]", location_aref, 1);
 }

data/ext/rbs_extension/legacy_location.h

@@ -15,9 +15,9 @@ SUPPRESS_RUBY_HEADER_DIAGNOSTICS_END
 extern VALUE RBS_Location;
 
 typedef struct {
-  VALUE buffer;
-  rbs_loc_range rg;
-  rbs_loc_children *children; // NULL when no children is allocated
+    VALUE buffer;
+    rbs_loc_range rg;
+    rbs_loc_children *children; // NULL when no children is allocated
 } rbs_loc;
 
 /**