Implement rb_enc_mbcput

Author: bjfish

lib/cext/include/ruby/onigmo.h

@@ -373,10 +373,12 @@ int onigenc_mbclen_approximate(const OnigUChar* p,const OnigUChar* e, const stru
 #ifdef TRUFFLERUBY
 int rb_tr_code_to_mbclen(OnigCodePoint code, OnigEncodingType *encoding);
 #define ONIGENC_CODE_TO_MBCLEN(enc,code)       rb_tr_code_to_mbclen(code,enc)
+int rb_tr_code_to_mbc(OnigCodePoint code, UChar *buf, OnigEncoding enc);
+#define ONIGENC_CODE_TO_MBC(enc,code,buf)      rb_tr_code_to_mbc(code,buf,enc)
 #else
 #define ONIGENC_CODE_TO_MBCLEN(enc,code)       (enc)->code_to_mbclen(code,enc)
-#endif
 #define ONIGENC_CODE_TO_MBC(enc,code,buf)      (enc)->code_to_mbc(code,buf,enc)
+#endif
 #define ONIGENC_PROPERTY_NAME_TO_CTYPE(enc,p,end) \
   (enc)->property_name_to_ctype(enc,p,end)
 

spec/ruby/optional/capi/encoding_spec.rb

@@ -139,6 +139,15 @@
     end
   end
 
+  describe "rb_enc_mbcput" do
+    it "writes the correct bytes to the buffer" do
+      @s.rb_enc_mbcput(0x24, Encoding::UTF_8).should == "$"
+      @s.rb_enc_mbcput(0xA2, Encoding::UTF_8).should == "¢"
+      @s.rb_enc_mbcput(0x20AC, Encoding::UTF_8).should == "€"
+      @s.rb_enc_mbcput(0x24B62, Encoding::UTF_8).should == "𤭢"
+    end
+  end
+
   describe "rb_usascii_encoding" do
     it "returns the encoding for Encoding::US_ASCII" do
       @s.rb_usascii_encoding.should == "US-ASCII"

spec/ruby/optional/capi/ext/encoding_spec.c

@@ -127,6 +127,16 @@ static VALUE encoding_spec_rb_enc_mbc_to_codepoint(VALUE self, VALUE str, VALUE
   return INT2FIX(rb_enc_mbc_to_codepoint(p, e, rb_enc_get(str)));
 }
 
+static VALUE encoding_spec_rb_enc_mbcput(VALUE self, VALUE code, VALUE encoding) {
+  unsigned int c = FIX2UINT(code);
+  rb_encoding *enc = rb_to_encoding(encoding);
+  char *buf = (char*) malloc(10);
+  int len = rb_enc_mbcput(c, buf, enc);
+  VALUE str = rb_enc_str_new(buf, len, enc);
+  free(buf);
+  return str;
+}
+
 static VALUE encoding_spec_rb_enc_from_encoding(VALUE self, VALUE name) {
   return rb_enc_from_encoding(rb_enc_find(RSTRING_PTR(name)));
 }
@@ -324,6 +334,7 @@ void Init_encoding_spec(void) {
   rb_define_method(cls, "rb_enc_isspace", encoding_spec_rb_enc_isspace, 2);
   rb_define_method(cls, "rb_enc_from_index", encoding_spec_rb_enc_from_index, 1);
   rb_define_method(cls, "rb_enc_mbc_to_codepoint", encoding_spec_rb_enc_mbc_to_codepoint, 2);
+  rb_define_method(cls, "rb_enc_mbcput", encoding_spec_rb_enc_mbcput, 2);
   rb_define_method(cls, "rb_enc_from_encoding", encoding_spec_rb_enc_from_encoding, 1);
   rb_define_method(cls, "rb_enc_get", encoding_spec_rb_enc_get, 1);
   rb_define_method(cls, "rb_enc_precise_mbclen", encoding_spec_rb_enc_precise_mbclen, 2);

src/main/c/cext/encoding.c

@@ -366,3 +366,13 @@ int rb_tr_enc_mbc_case_fold(rb_encoding *enc, int flag, const UChar** p, const U
    }
    return result_len;
 }
+
+int rb_tr_code_to_mbc(OnigCodePoint code, UChar *buf, OnigEncoding enc) {
+    VALUE result_str = rb_tr_wrap(polyglot_invoke(RUBY_CEXT, "rb_tr_code_to_mbc",
+            rb_tr_unwrap(rb_enc_from_encoding(enc)), code));
+   int result_len = RSTRING_LEN(result_str);
+   if (result_len > 0) {
+     strncpy((char *)buf, RSTRING_PTR(result_str), result_len);
+   }
+   return result_len;
+}

src/main/java/org/truffleruby/cext/CExtNodes.java

@@ -1310,6 +1310,25 @@ protected int getCacheLimit() {
 
     }
 
+    @CoreMethod(names = "rb_tr_code_to_mbc", onSingleton = true, required = 2, lowerFixnum = 2)
+    public abstract static class RbTrMbcPutNode extends CoreMethodArrayArgumentsNode {
+
+        @Specialization(guards = "isRubyEncoding(enc)")
+        protected Object rbTrEncMbcPut(DynamicObject enc, int code) {
+            final Encoding encoding = EncodingOperations.getEncoding(enc);
+            final byte buf[] = new byte[org.jcodings.Config.ENC_CODE_TO_MBC_MAXLEN];
+            final int resultLength = encoding.codeToMbc(code, buf, 0);
+            final byte result[] = new byte[resultLength];
+            if (resultLength > 0) {
+                System.arraycopy(buf, 0, result, 0, resultLength);
+            }
+            return StringOperations.createString(
+                    getContext(),
+                    RopeOperations.create(result, USASCIIEncoding.INSTANCE, CodeRange.CR_UNKNOWN));
+        }
+
+    }
+
     @CoreMethod(names = "rb_enc_mbmaxlen", onSingleton = true, required = 1)
     public abstract static class RbEncMaxLenNode extends CoreMethodArrayArgumentsNode {