1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

//! Utilities for [`AdtDef`].

use rustc_hir::def_id::DefId;
use rustc_middle::ty::{AdtDef, FieldDef, TyCtxt};

/// Extension trait for [`AdtDef`].
pub trait AdtDefExt<'tcx> {
  type AllVisibleFieldsIter: Iterator<Item = &'tcx FieldDef>;

  /// Returns an iterator over all the fields of the ADT that are visible
  /// from `module`.
  fn all_visible_fields(
    self,
    module: DefId,
    tcx: TyCtxt<'tcx>,
  ) -> Self::AllVisibleFieldsIter;
}

impl<'tcx> AdtDefExt<'tcx> for AdtDef<'tcx> {
  type AllVisibleFieldsIter = impl Iterator<Item = &'tcx FieldDef>;
  fn all_visible_fields(
    self,
    module: DefId,
    tcx: TyCtxt<'tcx>,
  ) -> Self::AllVisibleFieldsIter {
    self
      .all_fields()
      .filter(move |field| field.vis.is_accessible_from(module, tcx))
  }
}