language-servers/erg
1
0
Fork 0
mirror of https://github.com/erg-lang/erg.git synced 2025-09-27 03:49:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: ref/ref! with type specification

Browse source
This commit is contained in:
Shunsuke Shibayama 2024-12-27 02:46:58 +09:00
parent 6d2c399d67
commit 1fe189e4d3
4 changed files with 65 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

60
crates/erg_parser/convert.rs
View file

@ -521,27 +521,63 @@ impl Parser {
Expr::UnaryOp(unary) => match unary.op.kind {
TokenKind::RefOp => {
let var = unary.args.into_iter().next().unwrap();
let Expr::Accessor(Accessor::Ident(var)) = *var else {
let err = ParseError::simple_syntax_error(line!() as usize, var.loc());
self.errs.push(err);
debug_exit_info!(self);
return Err(());
let (var, t_spec) = match *var {
Expr::Accessor(Accessor::Ident(var)) => (var, None),
Expr::TypeAscription(tasc) => {
let var = match *tasc.expr {
Expr::Accessor(Accessor::Ident(var)) => var,
_ => {
let err = ParseError::simple_syntax_error(
line!() as usize,
tasc.loc(),
);
self.errs.push(err);
debug_exit_info!(self);
return Err(());
}
};
(var, Some(tasc.t_spec))
}
_ => {
let err = ParseError::simple_syntax_error(line!() as usize, var.loc());
self.errs.push(err);
debug_exit_info!(self);
return Err(());
}
};
let pat = ParamPattern::Ref(var.name);
let param = NonDefaultParamSignature::new(pat, None);
let param = NonDefaultParamSignature::new(pat, t_spec);
debug_exit_info!(self);
Ok(param)
}
TokenKind::RefMutOp => {
let var = unary.args.into_iter().next().unwrap();
let Expr::Accessor(Accessor::Ident(var)) = *var else {
let err = ParseError::simple_syntax_error(line!() as usize, var.loc());
self.errs.push(err);
debug_exit_info!(self);
return Err(());
let (var, t_spec) = match *var {
Expr::Accessor(Accessor::Ident(var)) => (var, None),
Expr::TypeAscription(tasc) => {
let var = match *tasc.expr {
Expr::Accessor(Accessor::Ident(var)) => var,
_ => {
let err = ParseError::simple_syntax_error(
line!() as usize,
tasc.loc(),
);
self.errs.push(err);
debug_exit_info!(self);
return Err(());
}
};
(var, Some(tasc.t_spec))
}
_ => {
let err = ParseError::simple_syntax_error(line!() as usize, var.loc());
self.errs.push(err);
debug_exit_info!(self);
return Err(());
}
};
let pat = ParamPattern::RefMut(var.name);
let param = NonDefaultParamSignature::new(pat, None);
let param = NonDefaultParamSignature::new(pat, t_spec);
debug_exit_info!(self);
Ok(param)
}