language-servers/erg
1
0
Fork 0
mirror of https://github.com/erg-lang/erg.git synced 2025-08-31 15:48:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add Array.update_nth!

Browse source
This commit is contained in:
Shunsuke Shibayama 2023-08-28 15:33:58 +09:00
parent 07e37c878c
commit 74bca70017
7 changed files with 87 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

7
crates/erg_parser/ast.rs
View file

@ -3321,6 +3321,13 @@ impl Identifier {
)
}
pub fn public_with_loc(dot: Token, name: Str, loc: Location) -> Self {
Self::new(
VisModifierSpec::Public(dot),
VarName::from_str_and_loc(name, loc),
)
}
pub fn public_from_token(dot: Token, symbol: Token) -> Self {
Self::new(VisModifierSpec::Public(dot), VarName::new(symbol))
}

12
crates/erg_parser/desugar.rs
View file

@ -1465,14 +1465,14 @@ impl Desugarer {
match acc {
// x[y] => x.__getitem__(y)
Accessor::Subscr(subscr) => {
let loc = subscr.loc();
let args = Args::single(PosArg::new(Self::rec_desugar_acc(*subscr.index)));
let line = subscr.obj.ln_begin().unwrap_or(1);
let call = Call::new(
Self::rec_desugar_acc(*subscr.obj),
Some(Identifier::public_with_line(
Some(Identifier::public_with_loc(
DOT,
Str::ever("__getitem__"),
line,
loc,
)),
args,
);
@ -1480,14 +1480,14 @@ impl Desugarer {
}
// x.0 => x.__Tuple_getitem__(0)
Accessor::TupleAttr(tattr) => {
let loc = tattr.loc();
let args = Args::single(PosArg::new(Expr::Literal(tattr.index)));
let line = tattr.obj.ln_begin().unwrap_or(1);
let call = Call::new(
Self::rec_desugar_acc(*tattr.obj),
Some(Identifier::public_with_line(
Some(Identifier::public_with_loc(
DOT,
Str::ever("__Tuple_getitem__"),
line,
loc,
)),
args,
);