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
60
61
62
63
64
65

use either::Either;
use prisma_value::PrismaValue;

use crate::{DefaultKind, DefaultValue, ViewColumnId, ViewColumnWalker, ViewDefaultValueId, Walker};

use super::DefaultValueWalker;

/// Traverse a view column's default value.
pub type ViewDefaultValueWalker<'a> = Walker<'a, ViewDefaultValueId>;

impl<'a> ViewDefaultValueWalker<'a> {
    /// Coarsen the walker into a generic column default version.
    pub fn coarsen(self) -> DefaultValueWalker<'a> {
        self.walk(Either::Right(self.id))
    }

    /// The column where the default value is located.
    pub fn column(self) -> ViewColumnWalker<'a> {
        self.walk(self.get().0)
    }

    /// Return a value if a constant.
    pub fn as_value(self) -> Option<&'a PrismaValue> {
        self.coarsen().as_value()
    }

    /// If the value is a squence, return it
    pub fn as_sequence(self) -> Option<&'a str> {
        self.coarsen().as_sequence()
    }

    /// True if a constant value
    pub fn is_value(&self) -> bool {
        self.coarsen().is_value()
    }

    /// True if `now()`
    pub fn is_now(&self) -> bool {
        self.coarsen().is_now()
    }

    /// True if referencing a sequence
    pub fn is_sequence(&self) -> bool {
        self.coarsen().is_sequence()
    }

    /// True if value generation is handled in the database
    pub fn is_db_generated(&self) -> bool {
        self.coarsen().is_db_generated()
    }

    /// The value kind enumerator
    pub fn kind(self) -> &'a DefaultKind {
        self.coarsen().kind()
    }

    /// The name of the default value constraint.
    pub fn constraint_name(self) -> Option<&'a str> {
        self.coarsen().constraint_name()
    }

    fn get(self) -> &'a (ViewColumnId, DefaultValue) {
        &self.schema.view_default_values[self.id.0 as usize]
    }
}