1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18

use rustc_hir::def::DefKind;
use rustc_query_system::query::SimpleDefKind;

/// Convert a [`DefKind`] to a [`SimpleDefKind`].
///
/// *See [`SimpleDefKind`]'s docs for more information.*
pub(crate) fn def_kind_to_simple_def_kind(def_kind: DefKind) -> SimpleDefKind {
    match def_kind {
        DefKind::Struct => SimpleDefKind::Struct,
        DefKind::Enum => SimpleDefKind::Enum,
        DefKind::Union => SimpleDefKind::Union,
        DefKind::Trait => SimpleDefKind::Trait,
        DefKind::TyAlias => SimpleDefKind::TyAlias,
        DefKind::TraitAlias => SimpleDefKind::TraitAlias,

        _ => SimpleDefKind::Other,
    }
}