String strip! family

.gitignore

@@ -6,6 +6,7 @@ coverage-report
 
 # Editor tools
 .ccls
+.ccls-cache
 compile_commands.json
 tags
 .gdb_history

include/natalie/string_object.hpp

@@ -152,15 +152,18 @@ class StringObject : public Object {
     Value force_encoding(Env *, Value);
     Value ljust(Env *, Value, Value);
     Value lstrip(Env *) const;
+    Value lstrip_bang(Env *);
     Value match(Env *, Value);
     Value mul(Env *, Value) const;
     Value ord(Env *);
     Value ref(Env *, Value);
     Value reverse(Env *);
     Value rstrip(Env *) const;
+    Value rstrip_bang(Env *);
     Value size(Env *);
     Value split(Env *, Value, Value);
     Value strip(Env *) const;
+    Value strip_bang(Env *);
     Value to_i(Env *, Value) const;
     Value upcase(Env *);
     Value uplus(Env *);

lib/natalie/compiler/binding_gen.rb

@@ -742,14 +742,17 @@ def generate_name
 gen.binding('String', 'length', 'StringObject', 'length', argc: 0, pass_env: false, pass_block: false, return_type: :size_t)
 gen.binding('String', 'ljust', 'StringObject', 'ljust', argc: 1..2, pass_env: true, pass_block: false, return_type: :Object)
 gen.binding('String', 'lstrip', 'StringObject', 'lstrip', argc: 0, pass_env: true, pass_block: false, return_type: :Object)
+gen.binding('String', 'lstrip!', 'StringObject', 'lstrip_bang', argc: 0, pass_env: true, pass_block: false, return_type: :Object)
 gen.binding('String', 'match', 'StringObject', 'match', argc: 1, pass_env: true, pass_block: false, return_type: :Object)
 gen.binding('String', 'ord', 'StringObject', 'ord', argc: 0, pass_env: true, pass_block: false, return_type: :Object)
 gen.binding('String', 'reverse', 'StringObject', 'reverse', argc: 0, pass_env: true, pass_block: false, return_type: :Object)
 gen.binding('String', 'rstrip', 'StringObject', 'rstrip', argc: 0, pass_env: true, pass_block: false, return_type: :Object)
+gen.binding('String', 'rstrip!', 'StringObject', 'rstrip_bang', argc: 0, pass_env: true, pass_block: false, return_type: :Object)
 gen.binding('String', 'size', 'StringObject', 'size', argc: 0, pass_env: true, pass_block: false, return_type: :Object)
 gen.binding('String', 'split', 'StringObject', 'split', argc: 0..2, pass_env: true, pass_block: false, return_type: :Object)
 gen.binding('String', 'start_with?', 'StringObject', 'start_with', argc: 1, pass_env: true, pass_block: false, return_type: :bool)
 gen.binding('String', 'strip', 'StringObject', 'strip', argc: 0, pass_env: true, pass_block: false, return_type: :Object)
+gen.binding('String', 'strip!', 'StringObject', 'strip_bang', argc: 0, pass_env: true, pass_block: false, return_type: :Object)
 gen.binding('String', 'sub', 'StringObject', 'sub', argc: 1..2, pass_env: true, pass_block: true, return_type: :Object)
 gen.binding('String', 'succ', 'StringObject', 'successive', argc: 0, pass_env: true, pass_block: false, return_type: :Object)
 gen.binding('String', 'to_i', 'StringObject', 'to_i', argc: 0..1, pass_env: true, pass_block: false, return_type: :Object)

spec/core/string/lstrip_spec.rb

@@ -28,8 +28,7 @@
   end
 end
 
-# NATFIXME: implement mutating lstrip! method
-xdescribe "String#lstrip!" do
+describe "String#lstrip!" do
   it "modifies self in place and returns self" do
     a = "  hello  "
     a.lstrip!.should equal(a)

spec/core/string/rstrip_spec.rb

@@ -26,8 +26,7 @@
   end
 end
 
-# NATFIXME: implement mutating rstrip! method
-xdescribe "String#rstrip!" do
+describe "String#rstrip!" do
   it "modifies self in place and returns self" do
     a = "  hello  "
     a.rstrip!.should equal(a)

spec/core/string/strip_spec.rb

@@ -26,8 +26,7 @@
   end
 end
 
-# NATFIXME: implement mutating strip! method
-xdescribe "String#strip!" do
+describe "String#strip!" do
   it "modifies self in place and returns self" do
     a = "   hello   "
     a.strip!.should equal(a)

src/string_object.cpp

@@ -652,6 +652,14 @@ Value StringObject::strip(Env *env) const {
     }
 }
 
+Value StringObject::strip_bang(Env *env) {
+    // right side needs to go first beacuse then we have less to move in
+    // on the left side
+    auto r = rstrip_bang(env);
+    auto l = lstrip_bang(env);
+    return l->is_nil() && r->is_nil() ? Value(NilObject::the()) : Value(this);
+}
+
 Value StringObject::lstrip(Env *env) const {
     if (length() == 0)
         return new StringObject {};
@@ -671,6 +679,32 @@ Value StringObject::lstrip(Env *env) const {
     }
 }
 
+Value StringObject::lstrip_bang(Env *env) {
+    if (is_frozen()) {
+        auto frozenError = GlobalEnv::the()->Object()->const_find(env, SymbolObject::intern("FrozenError"))->as_class();
+        env->raise_exception(new ExceptionObject(frozenError));
+    }
+
+    if (length() == 0)
+        return NilObject::the();
+
+    assert(length() < NAT_INT_MAX);
+    nat_int_t first_char;
+    nat_int_t len = static_cast<nat_int_t>(length());
+    for (first_char = 0; first_char < len; first_char++) {
+        char c = c_str()[first_char];
+        if (!is_strippable_whitespace(c))
+            break;
+    }
+
+    if (first_char == 0)
+        return NilObject::the();
+
+    memcpy(&m_string[0], &m_string[0] + first_char, len - first_char);
+    m_string.truncate(len - first_char);
+    return this;
+}
+
 Value StringObject::rstrip(Env *env) const {
     if (length() == 0)
         return new StringObject {};
@@ -690,6 +724,31 @@ Value StringObject::rstrip(Env *env) const {
     }
 }
 
+Value StringObject::rstrip_bang(Env *env) {
+    if (is_frozen()) {
+        auto frozenError = GlobalEnv::the()->Object()->const_find(env, SymbolObject::intern("FrozenError"))->as_class();
+        env->raise_exception(new ExceptionObject(frozenError));
+    }
+
+    if (length() == 0)
+        return NilObject::the();
+
+    assert(length() < NAT_INT_MAX);
+    nat_int_t last_char;
+    nat_int_t len = static_cast<nat_int_t>(length());
+    for (last_char = len - 1; last_char >= 0; last_char--) {
+        char c = c_str()[last_char];
+        if (!is_strippable_whitespace(c))
+            break;
+    }
+
+    if (last_char == len - 1)
+        return NilObject::the();
+
+    m_string.truncate(last_char < 0 ? 0 : last_char + 1);
+    return this;
+}
+
 Value StringObject::downcase(Env *env) {
     auto ary = chars(env);
     auto str = new StringObject {};