summaryrefslogtreecommitdiff
path: root/dhall/src/syntax
diff options
context:
space:
mode:
Diffstat (limited to 'dhall/src/syntax')
-rw-r--r--dhall/src/syntax/text/printer.rs54
1 files changed, 33 insertions, 21 deletions
diff --git a/dhall/src/syntax/text/printer.rs b/dhall/src/syntax/text/printer.rs
index 9e90660..ccba385 100644
--- a/dhall/src/syntax/text/printer.rs
+++ b/dhall/src/syntax/text/printer.rs
@@ -146,6 +146,21 @@ where
f.write_str(close)
}
+fn fmt_label(label: &Label, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
+ // TODO: distinguish between reserved and nonreserved locations for quoting builtins
+ let s = String::from(label);
+ let is_reserved = match s.as_str() {
+ "let" | "in" | "if" | "then" | "else" | "Type" | "Kind" | "Sort"
+ | "True" | "False" | "Some" => true,
+ _ => Builtin::parse(&s).is_some(),
+ };
+ if !is_reserved && s.chars().all(|c| c.is_ascii_alphanumeric()) {
+ write!(f, "{}", s)
+ } else {
+ write!(f, "`{}`", s)
+ }
+}
+
/// Generic instance that delegates to subexpressions
impl<SE: Display + Clone> Display for ExprKind<SE> {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
@@ -153,16 +168,21 @@ impl<SE: Display + Clone> Display for ExprKind<SE> {
match self {
Var(a) => a.fmt(f)?,
Lam(a, b, c) => {
- write!(f, "λ({} : {}) → {}", a, b, c)?;
+ write!(f, "λ(")?;
+ fmt_label(a, f)?;
+ write!(f, " : {}) → {}", b, c)?;
}
Pi(a, b, c) if &String::from(a) == "_" => {
write!(f, "{} → {}", b, c)?;
}
Pi(a, b, c) => {
- write!(f, "∀({} : {}) → {}", a, b, c)?;
+ write!(f, "∀(")?;
+ fmt_label(a, f)?;
+ write!(f, " : {}) → {}", b, c)?;
}
Let(a, b, c, d) => {
- write!(f, "let {}", a)?;
+ write!(f, "let ")?;
+ fmt_label(a, f)?;
if let Some(b) = b {
write!(f, " : {}", b)?;
}
@@ -183,14 +203,16 @@ impl<SE: Display + Clone> Display for ExprKind<SE> {
}
RecordLit(a) if a.is_empty() => f.write_str("{=}")?,
RecordLit(a) => fmt_list("{ ", ", ", " }", a, f, |(k, v), f| {
- write!(f, "{} = {}", k, v)
+ fmt_label(k, f)?;
+ write!(f, " = {}", v)
})?,
RecordType(a) if a.is_empty() => f.write_str("{}")?,
RecordType(a) => fmt_list("{ ", ", ", " }", a, f, |(k, t), f| {
- write!(f, "{} : {}", k, t)
+ fmt_label(k, f)?;
+ write!(f, " : {}", t)
})?,
UnionType(a) => fmt_list("< ", " | ", " >", a, f, |(k, v), f| {
- write!(f, "{}", k)?;
+ fmt_label(k, f)?;
if let Some(v) = v {
write!(f, ": {}", v)?;
}
@@ -238,11 +260,12 @@ impl<SE: Display + Clone> Display for OpKind<SE> {
}
}
Field(a, b) => {
- write!(f, "{}.{}", a, b)?;
+ write!(f, "{}.", a)?;
+ fmt_label(b, f)?;
}
Projection(e, ls) => {
write!(f, "{}.", e)?;
- fmt_list("{ ", ", ", " }", ls, f, Display::fmt)?;
+ fmt_list("{ ", ", ", " }", ls, f, fmt_label)?;
}
ProjectionByExpr(a, b) => {
write!(f, "{}.({})", a, b)?;
@@ -379,18 +402,7 @@ impl Display for NaiveDouble {
impl Display for Label {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
- // TODO: distinguish between reserved and nonreserved locations for quoting builtins
- let s = String::from(self);
- let is_reserved = match s.as_str() {
- "let" | "in" | "if" | "then" | "else" | "Type" | "Kind"
- | "Sort" | "True" | "False" | "Some" => true,
- _ => Builtin::parse(&s).is_some(),
- };
- if !is_reserved && s.chars().all(|c| c.is_ascii_alphanumeric()) {
- write!(f, "{}", s)
- } else {
- write!(f, "`{}`", s)
- }
+ write!(f, "{}", String::from(self))
}
}
@@ -495,7 +507,7 @@ impl Display for Scheme {
impl Display for V {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
let V(x, n) = self;
- x.fmt(f)?;
+ fmt_label(x, f)?;
if *n != 0 {
write!(f, "@{}", n)?;
}