From b05014f6a9666b66935c33543edd9fcf451ae3a1 Mon Sep 17 00:00:00 2001
From: Yukihiro Matsumoto <matz@ruby-lang.org>
Date: Thu, 31 May 2012 16:31:33 +0900
Subject: reimplement String#capitalize

---
 src/string.c | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/string.c b/src/string.c
index 1511c4f88..3417b63d6 100644
--- a/src/string.c
+++ b/src/string.c
@@ -794,6 +794,22 @@ mrb_str_aref_m(mrb_state *mrb, mrb_value str)
 static mrb_value
 mrb_str_capitalize_bang(mrb_state *mrb, mrb_value str)
 {
+  char *s, *send;
+  int modify = 0;
+
+  if (RSTRING_LEN(str) == 0 || !RSTRING_PTR(str)) return mrb_nil_value();
+  s = RSTRING_PTR(str); send = RSTRING_END(str);
+  if (ISLOWER(*s)) {
+    *s = toupper(*s);
+    modify = 1;
+  }
+  while (++s < send) {
+    if (ISUPPER(*s)) {
+      *s = tolower(*s);
+      modify = 1;
+    }
+  }
+  if (modify) return str;
   return mrb_nil_value();
 }
 
-- 
cgit v1.2.3