From c96e093c724e85e1f2979cb01043b83670b4690e Mon Sep 17 00:00:00 2001
From: Shadowfacts <me@shadowfacts.net>
Date: Wed, 12 Aug 2020 19:17:14 -0400
Subject: [PATCH] Fix JSON pretty printer string escaping

---
 MongoView/JSONPrettyPrinter.swift | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/MongoView/JSONPrettyPrinter.swift b/MongoView/JSONPrettyPrinter.swift
index f9b42dc..7ed2b18 100644
--- a/MongoView/JSONPrettyPrinter.swift
+++ b/MongoView/JSONPrettyPrinter.swift
@@ -89,7 +89,23 @@ class JSONPrettyPrinter {
     }
     
     private func escape(_ str: String) -> String {
-        return str.replacingOccurrences(of: "\"", with: "\\\"")
+        var str = str
+        var index = str.startIndex
+        while index < str.endIndex {
+            let c = str[index]
+            if c == "\\" || c == "\"" {
+                str.replaceSubrange(index..<str.index(after: index), with: "\\\(c)")
+                index = str.index(after: index)
+            } else if c == "\n" {
+                str.replaceSubrange(index..<str.index(after: index), with: "\\n")
+                index = str.index(after: index)
+            } else if c == "\r" {
+                str.replaceSubrange(index..<str.index(after: index), with: "\\r")
+                index = str.index(after: index)
+            }
+            index = str.index(after: index)
+        }
+        return str
     }
 }