1use std::cell::RefCell;
178use std::ops::Not;
179use std::{iter, vec};
180
181pub(crate) use StaticFields::*;
182pub(crate) use SubstructureFields::*;
183use rustc_ast::token::{IdentIsRaw, LitKind, Token, TokenKind};
184use rustc_ast::tokenstream::{DelimSpan, Spacing, TokenTree};
185use rustc_ast::{
186 self as ast, AnonConst, AttrArgs, BindingMode, ByRef, DelimArgs, EnumDef, Expr, GenericArg,
187 GenericParamKind, Generics, Mutability, PatKind, Safety, VariantData,
188};
189use rustc_attr_ir::{Attribute, AttributeKind, ReprPacked};
190use rustc_attr_parsing::AttributeParser;
191use rustc_expand::base::{Annotatable, ExtCtxt};
192use rustc_span::{DUMMY_SP, Ident, Span, Symbol, kw, sym};
193use thin_vec::{ThinVec, thin_vec};
194use ty::{Bounds, Path, Ref, Self_, Ty};
195
196use crate::{deriving, diagnostics};
197
198pub(crate) mod ty;
199
200pub(crate) struct TraitDef<'a> {
201 pub span: Span,
203
204 pub path: Path,
206
207 pub skip_path_as_bound: bool,
209
210 pub needs_copy_as_bound_if_packed: bool,
212
213 pub additional_bounds: Vec<Ty>,
216
217 pub supports_unions: bool,
219
220 pub methods: Vec<MethodDef<'a>>,
221
222 pub associated_types: Vec<(Ident, Ty)>,
223
224 pub is_const: bool,
225
226 pub is_staged_api_crate: bool,
227
228 pub safety: Safety,
230
231 pub document: bool,
233}
234
235pub(crate) struct MethodDef<'a> {
236 pub name: Symbol,
238 pub generics: Bounds,
240
241 pub explicit_self: bool,
243
244 pub nonself_args: Vec<(Ty, Symbol)>,
246
247 pub ret_ty: Ty,
249
250 pub attributes: ast::AttrVec,
251
252 pub fieldless_variants_strategy: FieldlessVariantsStrategy,
253
254 pub combine_substructure: RefCell<CombineSubstructureFunc<'a>>,
255}
256
257#[derive(#[automatically_derived]
impl ::core::cmp::PartialEq for FieldlessVariantsStrategy {
#[inline]
fn eq(&self, other: &FieldlessVariantsStrategy) -> bool {
let __self_discr = ::core::intrinsics::discriminant_value(self);
let __arg1_discr = ::core::intrinsics::discriminant_value(other);
__self_discr == __arg1_discr
}
}PartialEq)]
259pub(crate) enum FieldlessVariantsStrategy {
260 Unify,
264 Default,
267 SpecializeIfAllVariantsFieldless,
271}
272
273pub(crate) struct Substructure<'a> {
275 pub type_ident: Ident,
277 pub nonselflike_args: &'a [Box<Expr>],
280 pub fields: &'a SubstructureFields<'a>,
281}
282
283pub(crate) struct FieldInfo {
285 pub span: Span,
286 pub name: Option<Ident>,
289 pub self_expr: Box<Expr>,
292 pub other_selflike_exprs: Vec<Box<Expr>>,
295 pub maybe_scalar: bool,
296}
297
298#[derive(#[automatically_derived]
impl ::core::marker::Copy for IsTuple { }Copy, #[automatically_derived]
impl ::core::clone::Clone for IsTuple {
#[inline]
fn clone(&self) -> IsTuple { *self }
}Clone)]
299pub(crate) enum IsTuple {
300 No,
301 Yes,
302}
303
304pub(crate) enum StaticFields<'a> {
306 Unnamed(Vec<Span>, IsTuple),
308 Named(Vec<(Ident, Span, Option<&'a AnonConst>)>),
310}
311
312pub(crate) enum SubstructureFields<'a> {
314 Struct(&'a ast::VariantData, Vec<FieldInfo>),
316
317 AllFieldlessEnum(&'a ast::EnumDef),
321
322 EnumMatching(&'a ast::Variant, Vec<FieldInfo>),
326
327 EnumDiscr(FieldInfo, Option<Box<Expr>>),
331
332 StaticStruct(&'a ast::VariantData, StaticFields<'a>),
334
335 StaticEnum(&'a ast::EnumDef),
337}
338
339pub(crate) type CombineSubstructureFunc<'a> =
342 Box<dyn FnMut(&ExtCtxt<'_>, Span, &Substructure<'_>) -> BlockOrExpr + 'a>;
343
344pub(crate) fn combine_substructure(
345 f: CombineSubstructureFunc<'_>,
346) -> RefCell<CombineSubstructureFunc<'_>> {
347 RefCell::new(f)
348}
349
350struct TypeParameter {
351 bound_generic_params: ThinVec<ast::GenericParam>,
352 ty: Box<ast::Ty>,
353}
354
355pub(crate) struct BlockOrExpr(ThinVec<ast::Stmt>, Option<Box<Expr>>);
362
363impl BlockOrExpr {
364 pub(crate) fn new_stmts(stmts: ThinVec<ast::Stmt>) -> BlockOrExpr {
365 BlockOrExpr(stmts, None)
366 }
367
368 pub(crate) fn new_expr(expr: Box<Expr>) -> BlockOrExpr {
369 BlockOrExpr(ThinVec::new(), Some(expr))
370 }
371
372 pub(crate) fn new_mixed(stmts: ThinVec<ast::Stmt>, expr: Option<Box<Expr>>) -> BlockOrExpr {
373 BlockOrExpr(stmts, expr)
374 }
375
376 fn into_block(mut self, cx: &ExtCtxt<'_>, span: Span) -> Box<ast::Block> {
378 if let Some(expr) = self.1 {
379 self.0.push(cx.stmt_expr(expr));
380 }
381 cx.block(span, self.0)
382 }
383
384 fn into_expr(self, cx: &ExtCtxt<'_>, span: Span) -> Box<Expr> {
386 if self.0.is_empty() {
387 match self.1 {
388 None => cx.expr_block(cx.block(span, ThinVec::new())),
389 Some(expr) => expr,
390 }
391 } else if let [stmt] = self.0.as_slice()
392 && let ast::StmtKind::Expr(expr) = &stmt.kind
393 && self.1.is_none()
394 {
395 expr.clone()
397 } else {
398 cx.expr_block(self.into_block(cx, span))
400 }
401 }
402}
403
404fn find_type_parameters(
409 ty: &ast::Ty,
410 ty_param_names: &[Symbol],
411 cx: &ExtCtxt<'_>,
412) -> Vec<TypeParameter> {
413 use rustc_ast::visit;
414
415 struct Visitor<'a, 'b> {
416 cx: &'a ExtCtxt<'b>,
417 ty_param_names: &'a [Symbol],
418 bound_generic_params_stack: ThinVec<ast::GenericParam>,
419 type_params: Vec<TypeParameter>,
420 }
421
422 impl<'a, 'b> visit::Visitor<'a> for Visitor<'a, 'b> {
423 fn visit_ty(&mut self, ty: &'a ast::Ty) {
424 let stack_len = self.bound_generic_params_stack.len();
425 if let ast::TyKind::FnPtr(fn_ptr) = &ty.kind
426 && !fn_ptr.generic_params.is_empty()
427 {
428 self.bound_generic_params_stack.extend(fn_ptr.generic_params.iter().cloned());
431 }
432
433 if let ast::TyKind::Path(_, path) = &ty.kind
434 && let Some(segment) = path.segments.first()
435 && self.ty_param_names.contains(&segment.ident.name)
436 {
437 self.type_params.push(TypeParameter {
438 bound_generic_params: self.bound_generic_params_stack.clone(),
439 ty: Box::new(ty.clone()),
440 });
441 }
442
443 visit::walk_ty(self, ty);
444 self.bound_generic_params_stack.truncate(stack_len);
445 }
446
447 fn visit_poly_trait_ref(&mut self, trait_ref: &'a ast::PolyTraitRef) {
449 let stack_len = self.bound_generic_params_stack.len();
450 self.bound_generic_params_stack.extend(trait_ref.bound_generic_params.iter().cloned());
451
452 visit::walk_poly_trait_ref(self, trait_ref);
453
454 self.bound_generic_params_stack.truncate(stack_len);
455 }
456
457 fn visit_mac_call(&mut self, mac: &ast::MacCall) {
458 self.cx.dcx().emit_err(diagnostics::DeriveMacroCall { span: mac.span() });
459 }
460 }
461
462 let mut visitor = Visitor {
463 cx,
464 ty_param_names,
465 bound_generic_params_stack: ThinVec::new(),
466 type_params: Vec::new(),
467 };
468 visit::Visitor::visit_ty(&mut visitor, ty);
469
470 visitor.type_params
471}
472
473impl<'a> TraitDef<'a> {
474 pub(crate) fn expand(
475 self,
476 cx: &ExtCtxt<'_>,
477 mitem: &ast::MetaItem,
478 item: &'a Annotatable,
479 push: &mut dyn FnMut(Annotatable),
480 ) {
481 self.expand_ext(cx, mitem, item, push, false);
482 }
483
484 pub(crate) fn expand_ext(
485 self,
486 cx: &ExtCtxt<'_>,
487 mitem: &ast::MetaItem,
488 item: &'a Annotatable,
489 push: &mut dyn FnMut(Annotatable),
490 from_scratch: bool,
491 ) {
492 match item {
493 Annotatable::Item(item) => {
494 let is_packed = #[allow(non_exhaustive_omitted_patterns)] match AttributeParser::parse_limited_sym(cx.sess,
&item.attrs, &[sym::repr]) {
Some(Attribute::Parsed(AttributeKind::Repr { reprs, .. })) if
reprs.iter().any(|(x, _)|
#[allow(non_exhaustive_omitted_patterns)] match x {
ReprPacked(..) => true,
_ => false,
}) => true,
_ => false,
}matches!(
495 AttributeParser::parse_limited_sym(cx.sess, &item.attrs, &[sym::repr]),
496 Some(Attribute::Parsed(AttributeKind::Repr { reprs, .. })) if reprs.iter().any(|(x, _)| matches!(x, ReprPacked(..)))
497 );
498
499 let newitem = match &item.kind {
500 ast::ItemKind::Struct(ident, generics, struct_def) => self.expand_struct_def(
501 cx,
502 struct_def,
503 *ident,
504 generics,
505 from_scratch,
506 is_packed,
507 ),
508 ast::ItemKind::Enum(ident, generics, enum_def) => {
509 self.expand_enum_def(cx, enum_def, *ident, generics, from_scratch)
515 }
516 ast::ItemKind::Union(ident, generics, struct_def) => {
517 if self.supports_unions {
518 self.expand_struct_def(
519 cx,
520 struct_def,
521 *ident,
522 generics,
523 from_scratch,
524 is_packed,
525 )
526 } else {
527 cx.dcx().emit_err(diagnostics::DeriveUnion { span: mitem.span });
528 return;
529 }
530 }
531 _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
532 };
533 let mut attrs = newitem.attrs.clone();
536 attrs.extend(
537 item.attrs
538 .iter()
539 .filter(|a| {
540 a.has_any_name(&[
541 sym::allow,
542 sym::warn,
543 sym::deny,
544 sym::forbid,
545 sym::stable,
546 sym::unstable,
547 ])
548 })
549 .cloned(),
550 );
551 push(Annotatable::Item(Box::new(ast::Item { attrs, ..(*newitem).clone() })))
552 }
553 _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
554 }
555 }
556
557 fn create_derived_impl(
593 &self,
594 cx: &ExtCtxt<'_>,
595 type_ident: Ident,
596 generics: &Generics,
597 field_tys: Vec<&ast::Ty>,
598 methods: Vec<Box<ast::AssocItem>>,
599 is_packed: bool,
600 ) -> Box<ast::Item> {
601 let trait_path = self.path.to_path(cx, self.span, type_ident, generics);
602
603 let associated_types = self.associated_types.iter().map(|&(ident, ref type_def)| {
605 Box::new(ast::AssocItem {
606 id: ast::DUMMY_NODE_ID,
607 span: self.span,
608 vis: ast::Visibility {
609 span: self.span.shrink_to_lo(),
610 kind: ast::VisibilityKind::Inherited,
611 },
612 attrs: ast::AttrVec::new(),
613 kind: ast::AssocItemKind::Type(Box::new(ast::TyAlias {
614 defaultness: ast::Defaultness::Implicit,
615 ident,
616 generics: Generics::default(),
617 after_where_clause: ast::WhereClause::default(),
618 bounds: ThinVec::new(),
619 ty: Some(type_def.to_ty(cx, self.span, type_ident, generics)),
620 })),
621 tokens: None,
622 })
623 });
624
625 let mut where_clause = ast::WhereClause::default();
626 where_clause.span = generics.where_clause.span;
627 let ctxt = self.span.ctxt();
628 let span = generics.span.with_ctxt(ctxt);
629
630 let params: ThinVec<_> = generics
632 .params
633 .iter()
634 .map(|param| match ¶m.kind {
635 GenericParamKind::Lifetime { .. } => param.clone(),
636 GenericParamKind::Type { .. } => {
637 let span = param.ident.span.with_ctxt(ctxt);
640 let bounds: ThinVec<_> = self
641 .additional_bounds
642 .iter()
643 .map(|p| {
644 cx.trait_bound(p.to_path(cx, span, type_ident, generics), self.is_const)
645 })
646 .chain(
647 self.skip_path_as_bound.not().then(|| {
649 let mut trait_path = trait_path.clone();
650 trait_path.span = span;
651 cx.trait_bound(trait_path, self.is_const)
652 }),
653 )
654 .chain({
655 if is_packed && self.needs_copy_as_bound_if_packed {
657 let p = generic::ty::Path::new({
::alloc::boxed::box_assume_init_into_vec_unsafe(::alloc::intrinsics::write_box_via_move(::alloc::boxed::Box::new_uninit(),
[sym::marker, sym::Copy]))
})deriving::path_std!(marker::Copy);
658 Some(cx.trait_bound(
659 p.to_path(cx, span, type_ident, generics),
660 self.is_const,
661 ))
662 } else {
663 None
664 }
665 })
666 .chain(
667 param.bounds.iter().cloned(),
669 )
670 .collect();
671
672 cx.typaram(span, param.ident, bounds, None)
673 }
674 GenericParamKind::Const { ty, span, .. } => {
675 let const_nodefault_kind = GenericParamKind::Const {
676 ty: ty.clone(),
677 span: span.with_ctxt(ctxt),
678
679 default: None,
681 };
682 let mut param_clone = param.clone();
683 param_clone.kind = const_nodefault_kind;
684 param_clone
685 }
686 })
687 .map(|mut param| {
688 param.attrs.clear();
691 param
692 })
693 .collect();
694
695 where_clause.predicates.extend(generics.where_clause.predicates.iter().map(|clause| {
697 ast::WherePredicate {
698 attrs: clause.attrs.clone(),
699 kind: clause.kind.clone(),
700 id: ast::DUMMY_NODE_ID,
701 span: clause.span.with_ctxt(ctxt),
702 is_placeholder: false,
703 }
704 }));
705
706 let ty_param_names: Vec<Symbol> = params
707 .iter()
708 .filter(|param| #[allow(non_exhaustive_omitted_patterns)] match param.kind {
ast::GenericParamKind::Type { .. } => true,
_ => false,
}matches!(param.kind, ast::GenericParamKind::Type { .. }))
709 .map(|ty_param| ty_param.ident.name)
710 .collect();
711
712 if !ty_param_names.is_empty() {
713 for field_ty in field_tys {
714 let field_ty_params = find_type_parameters(&field_ty, &ty_param_names, cx);
715
716 for field_ty_param in field_ty_params {
717 if let ast::TyKind::Path(_, p) = &field_ty_param.ty.kind
719 && let [sole_segment] = &*p.segments
720 && ty_param_names.contains(&sole_segment.ident.name)
721 {
722 continue;
723 }
724 let mut bounds: ThinVec<_> = self
725 .additional_bounds
726 .iter()
727 .map(|p| {
728 cx.trait_bound(
729 p.to_path(cx, self.span, type_ident, generics),
730 self.is_const,
731 )
732 })
733 .collect();
734
735 if !self.skip_path_as_bound {
737 bounds.push(cx.trait_bound(trait_path.clone(), self.is_const));
738 }
739
740 if is_packed && self.needs_copy_as_bound_if_packed {
742 let p = generic::ty::Path::new({
::alloc::boxed::box_assume_init_into_vec_unsafe(::alloc::intrinsics::write_box_via_move(::alloc::boxed::Box::new_uninit(),
[sym::marker, sym::Copy]))
})deriving::path_std!(marker::Copy);
743 bounds.push(cx.trait_bound(
744 p.to_path(cx, self.span, type_ident, generics),
745 self.is_const,
746 ));
747 }
748
749 if !bounds.is_empty() {
750 let predicate = ast::WhereBoundPredicate {
751 bound_generic_params: field_ty_param.bound_generic_params,
752 bounded_ty: field_ty_param.ty,
753 bounds,
754 };
755
756 let kind = ast::WherePredicateKind::BoundPredicate(predicate);
757 let predicate = ast::WherePredicate {
758 attrs: ThinVec::new(),
759 kind,
760 id: ast::DUMMY_NODE_ID,
761 span: self.span,
762 is_placeholder: false,
763 };
764 where_clause.predicates.push(predicate);
765 }
766 }
767 }
768 }
769
770 let trait_generics = Generics { params, where_clause, span };
771
772 let trait_ref = cx.trait_ref(trait_path);
774
775 let self_params: Vec<_> = generics
776 .params
777 .iter()
778 .map(|param| match param.kind {
779 GenericParamKind::Lifetime { .. } => {
780 GenericArg::Lifetime(cx.lifetime(param.ident.span.with_ctxt(ctxt), param.ident))
781 }
782 GenericParamKind::Type { .. } => {
783 GenericArg::Type(cx.ty_ident(param.ident.span.with_ctxt(ctxt), param.ident))
784 }
785 GenericParamKind::Const { .. } => {
786 GenericArg::Const(cx.const_ident(param.ident.span.with_ctxt(ctxt), param.ident))
787 }
788 })
789 .collect();
790
791 let path =
793 cx.path_all(type_ident.span.with_ctxt(ctxt), false, ::alloc::boxed::box_assume_init_into_vec_unsafe(::alloc::intrinsics::write_box_via_move(::alloc::boxed::Box::new_uninit(),
[type_ident]))vec![type_ident], self_params);
794 let self_type = cx.ty_path(path);
795 let rustc_const_unstable =
796 cx.path_ident(self.span, Ident::new(sym::rustc_const_unstable, self.span));
797
798 let mut attrs = {
let len = [()].len();
let mut vec = ::thin_vec::ThinVec::with_capacity(len);
vec.push(cx.attr_word(sym::automatically_derived, self.span));
vec
}thin_vec![cx.attr_word(sym::automatically_derived, self.span),];
799
800 if self.is_const && self.is_staged_api_crate {
804 attrs.push(
805 cx.attr_nested(
806 rustc_ast::AttrItem {
807 unsafety: Safety::Default,
808 path: rustc_const_unstable,
809 args: AttrArgs::Delimited(DelimArgs {
810 dspan: DelimSpan::from_single(self.span),
811 delim: rustc_ast::token::Delimiter::Parenthesis,
812 tokens: [
813 TokenKind::Ident(sym::feature, IdentIsRaw::No),
814 TokenKind::Eq,
815 TokenKind::lit(LitKind::Str, sym::derive_const, None),
816 TokenKind::Comma,
817 TokenKind::Ident(sym::issue, IdentIsRaw::No),
818 TokenKind::Eq,
819 TokenKind::lit(LitKind::Str, sym::derive_const_issue, None),
820 ]
821 .into_iter()
822 .map(|kind| {
823 TokenTree::Token(Token { kind, span: self.span }, Spacing::Alone)
824 })
825 .collect(),
826 }),
827 span: self.span,
828 },
829 self.span,
830 ),
831 )
832 }
833
834 if !self.document {
835 attrs.push(cx.attr_nested_word(sym::doc, sym::hidden, self.span));
836 }
837
838 cx.item(
839 self.span,
840 attrs,
841 ast::ItemKind::Impl(ast::Impl {
842 generics: trait_generics,
843 of_trait: Some(Box::new(ast::TraitImplHeader {
844 safety: self.safety,
845 polarity: ast::ImplPolarity::Positive,
846 defaultness: ast::Defaultness::Implicit,
847 trait_ref,
848 })),
849 constness: if self.is_const { ast::Const::Yes(DUMMY_SP) } else { ast::Const::No },
850 self_ty: self_type,
851 items: methods.into_iter().chain(associated_types).collect(),
852 }),
853 )
854 }
855
856 fn expand_struct_def(
857 &self,
858 cx: &ExtCtxt<'_>,
859 struct_def: &'a VariantData,
860 type_ident: Ident,
861 generics: &Generics,
862 from_scratch: bool,
863 is_packed: bool,
864 ) -> Box<ast::Item> {
865 let field_tys = Vec::from_iter(struct_def.fields().iter().map(|field| &*field.ty));
866
867 let methods = self
868 .methods
869 .iter()
870 .map(|method_def| {
871 let (explicit_self, selflike_args, nonselflike_args, nonself_arg_tys) =
872 method_def.extract_arg_details(cx, self, type_ident, generics);
873
874 let body = if from_scratch || method_def.is_static() {
875 method_def.expand_static_struct_method_body(
876 cx,
877 self,
878 struct_def,
879 type_ident,
880 &nonselflike_args,
881 )
882 } else {
883 method_def.expand_struct_method_body(
884 cx,
885 self,
886 struct_def,
887 type_ident,
888 &selflike_args,
889 &nonselflike_args,
890 is_packed,
891 )
892 };
893
894 method_def.create_method(
895 cx,
896 self,
897 type_ident,
898 generics,
899 explicit_self,
900 nonself_arg_tys,
901 body,
902 )
903 })
904 .collect();
905
906 self.create_derived_impl(cx, type_ident, generics, field_tys, methods, is_packed)
907 }
908
909 fn expand_enum_def(
910 &self,
911 cx: &ExtCtxt<'_>,
912 enum_def: &'a EnumDef,
913 type_ident: Ident,
914 generics: &Generics,
915 from_scratch: bool,
916 ) -> Box<ast::Item> {
917 let field_tys = Vec::from_iter(
918 enum_def
919 .variants
920 .iter()
921 .flat_map(|variant| variant.data.fields())
922 .map(|field| &*field.ty),
923 );
924
925 let methods = self
926 .methods
927 .iter()
928 .map(|method_def| {
929 let (explicit_self, selflike_args, nonselflike_args, nonself_arg_tys) =
930 method_def.extract_arg_details(cx, self, type_ident, generics);
931
932 let body = if from_scratch || method_def.is_static() {
933 method_def.expand_static_enum_method_body(
934 cx,
935 self,
936 enum_def,
937 type_ident,
938 &nonselflike_args,
939 )
940 } else {
941 method_def.expand_enum_method_body(
942 cx,
943 self,
944 enum_def,
945 type_ident,
946 selflike_args,
947 &nonselflike_args,
948 )
949 };
950
951 method_def.create_method(
952 cx,
953 self,
954 type_ident,
955 generics,
956 explicit_self,
957 nonself_arg_tys,
958 body,
959 )
960 })
961 .collect();
962
963 let is_packed = false; self.create_derived_impl(cx, type_ident, generics, field_tys, methods, is_packed)
965 }
966}
967
968impl<'a> MethodDef<'a> {
969 fn call_substructure_method(
970 &self,
971 cx: &ExtCtxt<'_>,
972 trait_: &TraitDef<'_>,
973 type_ident: Ident,
974 nonselflike_args: &[Box<Expr>],
975 fields: &SubstructureFields<'_>,
976 ) -> BlockOrExpr {
977 let span = trait_.span;
978 let substructure = Substructure { type_ident, nonselflike_args, fields };
979 let mut f = self.combine_substructure.borrow_mut();
980 let f: &mut CombineSubstructureFunc<'_> = &mut *f;
981 f(cx, span, &substructure)
982 }
983
984 fn is_static(&self) -> bool {
985 !self.explicit_self
986 }
987
988 fn extract_arg_details(
996 &self,
997 cx: &ExtCtxt<'_>,
998 trait_: &TraitDef<'_>,
999 type_ident: Ident,
1000 generics: &Generics,
1001 ) -> (Option<ast::ExplicitSelf>, ThinVec<Box<Expr>>, Vec<Box<Expr>>, Vec<(Ident, Box<ast::Ty>)>)
1002 {
1003 let mut selflike_args = ThinVec::new();
1004 let mut nonselflike_args = Vec::new();
1005 let mut nonself_arg_tys = Vec::new();
1006 let span = trait_.span;
1007
1008 let explicit_self = self.explicit_self.then(|| {
1009 let (self_expr, explicit_self) = ty::get_explicit_self(cx, span);
1010 selflike_args.push(self_expr);
1011 explicit_self
1012 });
1013
1014 for (ty, name) in self.nonself_args.iter() {
1015 let ast_ty = ty.to_ty(cx, span, type_ident, generics);
1016 let ident = Ident::new(*name, span);
1017 nonself_arg_tys.push((ident, ast_ty));
1018
1019 let arg_expr = cx.expr_ident(span, ident);
1020
1021 match ty {
1022 Ref(Self_, _) if !self.is_static() => selflike_args.push(arg_expr),
1024 Self_ => cx.dcx().span_bug(span, "`Self` in non-return position"),
1025 _ => nonselflike_args.push(arg_expr),
1026 }
1027 }
1028
1029 (explicit_self, selflike_args, nonselflike_args, nonself_arg_tys)
1030 }
1031
1032 fn create_method(
1033 &self,
1034 cx: &ExtCtxt<'_>,
1035 trait_: &TraitDef<'_>,
1036 type_ident: Ident,
1037 generics: &Generics,
1038 explicit_self: Option<ast::ExplicitSelf>,
1039 nonself_arg_tys: Vec<(Ident, Box<ast::Ty>)>,
1040 body: BlockOrExpr,
1041 ) -> Box<ast::AssocItem> {
1042 let span = trait_.span;
1043 let fn_generics = self.generics.to_generics(cx, span, type_ident, generics);
1045
1046 let args = {
1047 let self_arg = explicit_self.map(|explicit_self| {
1048 let ident = Ident::new(kw::SelfLower, span);
1049 ast::Param::from_self(ast::AttrVec::default(), explicit_self, ident)
1050 });
1051 let nonself_args =
1052 nonself_arg_tys.into_iter().map(|(name, ty)| cx.param(span, name, ty));
1053 self_arg.into_iter().chain(nonself_args).collect()
1054 };
1055
1056 let ret_type = if let Ty::Unit = &self.ret_ty {
1057 ast::FnRetTy::Default(span)
1058 } else {
1059 ast::FnRetTy::Ty(self.ret_ty.to_ty(cx, span, type_ident, generics))
1060 };
1061
1062 let method_ident = Ident::new(self.name, span);
1063 let fn_decl = cx.fn_decl(args, ret_type);
1064 let body_block = body.into_block(cx, span);
1065
1066 let trait_lo_sp = span.shrink_to_lo();
1067
1068 let sig = ast::FnSig { header: ast::FnHeader::default(), decl: fn_decl, span };
1069 let defaultness = ast::Defaultness::Implicit;
1070
1071 Box::new(ast::AssocItem {
1073 id: ast::DUMMY_NODE_ID,
1074 attrs: self.attributes.clone(),
1075 span,
1076 vis: ast::Visibility { span: trait_lo_sp, kind: ast::VisibilityKind::Inherited },
1077 kind: ast::AssocItemKind::Fn(Box::new(ast::Fn {
1078 defaultness,
1079 sig,
1080 ident: method_ident,
1081 generics: fn_generics,
1082 contract: None,
1083 body: Some(body_block),
1084 define_opaque: None,
1085 eii_impl: None,
1086 })),
1087 tokens: None,
1088 })
1089 }
1090
1091 fn expand_struct_method_body<'b>(
1127 &self,
1128 cx: &ExtCtxt<'_>,
1129 trait_: &TraitDef<'b>,
1130 struct_def: &'b VariantData,
1131 type_ident: Ident,
1132 selflike_args: &[Box<Expr>],
1133 nonselflike_args: &[Box<Expr>],
1134 is_packed: bool,
1135 ) -> BlockOrExpr {
1136 if !(selflike_args.len() == 1 || selflike_args.len() == 2) {
::core::panicking::panic("assertion failed: selflike_args.len() == 1 || selflike_args.len() == 2")
};assert!(selflike_args.len() == 1 || selflike_args.len() == 2);
1137
1138 let selflike_fields =
1139 trait_.create_struct_field_access_fields(cx, selflike_args, struct_def, is_packed);
1140 self.call_substructure_method(
1141 cx,
1142 trait_,
1143 type_ident,
1144 nonselflike_args,
1145 &Struct(struct_def, selflike_fields),
1146 )
1147 }
1148
1149 fn expand_static_struct_method_body(
1150 &self,
1151 cx: &ExtCtxt<'_>,
1152 trait_: &TraitDef<'a>,
1153 struct_def: &'a VariantData,
1154 type_ident: Ident,
1155 nonselflike_args: &[Box<Expr>],
1156 ) -> BlockOrExpr {
1157 let summary = trait_.summarise_struct(cx, struct_def);
1158
1159 self.call_substructure_method(
1160 cx,
1161 trait_,
1162 type_ident,
1163 nonselflike_args,
1164 &StaticStruct(struct_def, summary),
1165 )
1166 }
1167
1168 fn expand_enum_method_body<'b>(
1204 &self,
1205 cx: &ExtCtxt<'_>,
1206 trait_: &TraitDef<'b>,
1207 enum_def: &'b EnumDef,
1208 type_ident: Ident,
1209 mut selflike_args: ThinVec<Box<Expr>>,
1210 nonselflike_args: &[Box<Expr>],
1211 ) -> BlockOrExpr {
1212 if !!selflike_args.is_empty() {
{
::core::panicking::panic_fmt(format_args!("static methods must use `expand_static_enum_method_body`"));
}
};assert!(
1213 !selflike_args.is_empty(),
1214 "static methods must use `expand_static_enum_method_body`",
1215 );
1216
1217 let span = trait_.span;
1218 let variants = &enum_def.variants;
1219
1220 let unify_fieldless_variants =
1222 self.fieldless_variants_strategy == FieldlessVariantsStrategy::Unify;
1223
1224 if variants.is_empty() {
1228 selflike_args.truncate(1);
1229 let match_arg = cx.expr_deref(span, selflike_args.pop().unwrap());
1230 let match_arms = ThinVec::new();
1231 let expr = cx.expr_match(span, match_arg, match_arms);
1232 return BlockOrExpr(ThinVec::new(), Some(expr));
1233 }
1234
1235 let prefixes = iter::once("__self".to_string())
1236 .chain(
1237 selflike_args
1238 .iter()
1239 .enumerate()
1240 .skip(1)
1241 .map(|(arg_count, _selflike_arg)| ::alloc::__export::must_use({
::alloc::fmt::format(format_args!("__arg{0}", arg_count))
})format!("__arg{arg_count}")),
1242 )
1243 .collect::<Vec<String>>();
1244
1245 let get_discr_pieces = |cx: &ExtCtxt<'_>| {
1254 let discr_idents: Vec<_> = prefixes
1255 .iter()
1256 .map(|name| Ident::from_str_and_span(&::alloc::__export::must_use({
::alloc::fmt::format(format_args!("{0}_discr", name))
})format!("{name}_discr"), span))
1257 .collect();
1258
1259 let mut discr_exprs: Vec<_> = discr_idents
1260 .iter()
1261 .map(|&ident| cx.expr_addr_of(span, cx.expr_ident(span, ident)))
1262 .collect();
1263
1264 let self_expr = discr_exprs.remove(0);
1265 let other_selflike_exprs = discr_exprs;
1266 let discr_field =
1267 FieldInfo { span, name: None, self_expr, other_selflike_exprs, maybe_scalar: true };
1268
1269 let discr_let_stmts: ThinVec<_> = iter::zip(&discr_idents, &selflike_args)
1270 .map(|(&ident, selflike_arg)| {
1271 let variant_value = deriving::call_intrinsic(
1272 cx,
1273 span,
1274 sym::discriminant_value,
1275 {
let len = [()].len();
let mut vec = ::thin_vec::ThinVec::with_capacity(len);
vec.push(selflike_arg.clone());
vec
}thin_vec![selflike_arg.clone()],
1276 );
1277 cx.stmt_let(span, false, ident, variant_value)
1278 })
1279 .collect();
1280
1281 (discr_field, discr_let_stmts)
1282 };
1283
1284 let all_fieldless = variants.iter().all(|v| v.data.fields().is_empty());
1287 if all_fieldless {
1288 if variants.len() > 1 {
1289 match self.fieldless_variants_strategy {
1290 FieldlessVariantsStrategy::Unify => {
1291 let (discr_field, mut discr_let_stmts) = get_discr_pieces(cx);
1295 let mut discr_check = self.call_substructure_method(
1296 cx,
1297 trait_,
1298 type_ident,
1299 nonselflike_args,
1300 &EnumDiscr(discr_field, None),
1301 );
1302 discr_let_stmts.append(&mut discr_check.0);
1303 return BlockOrExpr(discr_let_stmts, discr_check.1);
1304 }
1305 FieldlessVariantsStrategy::SpecializeIfAllVariantsFieldless => {
1306 return self.call_substructure_method(
1307 cx,
1308 trait_,
1309 type_ident,
1310 nonselflike_args,
1311 &AllFieldlessEnum(enum_def),
1312 );
1313 }
1314 FieldlessVariantsStrategy::Default => (),
1315 }
1316 } else if let [variant] = variants.as_slice() {
1317 return self.call_substructure_method(
1320 cx,
1321 trait_,
1322 type_ident,
1323 nonselflike_args,
1324 &EnumMatching(variant, Vec::new()),
1325 );
1326 }
1327 }
1328
1329 let mut match_arms: ThinVec<ast::Arm> = variants
1335 .iter()
1336 .filter(|&v| !(unify_fieldless_variants && v.data.fields().is_empty()))
1337 .map(|variant| {
1338 let fields = trait_.create_struct_pattern_fields(cx, &variant.data, &prefixes);
1342
1343 let sp = variant.span.with_ctxt(trait_.span.ctxt());
1344 let variant_path = cx.path(sp, ::alloc::boxed::box_assume_init_into_vec_unsafe(::alloc::intrinsics::write_box_via_move(::alloc::boxed::Box::new_uninit(),
[type_ident, variant.ident]))vec![type_ident, variant.ident]);
1345 let by_ref = ByRef::No; let mut subpats = trait_.create_struct_patterns(
1347 cx,
1348 variant_path,
1349 &variant.data,
1350 &prefixes,
1351 by_ref,
1352 );
1353
1354 let single_pat = if subpats.len() == 1 {
1356 subpats.pop().unwrap()
1357 } else {
1358 cx.pat_tuple(span, subpats)
1359 };
1360
1361 let substructure = EnumMatching(variant, fields);
1370 let arm_expr = self
1371 .call_substructure_method(
1372 cx,
1373 trait_,
1374 type_ident,
1375 nonselflike_args,
1376 &substructure,
1377 )
1378 .into_expr(cx, span);
1379
1380 cx.arm(span, single_pat, arm_expr)
1381 })
1382 .collect();
1383
1384 let first_fieldless = variants.iter().find(|v| v.data.fields().is_empty());
1386 let default = match first_fieldless {
1387 Some(v) if unify_fieldless_variants => {
1388 Some(
1392 self.call_substructure_method(
1393 cx,
1394 trait_,
1395 type_ident,
1396 nonselflike_args,
1397 &EnumMatching(v, Vec::new()),
1398 )
1399 .into_expr(cx, span),
1400 )
1401 }
1402 _ if variants.len() > 1 && selflike_args.len() > 1 => {
1403 Some(deriving::call_unreachable(cx, span))
1407 }
1408 _ => None,
1409 };
1410 if let Some(arm) = default {
1411 match_arms.push(cx.arm(span, cx.pat_wild(span), arm));
1412 }
1413
1414 let get_match_expr = |mut selflike_args: ThinVec<Box<Expr>>| {
1423 let match_arg = if selflike_args.len() == 1 {
1424 selflike_args.pop().unwrap()
1425 } else {
1426 cx.expr(span, ast::ExprKind::Tup(selflike_args))
1427 };
1428 cx.expr_match(span, match_arg, match_arms)
1429 };
1430
1431 if unify_fieldless_variants && variants.len() > 1 {
1435 let (discr_field, mut discr_let_stmts) = get_discr_pieces(cx);
1436
1437 let mut discr_check_plus_match = self.call_substructure_method(
1439 cx,
1440 trait_,
1441 type_ident,
1442 nonselflike_args,
1443 &EnumDiscr(discr_field, Some(get_match_expr(selflike_args))),
1444 );
1445 discr_let_stmts.append(&mut discr_check_plus_match.0);
1446 BlockOrExpr(discr_let_stmts, discr_check_plus_match.1)
1447 } else {
1448 BlockOrExpr(ThinVec::new(), Some(get_match_expr(selflike_args)))
1449 }
1450 }
1451
1452 fn expand_static_enum_method_body(
1453 &self,
1454 cx: &ExtCtxt<'_>,
1455 trait_: &TraitDef<'_>,
1456 enum_def: &EnumDef,
1457 type_ident: Ident,
1458 nonselflike_args: &[Box<Expr>],
1459 ) -> BlockOrExpr {
1460 self.call_substructure_method(
1461 cx,
1462 trait_,
1463 type_ident,
1464 nonselflike_args,
1465 &StaticEnum(enum_def),
1466 )
1467 }
1468}
1469
1470impl<'a> TraitDef<'a> {
1472 fn summarise_struct(&self, cx: &ExtCtxt<'_>, struct_def: &'a VariantData) -> StaticFields<'a> {
1473 let mut named_idents = Vec::new();
1474 let mut just_spans = Vec::new();
1475 for field in struct_def.fields() {
1476 let sp = field.span.with_ctxt(self.span.ctxt());
1477 match field.ident {
1478 Some(ident) => named_idents.push((ident, sp, field.default_value())),
1479 _ => just_spans.push(sp),
1480 }
1481 }
1482
1483 let is_tuple = match struct_def {
1484 ast::VariantData::Tuple(..) => IsTuple::Yes,
1485 _ => IsTuple::No,
1486 };
1487 match (just_spans.is_empty(), named_idents.is_empty()) {
1488 (false, false) => cx
1489 .dcx()
1490 .span_bug(self.span, "a struct with named and unnamed fields in generic `derive`"),
1491 (_, false) => Named(named_idents),
1493 (false, _) => Unnamed(just_spans, is_tuple),
1495 _ => Named(Vec::new()),
1497 }
1498 }
1499
1500 fn create_struct_patterns(
1501 &self,
1502 cx: &ExtCtxt<'_>,
1503 struct_path: ast::Path,
1504 struct_def: &'a VariantData,
1505 prefixes: &[String],
1506 by_ref: ByRef,
1507 ) -> ThinVec<ast::Pat> {
1508 prefixes
1509 .iter()
1510 .map(|prefix| {
1511 let pieces_iter =
1512 struct_def.fields().iter().enumerate().map(|(i, struct_field)| {
1513 let sp = struct_field.span.with_ctxt(self.span.ctxt());
1514 let ident = self.mk_pattern_ident(prefix, i);
1515 let path = ident.with_span_pos(sp);
1516 (
1517 sp,
1518 struct_field.ident,
1519 cx.pat(
1520 path.span,
1521 PatKind::Ident(BindingMode(by_ref, Mutability::Not), path, None),
1522 ),
1523 )
1524 });
1525
1526 let struct_path = struct_path.clone();
1527 match *struct_def {
1528 VariantData::Struct { .. } => {
1529 let field_pats = pieces_iter
1530 .map(|(sp, ident, pat)| {
1531 if ident.is_none() {
1532 cx.dcx().span_bug(
1533 sp,
1534 "a braced struct with unnamed fields in `derive`",
1535 );
1536 }
1537 ast::PatField {
1538 ident: ident.unwrap(),
1539 is_shorthand: false,
1540 attrs: ast::AttrVec::new(),
1541 id: ast::DUMMY_NODE_ID,
1542 span: pat.span.with_ctxt(self.span.ctxt()),
1543 pat: Box::new(pat),
1544 is_placeholder: false,
1545 }
1546 })
1547 .collect();
1548 cx.pat_struct(self.span, struct_path, field_pats)
1549 }
1550 VariantData::Tuple(..) => {
1551 let subpats = pieces_iter.map(|(_, _, subpat)| subpat).collect();
1552 cx.pat_tuple_struct(self.span, struct_path, subpats)
1553 }
1554 VariantData::Unit(..) => cx.pat_path(self.span, struct_path),
1555 }
1556 })
1557 .collect()
1558 }
1559
1560 fn create_fields<F>(&self, struct_def: &'a VariantData, mk_exprs: F) -> Vec<FieldInfo>
1561 where
1562 F: Fn(usize, &ast::FieldDef, Span) -> Vec<Box<ast::Expr>>,
1563 {
1564 struct_def
1565 .fields()
1566 .iter()
1567 .enumerate()
1568 .map(|(i, struct_field)| {
1569 let sp = struct_field.span.with_ctxt(self.span.ctxt());
1572 let mut exprs: Vec<_> = mk_exprs(i, struct_field, sp);
1573 let self_expr = exprs.remove(0);
1574 let other_selflike_exprs = exprs;
1575 FieldInfo {
1576 span: sp.with_ctxt(self.span.ctxt()),
1577 name: struct_field.ident,
1578 self_expr,
1579 other_selflike_exprs,
1580 maybe_scalar: struct_field.ty.peel_refs().kind.maybe_scalar(),
1581 }
1582 })
1583 .collect()
1584 }
1585
1586 fn mk_pattern_ident(&self, prefix: &str, i: usize) -> Ident {
1587 Ident::from_str_and_span(&::alloc::__export::must_use({
::alloc::fmt::format(format_args!("{0}_{1}", prefix, i))
})format!("{prefix}_{i}"), self.span)
1588 }
1589
1590 fn create_struct_pattern_fields(
1591 &self,
1592 cx: &ExtCtxt<'_>,
1593 struct_def: &'a VariantData,
1594 prefixes: &[String],
1595 ) -> Vec<FieldInfo> {
1596 self.create_fields(struct_def, |i, _struct_field, sp| {
1597 prefixes
1598 .iter()
1599 .map(|prefix| {
1600 let ident = self.mk_pattern_ident(prefix, i);
1601 cx.expr_path(cx.path_ident(sp, ident))
1602 })
1603 .collect()
1604 })
1605 }
1606
1607 fn create_struct_field_access_fields(
1608 &self,
1609 cx: &ExtCtxt<'_>,
1610 selflike_args: &[Box<Expr>],
1611 struct_def: &'a VariantData,
1612 is_packed: bool,
1613 ) -> Vec<FieldInfo> {
1614 self.create_fields(struct_def, |i, struct_field, sp| {
1615 selflike_args
1616 .iter()
1617 .map(|selflike_arg| {
1618 let mut field_expr = cx.expr(
1623 sp,
1624 ast::ExprKind::Field(
1625 selflike_arg.clone(),
1626 struct_field.ident.unwrap_or_else(|| {
1627 Ident::from_str_and_span(&i.to_string(), struct_field.span)
1628 }),
1629 ),
1630 );
1631 if is_packed {
1632 field_expr = cx.expr_block(
1635 cx.block(struct_field.span, {
let len = [()].len();
let mut vec = ::thin_vec::ThinVec::with_capacity(len);
vec.push(cx.stmt_expr(field_expr));
vec
}thin_vec![cx.stmt_expr(field_expr)]),
1636 );
1637 }
1638 cx.expr_addr_of(sp, field_expr)
1639 })
1640 .collect()
1641 })
1642 }
1643}
1644
1645pub(crate) enum CsFold<'a> {
1649 Single(&'a FieldInfo),
1652
1653 Combine(Span, Box<Expr>, Box<Expr>),
1656
1657 Fieldless,
1659}
1660
1661pub(crate) fn cs_fold<F>(
1664 use_foldl: bool,
1665 cx: &ExtCtxt<'_>,
1666 trait_span: Span,
1667 substructure: &Substructure<'_>,
1668 mut f: F,
1669) -> Box<Expr>
1670where
1671 F: FnMut(&ExtCtxt<'_>, CsFold<'_>) -> Box<Expr>,
1672{
1673 match substructure.fields {
1674 EnumMatching(.., all_fields) | Struct(_, all_fields) => {
1675 if all_fields.is_empty() {
1676 return f(cx, CsFold::Fieldless);
1677 }
1678
1679 let (base_field, rest) = if use_foldl {
1680 all_fields.split_first().unwrap()
1681 } else {
1682 all_fields.split_last().unwrap()
1683 };
1684
1685 let base_expr = f(cx, CsFold::Single(base_field));
1686
1687 let op = |old, field: &FieldInfo| {
1688 let new = f(cx, CsFold::Single(field));
1689 f(cx, CsFold::Combine(field.span, old, new))
1690 };
1691
1692 if use_foldl {
1693 rest.iter().fold(base_expr, op)
1694 } else {
1695 rest.iter().rfold(base_expr, op)
1696 }
1697 }
1698 EnumDiscr(discr_field, match_expr) => {
1699 let discr_check_expr = f(cx, CsFold::Single(discr_field));
1700 if let Some(match_expr) = match_expr {
1701 if use_foldl {
1702 f(cx, CsFold::Combine(trait_span, discr_check_expr, match_expr.clone()))
1703 } else {
1704 f(cx, CsFold::Combine(trait_span, match_expr.clone(), discr_check_expr))
1705 }
1706 } else {
1707 discr_check_expr
1708 }
1709 }
1710 StaticEnum(..) | StaticStruct(..) => {
1711 cx.dcx().span_bug(trait_span, "static function in `derive`")
1712 }
1713 AllFieldlessEnum(..) => cx.dcx().span_bug(trait_span, "fieldless enum in `derive`"),
1714 }
1715}