fix highlight background and add paddings to the trailing of @You

Session/Utilities/MentionUtilities.swift

@@ -110,6 +110,7 @@ public enum MentionUtilities {
                 result.addAttribute(.currentUserMentionBackgroundCornerRadius, value: (8 * sizeDiff), range: mention.range)
                 result.addAttribute(.currentUserMentionBackgroundPadding, value: (3 * sizeDiff), range: mention.range)
                 result.addAttribute(.currentUserMentionBackgroundColor, value: primaryColor.color, range: mention.range)
+                result.addAttribute(.kern, value: (3 * sizeDiff), range: NSRange(location: mention.range.upperBound, length: 1))
             }
             
             switch (location, mention.isCurrentUser, theme.interfaceStyle) {

SessionUIKit/Components/HighlightMentionBackgroundView.swift

@@ -85,6 +85,9 @@ public class HighlightMentionBackgroundView: UIView {
 
         var origins = [CGPoint](repeating: .zero, count: lines.count)
         CTFrameGetLineOrigins(frame, CFRangeMake(0, 0), &origins)
+
+        var currentMentionBounds: CGRect? = nil  // Store mention bounding box
+        var lastMentionBackgroundColor: UIColor = .clear
         
         for lineIndex in 0..<lines.count {
             let line = lines[lineIndex]
@@ -98,9 +101,18 @@ public class HighlightMentionBackgroundView: UIView {
                 let attributes: NSDictionary = CTRunGetAttributes(run)
                 
                 guard let mentionBackgroundColor: UIColor = attributes.value(forKey: NSAttributedString.Key.currentUserMentionBackgroundColor.rawValue) as? UIColor else {
+                    if let currentBounds = currentMentionBounds {
+                        // Draw a single background rectangle for the mention
+                        let path = UIBezierPath(roundedRect: currentBounds, cornerRadius: 6)
+                        lastMentionBackgroundColor.setFill()
+                        path.fill()
+                        currentMentionBounds = nil // Reset mention bounds
+                    }
+                    lastMentionBackgroundColor = .clear
                     continue
                 }
                 
+                lastMentionBackgroundColor = mentionBackgroundColor
                 let maybeCornerRadius: CGFloat? = (attributes
                     .value(forKey: NSAttributedString.Key.currentUserMentionBackgroundCornerRadius.rawValue) as? CGFloat)
                 let maybePadding: CGFloat? = (attributes
@@ -139,12 +151,23 @@ public class HighlightMentionBackgroundView: UIView {
                     runDescent -
                     extraYOffset
                 )
-                
-                let path = UIBezierPath(roundedRect: runBounds, cornerRadius: (maybeCornerRadius ?? 0))
-                mentionBackgroundColor.setFill()
-                path.fill()
+
+                if currentMentionBounds == nil {
+                    // Start tracking mention bounds
+                    currentMentionBounds = runBounds
+                } else {
+                    // Expand bounds to include this part of the mention
+                    currentMentionBounds = currentMentionBounds!.union(runBounds)
+                }
             }
         }
+
+        // Final mention draw (in case the loop ends with a mention active)
+        if let currentBounds = currentMentionBounds {
+            let path = UIBezierPath(roundedRect: currentBounds, cornerRadius: 6)
+            lastMentionBackgroundColor.setFill()
+            path.fill()
+        }
     }
 }