language-servers/erg
1
0
Fork 0
mirror of https://github.com/erg-lang/erg.git synced 2025-08-04 18:58:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add const Str::{endswith, find, join, startswith}

Browse source
This commit is contained in:
Shunsuke Shibayama 2024-01-30 12:37:07 +09:00
parent 284146843b
commit d6e6914662
2 changed files with 132 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

72
crates/erg_compiler/context/initialize/classes.rs
View file

@ -847,18 +847,35 @@ impl Context {
Immutable,
Visibility::BUILTIN_PUBLIC,
);
str_.register_builtin_py_impl(
let t_s_startswith = fn1_met(Str, Str, Bool);
let s_startswith = ValueObj::Subr(ConstSubr::Builtin(BuiltinConstSubr::new(
FUNC_STARTSWITH,
str_startswith,
t_s_startswith.clone(),
None,
)));
str_.register_py_builtin_const(
FUNC_STARTSWITH,
fn1_met(Str, Str, Bool),
Immutable,
Visibility::BUILTIN_PUBLIC,
Some(t_s_startswith),
s_startswith,
Some(FUNC_STARTSWITH),
None,
);
str_.register_py_builtin(
let t_s_endswith = fn1_met(Str, Str, Bool);
let s_endswith = ValueObj::Subr(ConstSubr::Builtin(BuiltinConstSubr::new(
FUNC_ENDSWITH,
fn1_met(Str, Str, Bool),
str_endswith,
t_s_endswith.clone(),
None,
)));
str_.register_py_builtin_const(
FUNC_ENDSWITH,
Visibility::BUILTIN_PUBLIC,
Some(t_s_endswith),
s_endswith,
Some(FUNC_ENDSWITH),
69,
Some(69),
);
str_.register_builtin_py_impl(
FUNC_SPLIT,
@ -888,12 +905,20 @@ impl Context {
Visibility::BUILTIN_PUBLIC,
Some(FUNC_SPLITLINES),
);
str_.register_builtin_py_impl(
let t_s_join = fn1_met(Str, poly(ITERABLE, vec![ty_tp(Str)]), Str);
let s_join = ValueObj::Subr(ConstSubr::Builtin(BuiltinConstSubr::new(
FUNC_JOIN,
str_join,
t_s_join.clone(),
None,
)));
str_.register_py_builtin_const(
FUNC_JOIN,
fn1_met(Str, poly(ITERABLE, vec![ty_tp(Str)]), Str),
Immutable,
Visibility::BUILTIN_PUBLIC,
Some(t_s_join),
s_join,
Some(FUNC_JOIN),
None,
);
str_.register_py_builtin(
FUNC_INDEX,
@ -922,18 +947,27 @@ impl Context {
Visibility::BUILTIN_PUBLIC,
Some(FUNC_RINDEX),
);
str_.register_py_builtin(
let t_s_find = fn_met(
Str,
vec![kw(KW_SUB, Str)],
None,
vec![kw(KW_START, Nat), kw(KW_END, Nat)],
None,
or(Nat, v_enum(set! {(-1).into()})),
);
let s_find = ValueObj::Subr(ConstSubr::Builtin(BuiltinConstSubr::new(
FUNC_FIND,
fn_met(
Str,
vec![kw(KW_SUB, Str)],
None,
vec![kw(KW_START, Nat), kw(KW_END, Nat)],
None,
or(Nat, v_enum(set! {(-1).into()})),
),
str_find,
t_s_find.clone(),
None,
)));
str_.register_py_builtin_const(
FUNC_FIND,
Visibility::BUILTIN_PUBLIC,
Some(t_s_find),
s_find,
Some(FUNC_FIND),
93,
Some(93),
);
str_.register_builtin_py_impl(
FUNC_RFIND,