-
Notifications
You must be signed in to change notification settings - Fork 49
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
711ce92
commit 1233cf3
Showing
3 changed files
with
92 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,86 @@ | ||
| // | ||
| // UIViewController.swift | ||
| // PostHog | ||
| // | ||
| // Inspired by | ||
| // https://raw.githubusercontent.com/segmentio/analytics-swift/e613e09aa1b97144126a923ec408374f914a6f2e/Examples/other_plugins/UIKitScreenTracking.swift | ||
| // | ||
| // Created by Manoel Aranda Neto on 23.10.23. | ||
| // | ||
|
|
||
| import Foundation | ||
| import UIKit | ||
|
|
||
| extension UIViewController { | ||
| static func swizzle(forClass: AnyClass, original: Selector, new: Selector) { | ||
| guard let originalMethod = class_getInstanceMethod(forClass, original) else { return } | ||
| guard let swizzledMethod = class_getInstanceMethod(forClass, new) else { return } | ||
| method_exchangeImplementations(originalMethod, swizzledMethod) | ||
| } | ||
|
|
||
| static func swizzleScreenView() { | ||
| UIViewController.swizzle(forClass: UIViewController.self, | ||
| original: #selector(UIViewController.viewDidAppear(_:)), | ||
| new: #selector(UIViewController.viewDidApperOverride)) | ||
| } | ||
|
|
||
| private func activeController() -> UIViewController? { | ||
| // if a view is being dismissed, this will return nil | ||
| if let root = viewIfLoaded?.window?.rootViewController { | ||
| return root | ||
| } else if #available(iOS 13.0, *) { | ||
| // preferred way to get active controller in ios 13+ | ||
| for scene in UIApplication.shared.connectedScenes where scene.activationState == .foregroundActive { | ||
| let windowScene = scene as? UIWindowScene | ||
| let sceneDelegate = windowScene?.delegate as? UIWindowSceneDelegate | ||
| if let target = sceneDelegate, let window = target.window { | ||
| return window?.rootViewController | ||
| } | ||
| } | ||
| } else { | ||
| // this was deprecated in ios 13.0 | ||
| return UIApplication.shared.keyWindow?.rootViewController | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| private func captureScreenView() { | ||
| var rootController = viewIfLoaded?.window?.rootViewController | ||
| if rootController == nil { | ||
| rootController = activeController() | ||
| } | ||
| guard let top = findVisibleViewController(activeController()) else { return } | ||
|
|
||
| var name = String(describing: top.classForCoder).replacingOccurrences(of: "ViewController", with: "") | ||
|
|
||
| if name.count == 0 { | ||
| name = top.title ?? "Unknown" | ||
| } | ||
|
|
||
| if name != "Unknown" { | ||
| PostHogSDK.shared.capture(name) | ||
| } | ||
| } | ||
|
|
||
| @objc func viewDidApperOverride(animated: Bool) { | ||
| captureScreenView() | ||
| // it looks like we're calling ourselves, but we're actually | ||
| // calling the original implementation of viewDidAppear since it's been swizzled. | ||
| viewDidApperOverride(animated: animated) | ||
| } | ||
|
|
||
| private func findVisibleViewController(_ controller: UIViewController?) -> UIViewController? { | ||
| if let navigationController = controller as? UINavigationController { | ||
| return findVisibleViewController(navigationController.visibleViewController) | ||
| } | ||
| if let tabController = controller as? UITabBarController { | ||
| if let selected = tabController.selectedViewController { | ||
| return findVisibleViewController(selected) | ||
| } | ||
| } | ||
| if let presented = controller?.presentedViewController { | ||
| return findVisibleViewController(presented) | ||
| } | ||
| return controller | ||
| } | ||
| } |