implement Symbol#succ

MacRuby · Jun 23, 2012 · c9e48eb · c9e48eb
1 parent 64299cd
commit c9e48eb
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 6 deletions.
diff --git a/include/ruby/intern.h b/include/ruby/intern.h
@@ -562,6 +562,7 @@ VALUE rb_str_intern(VALUE);
 VALUE rb_sym_to_s(VALUE);
 VALUE rb_str_length(VALUE);
 VALUE rb_str_match(VALUE, VALUE);
+VALUE rb_str_succ(VALUE);
 
 // Return a string object appropriate for bstr_ calls. This does nothing for
 // data/binary RubyStrings.

diff --git a/spec/frozen/tags/macruby/core/symbol/succ_tags.txt b/spec/frozen/tags/macruby/core/symbol/succ_tags.txt
diff --git a/string.c b/string.c
@@ -5127,6 +5127,12 @@ rstr_succ(VALUE str, SEL sel)
     return newstr;
 }
 
+VALUE
+rb_str_succ(VALUE str)
+{
+    return rstr_succ(str, 0);
+}
+
 /*
  *  call-seq:
  *     str.succ!   => str

diff --git a/symbol.c b/symbol.c
@@ -323,6 +323,20 @@ Init_PreSymbol(void)
     }
 }
 
+/*
+ * call-seq:
+ *
+ *   sym.succ
+ *
+ * Same as <code>sym.to_s.succ.intern</code>.
+ */
+
+static VALUE
+rsym_succ(VALUE sym, SEL sel)
+{
+    return rb_str_intern(rb_str_succ(rb_sym_to_s(sym)));
+}
+
 /*
  * call-seq:
  *
@@ -818,6 +832,8 @@ Init_Symbol(void)
     rb_objc_define_method(rb_cSymbol, "downcase", rsym_downcase, 0);
     rb_objc_define_method(rb_cSymbol, "swapcase", rsym_swapcase, 0);
     rb_objc_define_method(rb_cSymbol, "capitalize", rsym_capitalize, 0);
+    rb_objc_define_method(rb_cSymbol, "succ", rsym_succ, 0);
+    rb_objc_define_method(rb_cSymbol, "next", rsym_succ, 0);
 
     // Cocoa primitives.
     rb_objc_install_method2((Class)rb_cSymbol, "copy",