Skip to main content

rustc_ty_utils/
common_traits.rs

1//! Queries for checking whether a type implements one of a few common traits.
2
3use rustc_hir::lang_items::LangItem;
4use rustc_infer::infer::TyCtxtInferExt;
5use rustc_middle::query::Providers;
6use rustc_middle::ty::{self, Ty, TyCtxt};
7use rustc_span::DUMMY_SP;
8use rustc_trait_selection::traits;
9
10fn is_copy_raw<'tcx>(tcx: TyCtxt<'tcx>, query: ty::PseudoCanonicalInput<'tcx, Ty<'tcx>>) -> bool {
11    is_trait_raw(tcx, query, LangItem::Copy)
12}
13
14fn is_use_cloned_raw<'tcx>(
15    tcx: TyCtxt<'tcx>,
16    query: ty::PseudoCanonicalInput<'tcx, Ty<'tcx>>,
17) -> bool {
18    is_trait_raw(tcx, query, LangItem::UseCloned)
19}
20
21fn is_sized_raw<'tcx>(tcx: TyCtxt<'tcx>, query: ty::PseudoCanonicalInput<'tcx, Ty<'tcx>>) -> bool {
22    is_trait_raw(tcx, query, LangItem::Sized)
23}
24
25fn is_freeze_raw<'tcx>(tcx: TyCtxt<'tcx>, query: ty::PseudoCanonicalInput<'tcx, Ty<'tcx>>) -> bool {
26    is_trait_raw(tcx, query, LangItem::Freeze)
27}
28
29fn is_unsafe_unpin_raw<'tcx>(
30    tcx: TyCtxt<'tcx>,
31    query: ty::PseudoCanonicalInput<'tcx, Ty<'tcx>>,
32) -> bool {
33    is_trait_raw(tcx, query, LangItem::UnsafeUnpin)
34}
35
36fn is_unpin_raw<'tcx>(tcx: TyCtxt<'tcx>, query: ty::PseudoCanonicalInput<'tcx, Ty<'tcx>>) -> bool {
37    is_trait_raw(tcx, query, LangItem::Unpin)
38}
39
40fn is_async_drop_raw<'tcx>(
41    tcx: TyCtxt<'tcx>,
42    query: ty::PseudoCanonicalInput<'tcx, Ty<'tcx>>,
43) -> bool {
44    is_trait_raw(tcx, query, LangItem::AsyncDrop)
45}
46
47fn is_trait_raw<'tcx>(
48    tcx: TyCtxt<'tcx>,
49    query: ty::PseudoCanonicalInput<'tcx, Ty<'tcx>>,
50    item: LangItem,
51) -> bool {
52    let (infcx, param_env) = tcx.infer_ctxt().build_with_typing_env(query.typing_env);
53    let trait_def_id = tcx.require_lang_item(item, DUMMY_SP);
54    traits::type_known_to_meet_bound_modulo_regions(&infcx, param_env, query.value, trait_def_id)
55}
56
57pub(crate) fn provide(providers: &mut Providers) {
58    *providers = Providers {
59        is_copy_raw,
60        is_use_cloned_raw,
61        is_sized_raw,
62        is_freeze_raw,
63        is_unsafe_unpin_raw,
64        is_unpin_raw,
65        is_async_drop_raw,
66        ..*providers
67    };
68}