snapshotOnWindow.swift

func createSnapshot(from icon: Icon) -> UIImage {
  // Create view and add as subview on the window
  let window: UIWindow = UIApplication.shared.keyWindow ?? UIWindow()
  let iconView = IconView(icon: icon)
  window.addSubview(iconView)
  
  // Set a default size for all the snapshots
  iconView.frame = CGRect(origin: .zero, size: dataImageStorageSize)
  
  // Take a snapshot
  let bounds = iconView.bounds
  let renderer = UIGraphicsImageRenderer(bounds: bounds)
  let image = renderer.image { _ in
     iconView.drawHierarchy(in: bounds, afterScreenUpdates: true)
  }
  return image
}