summaryrefslogtreecommitdiffhomepage
path: root/mrbgems/mruby-string-utf8
diff options
context:
space:
mode:
Diffstat (limited to 'mrbgems/mruby-string-utf8')
-rw-r--r--mrbgems/mruby-string-utf8/src/string.c34
-rw-r--r--mrbgems/mruby-string-utf8/test/string.rb16
2 files changed, 50 insertions, 0 deletions
diff --git a/mrbgems/mruby-string-utf8/src/string.c b/mrbgems/mruby-string-utf8/src/string.c
index 5379dd3aa..c377726a2 100644
--- a/mrbgems/mruby-string-utf8/src/string.c
+++ b/mrbgems/mruby-string-utf8/src/string.c
@@ -247,6 +247,38 @@ mrb_str_aref_m(mrb_state *mrb, mrb_value str)
return mrb_str_aref(mrb, str, a1);
}
+static mrb_value
+mrb_str_reverse_bang(mrb_state *mrb, mrb_value str)
+{
+ int utf8_len = mrb_utf8_strlen(str);
+ if (utf8_len > 1) {
+ int len = RSTRING_LEN(str);
+ char *buf = (char *)mrb_malloc(mrb, len);
+ unsigned char* p = (unsigned char*)buf;
+ unsigned char* e = (unsigned char*)buf + len;
+ unsigned char* r = (unsigned char*)RSTRING_END(str);
+
+ memcpy(buf, RSTRING_PTR(str), len);
+ mrb_str_modify(mrb, mrb_str_ptr(str));
+
+ while (p<e) {
+ int clen = utf8len(p);
+ r -= clen;
+ memcpy(r, p, clen);
+ p += clen;
+ }
+ mrb_free(mrb, buf);
+ }
+
+ return str;
+}
+
+static mrb_value
+mrb_str_reverse(mrb_state *mrb, mrb_value str)
+{
+ return mrb_str_reverse_bang(mrb, mrb_str_dup(mrb, str));
+}
+
void
mrb_mruby_string_utf8_gem_init(mrb_state* mrb)
{
@@ -255,6 +287,8 @@ mrb_mruby_string_utf8_gem_init(mrb_state* mrb)
mrb_define_method(mrb, s, "size", mrb_str_size, MRB_ARGS_NONE());
mrb_define_method(mrb, s, "[]", mrb_str_aref_m, MRB_ARGS_ANY());
mrb_define_method(mrb, s, "slice", mrb_str_aref_m, MRB_ARGS_ANY());
+ mrb_define_method(mrb, s, "reverse", mrb_str_reverse, MRB_ARGS_NONE());
+ mrb_define_method(mrb, s, "reverse!", mrb_str_reverse_bang, MRB_ARGS_NONE());
}
void
diff --git a/mrbgems/mruby-string-utf8/test/string.rb b/mrbgems/mruby-string-utf8/test/string.rb
index 4ee815a36..fc21c0a93 100644
--- a/mrbgems/mruby-string-utf8/test/string.rb
+++ b/mrbgems/mruby-string-utf8/test/string.rb
@@ -9,3 +9,19 @@ assert('String#[]') do
assert_equal "んに", "こんにちわ世界"[1,2]
assert_equal "世", "こんにちわ世界"["世"]
end
+
+assert('String#reverse', '15.2.10.5.29') do
+ a = 'こんにちわ世界!'
+ a.reverse
+
+ assert_equal 'こんにちわ世界!', a
+ assert_equal '!界世わちにんこ', 'こんにちわ世界!'.reverse
+end
+
+assert('String#reverse!', '15.2.10.5.30') do
+ a = 'こんにちわ世界!'
+ a.reverse!
+
+ assert_equal '!界世わちにんこ', a
+ assert_equal '!界世わちにんこ', 'こんにちわ世界!'.reverse!
+end