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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

use crate::inherent::{AdtDef, IntoKind, Ty};
use crate::lang_items::TraitSolverLangItem::{EffectsMaybe, EffectsNoRuntime, EffectsRuntime};
use crate::Interner;

#[derive(Clone, Copy, PartialEq, Eq)]
pub enum EffectKind {
    Maybe,
    Runtime,
    NoRuntime,
}

impl EffectKind {
    pub fn try_from_def_id<I: Interner>(tcx: I, def_id: I::DefId) -> Option<EffectKind> {
        if tcx.is_lang_item(def_id, EffectsMaybe) {
            Some(EffectKind::Maybe)
        } else if tcx.is_lang_item(def_id, EffectsRuntime) {
            Some(EffectKind::Runtime)
        } else if tcx.is_lang_item(def_id, EffectsNoRuntime) {
            Some(EffectKind::NoRuntime)
        } else {
            None
        }
    }

    pub fn to_def_id<I: Interner>(self, tcx: I) -> I::DefId {
        let lang_item = match self {
            EffectKind::Maybe => EffectsMaybe,
            EffectKind::NoRuntime => EffectsNoRuntime,
            EffectKind::Runtime => EffectsRuntime,
        };

        tcx.require_lang_item(lang_item)
    }

    pub fn try_from_ty<I: Interner>(tcx: I, ty: I::Ty) -> Option<EffectKind> {
        if let crate::Adt(def, _) = ty.kind() {
            Self::try_from_def_id(tcx, def.def_id())
        } else {
            None
        }
    }

    pub fn to_ty<I: Interner>(self, tcx: I) -> I::Ty {
        I::Ty::new_adt(tcx, tcx.adt_def(self.to_def_id(tcx)), Default::default())
    }

    /// Returns an intersection between two effect kinds. If one effect kind
    /// is more permissive than the other (e.g. `Maybe` vs `Runtime`), this
    /// returns the less permissive effect kind (`Runtime`).
    pub fn intersection(a: Self, b: Self) -> Option<Self> {
        use EffectKind::*;
        match (a, b) {
            (Maybe, x) | (x, Maybe) => Some(x),
            (Runtime, Runtime) => Some(Runtime),
            (NoRuntime, NoRuntime) => Some(NoRuntime),
            (Runtime, NoRuntime) | (NoRuntime, Runtime) => None,
        }
    }
}