import Foundation import UIKit import AsyncDisplayKit import SwiftSignalKit import UIKitRuntimeUtils private let separatorHeight: CGFloat = 1.0 / UIScreen.main.scale private func tabBarItemImage(_ image: UIImage?, title: String, backgroundColor: UIColor, tintColor: UIColor, horizontal: Bool, imageMode: Bool, centered: Bool = false) -> (UIImage, CGFloat) { let font = horizontal ? Font.regular(13.0) : Font.medium(10.0) let titleSize = (title as NSString).boundingRect(with: CGSize(width: CGFloat.greatestFiniteMagnitude, height: CGFloat.greatestFiniteMagnitude), options: [.usesLineFragmentOrigin], attributes: [NSAttributedString.Key.font: font], context: nil).size let imageSize: CGSize if let image = image { if horizontal { let factor: CGFloat = 0.8 imageSize = CGSize(width: floor(image.size.width * factor), height: floor(image.size.height * factor)) } else { imageSize = image.size } } else { imageSize = CGSize() } let horizontalSpacing: CGFloat = 4.0 let size: CGSize let contentWidth: CGFloat if horizontal { let width = max(1.0, centered ? imageSize.width : ceil(titleSize.width) + horizontalSpacing + imageSize.width) size = CGSize(width: width, height: 34.0) contentWidth = size.width } else { let width = max(1.0, centered ? imageSize.width : max(ceil(titleSize.width), imageSize.width), 1.0) size = CGSize(width: width, height: 45.0) contentWidth = imageSize.width } UIGraphicsBeginImageContextWithOptions(size, false, 0.0) if let context = UIGraphicsGetCurrentContext() { context.setFillColor(backgroundColor.cgColor) context.fill(CGRect(origin: CGPoint(), size: size)) if let image = image, imageMode { if horizontal { let imageRect = CGRect(origin: CGPoint(x: 0.0, y: floor((size.height - imageSize.height) / 2.0)), size: imageSize) context.saveGState() context.translateBy(x: imageRect.midX, y: imageRect.midY) context.scaleBy(x: 1.0, y: -1.0) context.translateBy(x: -imageRect.midX, y: -imageRect.midY) if image.renderingMode == .alwaysOriginal { context.draw(image.cgImage!, in: imageRect) } else { context.clip(to: imageRect, mask: image.cgImage!) context.setFillColor(tintColor.cgColor) context.fill(imageRect) } context.restoreGState() } else { let imageRect = CGRect(origin: CGPoint(x: floorToScreenPixels((size.width - imageSize.width) / 2.0), y: centered ? floor((size.height - imageSize.height) / 2.0) : 0.0), size: imageSize) context.saveGState() context.translateBy(x: imageRect.midX, y: imageRect.midY) context.scaleBy(x: 1.0, y: -1.0) context.translateBy(x: -imageRect.midX, y: -imageRect.midY) if image.renderingMode == .alwaysOriginal { context.draw(image.cgImage!, in: imageRect) } else { context.clip(to: imageRect, mask: image.cgImage!) context.setFillColor(tintColor.cgColor) context.fill(imageRect) } context.restoreGState() } } } if !imageMode { if horizontal { (title as NSString).draw(at: CGPoint(x: imageSize.width + horizontalSpacing, y: floor((size.height - titleSize.height) / 2.0)), withAttributes: [NSAttributedString.Key.font: font, NSAttributedString.Key.foregroundColor: tintColor]) } else { (title as NSString).draw(at: CGPoint(x: floorToScreenPixels((size.width - titleSize.width) / 2.0), y: size.height - titleSize.height - 1.0), withAttributes: [NSAttributedString.Key.font: font, NSAttributedString.Key.foregroundColor: tintColor]) } } let resultImage = UIGraphicsGetImageFromCurrentImageContext() UIGraphicsEndImageContext() return (resultImage!, contentWidth) } private let badgeFont = Font.regular(13.0) public enum TabBarItemSwipeDirection { case left case right } private final class TabBarItemNode: ASDisplayNode { let extractedContainerNode: ContextExtractedContentContainingNode let containerNode: ContextControllerSourceNode let imageNode: ASImageNode let textImageNode: ASImageNode let contextImageNode: ASImageNode let contextTextImageNode: ASImageNode var contentWidth: CGFloat? var swiped: ((TabBarItemSwipeDirection) -> Void)? override init() { self.extractedContainerNode = ContextExtractedContentContainingNode() self.containerNode = ContextControllerSourceNode() self.imageNode = ASImageNode() self.imageNode.isUserInteractionEnabled = false self.imageNode.displayWithoutProcessing = true self.imageNode.displaysAsynchronously = false self.imageNode.isAccessibilityElement = false self.textImageNode = ASImageNode() self.textImageNode.isUserInteractionEnabled = false self.textImageNode.displayWithoutProcessing = true self.textImageNode.displaysAsynchronously = false self.textImageNode.isAccessibilityElement = false self.contextImageNode = ASImageNode() self.contextImageNode.isUserInteractionEnabled = false self.contextImageNode.displayWithoutProcessing = true self.contextImageNode.displaysAsynchronously = false self.contextImageNode.isAccessibilityElement = false self.contextImageNode.alpha = 0.0 self.contextTextImageNode = ASImageNode() self.contextTextImageNode.isUserInteractionEnabled = false self.contextTextImageNode.displayWithoutProcessing = true self.contextTextImageNode.displaysAsynchronously = false self.contextTextImageNode.isAccessibilityElement = false self.contextTextImageNode.alpha = 0.0 super.init() self.isAccessibilityElement = true self.extractedContainerNode.contentNode.addSubnode(self.textImageNode) self.extractedContainerNode.contentNode.addSubnode(self.imageNode) self.extractedContainerNode.contentNode.addSubnode(self.contextTextImageNode) self.extractedContainerNode.contentNode.addSubnode(self.contextImageNode) self.containerNode.addSubnode(self.extractedContainerNode) self.containerNode.targetNodeForActivationProgress = self.extractedContainerNode.contentNode self.addSubnode(self.containerNode) self.extractedContainerNode.willUpdateIsExtractedToContextPreview = { [weak self] isExtracted, transition in guard let strongSelf = self else { return } transition.updateAlpha(node: strongSelf.imageNode, alpha: isExtracted ? 0.0 : 1.0) transition.updateAlpha(node: strongSelf.textImageNode, alpha: isExtracted ? 0.0 : 1.0) transition.updateAlpha(node: strongSelf.contextImageNode, alpha: isExtracted ? 1.0 : 0.0) transition.updateAlpha(node: strongSelf.contextTextImageNode, alpha: isExtracted ? 1.0 : 0.0) } let leftSwipe = UISwipeGestureRecognizer(target: self, action: #selector(self.swipeGesture(_:))) leftSwipe.direction = .left self.containerNode.view.addGestureRecognizer(leftSwipe) let rightSwipe = UISwipeGestureRecognizer(target: self, action: #selector(self.swipeGesture(_:))) rightSwipe.direction = .right self.containerNode.view.addGestureRecognizer(rightSwipe) } @objc private func swipeGesture(_ gesture: UISwipeGestureRecognizer) { if case .ended = gesture.state { self.containerNode.cancelGesture() switch gesture.direction { case .left: self.swiped?(.left) default: self.swiped?(.right) } } } } private final class TabBarNodeContainer { let item: UITabBarItem let updateBadgeListenerIndex: Int let updateTitleListenerIndex: Int let updateImageListenerIndex: Int let updateSelectedImageListenerIndex: Int let imageNode: TabBarItemNode let badgeContainerNode: ASDisplayNode let badgeBackgroundNode: ASImageNode let badgeTextNode: ImmediateTextNode var badgeValue: String? var appliedBadgeValue: String? var titleValue: String? var appliedTitleValue: String? var imageValue: UIImage? var appliedImageValue: UIImage? var selectedImageValue: UIImage? var appliedSelectedImageValue: UIImage? init(item: TabBarNodeItem, imageNode: TabBarItemNode, updateBadge: @escaping (String) -> Void, updateTitle: @escaping (String, Bool) -> Void, updateImage: @escaping (UIImage?) -> Void, updateSelectedImage: @escaping (UIImage?) -> Void, contextAction: @escaping (ContextExtractedContentContainingNode, ContextGesture) -> Void, swipeAction: @escaping (TabBarItemSwipeDirection) -> Void) { self.item = item.item self.imageNode = imageNode self.imageNode.isAccessibilityElement = true self.imageNode.accessibilityTraits = .button self.badgeContainerNode = ASDisplayNode() self.badgeContainerNode.isUserInteractionEnabled = false self.badgeContainerNode.isAccessibilityElement = false self.badgeBackgroundNode = ASImageNode() self.badgeBackgroundNode.isUserInteractionEnabled = false self.badgeBackgroundNode.displayWithoutProcessing = true self.badgeBackgroundNode.displaysAsynchronously = false self.badgeBackgroundNode.isAccessibilityElement = false self.badgeTextNode = ImmediateTextNode() self.badgeTextNode.maximumNumberOfLines = 1 self.badgeTextNode.isUserInteractionEnabled = false self.badgeTextNode.displaysAsynchronously = false self.badgeTextNode.isAccessibilityElement = false self.badgeContainerNode.addSubnode(self.badgeBackgroundNode) self.badgeContainerNode.addSubnode(self.badgeTextNode) self.badgeValue = item.item.badgeValue ?? "" self.updateBadgeListenerIndex = UITabBarItem_addSetBadgeListener(item.item, { value in updateBadge(value ?? "") }) self.titleValue = item.item.title self.updateTitleListenerIndex = item.item.addSetTitleListener { value, animated in updateTitle(value ?? "", animated) } self.imageValue = item.item.image self.updateImageListenerIndex = item.item.addSetImageListener { value in updateImage(value) } self.selectedImageValue = item.item.selectedImage self.updateSelectedImageListenerIndex = item.item.addSetSelectedImageListener { value in updateSelectedImage(value) } imageNode.containerNode.activated = { [weak self] gesture in guard let strongSelf = self else { return } contextAction(strongSelf.imageNode.extractedContainerNode, gesture) } imageNode.swiped = { [weak self] direction in swipeAction(direction) } imageNode.containerNode.isGestureEnabled = item.hasContext } deinit { self.item.removeSetBadgeListener(self.updateBadgeListenerIndex) self.item.removeSetTitleListener(self.updateTitleListenerIndex) self.item.removeSetImageListener(self.updateImageListenerIndex) self.item.removeSetSelectedImageListener(self.updateSelectedImageListenerIndex) } } final class TabBarNodeItem { let item: UITabBarItem let hasContext: Bool init(item: UITabBarItem, hasContext: Bool) { self.item = item self.hasContext = hasContext } } class TabBarNode: ASDisplayNode { var tabBarItems: [TabBarNodeItem] = [] { didSet { self.reloadTabBarItems() } } var selectedIndex: Int? { didSet { if self.selectedIndex != oldValue { if let oldValue = oldValue { self.updateNodeImage(oldValue, layout: true) } if let selectedIndex = self.selectedIndex { self.updateNodeImage(selectedIndex, layout: true) } } } } private let itemSelected: (Int, Bool, [ASDisplayNode]) -> Void private let contextAction: (Int, ContextExtractedContentContainingNode, ContextGesture) -> Void private let swipeAction: (Int, TabBarItemSwipeDirection) -> Void private var theme: TabBarControllerTheme private var validLayout: (CGSize, CGFloat, CGFloat, CGFloat)? private var horizontal: Bool = false private var centered: Bool = false private var badgeImage: UIImage let separatorNode: ASDisplayNode private var tabBarNodeContainers: [TabBarNodeContainer] = [] private var tapRecognizer: TapLongTapOrDoubleTapGestureRecognizer? init(theme: TabBarControllerTheme, itemSelected: @escaping (Int, Bool, [ASDisplayNode]) -> Void, contextAction: @escaping (Int, ContextExtractedContentContainingNode, ContextGesture) -> Void, swipeAction: @escaping (Int, TabBarItemSwipeDirection) -> Void) { self.itemSelected = itemSelected self.contextAction = contextAction self.swipeAction = swipeAction self.theme = theme self.separatorNode = ASDisplayNode() self.separatorNode.backgroundColor = theme.tabBarSeparatorColor self.separatorNode.isOpaque = true self.separatorNode.isLayerBacked = true self.badgeImage = generateStretchableFilledCircleImage(diameter: 18.0, color: theme.tabBarBadgeBackgroundColor, strokeColor: theme.tabBarBadgeStrokeColor, strokeWidth: 1.0, backgroundColor: nil)! super.init() self.isAccessibilityContainer = false self.isOpaque = true self.backgroundColor = theme.tabBarBackgroundColor self.addSubnode(self.separatorNode) } override func didLoad() { super.didLoad() let recognizer = TapLongTapOrDoubleTapGestureRecognizer(target: self, action: #selector(self.tapLongTapOrDoubleTapGesture(_:))) recognizer.tapActionAtPoint = { _ in return .keepWithSingleTap } self.tapRecognizer = recognizer self.view.addGestureRecognizer(recognizer) } @objc private func tapLongTapOrDoubleTapGesture(_ recognizer: TapLongTapOrDoubleTapGestureRecognizer) { switch recognizer.state { case .ended: if let (gesture, location) = recognizer.lastRecognizedGestureAndLocation { if case .tap = gesture { self.tapped(at: location, longTap: false) } } default: break } } func updateTheme(_ theme: TabBarControllerTheme) { if self.theme !== theme { self.theme = theme self.separatorNode.backgroundColor = theme.tabBarSeparatorColor self.backgroundColor = theme.tabBarBackgroundColor self.badgeImage = generateStretchableFilledCircleImage(diameter: 18.0, color: theme.tabBarBadgeBackgroundColor, strokeColor: theme.tabBarBadgeStrokeColor, strokeWidth: 1.0, backgroundColor: nil)! for container in self.tabBarNodeContainers { if let attributedText = container.badgeTextNode.attributedText, !attributedText.string.isEmpty { container.badgeTextNode.attributedText = NSAttributedString(string: attributedText.string, font: badgeFont, textColor: self.theme.tabBarBadgeTextColor) } } for i in 0 ..< self.tabBarItems.count { self.updateNodeImage(i, layout: false) self.tabBarNodeContainers[i].badgeBackgroundNode.image = self.badgeImage } if let validLayout = self.validLayout { self.updateLayout(size: validLayout.0, leftInset: validLayout.1, rightInset: validLayout.2, bottomInset: validLayout.3, transition: .immediate) } } } func sourceNodesForController(at index: Int) -> [ASDisplayNode]? { let container = self.tabBarNodeContainers[index] return [container.imageNode.imageNode, container.imageNode.textImageNode, container.badgeContainerNode] } private func reloadTabBarItems() { for node in self.tabBarNodeContainers { node.imageNode.removeFromSupernode() } self.centered = self.theme.tabBarTextColor == .clear var tabBarNodeContainers: [TabBarNodeContainer] = [] for i in 0 ..< self.tabBarItems.count { let item = self.tabBarItems[i] let node = TabBarItemNode() let container = TabBarNodeContainer(item: item, imageNode: node, updateBadge: { [weak self] value in self?.updateNodeBadge(i, value: value) }, updateTitle: { [weak self] _, _ in self?.updateNodeImage(i, layout: true) }, updateImage: { [weak self] _ in self?.updateNodeImage(i, layout: true) }, updateSelectedImage: { [weak self] _ in self?.updateNodeImage(i, layout: true) }, contextAction: { [weak self] node, gesture in self?.tapRecognizer?.cancel() self?.contextAction(i, node, gesture) }, swipeAction: { [weak self] direction in self?.swipeAction(i, direction) }) if let selectedIndex = self.selectedIndex, selectedIndex == i { let (textImage, contentWidth) = tabBarItemImage(item.item.selectedImage, title: item.item.title ?? "", backgroundColor: .clear, tintColor: self.theme.tabBarSelectedTextColor, horizontal: self.horizontal, imageMode: false, centered: self.centered) let (image, imageContentWidth) = tabBarItemImage(item.item.selectedImage, title: item.item.title ?? "", backgroundColor: .clear, tintColor: self.theme.tabBarSelectedTextColor, horizontal: self.horizontal, imageMode: true, centered: self.centered) node.textImageNode.image = textImage node.imageNode.image = image node.accessibilityLabel = item.item.title node.contentWidth = max(contentWidth, imageContentWidth) } else { let (textImage, contentWidth) = tabBarItemImage(item.item.image, title: item.item.title ?? "", backgroundColor: .clear, tintColor: self.theme.tabBarTextColor, horizontal: self.horizontal, imageMode: false, centered: self.centered) let (image, imageContentWidth) = tabBarItemImage(item.item.image, title: item.item.title ?? "", backgroundColor: .clear, tintColor: self.theme.tabBarIconColor, horizontal: self.horizontal, imageMode: true, centered: self.centered) node.textImageNode.image = textImage node.accessibilityLabel = item.item.title node.imageNode.image = image node.contentWidth = max(contentWidth, imageContentWidth) } container.badgeBackgroundNode.image = self.badgeImage node.extractedContainerNode.contentNode.addSubnode(container.badgeContainerNode) tabBarNodeContainers.append(container) self.addSubnode(node) } self.tabBarNodeContainers = tabBarNodeContainers self.setNeedsLayout() } private func updateNodeImage(_ index: Int, layout: Bool) { if index < self.tabBarNodeContainers.count && index < self.tabBarItems.count { let node = self.tabBarNodeContainers[index].imageNode let item = self.tabBarItems[index] self.centered = self.theme.tabBarTextColor == .clear let previousImageSize = node.imageNode.image?.size ?? CGSize() let previousTextImageSize = node.textImageNode.image?.size ?? CGSize() if let selectedIndex = self.selectedIndex, selectedIndex == index { let (textImage, contentWidth) = tabBarItemImage(item.item.selectedImage, title: item.item.title ?? "", backgroundColor: .clear, tintColor: self.theme.tabBarSelectedTextColor, horizontal: self.horizontal, imageMode: false, centered: self.centered) let (image, imageContentWidth) = tabBarItemImage(item.item.selectedImage, title: item.item.title ?? "", backgroundColor: .clear, tintColor: self.theme.tabBarSelectedIconColor, horizontal: self.horizontal, imageMode: true, centered: self.centered) let (contextTextImage, _) = tabBarItemImage(item.item.image, title: item.item.title ?? "", backgroundColor: .clear, tintColor: self.theme.tabBarExtractedTextColor, horizontal: self.horizontal, imageMode: false, centered: self.centered) let (contextImage, _) = tabBarItemImage(item.item.image, title: item.item.title ?? "", backgroundColor: .clear, tintColor: self.theme.tabBarExtractedIconColor, horizontal: self.horizontal, imageMode: true, centered: self.centered) node.textImageNode.image = textImage node.accessibilityLabel = item.item.title node.imageNode.image = image node.contextTextImageNode.image = contextTextImage node.contextImageNode.image = contextImage node.contentWidth = max(contentWidth, imageContentWidth) } else { let (textImage, contentWidth) = tabBarItemImage(item.item.image, title: item.item.title ?? "", backgroundColor: .clear, tintColor: self.theme.tabBarTextColor, horizontal: self.horizontal, imageMode: false, centered: self.centered) let (image, imageContentWidth) = tabBarItemImage(item.item.image, title: item.item.title ?? "", backgroundColor: .clear, tintColor: self.theme.tabBarIconColor, horizontal: self.horizontal, imageMode: true, centered: self.centered) let (contextTextImage, _) = tabBarItemImage(item.item.image, title: item.item.title ?? "", backgroundColor: .clear, tintColor: self.theme.tabBarExtractedTextColor, horizontal: self.horizontal, imageMode: false, centered: self.centered) let (contextImage, _) = tabBarItemImage(item.item.image, title: item.item.title ?? "", backgroundColor: .clear, tintColor: self.theme.tabBarExtractedIconColor, horizontal: self.horizontal, imageMode: true, centered: self.centered) node.textImageNode.image = textImage node.accessibilityLabel = item.item.title node.imageNode.image = image node.contextTextImageNode.image = contextTextImage node.contextImageNode.image = contextImage node.contentWidth = max(contentWidth, imageContentWidth) } let updatedImageSize = node.imageNode.image?.size ?? CGSize() let updatedTextImageSize = node.textImageNode.image?.size ?? CGSize() if previousImageSize != updatedImageSize || previousTextImageSize != updatedTextImageSize { if let validLayout = self.validLayout, layout { self.updateLayout(size: validLayout.0, leftInset: validLayout.1, rightInset: validLayout.2, bottomInset: validLayout.3, transition: .immediate) } } } } private func updateNodeBadge(_ index: Int, value: String) { self.tabBarNodeContainers[index].badgeValue = value if self.tabBarNodeContainers[index].badgeValue != self.tabBarNodeContainers[index].appliedBadgeValue { if let validLayout = self.validLayout { self.updateLayout(size: validLayout.0, leftInset: validLayout.1, rightInset: validLayout.2, bottomInset: validLayout.3, transition: .immediate) } } } private func updateNodeTitle(_ index: Int, value: String) { self.tabBarNodeContainers[index].titleValue = value if self.tabBarNodeContainers[index].titleValue != self.tabBarNodeContainers[index].appliedTitleValue { if let validLayout = self.validLayout { self.updateLayout(size: validLayout.0, leftInset: validLayout.1, rightInset: validLayout.2, bottomInset: validLayout.3, transition: .immediate) } } } func updateLayout(size: CGSize, leftInset: CGFloat, rightInset: CGFloat, bottomInset: CGFloat, transition: ContainedViewLayoutTransition) { self.validLayout = (size, leftInset, rightInset, bottomInset) transition.updateFrame(node: self.separatorNode, frame: CGRect(origin: CGPoint(x: 0.0, y: -separatorHeight), size: CGSize(width: size.width, height: separatorHeight))) let horizontal = !leftInset.isZero if self.horizontal != horizontal { self.horizontal = horizontal for i in 0 ..< self.tabBarItems.count { self.updateNodeImage(i, layout: false) } } if self.tabBarNodeContainers.count != 0 { let distanceBetweenNodes = size.width / CGFloat(self.tabBarNodeContainers.count) let internalWidth = distanceBetweenNodes * CGFloat(self.tabBarNodeContainers.count - 1) let leftNodeOriginX = (size.width - internalWidth) / 2.0 for i in 0 ..< self.tabBarNodeContainers.count { let container = self.tabBarNodeContainers[i] let node = container.imageNode let nodeSize = node.textImageNode.image?.size ?? CGSize() let originX = floor(leftNodeOriginX + CGFloat(i) * distanceBetweenNodes - nodeSize.width / 2.0) let horizontalHitTestInset = distanceBetweenNodes / 2.0 - nodeSize.width / 2.0 let nodeFrame = CGRect(origin: CGPoint(x: originX, y: 3.0), size: nodeSize) transition.updateFrame(node: node, frame: nodeFrame) node.extractedContainerNode.frame = CGRect(origin: CGPoint(), size: nodeFrame.size) node.extractedContainerNode.contentNode.frame = node.extractedContainerNode.bounds node.extractedContainerNode.contentRect = node.extractedContainerNode.bounds node.containerNode.frame = CGRect(origin: CGPoint(), size: nodeFrame.size) node.hitTestSlop = UIEdgeInsets(top: -3.0, left: -horizontalHitTestInset, bottom: -3.0, right: -horizontalHitTestInset) node.containerNode.hitTestSlop = UIEdgeInsets(top: -3.0, left: -horizontalHitTestInset, bottom: -3.0, right: -horizontalHitTestInset) node.imageNode.frame = CGRect(origin: CGPoint(), size: nodeFrame.size) node.textImageNode.frame = CGRect(origin: CGPoint(), size: nodeFrame.size) node.contextImageNode.frame = CGRect(origin: CGPoint(), size: nodeFrame.size) node.contextTextImageNode.frame = CGRect(origin: CGPoint(), size: nodeFrame.size) if container.badgeValue != container.appliedBadgeValue { container.appliedBadgeValue = container.badgeValue if let badgeValue = container.badgeValue, !badgeValue.isEmpty { container.badgeTextNode.attributedText = NSAttributedString(string: badgeValue, font: badgeFont, textColor: self.theme.tabBarBadgeTextColor) container.badgeContainerNode.isHidden = false } else { container.badgeContainerNode.isHidden = true } } if !container.badgeContainerNode.isHidden { var hasSingleLetterValue: Bool = false if let string = container.badgeTextNode.attributedText?.string { hasSingleLetterValue = string.count == 1 } let badgeSize = container.badgeTextNode.updateLayout(CGSize(width: 200.0, height: 100.0)) let backgroundSize = CGSize(width: hasSingleLetterValue ? 18.0 : max(18.0, badgeSize.width + 10.0 + 1.0), height: 18.0) let backgroundFrame: CGRect if horizontal { backgroundFrame = CGRect(origin: CGPoint(x: 15.0, y: 0.0), size: backgroundSize) } else { let contentWidth: CGFloat = 25.0 backgroundFrame = CGRect(origin: CGPoint(x: floor(node.frame.width / 2.0) + contentWidth - backgroundSize.width - 5.0, y: self.centered ? 6.0 : -1.0), size: backgroundSize) } transition.updateFrame(node: container.badgeContainerNode, frame: backgroundFrame) container.badgeBackgroundNode.frame = CGRect(origin: CGPoint(), size: backgroundFrame.size) let scaleFactor: CGFloat = horizontal ? 0.8 : 1.0 container.badgeContainerNode.subnodeTransform = CATransform3DMakeScale(scaleFactor, scaleFactor, 1.0) container.badgeTextNode.frame = CGRect(origin: CGPoint(x: floorToScreenPixels((backgroundFrame.size.width - badgeSize.width) / 2.0), y: 1.0), size: badgeSize) } } } } private func tapped(at location: CGPoint, longTap: Bool) { if let bottomInset = self.validLayout?.3 { if location.y > self.bounds.size.height - bottomInset { return } var closestNode: (Int, CGFloat)? for i in 0 ..< self.tabBarNodeContainers.count { let node = self.tabBarNodeContainers[i].imageNode let distance = abs(location.x - node.position.x) if let previousClosestNode = closestNode { if previousClosestNode.1 > distance { closestNode = (i, distance) } } else { closestNode = (i, distance) } } if let closestNode = closestNode { let container = self.tabBarNodeContainers[closestNode.0] self.itemSelected(closestNode.0, longTap, [container.imageNode.imageNode, container.imageNode.textImageNode, container.badgeContainerNode]) } } } }