use erg_common::consts::PYTHON_MODE;
use erg_common::error::Location;
use erg_common::fresh::FreshNameGenerator;
use erg_common::traits::{Locational, Stream};
use erg_common::{debug_power_assert, Str};
use erg_common::{enum_unwrap, get_hash, log, set};
use crate::ast::{
Accessor, Args, BinOp, Block, Call, ClassAttr, ClassAttrs, ClassDef, Compound, ConstExpr,
DataPack, Def, DefBody, DefId, DefaultParamSignature, Dict, Dummy, Expr, GuardClause,
Identifier, InlineModule, KeyValue, KwArg, Lambda, LambdaSignature, List, ListComprehension,
ListTypeSpec, ListWithLength, Literal, Methods, MixedRecord, Module, NonDefaultParamSignature,
NormalDict, NormalList, NormalRecord, NormalSet, NormalTuple, ParamPattern, ParamRecordAttr,
ParamTuplePattern, Params, PatchDef, PosArg, ReDef, Record, RecordAttrOrIdent, RecordAttrs,
RecordTypeSpec, Set as astSet, SetComprehension, SetWithLength, Signature, SubrSignature,
Tuple, TupleTypeSpec, TypeAppArgs, TypeAppArgsKind, TypeBoundSpecs, TypeSpec, TypeSpecWithOp,
UnaryOp, VarName, VarPattern, VarRecordAttr, VarSignature, VisModifierSpec, AST,
};
use crate::token::{Token, TokenKind, COLON, DOT};
pub fn symop_to_dname(op: &str) -> Option<&'static str> {
match op {
"`_+_`" => Some("__add__"),
"`_-_`" => Some("__sub__"),
"`*`" | "`cross`" => Some("__mul__"),
"`/`" => Some("__div__"),
"`//`" => Some("__floordiv__"),
"`**`" => Some("__pow__"),
"`%`" => Some("__mod__"),
"`@`" | "`dot`" => Some("__matmul__"),
"`&&`" => Some("__and__"),
"`||`" => Some("__or__"),
"`^`" => Some("__xor__"),
"`==`" => Some("__eq__"),
"`!=`" => Some("__ne__"),
"`<`" => Some("__lt__"),
"`<=`" => Some("__le__"),
"`>`" => Some("__gt__"),
"`>=`" => Some("__ge__"),
"`<<`" => Some("__lshift__"),
"`>>`" => Some("__rshift__"),
"`+_`" => Some("__pos__"),
"`-_`" => Some("__neg__"),
"`~`" => Some("__invert__"),
"`!`" => Some("__mutate__"),
"`...`" => Some("__spread__"),
_ => None,
}
}
#[derive(Debug, Clone, PartialEq, Eq)]
enum BufIndex<'i> {
List(usize),
Tuple(usize),
Record(&'i Identifier),
}
#[derive(Debug)]
pub struct Desugarer {
var_gen: FreshNameGenerator,
}
impl Desugarer {
pub fn new() -> Desugarer {
Self {
var_gen: FreshNameGenerator::new("desugar"),
}
}
pub fn desugar(&mut self, module: Module) -> Module {
log!(info "the desugaring process has started.");
let module = self.desugar_multiple_pattern_def(module);
let module = self.desugar_pattern_in_module(module);
let module = Self::desugar_shortened_record(module);
let module = Self::desugar_acc(module);
let module = Self::desugar_operator(module);
let module = Self::desugar_comprehension(module);
log!(info "AST (desugared):\n{module}");
log!(info "the desugaring process has completed.");
module
}
pub fn desugar_simple_expr(expr: Expr) -> Expr {
let expr = Self::rec_desugar_shortened_record(expr);
let expr = Self::rec_desugar_lambda_pattern(&mut Desugarer::new(), expr);
Self::rec_desugar_acc(expr)
}
fn desugar_all_chunks(module: Module, desugar: impl Fn(Expr) -> Expr) -> Module {
module.into_iter().map(desugar).collect()
}
fn desugar_args(mut desugar: impl FnMut(Expr) -> Expr, args: Args) -> Args {
let (pos_args, var_args, kw_args, ke_var, paren) = args.deconstruct();
let pos_args = pos_args
.into_iter()
.map(|arg| PosArg::new(desugar(arg.expr)))
.collect();
let var_args = var_args.map(|arg| PosArg::new(desugar(arg.expr)));
let kw_args = kw_args
.into_iter()
.map(|arg| {
KwArg::new(arg.keyword, arg.t_spec, desugar(arg.expr)) })
.collect();
let kw_var_args = ke_var.map(|arg| PosArg::new(desugar(arg.expr)));
Args::new(pos_args, var_args, kw_args, kw_var_args, paren)
}
fn perform_desugar_acc(mut desugar: impl FnMut(Expr) -> Expr, acc: Accessor) -> Accessor {
match acc {
Accessor::Ident(ident) => Accessor::Ident(ident),
Accessor::Attr(attr) => desugar(*attr.obj).attr(attr.ident),
Accessor::TupleAttr(tup) => {
let obj = desugar(*tup.obj);
obj.tuple_attr(tup.index)
}
Accessor::Subscr(sub) => {
let obj = desugar(*sub.obj);
let index = desugar(*sub.index);
obj.subscr(index, sub.r_sqbr)
}
Accessor::TypeApp(tapp) => {
let obj = desugar(*tapp.obj);
let args = match tapp.type_args.args {
TypeAppArgsKind::Args(args) => {
TypeAppArgsKind::Args(Self::desugar_args(desugar, args))
}
other => other,
};
let type_args =
TypeAppArgs::new(tapp.type_args.l_vbar, args, tapp.type_args.r_vbar);
obj.type_app(type_args)
}
}
}
fn perform_desugar(mut desugar: impl FnMut(Expr) -> Expr, expr: Expr) -> Expr {
match expr {
Expr::Literal(_) => expr,
Expr::Record(record) => match record {
Record::Normal(rec) => {
let mut new_attrs = vec![];
for attr in rec.attrs {
new_attrs.push(enum_unwrap!(desugar(Expr::Def(attr)), Expr::Def));
}
Expr::Record(Record::Normal(NormalRecord::new(
rec.l_brace,
rec.r_brace,
RecordAttrs::new(new_attrs),
)))
}
Record::Mixed(mixed) => {
let mut new_attrs = vec![];
for attr in mixed.attrs {
match attr {
RecordAttrOrIdent::Attr(attr) => {
let attr = RecordAttrOrIdent::Attr(enum_unwrap!(
desugar(Expr::Def(attr)),
Expr::Def
));
new_attrs.push(attr);
}
RecordAttrOrIdent::Ident(ident) => {
new_attrs.push(RecordAttrOrIdent::Ident(ident));
}
}
}
Expr::Record(Record::Mixed(MixedRecord::new(
mixed.l_brace,
mixed.r_brace,
new_attrs,
)))
}
},
Expr::DataPack(pack) => {
let class = desugar(*pack.class);
let Expr::Record(args) = desugar(Expr::Record(pack.args)) else {
unreachable!()
};
Expr::DataPack(DataPack::new(class, pack.connector, args))
}
Expr::List(list) => match list {
List::Normal(lis) => {
let (elems, ..) = lis.elems.deconstruct();
let elems = elems
.into_iter()
.map(|elem| PosArg::new(desugar(elem.expr)))
.collect();
let elems = Args::pos_only(elems, None);
let lis = NormalList::new(lis.l_sqbr, lis.r_sqbr, elems);
Expr::List(List::Normal(lis))
}
List::WithLength(lis) => {
let elem = PosArg::new(desugar(lis.elem.expr));
let len = desugar(*lis.len);
let lis = ListWithLength::new(lis.l_sqbr, lis.r_sqbr, elem, len);
Expr::List(List::WithLength(lis))
}
List::Comprehension(lis) => {
let layout = lis.layout.map(|ex| desugar(*ex));
let generators = lis
.generators
.into_iter()
.map(|(ident, gen)| (ident, desugar(gen)))
.collect();
let guard = lis.guard.map(|ex| desugar(*ex));
let lis =
ListComprehension::new(lis.l_sqbr, lis.r_sqbr, layout, generators, guard);
Expr::List(List::Comprehension(lis))
}
},
Expr::Tuple(tuple) => match tuple {
Tuple::Normal(tup) => {
let (elems, _, _, _, paren) = tup.elems.deconstruct();
let elems = elems
.into_iter()
.map(|elem| PosArg::new(desugar(elem.expr)))
.collect();
let new_tup = Args::pos_only(elems, paren);
let tup = NormalTuple::new(new_tup);
Expr::Tuple(Tuple::Normal(tup))
}
},
Expr::Set(set) => match set {
astSet::Normal(set) => {
let (elems, ..) = set.elems.deconstruct();
let elems = elems
.into_iter()
.map(|elem| PosArg::new(desugar(elem.expr)))
.collect();
let elems = Args::pos_only(elems, None);
let set = NormalSet::new(set.l_brace, set.r_brace, elems);
Expr::Set(astSet::Normal(set))
}
astSet::WithLength(set) => {
let elem = PosArg::new(desugar(set.elem.expr));
let len = desugar(*set.len);
let set = SetWithLength::new(set.l_brace, set.r_brace, elem, len);
Expr::Set(astSet::WithLength(set))
}
astSet::Comprehension(set) => {
let elem = set.layout.map(|ex| desugar(*ex));
let mut new_generators = vec![];
for (ident, gen) in set.generators.into_iter() {
new_generators.push((ident, desugar(gen)));
}
let new_guard = set.guard.map(|ex| desugar(*ex));
let set = SetComprehension::new(
set.l_brace,
set.r_brace,
elem,
new_generators,
new_guard,
);
Expr::Set(astSet::Comprehension(set))
}
},
Expr::Dict(dict) => match dict {
Dict::Normal(dic) => {
let new_kvs = dic
.kvs
.into_iter()
.map(|elem| {
let key = desugar(elem.key);
let value = desugar(elem.value);
KeyValue::new(key, value)
})
.collect();
let tup = NormalDict::new(dic.l_brace, dic.r_brace, new_kvs);
Expr::Dict(Dict::Normal(tup))
}
_ => todo!("dict comprehension"),
},
Expr::BinOp(binop) => {
let mut args = binop.args.into_iter();
let lhs = desugar(*args.next().unwrap());
let rhs = desugar(*args.next().unwrap());
Expr::BinOp(BinOp::new(binop.op, lhs, rhs))
}
Expr::UnaryOp(unaryop) => {
let mut args = unaryop.args.into_iter();
let expr = desugar(*args.next().unwrap());
Expr::UnaryOp(UnaryOp::new(unaryop.op, expr))
}
Expr::Call(call) => {
let obj = desugar(*call.obj);
let args = Self::desugar_args(desugar, call.args);
Expr::Call(Call::new(obj, call.attr_name, args))
}
Expr::Def(mut def) => {
let mut chunks = vec![];
for chunk in def.body.block.into_iter() {
chunks.push(desugar(chunk));
}
if let Some(t_op) = def.sig.t_spec_op_mut() {
*t_op.t_spec_as_expr = desugar(*t_op.t_spec_as_expr.clone());
}
if let Signature::Subr(mut subr) = def.sig {
subr.params = Self::perform_desugar_params(desugar, subr.params);
def.sig = Signature::Subr(subr);
}
let body = DefBody::new(def.body.op, Block::new(chunks), def.body.id);
Expr::Def(Def::new(def.sig, body))
}
Expr::ClassDef(class_def) => {
let Expr::Def(def) = desugar(Expr::Def(class_def.def)) else {
unreachable!()
};
let methods = class_def
.methods_list
.into_iter()
.map(|method| enum_unwrap!(desugar(Expr::Methods(method)), Expr::Methods))
.collect();
Expr::ClassDef(ClassDef::new(def, methods))
}
Expr::PatchDef(class_def) => {
let Expr::Def(def) = desugar(Expr::Def(class_def.def)) else {
unreachable!()
};
let methods = class_def
.methods_list
.into_iter()
.map(|method| enum_unwrap!(desugar(Expr::Methods(method)), Expr::Methods))
.collect();
Expr::PatchDef(PatchDef::new(def, methods))
}
Expr::ReDef(mut redef) => {
let expr = desugar(*redef.expr);
if let Some(t_op) = &mut redef.t_spec {
*t_op.t_spec_as_expr = desugar(*t_op.t_spec_as_expr.clone());
}
let attr = Self::perform_desugar_acc(desugar, redef.attr);
Expr::ReDef(ReDef::new(attr, redef.t_spec, expr))
}
Expr::Lambda(mut lambda) => {
let mut chunks = vec![];
for chunk in lambda.body.into_iter() {
chunks.push(desugar(chunk));
}
if let Some(t_op) = &mut lambda.sig.return_t_spec {
*t_op.t_spec_as_expr = desugar(*t_op.t_spec_as_expr.clone());
}
lambda.sig.params = Self::perform_desugar_params(desugar, lambda.sig.params);
let body = Block::new(chunks);
Expr::Lambda(Lambda::new(lambda.sig, lambda.op, body, lambda.id))
}
Expr::TypeAscription(tasc) => {
let expr = desugar(*tasc.expr);
let t_spec_as_expr = desugar(*tasc.t_spec.t_spec_as_expr);
let t_spec =
TypeSpecWithOp::new(tasc.t_spec.op, tasc.t_spec.t_spec, t_spec_as_expr);
expr.type_asc_expr(t_spec)
}
Expr::Methods(method_defs) => {
let mut new_attrs = vec![];
for attr in method_defs.attrs.into_iter() {
match attr {
ClassAttr::Def(def) => {
let mut chunks = vec![];
for chunk in def.body.block.into_iter() {
chunks.push(desugar(chunk));
}
let body = DefBody::new(def.body.op, Block::new(chunks), def.body.id);
new_attrs.push(ClassAttr::Def(Def::new(def.sig, body)));
}
ClassAttr::Decl(decl) => {
let expr = desugar(*decl.expr);
let t_spec_as_expr = desugar(*decl.t_spec.t_spec_as_expr);
let t_spec = TypeSpecWithOp::new(
decl.t_spec.op,
decl.t_spec.t_spec,
t_spec_as_expr,
);
new_attrs.push(ClassAttr::Decl(expr.type_asc(t_spec)));
}
ClassAttr::Doc(doc) => {
new_attrs.push(ClassAttr::Doc(doc));
}
}
}
let new_attrs = ClassAttrs::from(new_attrs);
Expr::Methods(Methods::new(
method_defs.id,
method_defs.class,
*method_defs.class_as_expr,
method_defs.vis,
new_attrs,
))
}
Expr::Accessor(acc) => Expr::Accessor(Self::perform_desugar_acc(desugar, acc)),
Expr::Compound(exprs) => {
let mut chunks = vec![];
for chunk in exprs.into_iter() {
chunks.push(desugar(chunk));
}
Expr::Compound(Compound::new(chunks))
}
Expr::InlineModule(inline) => {
let mut chunks = vec![];
for chunk in inline.ast.module.into_iter() {
chunks.push(desugar(chunk));
}
let ast = AST::new(inline.ast.name, Module::new(chunks));
Expr::InlineModule(InlineModule::new(inline.input, ast, inline.import))
}
Expr::Dummy(exprs) => {
let loc = exprs.loc;
let mut chunks = vec![];
for chunk in exprs.into_iter() {
chunks.push(desugar(chunk));
}
Expr::Dummy(Dummy::new(loc, chunks))
}
}
}
fn perform_desugar_params(mut desugar: impl FnMut(Expr) -> Expr, mut params: Params) -> Params {
let mut non_defaults = vec![];
for mut non_default in params.non_defaults.into_iter() {
non_default.t_spec = non_default.t_spec.map(|t_spec| {
TypeSpecWithOp::new(t_spec.op, t_spec.t_spec, desugar(*t_spec.t_spec_as_expr))
});
non_defaults.push(non_default);
}
params.var_params = params.var_params.map(|mut var_params| {
var_params.t_spec = var_params.t_spec.map(|t_spec| {
TypeSpecWithOp::new(t_spec.op, t_spec.t_spec, desugar(*t_spec.t_spec_as_expr))
});
var_params
});
let mut defaults = vec![];
for mut default in params.defaults.into_iter() {
let default_val = desugar(default.default_val);
default.sig.t_spec = default.sig.t_spec.map(|t_spec| {
TypeSpecWithOp::new(t_spec.op, t_spec.t_spec, desugar(*t_spec.t_spec_as_expr))
});
defaults.push(DefaultParamSignature {
default_val,
..default
});
}
let mut guards = vec![];
for guard in params.guards.into_iter() {
let guard = match guard {
GuardClause::Condition(cond) => GuardClause::Condition(desugar(cond)),
GuardClause::Bind(bind) => {
let bind = Expr::Def(bind);
let bind = enum_unwrap!(desugar(bind), Expr::Def);
GuardClause::Bind(bind)
}
};
guards.push(guard);
}
params.non_defaults = non_defaults;
params.defaults = defaults;
params.guards = guards;
params
}
fn desugar_multiple_pattern_def(&self, module: Module) -> Module {
let mut new = Module::with_capacity(module.len());
for chunk in module.into_iter() {
match chunk {
Expr::Def(def) if def.is_subr() => {
if let Some(Expr::Def(previous)) = new.last() {
if previous.is_subr() && previous.sig.name_as_str() == def.sig.name_as_str()
{
let Some(Expr::Def(previous)) = new.pop() else {
unreachable!()
};
let name = def.sig.ident().unwrap().clone();
let id = def.body.id;
let op = def.body.op.clone();
let (call, return_t_spec) = if previous.body.block.len() == 1
&& previous.body.block.first().unwrap().is_match_call()
{
self.add_arg_to_match_call(previous, def)
} else {
self.gen_match_call(previous, def)
};
let params = match &call.args.pos_args().iter().next().unwrap().expr {
Expr::Tuple(Tuple::Normal(tup)) => {
let mut params = vec![];
for arg in tup.elems.pos_args().iter() {
match &arg.expr {
Expr::Accessor(Accessor::Ident(ident)) => {
let param_name = ident.inspect();
let col_begin = name.col_end().unwrap_or(0) + 1;
let col_end =
col_begin + param_name.chars().count() as u32;
let param = VarName::new(Token::new_fake(
TokenKind::Symbol,
param_name,
name.ln_begin().unwrap_or(1),
col_begin,
col_end,
));
let param = NonDefaultParamSignature::new(
ParamPattern::VarName(param),
None,
);
params.push(param);
}
_ => unreachable!(),
}
}
Params::new(params, None, vec![], None, None)
}
Expr::Accessor(Accessor::Ident(ident)) => {
let param_name = ident.inspect();
let col_begin = name.col_end().unwrap_or(0) + 1; let col_end = col_begin + param_name.chars().count() as u32;
let param = VarName::new(Token::new_fake(
TokenKind::Symbol,
param_name,
name.ln_begin().unwrap_or(1),
col_begin,
col_end,
));
let param = NonDefaultParamSignature::new(
ParamPattern::VarName(param),
None,
);
Params::single(param)
}
_ => unreachable!(),
};
let sig = Signature::Subr(SubrSignature::new(
set! {},
name,
TypeBoundSpecs::empty(),
params,
return_t_spec,
));
let body = DefBody::new(op, Block::new(vec![Expr::Call(call)]), id);
let def = Def::new(sig, body);
new.push(Expr::Def(def));
} else {
new.push(Expr::Def(def));
}
} else {
new.push(Expr::Def(def));
}
}
other => {
new.push(other);
}
}
}
new
}
fn add_arg_to_match_call(&self, mut previous: Def, def: Def) -> (Call, Option<TypeSpecWithOp>) {
let op = Token::from_str(TokenKind::FuncArrow, "->");
let Expr::Call(mut call) = previous.body.block.remove(0) else {
unreachable!()
};
let Signature::Subr(sig) = def.sig else {
unreachable!()
};
let return_t_spec = sig.return_t_spec;
let first_arg = sig.params.non_defaults.first().unwrap();
if let Some(name) = first_arg.inspect() {
call.args.remove_pos(0);
let arg = PosArg::new(Expr::local(
name,
first_arg.ln_begin().unwrap_or(1),
first_arg.col_begin().unwrap_or(0),
first_arg.col_end().unwrap_or(0),
));
call.args.insert_pos(0, arg);
}
let params = if sig.params.len() == 1 {
sig.params
} else {
let pat = ParamPattern::Tuple(ParamTuplePattern::new(sig.params));
Params::single(NonDefaultParamSignature::new(pat, None))
};
let sig = LambdaSignature::new(params, return_t_spec.clone(), sig.bounds);
let new_branch = Lambda::new(sig, op, def.body.block, def.body.id);
call.args.push_pos(PosArg::new(Expr::Lambda(new_branch)));
(call, return_t_spec)
}
fn gen_match_call(&self, previous: Def, def: Def) -> (Call, Option<TypeSpecWithOp>) {
let op = Token::from_str(TokenKind::FuncArrow, "->");
let Signature::Subr(prev_sig) = previous.sig else {
unreachable!()
};
let params_len = prev_sig.params.len();
let params = if params_len == 1 {
prev_sig.params
} else {
let pat = ParamPattern::Tuple(ParamTuplePattern::new(prev_sig.params));
Params::single(NonDefaultParamSignature::new(pat, None))
};
let match_symbol = Expr::static_local("match");
let sig = LambdaSignature::new(params, prev_sig.return_t_spec, prev_sig.bounds);
let first_branch = Lambda::new(sig, op.clone(), previous.body.block, previous.body.id);
let Signature::Subr(sig) = def.sig else {
unreachable!()
};
let params = if sig.params.len() == 1 {
sig.params
} else {
let pat = ParamPattern::Tuple(ParamTuplePattern::new(sig.params));
Params::single(NonDefaultParamSignature::new(pat, None))
};
let return_t_spec = sig.return_t_spec;
let sig = LambdaSignature::new(params, return_t_spec.clone(), sig.bounds);
let second_branch = Lambda::new(sig, op, def.body.block, def.body.id);
let first_arg = if params_len == 1 {
Expr::dummy_local(&self.var_gen.fresh_varname())
} else {
let args = (0..params_len)
.map(|_| PosArg::new(Expr::dummy_local(&self.var_gen.fresh_varname())));
Expr::Tuple(Tuple::Normal(NormalTuple::new(Args::pos_only(
args.collect(),
None,
))))
};
let args = Args::pos_only(
vec![
PosArg::new(first_arg), PosArg::new(Expr::Lambda(first_branch)),
PosArg::new(Expr::Lambda(second_branch)),
],
None,
);
let call = match_symbol.call(args);
(call, return_t_spec)
}
fn _desugar_literal_pattern(&self, _mod: Module) -> Module {
todo!()
}
fn gen_buf_name_and_sig(
&mut self,
loc: Location,
t_spec: Option<TypeSpecWithOp>,
) -> (Str, Signature) {
let buf_name = self.var_gen.fresh_varname();
let buf_sig = Signature::Var(VarSignature::new(
VarPattern::Ident(Identifier::private_with_loc(Str::rc(&buf_name), loc)),
t_spec,
));
(buf_name, buf_sig)
}
fn gen_buf_nd_param(&mut self, loc: Location) -> (Str, ParamPattern) {
let buf_name = self.var_gen.fresh_varname();
let pat = ParamPattern::VarName(VarName::from_str_and_loc(buf_name.clone(), loc));
(buf_name, pat)
}
fn rec_desugar_lambda_pattern(&mut self, expr: Expr) -> Expr {
match expr {
Expr::Lambda(mut lambda) => {
self.desugar_params_patterns(&mut lambda.sig.params);
lambda.body = self.desugar_pattern_in_block(lambda.body);
Expr::Lambda(lambda)
}
expr => Self::perform_desugar(|ex| self.rec_desugar_lambda_pattern(ex), expr),
}
}
fn desugar_pattern_in_module(&mut self, module: Module) -> Module {
#[allow(clippy::useless_conversion)]
Module::new(self.desugar_pattern(module.into_iter()))
}
fn desugar_pattern_in_block(&mut self, block: Block) -> Block {
#[allow(clippy::useless_conversion)]
Block::new(self.desugar_pattern(block.into_iter()))
}
fn desugar_pattern_in_body(&mut self, body: DefBody) -> DefBody {
let block = body
.block
.into_iter()
.map(|ex| self.rec_desugar_lambda_pattern(ex))
.collect();
let block = self.desugar_pattern_in_block(block);
DefBody::new(body.op, block, body.id)
}
fn desugar_def_pattern(&mut self, def: Def, new: &mut Vec<Expr>) {
match def {
Def {
sig: Signature::Var(mut v),
body,
} => match &v.pat {
VarPattern::Tuple(tup) => {
let (buf_name, buf_sig) = self.gen_buf_name_and_sig(v.loc(), v.t_spec);
let body = self.desugar_pattern_in_body(body);
let buf_def = Def::new(buf_sig, body);
new.push(Expr::Def(buf_def));
for (n, elem) in tup.elems.iter().enumerate() {
self.desugar_nested_var_pattern(new, elem, &buf_name, BufIndex::Tuple(n));
}
let elems_len = tup.elems.len();
if let Some(var) = tup.elems.starred.as_ref() {
self.desugar_rest_values(new, var, &buf_name, elems_len);
}
}
VarPattern::List(lis) => {
let (buf_name, buf_sig) = self.gen_buf_name_and_sig(v.loc(), v.t_spec);
let body = self.desugar_pattern_in_body(body);
let buf_def = Def::new(buf_sig, body);
new.push(Expr::Def(buf_def));
for (n, elem) in lis.elems.iter().enumerate() {
self.desugar_nested_var_pattern(new, elem, &buf_name, BufIndex::List(n));
}
let elems_len = lis.elems.len();
if let Some(var) = lis.elems.starred.as_ref() {
self.desugar_rest_values(new, var, &buf_name, elems_len);
}
}
VarPattern::Record(rec) => {
let (buf_name, buf_sig) = self.gen_buf_name_and_sig(v.loc(), v.t_spec);
let body = self.desugar_pattern_in_body(body);
let buf_def = Def::new(buf_sig, body);
new.push(Expr::Def(buf_def));
for VarRecordAttr { lhs, rhs } in rec.attrs.iter() {
self.desugar_nested_var_pattern(new, rhs, &buf_name, BufIndex::Record(lhs));
}
}
VarPattern::DataPack(pack) => {
let t_spec =
TypeSpecWithOp::new(COLON, pack.class.clone(), *pack.class_as_expr.clone());
let (buf_name, buf_sig) = self.gen_buf_name_and_sig(
v.loc(),
Some(t_spec), );
let body = self.desugar_pattern_in_body(body);
let buf_def = Def::new(buf_sig, body);
new.push(Expr::Def(buf_def));
for VarRecordAttr { lhs, rhs } in pack.args.attrs.iter() {
self.desugar_nested_var_pattern(new, rhs, &buf_name, BufIndex::Record(lhs));
}
}
VarPattern::Ident(_) | VarPattern::Discard(_) | VarPattern::Glob(_) => {
if let VarPattern::Ident(ident) = v.pat {
v.pat = VarPattern::Ident(Self::desugar_ident(ident));
}
let body = self.desugar_pattern_in_body(body);
let def = Def::new(Signature::Var(v), body);
new.push(Expr::Def(def));
}
},
Def {
sig: Signature::Subr(mut subr),
body,
} => {
subr.ident = Self::desugar_ident(subr.ident);
self.desugar_params_patterns(&mut subr.params);
let body = self.desugar_pattern_in_body(body);
let def = Def::new(Signature::Subr(subr), body);
new.push(Expr::Def(def));
}
}
}
fn desugar_pattern_of_methods(&mut self, methods: Methods) -> Methods {
let mut new_attrs = Vec::with_capacity(methods.attrs.len());
for attr in methods.attrs.into_iter() {
match attr {
ClassAttr::Def(def) => {
let mut new = vec![];
self.desugar_def_pattern(def, &mut new);
let Expr::Def(def) = new.remove(0) else {
todo!("{new:?}")
};
new_attrs.push(ClassAttr::Def(def));
}
_ => {
new_attrs.push(attr);
}
}
}
Methods::new(
methods.id,
methods.class,
*methods.class_as_expr,
methods.vis,
ClassAttrs::from(new_attrs),
)
}
fn desugar_pattern<I>(&mut self, chunks: I) -> Vec<Expr>
where
I: IntoIterator<Item = Expr> + ExactSizeIterator,
{
let mut new = Vec::with_capacity(chunks.len());
for chunk in chunks.into_iter() {
match chunk {
Expr::Def(def) => {
self.desugar_def_pattern(def, &mut new);
}
Expr::ClassDef(class_def) => {
let methods = class_def
.methods_list
.into_iter()
.map(|methods| self.desugar_pattern_of_methods(methods))
.collect();
new.push(Expr::ClassDef(ClassDef::new(class_def.def, methods)));
}
Expr::PatchDef(patch_def) => {
let methods = patch_def
.methods_list
.into_iter()
.map(|methods| self.desugar_pattern_of_methods(methods))
.collect();
new.push(Expr::PatchDef(PatchDef::new(patch_def.def, methods)));
}
Expr::Methods(methods) => {
let methods = self.desugar_pattern_of_methods(methods);
new.push(Expr::Methods(methods));
}
Expr::Dummy(dummy) => {
let loc = dummy.loc;
let new_dummy = self.desugar_pattern(dummy.into_iter());
new.push(Expr::Dummy(Dummy::new(loc, new_dummy)));
}
Expr::Compound(compound) => {
let new_compound = self.desugar_pattern(compound.into_iter());
new.push(Expr::Compound(Compound::new(new_compound)));
}
other => {
new.push(self.rec_desugar_lambda_pattern(other));
}
}
}
new
}
fn desugar_params_patterns(&mut self, params: &mut Params) {
let mut guards = vec![];
for param in params.non_defaults.iter_mut() {
guards.extend(self.desugar_nd_param(param));
}
if let Some(var_params) = params.var_params.as_mut() {
guards.extend(self.desugar_nd_param(var_params));
}
for param in params.defaults.iter_mut() {
guards.extend(self.desugar_nd_param(&mut param.sig));
}
params.extend_guards(guards);
}
fn desugar_nested_var_pattern(
&mut self,
new_module: &mut Vec<Expr>,
sig: &VarSignature,
buf_name: &str,
buf_index: BufIndex,
) {
let obj = Expr::local(
buf_name,
sig.ln_begin().unwrap_or(1),
sig.col_begin().unwrap_or(0),
sig.col_end().unwrap_or(0),
);
let acc = match buf_index {
BufIndex::Tuple(n) => obj.tuple_attr(Literal::nat(n, sig.ln_begin().unwrap_or(1))),
BufIndex::List(n) => {
let r_brace = Token::new(
TokenKind::RBrace,
"]",
sig.ln_begin().unwrap_or(1),
sig.col_begin().unwrap_or(0),
);
obj.subscr(
Expr::Literal(Literal::nat(n, sig.ln_begin().unwrap_or(1))),
r_brace,
)
}
BufIndex::Record(attr) => {
let attr = Identifier::new(VisModifierSpec::Auto, attr.name.clone());
obj.attr(attr)
}
};
let id = DefId(get_hash(&(&acc, buf_name)));
let block = Block::new(vec![Expr::Accessor(acc)]);
let op = Token::from_str(TokenKind::Assign, "=");
let body = DefBody::new(op, block, id);
match &sig.pat {
VarPattern::Tuple(tup) => {
let (buf_name, buf_sig) = self.gen_buf_name_and_sig(sig.loc(), None);
let buf_def = Def::new(buf_sig, body);
new_module.push(Expr::Def(buf_def));
for (n, elem) in tup.elems.iter().enumerate() {
self.desugar_nested_var_pattern(
new_module,
elem,
&buf_name,
BufIndex::Tuple(n),
);
}
}
VarPattern::List(lis) => {
let (buf_name, buf_sig) = self.gen_buf_name_and_sig(sig.loc(), None);
let buf_def = Def::new(buf_sig, body);
new_module.push(Expr::Def(buf_def));
for (n, elem) in lis.elems.iter().enumerate() {
self.desugar_nested_var_pattern(new_module, elem, &buf_name, BufIndex::List(n));
}
}
VarPattern::Record(rec) => {
let (buf_name, buf_sig) = self.gen_buf_name_and_sig(sig.loc(), None);
let buf_def = Def::new(buf_sig, body);
new_module.push(Expr::Def(buf_def));
for VarRecordAttr { lhs, rhs } in rec.attrs.iter() {
self.desugar_nested_var_pattern(
new_module,
rhs,
&buf_name,
BufIndex::Record(lhs),
);
}
}
VarPattern::DataPack(pack) => {
let t_spec =
TypeSpecWithOp::new(COLON, pack.class.clone(), *pack.class_as_expr.clone());
let (buf_name, buf_sig) = self.gen_buf_name_and_sig(sig.loc(), Some(t_spec));
let buf_def = Def::new(buf_sig, body);
new_module.push(Expr::Def(buf_def));
for VarRecordAttr { lhs, rhs } in pack.args.attrs.iter() {
self.desugar_nested_var_pattern(
new_module,
rhs,
&buf_name,
BufIndex::Record(lhs),
);
}
}
VarPattern::Ident(_) | VarPattern::Discard(_) | VarPattern::Glob(_) => {
let def = Def::new(Signature::Var(sig.clone()), body);
new_module.push(Expr::Def(def));
}
}
}
fn desugar_rest_values(
&mut self,
new_module: &mut Vec<Expr>,
sig: &VarSignature,
buf_name: &str,
elems_len: usize,
) {
let obj = Expr::local(
buf_name,
sig.ln_begin().unwrap_or(1),
sig.col_begin().unwrap_or(0),
sig.col_end().unwrap_or(0),
);
let op = Token::from_str(TokenKind::Assign, "=");
let id = DefId(get_hash(&(&obj, buf_name)));
let start = Expr::Literal(Literal::nat(elems_len, sig.ln_begin().unwrap_or(1)));
#[cfg(target_pointer_width = "64")]
let max = 109521666047; #[cfg(not(target_pointer_width = "64"))]
let max = 100000;
let end = Expr::Literal(Literal::nat(max, sig.ln_begin().unwrap_or(1)));
let range = Token::new_with_loc(TokenKind::Closed, "..", sig.loc());
let acc = obj.subscr(
start.bin_op(range, end).into(),
Token::new_fake(TokenKind::RBrace, "]", 0, 0, 0),
);
let body = DefBody::new(op, Block::new(vec![Expr::Accessor(acc)]), id);
let starred = Def::new(Signature::Var(sig.clone()), body);
new_module.push(Expr::Def(starred));
}
fn desugar_shortened_record(module: Module) -> Module {
Self::desugar_all_chunks(module, Self::rec_desugar_shortened_record)
}
fn rec_desugar_shortened_record(expr: Expr) -> Expr {
match expr {
Expr::Record(Record::Mixed(record)) => {
let rec = Self::desugar_shortened_record_inner(record);
Expr::Record(Record::Normal(rec))
}
Expr::DataPack(pack) => {
if let Record::Mixed(rec) = pack.args {
let class = Self::rec_desugar_shortened_record(*pack.class);
let rec = Self::desugar_shortened_record_inner(rec);
let args = Record::Normal(rec);
Expr::DataPack(DataPack::new(class, pack.connector, args))
} else {
Expr::DataPack(pack)
}
}
expr => Self::perform_desugar(Self::rec_desugar_shortened_record, expr),
}
}
pub fn desugar_shortened_record_inner(record: MixedRecord) -> NormalRecord {
let attrs = record
.attrs
.into_iter()
.map(|attr_or_ident| match attr_or_ident {
RecordAttrOrIdent::Attr(def) => def,
RecordAttrOrIdent::Ident(ident) => {
let var = VarSignature::new(VarPattern::Ident(ident.clone()), None);
let sig = Signature::Var(var);
let body = DefBody::new(
Token::from_str(TokenKind::Assign, "="),
Block::new(vec![Expr::local(
ident.inspect(),
ident.ln_begin().unwrap_or(1),
ident.col_begin().unwrap_or(0),
ident.col_end().unwrap_or(0),
)]),
DefId(get_hash(&(&sig, ident.inspect()))),
);
Def::new(sig, body)
}
})
.collect();
let attrs = RecordAttrs::new(attrs);
NormalRecord::new(record.l_brace, record.r_brace, attrs)
}
fn dummy_list_expr(len: Literal) -> Expr {
let l_sqbr = Token {
content: "[".into(),
kind: TokenKind::LSqBr,
..len.token
};
let r_sqbr = Token {
content: "]".into(),
kind: TokenKind::RSqBr,
..len.token
};
let elem = Expr::local("Obj", l_sqbr.lineno, l_sqbr.col_begin, l_sqbr.col_end);
let list = List::WithLength(ListWithLength::new(
l_sqbr,
r_sqbr,
PosArg::new(elem),
Expr::Literal(len),
));
Expr::List(list)
}
fn dummy_set_expr(lit: Literal) -> Expr {
let l_brace = Token {
content: "{".into(),
kind: TokenKind::LBrace,
..lit.token
};
let r_brace = Token {
content: "}".into(),
kind: TokenKind::RBrace,
..lit.token
};
let args = Args::single(PosArg::new(Expr::Literal(lit)));
Expr::from(NormalSet::new(l_brace, r_brace, args))
}
fn len_guard(param_name: Str, n: usize, loc: &impl Locational) -> GuardClause {
let name = VarName::new(Token::new_fake(
TokenKind::Symbol,
param_name.clone(),
loc.ln_begin().unwrap_or(1),
loc.col_begin().unwrap_or(0),
loc.col_end().unwrap_or(0),
));
let eq = Token::from_str(TokenKind::DblEq, "==");
let len = Literal::nat(n, loc.ln_begin().unwrap_or(0));
let bin = Expr::from(
Identifier::private("len".into())
.call1(Expr::from(Identifier::private_from_varname(name))),
)
.bin_op(eq, len.into());
GuardClause::Condition(bin.into())
}
fn hasattr_guard<'i>(
param_name: Str,
mut keys: impl Iterator<Item = &'i Identifier>,
loc: &impl Locational,
) -> Option<GuardClause> {
let name = VarName::new(Token::new_fake(
TokenKind::Symbol,
param_name.clone(),
loc.ln_begin().unwrap_or(1),
loc.col_begin().unwrap_or(0),
loc.col_end().unwrap_or(0),
));
let name = Expr::from(Identifier::private_from_varname(name));
let key = keys.next()?;
let attr_name = Expr::from(Literal::str(
format!("\"{}\"", key.inspect().clone()),
key.ln_begin().unwrap_or(0),
));
let hasattr = Identifier::private("hasattr".into()).call2(name.clone(), attr_name);
let res = keys.fold(Expr::from(hasattr), |acc, key| {
let attr_name = Expr::from(Literal::str(
format!("\"{}\"", key.inspect().clone()),
key.ln_begin().unwrap_or(0),
));
let hasattr = Identifier::private("hasattr".into()).call2(name.clone(), attr_name);
acc.bin_op(Token::from_str(TokenKind::AndOp, "and"), hasattr.into())
.into()
});
Some(GuardClause::Condition(res))
}
fn type_guard(param_name: Str, class: &Expr, loc: &impl Locational) -> Option<GuardClause> {
let name = VarName::new(Token::new_fake(
TokenKind::Symbol,
param_name.clone(),
loc.ln_begin().unwrap_or(1),
loc.col_begin().unwrap_or(0),
loc.col_end().unwrap_or(0),
));
let in_op = class.clone().bin_op(
Token::dummy(TokenKind::ContainsOp, "contains"),
Identifier::private_from_varname(name).into(),
);
Some(GuardClause::Condition(in_op.into()))
}
fn desugar_nd_param(&mut self, param: &mut NonDefaultParamSignature) -> Vec<GuardClause> {
let mut guards = vec![];
let line = param.ln_begin().unwrap_or(1);
match &mut param.pat {
ParamPattern::VarName(v) => {
if let Some(t_spec) = param
.t_spec
.as_ref()
.map(|t_spec| t_spec.t_spec_as_expr.as_ref())
{
guards.extend(Self::type_guard(v.inspect().clone(), t_spec, v));
}
guards
}
ParamPattern::Lit(l) => {
let lit = l.clone();
let name = VarName::new(Token::new_fake(
TokenKind::Symbol,
self.var_gen.fresh_param_name(),
l.ln_begin().unwrap_or(1),
l.col_begin().unwrap_or(0),
l.col_end().unwrap_or(0),
));
let eq = Token::from_str(TokenKind::DblEq, "==");
let guard = Expr::from(Identifier::private_from_varname(name.clone()))
.bin_op(eq, Expr::from(lit.clone()));
guards.push(GuardClause::Condition(guard.into()));
param.pat = ParamPattern::VarName(name);
let l_brace = Token {
content: "{".into(),
kind: TokenKind::LBrace,
..lit.token
};
let r_brace = Token {
content: "}".into(),
kind: TokenKind::RBrace,
..lit.token
};
let t_spec = TypeSpec::enum_t_spec(vec![lit.clone()]);
let args = Args::single(PosArg::new(Expr::Literal(lit)));
let t_spec_as_expr = Expr::from(NormalSet::new(l_brace, r_brace, args));
let t_spec = TypeSpecWithOp::new(COLON, t_spec, t_spec_as_expr);
param.t_spec = Some(t_spec);
guards
}
ParamPattern::Tuple(tup) => {
let (buf_name, buf_param) = self.gen_buf_nd_param(tup.loc());
let mut ty_specs = vec![];
let mut ty_exprs = vec![];
guards.push(Self::len_guard(buf_name.clone(), tup.elems.len(), tup));
for (n, elem) in tup.elems.non_defaults.iter_mut().enumerate() {
let gs = self.desugar_nested_param_pattern(elem, &buf_name, BufIndex::Tuple(n));
guards.extend(gs);
let infer = Token::new_fake(TokenKind::Try, "?", line, 0, 0);
let ty_expr = elem
.t_spec
.as_ref()
.map(|ts| *ts.t_spec_as_expr.clone())
.unwrap_or(Expr::local(
"Obj",
infer.lineno,
infer.col_begin,
infer.col_end,
));
ty_exprs.push(PosArg::new(ty_expr));
ty_specs.push(
elem.t_spec
.as_ref()
.map(|ts| ts.t_spec.clone())
.unwrap_or(TypeSpec::Infer(infer))
.clone(),
);
}
if param.t_spec.is_none() {
let t_spec =
TypeSpec::Tuple(TupleTypeSpec::new(tup.elems.parens.clone(), ty_specs));
let t_spec_as_expr = Expr::from(NormalTuple::new(Args::pos_only(
ty_exprs,
tup.elems.parens.clone(),
)));
param.t_spec = Some(TypeSpecWithOp::new(COLON, t_spec, t_spec_as_expr));
}
param.pat = buf_param;
guards
}
ParamPattern::List(lis) => {
fn const_check(expr: &Expr) -> bool {
match &expr {
Expr::Accessor(Accessor::Ident(ident)) => ident.is_const(),
Expr::List(List::Normal(lis)) => lis
.elems
.pos_args()
.iter()
.all(|pos| const_check(&pos.expr)),
_ => true,
}
}
let expr = Expr::try_from(&*lis).and_then(|expr| {
if const_check(&expr) {
Ok(expr)
} else {
Err(())
}
});
let (buf_name, buf_param) = self.gen_buf_nd_param(lis.loc());
guards.push(Self::len_guard(buf_name.clone(), lis.elems.len(), lis));
for (n, elem) in lis.elems.non_defaults.iter_mut().enumerate() {
let gs = self.desugar_nested_param_pattern(elem, &buf_name, BufIndex::List(n));
guards.extend(gs);
}
if param.t_spec.is_none() {
let len = lis.elems.non_defaults.len();
let len = Literal::new(Token::new_fake(
TokenKind::NatLit,
len.to_string(),
line,
0,
0,
));
let infer = Token::new_fake(TokenKind::Try, "?", line, 0, 0);
let t_spec = ListTypeSpec::new(
TypeSpec::Infer(infer),
ConstExpr::Lit(len.clone()),
Some((lis.l_sqbr.clone(), lis.r_sqbr.clone())),
);
let t_spec_as_expr = if let Ok(expr) = expr {
Expr::Set(astSet::Normal(NormalSet::new(
Token::DUMMY,
Token::DUMMY,
Args::single(PosArg::new(expr)),
)))
} else {
Self::dummy_list_expr(len)
};
param.t_spec = Some(TypeSpecWithOp::new(
Token::dummy(TokenKind::Colon, ":"),
TypeSpec::List(t_spec),
t_spec_as_expr,
));
}
param.pat = buf_param;
guards
}
ParamPattern::Record(rec) => {
let (buf_name, buf_param) = self.gen_buf_nd_param(rec.loc());
guards.extend(Self::hasattr_guard(buf_name.clone(), rec.elems.keys(), rec));
for ParamRecordAttr { lhs, rhs } in rec.elems.iter_mut() {
let gs =
self.desugar_nested_param_pattern(rhs, &buf_name, BufIndex::Record(lhs));
guards.extend(gs);
}
if param.t_spec.is_none() {
let mut attrs = RecordAttrs::new(vec![]);
let mut tys = vec![];
for ParamRecordAttr { lhs, rhs } in rec.elems.iter() {
let lhs = Identifier {
vis: VisModifierSpec::Public(Token::DUMMY),
..lhs.clone()
};
let infer = Token::new_fake(TokenKind::Try, "?", line, 0, 0);
let expr = rhs
.t_spec
.as_ref()
.map(|ts| *ts.t_spec_as_expr.clone())
.unwrap_or(Expr::local(
"Obj",
infer.lineno,
infer.col_begin,
infer.col_end,
));
let attr =
Def::new(Signature::new_var(lhs.clone()), DefBody::new_single(expr));
attrs.push(attr);
tys.push((
lhs.clone(),
rhs.t_spec
.as_ref()
.map(|ts| ts.t_spec.clone())
.unwrap_or(TypeSpec::Infer(infer))
.clone(),
));
}
let t_spec = TypeSpec::Record(RecordTypeSpec::new(
Some((rec.l_brace.clone(), rec.r_brace.clone())),
tys,
));
let t_spec_as_expr = Expr::from(NormalRecord::new(
rec.l_brace.clone(),
rec.r_brace.clone(),
attrs,
));
param.t_spec = Some(TypeSpecWithOp::new(COLON, t_spec, t_spec_as_expr));
}
param.pat = buf_param;
guards
}
_ => guards,
}
}
fn desugar_nested_param_pattern(
&mut self,
sig: &mut NonDefaultParamSignature,
buf_name: &str,
buf_index: BufIndex,
) -> Vec<GuardClause> {
let mut guards = vec![];
let obj = Expr::local(
buf_name,
sig.ln_begin().unwrap_or(1),
sig.col_begin().unwrap_or(0),
sig.col_end().unwrap_or(0),
);
let acc = match buf_index {
BufIndex::Tuple(n) => obj.tuple_attr(Literal::nat(n, sig.ln_begin().unwrap_or(1))),
BufIndex::List(n) => {
let r_brace = Token::new(
TokenKind::RBrace,
"]",
sig.ln_begin().unwrap_or(1),
sig.col_begin().unwrap_or(0),
);
obj.subscr(
Expr::Literal(Literal::nat(n, sig.ln_begin().unwrap_or(1))),
r_brace,
)
}
BufIndex::Record(attr) => {
let attr = Identifier::new(VisModifierSpec::Auto, attr.name.clone());
obj.attr(attr)
}
};
let id = DefId(get_hash(&(&acc, buf_name)));
let block = Block::new(vec![Expr::Accessor(acc)]);
let op = Token::from_str(TokenKind::Assign, "=");
let mut body = DefBody::new(op, block, id);
let line = sig.ln_begin().unwrap_or(1);
match &mut sig.pat {
ParamPattern::Tuple(tup) => {
let (buf_name, buf_sig) = self.gen_buf_nd_param(tup.loc());
let ident = Identifier::private(Str::from(&buf_name));
let def = Def::new(
Signature::Var(VarSignature::new(
VarPattern::Ident(ident),
sig.t_spec.clone(),
)),
body,
);
guards.push(GuardClause::Bind(def));
guards.push(Self::len_guard(buf_name.clone(), tup.elems.len(), tup));
let mut ty_exprs = vec![];
let mut tys = vec![];
for (n, elem) in tup.elems.non_defaults.iter_mut().enumerate() {
let gs = self.desugar_nested_param_pattern(elem, &buf_name, BufIndex::Tuple(n));
guards.extend(gs);
let infer = Token::new_fake(TokenKind::Try, "?", line, 0, 0);
let ty_expr = elem
.t_spec
.as_ref()
.map(|ts| *ts.t_spec_as_expr.clone())
.unwrap_or(Expr::local(
"Obj",
infer.lineno,
infer.col_begin,
infer.col_end,
));
ty_exprs.push(PosArg::new(ty_expr));
tys.push(
elem.t_spec
.as_ref()
.map(|ts| ts.t_spec.clone())
.unwrap_or(TypeSpec::Infer(infer))
.clone(),
);
}
if sig.t_spec.is_none() {
let t_spec = TypeSpec::Tuple(TupleTypeSpec::new(tup.elems.parens.clone(), tys));
let t_spec_as_expr = Expr::from(NormalTuple::new(Args::pos_only(
ty_exprs,
tup.elems.parens.clone(),
)));
sig.t_spec = Some(TypeSpecWithOp::new(COLON, t_spec, t_spec_as_expr));
}
sig.pat = buf_sig;
guards
}
ParamPattern::List(lis) => {
let (buf_name, buf_sig) = self.gen_buf_nd_param(lis.loc());
let def = Def::new(
Signature::Var(VarSignature::new(
VarPattern::Ident(Identifier::private(Str::from(&buf_name))),
sig.t_spec.clone(),
)),
body,
);
guards.push(GuardClause::Bind(def));
guards.push(Self::len_guard(buf_name.clone(), lis.elems.len(), lis));
for (n, elem) in lis.elems.non_defaults.iter_mut().enumerate() {
let gs = self.desugar_nested_param_pattern(elem, &buf_name, BufIndex::List(n));
guards.extend(gs);
}
if sig.t_spec.is_none() {
let len = lis.elems.non_defaults.len();
let len = Literal::new(Token::new_fake(
TokenKind::NatLit,
len.to_string(),
line,
0,
0,
));
let infer = Token::new_fake(TokenKind::Try, "?", line, 0, 0);
let t_spec = ListTypeSpec::new(
TypeSpec::Infer(infer),
ConstExpr::Lit(len.clone()),
Some((lis.l_sqbr.clone(), lis.r_sqbr.clone())),
);
let t_spec_as_expr = Self::dummy_list_expr(len);
sig.t_spec = Some(TypeSpecWithOp::new(
COLON,
TypeSpec::List(t_spec),
t_spec_as_expr,
));
}
sig.pat = buf_sig;
guards
}
ParamPattern::Record(rec) => {
let (buf_name, buf_sig) = self.gen_buf_nd_param(rec.loc());
let def = Def::new(
Signature::Var(VarSignature::new(
VarPattern::Ident(Identifier::private(Str::from(&buf_name))),
sig.t_spec.clone(),
)),
body,
);
guards.push(GuardClause::Bind(def));
guards.extend(Self::hasattr_guard(buf_name.clone(), rec.elems.keys(), rec));
let mut attrs = RecordAttrs::new(vec![]);
let mut tys = vec![];
for ParamRecordAttr { lhs, rhs } in rec.elems.iter_mut() {
let lhs = Identifier {
vis: VisModifierSpec::Public(Token::DUMMY),
..lhs.clone()
};
let gs =
self.desugar_nested_param_pattern(rhs, &buf_name, BufIndex::Record(&lhs));
guards.extend(gs);
let infer = Token::new_fake(TokenKind::Try, "?", line, 0, 0);
let expr = rhs
.t_spec
.as_ref()
.map(|ts| *ts.t_spec_as_expr.clone())
.unwrap_or(Expr::local(
"Obj",
infer.lineno,
infer.col_begin,
infer.col_end,
));
let attr = Def::new(Signature::new_var(lhs.clone()), DefBody::new_single(expr));
attrs.push(attr);
tys.push((
lhs.clone(),
rhs.t_spec
.as_ref()
.map(|ts| ts.t_spec.clone())
.unwrap_or(TypeSpec::Infer(infer))
.clone(),
));
}
if sig.t_spec.is_none() {
let t_spec = TypeSpec::Record(RecordTypeSpec::new(
Some((rec.l_brace.clone(), rec.r_brace.clone())),
tys,
));
let t_spec_as_expr = Expr::from(NormalRecord::new(
rec.l_brace.clone(),
rec.r_brace.clone(),
attrs,
));
sig.t_spec = Some(TypeSpecWithOp::new(COLON, t_spec, t_spec_as_expr));
}
sig.pat = buf_sig;
guards
}
ParamPattern::VarName(name) => {
if let Some(t_spec) = sig
.t_spec
.as_ref()
.map(|t_spec| t_spec.t_spec_as_expr.as_ref())
{
guards.extend(Self::type_guard(name.inspect().clone(), t_spec, name));
}
let ident = Identifier::new(VisModifierSpec::Private, name.clone());
let v = VarSignature::new(VarPattern::Ident(ident), sig.t_spec.clone());
let def = Def::new(Signature::Var(v), body);
guards.push(GuardClause::Bind(def));
guards
}
ParamPattern::Lit(l) => {
let lhs = body.block.remove(0);
let rhs = Expr::Literal(l.clone());
let eq = Token::new_fake(TokenKind::DblEq, "==", l.ln_begin().unwrap_or(0), 0, 0);
let guard = Expr::BinOp(lhs.bin_op(eq, rhs));
guards.push(GuardClause::Condition(guard));
let lit = l.clone();
sig.pat = ParamPattern::VarName(VarName::new(Token::new_fake(
TokenKind::Symbol,
self.var_gen.fresh_param_name(),
l.ln_begin().unwrap_or(1),
l.col_begin().unwrap_or(1),
l.col_end().unwrap_or(1),
)));
let t_spec = TypeSpec::enum_t_spec(vec![lit.clone()]);
let t_spec_as_expr = Self::dummy_set_expr(lit);
sig.t_spec = Some(TypeSpecWithOp::new(COLON, t_spec, t_spec_as_expr));
guards
}
_ => guards,
}
}
fn _desugar_self(module: Module) -> Module {
Self::desugar_all_chunks(module, Self::_desugar_self_inner)
}
fn _desugar_self_inner(_expr: Expr) -> Expr {
todo!()
}
fn _desugar_refinement_pattern(_mod: Module) -> Module {
todo!()
}
fn desugar_acc(module: Module) -> Module {
Self::desugar_all_chunks(module, Self::rec_desugar_acc)
}
fn rec_desugar_acc(expr: Expr) -> Expr {
match expr {
Expr::Accessor(acc) => Self::desugar_acc_inner(acc),
expr => Self::perform_desugar(Self::rec_desugar_acc, expr),
}
}
fn desugar_acc_inner(acc: Accessor) -> Expr {
match acc {
Accessor::Subscr(subscr) => {
let loc = subscr.loc();
let args = Args::single(PosArg::new(Self::rec_desugar_acc(*subscr.index)));
let call = Call::new(
Self::rec_desugar_acc(*subscr.obj),
Some(Identifier::public_with_loc(
DOT,
Str::ever("__getitem__"),
loc,
)),
args,
);
Expr::Call(call)
}
Accessor::TupleAttr(tattr) => {
let loc = tattr.loc();
let args = Args::single(PosArg::new(Expr::Literal(tattr.index)));
let call = Call::new(
Self::rec_desugar_acc(*tattr.obj),
Some(Identifier::public_with_loc(
DOT,
Str::ever("__Tuple_getitem__"),
loc,
)),
args,
);
Expr::Call(call)
}
Accessor::TypeApp(mut tapp) => {
tapp.obj = Box::new(Self::rec_desugar_acc(*tapp.obj));
Expr::Accessor(Accessor::TypeApp(tapp))
}
Accessor::Attr(mut attr) => {
attr.obj = Box::new(Self::rec_desugar_acc(*attr.obj));
attr.ident = Self::desugar_ident(attr.ident);
Expr::Accessor(Accessor::Attr(attr))
}
Accessor::Ident(ident) => Expr::Accessor(Accessor::Ident(Self::desugar_ident(ident))),
}
}
fn desugar_ident(mut ident: Identifier) -> Identifier {
if let Some(name) = symop_to_dname(ident.inspect()) {
ident.name.rename(name.into());
}
ident
}
fn desugar_operator(module: Module) -> Module {
Self::desugar_all_chunks(module, Self::rec_desugar_operator)
}
fn rec_desugar_operator(expr: Expr) -> Expr {
match expr {
Expr::BinOp(bin) if bin.op.is(TokenKind::InOp) => {
let (mut op, lhs, rhs) = bin.deconstruct();
op.content = Str::from("contains");
op.kind = TokenKind::ContainsOp;
Expr::BinOp(BinOp::new(op, rhs, lhs))
}
Expr::BinOp(bin) if bin.op.is(TokenKind::NotInOp) => {
let (mut op, lhs, rhs) = bin.deconstruct();
op.content = Str::from("contains");
op.kind = TokenKind::ContainsOp;
let not = if PYTHON_MODE {
Identifier::public("not".into())
} else {
Identifier::private("not".into())
};
let bin = Expr::BinOp(BinOp::new(op, rhs, lhs));
Expr::Accessor(Accessor::Ident(not)).call1(bin)
}
expr => Self::perform_desugar(Self::rec_desugar_operator, expr),
}
}
fn desugar_comprehension(module: Module) -> Module {
Self::desugar_all_chunks(module, Self::rec_desugar_comprehension)
}
fn rec_desugar_comprehension(expr: Expr) -> Expr {
match expr {
Expr::List(List::Comprehension(mut comp)) => {
debug_power_assert!(comp.generators.len(), >, 0);
if comp.generators.len() != 1 {
return Expr::List(List::Comprehension(comp));
}
let (ident, iter) = comp.generators.remove(0);
let iterator = Self::desugar_layout_and_guard(ident, iter, comp.layout, comp.guard);
Identifier::auto("list".into())
.call1(iterator.into())
.into()
}
Expr::Dict(Dict::Comprehension(mut comp)) => {
debug_power_assert!(comp.generators.len(), >, 0);
if comp.generators.len() != 1 {
return Expr::Dict(Dict::Comprehension(comp));
}
let (ident, iter) = comp.generators.remove(0);
let params = Params::single(NonDefaultParamSignature::new(
ParamPattern::VarName(ident.name),
None,
));
let sig = LambdaSignature::new(params, None, TypeBoundSpecs::empty());
let tuple = Tuple::Normal(NormalTuple::new(Args::pos_only(
vec![PosArg::new(comp.kv.key), PosArg::new(comp.kv.value)],
None,
)));
let body = Block::new(vec![tuple.into()]);
let lambda = Lambda::new(sig, Token::DUMMY, body, DefId(0));
let map = Identifier::private("map".into()).call2(lambda.into(), iter);
Identifier::auto("dict".into()).call1(map.into()).into()
}
Expr::Set(astSet::Comprehension(mut comp)) => {
debug_power_assert!(comp.generators.len(), >, 0);
if comp.generators.len() != 1 {
return Expr::Set(astSet::Comprehension(comp));
}
let (ident, iter) = comp.generators.remove(0);
let iterator = Self::desugar_layout_and_guard(ident, iter, comp.layout, comp.guard);
Identifier::auto("set".into()).call1(iterator.into()).into()
}
expr => Self::perform_desugar(Self::rec_desugar_comprehension, expr),
}
}
fn desugar_layout_and_guard(
ident: Identifier,
iter: Expr,
layout: Option<Box<Expr>>,
guard: Option<Box<Expr>>,
) -> Call {
let params = Params::single(NonDefaultParamSignature::new(
ParamPattern::VarName(ident.name),
None,
));
let sig = LambdaSignature::new(params, None, TypeBoundSpecs::empty());
match (layout, guard) {
(Some(elem), Some(guard)) => {
let f_body = Block::new(vec![*guard]);
let f_lambda = Lambda::new(sig.clone(), Token::DUMMY, f_body, DefId(0));
let filter = Identifier::auto("filter".into()).call2(f_lambda.into(), iter);
let m_body = Block::new(vec![*elem]);
let m_lambda = Lambda::new(sig, Token::DUMMY, m_body, DefId(0));
Identifier::auto("map".into()).call2(m_lambda.into(), filter.into())
}
(Some(elem), None) => {
let body = Block::new(vec![*elem]);
let lambda = Lambda::new(sig, Token::DUMMY, body, DefId(0));
Identifier::auto("map".into()).call2(lambda.into(), iter)
}
(None, Some(guard)) => {
let body = Block::new(vec![*guard]);
let lambda = Lambda::new(sig, Token::DUMMY, body, DefId(0));
Identifier::auto("filter".into()).call2(lambda.into(), iter)
}
(None, None) => todo!(),
}
}
}
impl Default for Desugarer {
fn default() -> Self {
Self::new()
}
}