Display spaces as dimmed middle dots

It also works as a workaround for linebreakMode not being supported.

Author: intitni

Core/Sources/SuggestionWidget/SuggestionPanelView.swift

@@ -107,8 +107,11 @@ struct CodeBlock: View {
             ForEach(0..<viewModel.suggestion.count, id: \.self) { index in
                 Text("\(index + viewModel.startLineIndex + 1)")
                     .foregroundColor(Color.white.opacity(0.6))
-                BreakByCharText(viewModel.suggestion[index])
+                Text(AttributedString(viewModel.suggestion[index]))
+                    .foregroundColor(.white.opacity(0.1))
                     .frame(maxWidth: .infinity, alignment: .leading)
+                    .multilineTextAlignment(.leading)
+                    .lineSpacing(4)
             }
         }
         .foregroundColor(.white)
@@ -232,38 +235,3 @@ struct SuggestionPanelView_Preview: PreviewProvider {
         }
     }
 }
-
-// Super dirty hack to workaround that AttributedText doesn't support paragraphStyle.lineBreakMode
-struct BreakByCharText: View {
-    let attributedString: NSAttributedString
-    let attributedChars: [NSAttributedString]
-
-    init(_ attributedString: NSAttributedString) {
-        self.attributedString = attributedString
-        attributedChars = splitAttributedString(attributedString)
-    }
-
-    var body: some View {
-        LazyVGrid(
-            columns: [.init(.adaptive(minimum: 8), spacing: 0, alignment: .leading)],
-            alignment: .leading,
-            spacing: 4
-        ) {
-            ForEach(0..<attributedChars.endIndex, id: \.self) { index in
-                Text(AttributedString(attributedChars[index]))
-            }
-        }
-    }
-}
-
-private func splitAttributedString(_ inputString: NSAttributedString) -> [NSAttributedString] {
-    var output = [NSAttributedString]()
-    var start = 0
-    for char in inputString.string {
-        let range = NSMakeRange(start, char.utf16.count)
-        let attributedString = inputString.attributedSubstring(from: range)
-        output.append(attributedString)
-        start += range.length
-    }
-    return output
-}

Core/Sources/SuggestionWidget/SuggestionWidgetController.swift

@@ -362,29 +362,48 @@ func highlighted(code: String, language: String) -> [NSAttributedString] {
             [.font: NSFont.monospacedSystemFont(ofSize: 13, weight: .regular)],
             range: NSRange(location: 0, length: formatted.length)
         )
-        return splitAttributedString(formatted)
+        return convertToCodeLines(formatted)
     default:
         guard let highlighter = Highlightr() else {
-            return splitAttributedString(NSAttributedString(string: code))
+            return convertToCodeLines(NSAttributedString(string: code))
         }
         highlighter.setTheme(to: "atom-one-dark")
         highlighter.theme.setCodeFont(.monospacedSystemFont(ofSize: 13, weight: .regular))
         guard let formatted = highlighter.highlight(code, as: "swift") else {
-            return splitAttributedString(NSAttributedString(string: code))
+            return convertToCodeLines(NSAttributedString(string: code))
         }
-        return splitAttributedString(formatted)
+        return convertToCodeLines(formatted)
     }
 }
 
-private func splitAttributedString(_ inputString: NSAttributedString) -> [NSAttributedString] {
-    let input = inputString.string
+private func convertToCodeLines(_ formatedCode: NSAttributedString) -> [NSAttributedString] {
+    let input = formatedCode.string
     let separatedInput = input.components(separatedBy: "\n")
     var output = [NSAttributedString]()
     var start = 0
     for sub in separatedInput {
         let range = NSMakeRange(start, sub.utf16.count)
-        let attributedString = inputString.attributedSubstring(from: range)
-        output.append(attributedString)
+        let attributedString = formatedCode.attributedSubstring(from: range)
+        let mutable = NSMutableAttributedString(attributedString: attributedString)
+        // use regex to replace all spaces to a middle dot
+        do {
+            let regex = try NSRegularExpression(pattern: "[ ]*", options: [])
+            let result = regex.matches(
+                in: mutable.string,
+                range: NSRange(location: 0, length: mutable.mutableString.length)
+            )
+            for r in result {
+                let range = r.range
+                mutable.replaceCharacters(
+                    in: range,
+                    with: String(repeating: "·", count: range.length)
+                )
+                mutable.addAttributes([
+                    .foregroundColor: NSColor.white.withAlphaComponent(0.1),
+                ], range: range)
+            }
+        } catch {}
+        output.append(mutable)
         start += range.length + 1
     }
     return output