Merge pull request #54 from sue445/rb_ivar_set

Add rb_ivar_set

Author: sue445

README.md

@@ -239,7 +239,7 @@ ref. https://docs.ruby-lang.org/ja/latest/function/index.html
 * [ ] `rb_iv_set`
 * [ ] `rb_ivar_defined`
 * [x] `rb_ivar_get`
-* [ ] `rb_ivar_set`
+* [x] `rb_ivar_set`
 * [ ] `rb_jump_tag`
 * [ ] `rb_lastline_get`
 * [ ] `rb_lastline_set`

ruby-internal-intern-variable.go

@@ -15,3 +15,12 @@ import "C"
 func RbIvarGet(obj VALUE, name ID) VALUE {
 	return VALUE(C.rb_ivar_get(C.VALUE(obj), C.ID(name)))
 }
+
+// RbIvarSet calls `rb_ivar_set` in C
+//
+// Original definition is following
+//
+//	VALUE rb_ivar_set(VALUE obj, ID name, VALUE val)
+func RbIvarSet(obj VALUE, name ID, val VALUE) VALUE {
+	return VALUE(C.rb_ivar_set(C.VALUE(obj), C.ID(name), C.VALUE(val)))
+}

testdata/dummy/ext/dummy/dummy.go

@@ -11,6 +11,7 @@ VALUE rb_dummy_to_string(VALUE self, VALUE source);
 VALUE rb_dummy_max(VALUE self, VALUE a, VALUE b);
 
 VALUE rb_dummy_unit_kilobyte(VALUE self);
+VALUE rb_dummy_unit_increment(VALUE self);
 */
 import "C"
 
@@ -72,6 +73,20 @@ func rb_dummy_unit_kilobyte(self C.VALUE) C.VALUE {
 	return C.VALUE(ruby.INT2NUM(result))
 }
 
+//export rb_dummy_unit_increment
+func rb_dummy_unit_increment(self C.VALUE) C.VALUE {
+	sourceID := ruby.RbIntern("@source")
+	sourceValue := ruby.RbIvarGet(ruby.VALUE(self), sourceID)
+
+	sourceInt := ruby.NUM2INT(sourceValue)
+	sourceInt++
+
+	result := ruby.INT2NUM(sourceInt)
+	ruby.RbIvarSet(ruby.VALUE(self), sourceID, result)
+
+	return C.VALUE(result)
+}
+
 //export rb_dummy_max
 func rb_dummy_max(_ C.VALUE, a C.VALUE, b C.VALUE) C.VALUE {
 	aLong := ruby.NUM2LONG(ruby.VALUE(a))
@@ -108,6 +123,7 @@ func Init_dummy() {
 	// Dummy::Unit
 	rb_cUnit := ruby.RbDefineClassUnder(rb_mDummy, "Unit", ruby.VALUE(C.rb_cObject))
 	ruby.RbDefineMethod(rb_cUnit, "kilobyte", C.rb_dummy_unit_kilobyte, 0)
+	ruby.RbDefineMethod(rb_cUnit, "increment", C.rb_dummy_unit_increment, 0)
 }
 
 func main() {

testdata/dummy/lib/dummy/unit.rb

@@ -2,6 +2,8 @@
 
 module Dummy
   class Unit # rubocop:disable Style/Documentation
+    attr_accessor :source
+
     # @param source [Integer]
     def initialize(source)
       @source = source

testdata/dummy/sig/dummy.rbs

@@ -26,6 +26,8 @@ module Dummy
   class Unit
     @source: Integer
 
+    attr_accessor source: Integer
+
     def initialize: (Integer source) -> void
 
     def kilobyte: () -> Integer

testdata/dummy/spec/dummy/unit_spec.rb

@@ -7,4 +7,14 @@
       expect(unit.kilobyte).to eq 2048
     end
   end
+
+  describe "#increment" do
+    it "works" do
+      unit = Dummy::Unit.new(1)
+      actual = unit.increment
+
+      expect(actual).to eq 2
+      expect(unit.source).to eq 2
+    end
+  end
 end