From: Mark Thom <markjordanthom@gmail.com>
Date: Thu, 17 May 2018 02:34:53 +0000 (-0600)
Subject: print certain atoms in single quotes.
X-Git-Tag: v0.8.110~462
X-Git-Url: https://git.sagredo.dev/?a=commitdiff_plain;h=ddb79e6b3415d9243a1b37eb7356c563fc6e9532;p=scryer-prolog.git

print certain atoms in single quotes.
---

diff --git a/src/prolog/ast.rs b/src/prolog/ast.rs
index c5f24cb4..721ec05d 100644
--- a/src/prolog/ast.rs
+++ b/src/prolog/ast.rs
@@ -552,7 +552,11 @@ impl fmt::Display for Constant {
     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
         match self {
             &Constant::Atom(ref atom) =>
-                write!(f, "{}", atom),
+                if atom.as_str().chars().any(|c| ".$'\" ".contains(c)) {
+                    write!(f, "'{}'", atom.as_str())
+                } else {
+                    write!(f, "{}", atom.as_str())
+                },
             &Constant::Char(c) =>
                 write!(f, "'{}'", c as u8),
             &Constant::EmptyList =>