git push origin masterMerge branch 'suzukaze-add-string.clear2'

1 files changed, 33 insertions, 0 deletions

diff --git a/src/string.c b/src/string.c
index 842e83d76..6fa9271e1 100644
--- a/src/string.c
+++ b/src/string.c
@@ -2523,6 +2523,38 @@ mrb_str_bytes(mrb_state *mrb, mrb_value str)
   return a;
 }
 
+static inline void
+str_discard(mrb_state *mrb, mrb_value str) {
+  struct RString *s = mrb_str_ptr(str);
+
+  if (!STR_SHARED_P(s) && !STR_EMBED_P(s) && ((s->flags & MRB_STR_NOFREE) == 0)) {
+    mrb_free(mrb, s->as.heap.ptr);
+    RSTRING(str)->as.heap.ptr = 0;
+    RSTRING(str)->as.heap.len = 0;
+  }
+}
+
+/*
+ *  call-seq:
+ *     string.clear    ->  string
+ *
+ *  Makes string empty.
+ *
+ *     a = "abcde"
+ *     a.clear    #=> ""
+ */
+static mrb_value
+mrb_str_clear(mrb_state *mrb, mrb_value str)
+{
+  struct RString *s = mrb_str_ptr(str);
+
+  str_discard(mrb, str);
+  STR_SET_EMBED_FLAG(s);
+  STR_SET_EMBED_LEN(s, 0);
+  RSTRING_PTR(str)[0] = '\0';
+  return str;
+}
+
 /* ---------------------------*/
 void
 mrb_init_string(mrb_state *mrb)
@@ -2574,4 +2606,5 @@ mrb_init_string(mrb_state *mrb)
   mrb_define_method(mrb, s, "upcase!",         mrb_str_upcase_bang,     MRB_ARGS_REQ(1)); /* 15.2.10.5.43 */
   mrb_define_method(mrb, s, "inspect",         mrb_str_inspect,         MRB_ARGS_NONE()); /* 15.2.10.5.46(x) */
   mrb_define_method(mrb, s, "bytes",           mrb_str_bytes,           MRB_ARGS_NONE());
+  mrb_define_method(mrb, s, "clear",           mrb_str_clear,           MRB_ARGS_NONE());
 }