fix: escape sequence

Ryu0118 · Sep 12, 2022 · 80c911a · 80c911a
1 parent 4e960a9
commit 80c911a
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 2 deletions.
diff --git a/Sources/MagicIBCore/Extensions/String+escapeSequence.swift b/Sources/MagicIBCore/Extensions/String+escapeSequence.swift
@@ -0,0 +1,22 @@
+//
+//  String+escapeSequence.swift
+//  
+//
+//  Created by Ryu on 2022/09/13.
+//
+
+import Foundation
+
+extension String {
+    func escapingSequence() -> String {
+        self.map {
+            if $0 == "\"" {
+                return "\\" + "\($0)"
+            }
+            else {
+                return "\($0)"
+            }
+        }
+        .joined()
+    }
+}
diff --git a/Sources/MagicIBCore/Generator/IBPropertyMapper+generator.swift b/Sources/MagicIBCore/Generator/IBPropertyMapper+generator.swift
@@ -23,7 +23,7 @@ extension IBPropertyMapper {
             return ".\(value)"
         case .string:
             guard let value = value as? String else { return nil }
-            return "\"\(value)\""
+            return "\"\(value.escapingSequence())\""
         default:
             guard let nonCustomizable = value as? NonCustomizable else { return nil }
             return nonCustomizable.getRightOperand()

diff --git a/Sources/MagicIBCore/IBFunctionMapper.swift b/Sources/MagicIBCore/IBFunctionMapper.swift
@@ -37,7 +37,7 @@ class IBFunctionMapper {
             return ".\(value)"
         case .string:
             guard let value = value as? String else { return nil }
-            return "\"\(value)\""
+            return "\"\(value.escapingSequence())\""
         default:
             guard let nonCustomizable = value as? NonCustomizable else { return nil }
             return nonCustomizable.getRightOperand()