mirror of
https://github.com/mruby/mruby
synced 2026-06-08 16:11:16 +00:00
mruby-string-ext: new method String#force_encoding
It takes a string instead of Encoding object for simplicity
This commit is contained in:
@@ -1359,6 +1359,29 @@ str_b(mrb_state *mrb, mrb_value self)
|
||||
return str;
|
||||
}
|
||||
|
||||
static mrb_value
|
||||
str_force_encoding(mrb_state *mrb, mrb_value self)
|
||||
{
|
||||
mrb_value enc;
|
||||
|
||||
mrb_get_args(mrb, "S", &enc);
|
||||
|
||||
struct RString *s = mrb_str_ptr(self);
|
||||
if (ENC_COMP_P(enc, ENC_ASCII_8BIT) ||
|
||||
ENC_COMP_P(enc, ENC_BINARY)) {
|
||||
s->flags |= MRB_STR_BINARY;
|
||||
}
|
||||
#ifdef MRB_UTF8_STRING
|
||||
else if (ENC_COMP_P(enc, ENC_UTF8)) {
|
||||
s->flags &= ~MRB_STR_BINARY;
|
||||
}
|
||||
#endif
|
||||
else {
|
||||
mrb_raisef(mrb, E_ARGUMENT_ERROR, "unknown encoding name - %v", enc);
|
||||
}
|
||||
return self;
|
||||
}
|
||||
|
||||
void
|
||||
mrb_mruby_string_ext_gem_init(mrb_state* mrb)
|
||||
{
|
||||
@@ -1400,6 +1423,7 @@ mrb_mruby_string_ext_gem_init(mrb_state* mrb)
|
||||
mrb_define_method_id(mrb, s, MRB_SYM_Q(valid_encoding), str_valid_enc_p, MRB_ARGS_NONE());
|
||||
mrb_define_method_id(mrb, s, MRB_SYM_Q(ascii_only), str_ascii_only_p, MRB_ARGS_NONE());
|
||||
mrb_define_method_id(mrb, s, MRB_SYM(b), str_b, MRB_ARGS_NONE());
|
||||
mrb_define_method_id(mrb, s, MRB_SYM(force_encoding), str_force_encoding, MRB_ARGS_REQ(1));
|
||||
|
||||
mrb_define_method_id(mrb, s, MRB_SYM(__lines), str_lines, MRB_ARGS_NONE());
|
||||
mrb_define_method_id(mrb, s, MRB_SYM(__codepoints), str_codepoints, MRB_ARGS_NONE());
|
||||
|
||||
Reference in New Issue
Block a user