summaryrefslogtreecommitdiff
path: root/dhall/src/syntax/ast/span.rs
blob: f9c7008f140c0d1caac8a730d328c2215f12b836 (plain)
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
use std::rc::Rc;

/// A location in the source text
#[derive(Debug, Clone)]
pub struct ParsedSpan {
    input: Rc<str>,
    /// # Safety
    ///
    /// Must be a valid character boundary index into `input`.
    start: usize,
    /// # Safety
    ///
    /// Must be a valid character boundary index into `input`.
    end: usize,
}

#[derive(Debug, Clone)]
pub enum Span {
    /// A location in the source text
    Parsed(ParsedSpan),
    /// For expressions obtained from decoding binary
    Decoded,
    /// For expressions constructed during normalization/typecheck
    Artificial,
}

impl Span {
    pub(crate) fn make(input: Rc<str>, sp: pest::Span) -> Self {
        Span::Parsed(ParsedSpan {
            input,
            start: sp.start(),
            end: sp.end(),
        })
    }

    /// Takes the union of the two spans, i.e. the range of input covered by the two spans plus any
    /// input between them. Assumes that the spans come from the same input. Fails if one of the
    /// spans does not point to an input location.
    pub fn union(&self, other: &Span) -> Self {
        use std::cmp::{max, min};
        use Span::*;
        match (self, other) {
            (Parsed(x), Parsed(y)) if Rc::ptr_eq(&x.input, &y.input) => {
                Parsed(ParsedSpan {
                    input: x.input.clone(),
                    start: min(x.start, y.start),
                    end: max(x.end, y.end),
                })
            }
            _ => panic!(
                "Tried to union incompatible spans: {:?} and {:?}",
                self, other
            ),
        }
    }

    /// Merges two spans assumed to point to a similar thing. If only one of them points to an
    /// input location, use that one.
    pub fn merge(&self, other: &Span) -> Self {
        use Span::*;
        match (self, other) {
            (Parsed(x), _) | (_, Parsed(x)) => Parsed(x.clone()),
            (Artificial, _) | (_, Artificial) => Artificial,
            (Decoded, Decoded) => Decoded,
        }
    }

    pub fn error(&self, message: impl Into<String>) -> String {
        use pest::error::{Error, ErrorVariant};
        use pest::Span;
        let message: String = message.into();
        let span = match self {
            self::Span::Parsed(span) => span,
            _ => return format!("[unknown location] {}", message),
        };
        let span = Span::new(&*span.input, span.start, span.end).unwrap();
        let err: ErrorVariant<!> = ErrorVariant::CustomError { message };
        let err = Error::new_from_span(err, span);
        format!("{}", err)
    }
}