Uses color of button for tool picker background

Author: simonbs

Sources/KeyboardToolbar/Helpers/UIImage+Helpers.swift

@@ -0,0 +1,18 @@
+import UIKit
+
+extension UIImage {
+    var firstPixelColor: UIColor? {
+        guard let inputImage = CIImage(image: self) else {
+            return nil
+        }
+        var bitmap = [UInt8](repeating: 0, count: 4)
+        let context = CIContext(options: [.workingColorSpace: kCFNull as Any])
+        let bounds = CGRect(x: 0, y: 0, width: size.width, height: size.height)
+        context.render(inputImage, toBitmap: &bitmap, rowBytes: 4, bounds: bounds, format: .RGBA8, colorSpace: nil)
+        let red = CGFloat(bitmap[0]) / 255
+        let green = CGFloat(bitmap[1]) / 255
+        let blue = CGFloat(bitmap[2]) / 255
+        let alpha = CGFloat(bitmap[3]) / 255
+        return UIColor(red: red, green: green, blue: blue, alpha: alpha)
+    }
+}

Sources/KeyboardToolbar/Helpers/UIView+Helpers.swift

@@ -5,25 +5,15 @@ extension UIView {
         guard let window = window else {
             return nil
         }
-        let rect = window.convert(frame, from: superview)
-        let image = UIGraphicsImageRenderer(bounds: rect).image { _ in
-            window.drawHierarchy(in: window.bounds, afterScreenUpdates: true)
+        let buttonFrame = window.convert(frame, from: superview)
+        let bounds = CGRect(x: buttonFrame.minX + point.x, y: buttonFrame.minY, width: 1, height: 1)
+        let imageRenderer = UIGraphicsImageRenderer(bounds: bounds)
+        let format = UIGraphicsImageRendererFormat()
+        format.scale = 1
+        format.opaque = true
+        let image = imageRenderer.image { _ in
+            window.drawHierarchy(in: window.bounds, afterScreenUpdates: false)
         }
-        return image.color(at: point)
-    }
-}
-
-private extension UIImage {
-    func color(at point: CGPoint) -> UIColor? {
-        guard let pixelData = cgImage?.dataProvider?.data else {
-            return nil
-        }
-        let data: UnsafePointer<UInt8> = CFDataGetBytePtr(pixelData)
-        let pixelInfo = ((Int(size.width) * Int(point.y)) + Int(point.x)) * 4
-        let r = CGFloat(data[pixelInfo]) / CGFloat(255)
-        let g = CGFloat(data[pixelInfo + 1]) / CGFloat(255)
-        let b = CGFloat(data[pixelInfo + 2]) / CGFloat(255)
-        let a = CGFloat(data[pixelInfo + 3]) / CGFloat(255)
-        return UIColor(red: r, green: g, blue: b, alpha: a)
+        return image.firstPixelColor
     }
 }

Sources/KeyboardToolbar/Views/KeyboardToolButton.swift

@@ -38,6 +38,12 @@ final class KeyboardToolButton: UIButton {
             }
         }
     }
+    private var toolPickerBackgroundColor: UIColor {
+        // Get the background color from a screenshot to ensure we get the right color for all keyboards.
+        // This is in particular relevant when using dark keyboards where both the keyboard and the buttons are transulcent.
+        let colorSamplePoint = CGPoint(x: frame.size.width / 2, y: 0)
+        return color(at: colorSamplePoint) ?? .black
+    }
 
     init(item: KeyboardToolGroupItem) {
         self.item = item
@@ -118,10 +124,7 @@ private extension KeyboardToolButton {
     }
 
     @objc private func touchDown(_ sender: UIButton, event: UIEvent) {
-        // Get the background color from a screenshot to ensure we get the right color for all keyboards.
-        // This is in particular relevant when using dark keyboards where both the keyboard and the buttons are transulcent.
-        let colorSamplePoint = CGPoint(x: bounds.midX, y: 2)
-        toolPickerBackgroundView.fillColor = color(at: colorSamplePoint) ?? .black
+        toolPickerBackgroundView.fillColor = toolPickerBackgroundColor
         UIDevice.current.playInputClick()
         cancelToolPickerTimer()
         guard !item.allTools.isEmpty else {