State tracking based - isa swizzle

Sources/KSCrashRecording/KSCrashAppStateTracker+Private.h

@@ -0,0 +1,13 @@
+#import "KSCrashAppStateTracker.h"
+
+NS_ASSUME_NONNULL_BEGIN
+
+@interface KSCrashAppStateTracker ()
+
+- (void)_setTransitionState:(KSCrashAppTransitionState)transitionState;
+
+@property (nonatomic, assign) BOOL proxied;
+
+@end
+
+NS_ASSUME_NONNULL_END

Sources/KSCrashRecording/KSCrashAppStateTracker.m

@@ -40,6 +40,7 @@
         case KSCrashAppTransitionStateStartupPrewarm: return "prewarm";
         case KSCrashAppTransitionStateActive: return "active";
         case KSCrashAppTransitionStateLaunching: return "launching";
+        case KSCrashAppTransitionStateLaunched: return "launched";
         case KSCrashAppTransitionStateBackground: return "background";
         case KSCrashAppTransitionStateTerminating: return "terminating";
         case KSCrashAppTransitionStateExiting: return "exiting";
@@ -60,6 +61,7 @@ bool ksapp_transition_state_is_user_perceptible(KSCrashAppTransitionState state)
 
         case KSCrashAppTransitionStateStartup:
         case KSCrashAppTransitionStateLaunching:
+        case KSCrashAppTransitionStateLaunched:
         case KSCrashAppTransitionStateForegrounding:
         case KSCrashAppTransitionStateActive:
         case KSCrashAppTransitionStateDeactivating:
@@ -109,8 +111,13 @@ @interface KSCrashAppStateTracker () {
     // transition state and observers protected by the lock
     os_unfair_lock _lock;
     KSCrashAppTransitionState _transitionState;
+    BOOL _transitionComplete;
     NSMutableArray<id<KSCrashAppStateTrackerObserving>> *_observers;
+    BOOL _proxied;
 }
+
+@property (nonatomic, assign) BOOL proxied;
+
 @end
 
 @implementation KSCrashAppStateTracker
@@ -156,6 +163,15 @@ - (void)dealloc
     [self stop];
 }
 
+- (void)setProxied:(BOOL)proxied
+{
+    _proxied = proxied;
+    if (proxied) {
+        [self stop];
+        _registrations = @[];
+    }
+}
+
 // Observers are either an object passed in that
 // implements `KSCrashAppStateTrackerObserving` or a block.
 // Both will be wrapped in a `KSCrashAppStateTrackerBlockObserver`.
@@ -233,20 +249,62 @@ - (KSCrashAppTransitionState)transitionState
     return ret;
 }
 
+- (void)_locked_completeState:(KSCrashAppTransitionState)transitionState
+{
+    if (_transitionState != transitionState || _transitionComplete) {
+        return;
+    }
+
+    _transitionComplete = YES;
+    NSLog(@"[AC] %s, completed: %d", ksapp_transition_state_to_string(transitionState), _transitionComplete);
+
+    // send delegate
+    // ie: didCompleteTransition or something...
+}
+
+- (void)_completeState:(KSCrashAppTransitionState)transitionState
+{
+    os_unfair_lock_lock(&_lock);
+    [self _locked_completeState:transitionState];
+    os_unfair_lock_unlock(&_lock);
+}
+
 - (void)_setTransitionState:(KSCrashAppTransitionState)transitionState
 {
+    BOOL completeOnNextLoop = NO;
     NSArray<id<KSCrashAppStateTrackerObserving>> *observers = nil;
     {
         os_unfair_lock_lock(&_lock);
+
         if (_transitionState != transitionState) {
+
+            // finalize the current state
+            [self _locked_completeState:_transitionState];
+
+            // move to the next state
             _transitionState = transitionState;
+            _transitionComplete = NO;
+            completeOnNextLoop = YES;
             observers = [_observers copy];
         }
+
         os_unfair_lock_unlock(&_lock);
     }
 
-    for (id<KSCrashAppStateTrackerObserving> obs in observers) {
-        [obs appStateTracker:self didTransitionToState:transitionState];
+    if (observers) {
+        NSLog(@"[AC] %s, completed: %d", ksapp_transition_state_to_string(transitionState), _transitionComplete);
+
+        for (id<KSCrashAppStateTrackerObserving> obs in observers) {
+            [obs appStateTracker:self didTransitionToState:transitionState];
+        }
+    }
+
+
+    if (completeOnNextLoop) {
+        __weak typeof(self)weakMe = self;
+        CFRunLoopPerformBlock(CFRunLoopGetCurrent(), kCFRunLoopCommonModes, ^{
+            [weakMe _locked_completeState:transitionState];
+        });
     }
 }
 
@@ -287,7 +345,7 @@ - (void)start
     _registrations = @[
 
         OBSERVE(_center, UIApplicationDidFinishLaunchingNotification, {
-            [weakMe _setTransitionState:KSCrashAppTransitionStateLaunching];
+            [weakMe _setTransitionState:KSCrashAppTransitionStateLaunched];
         }),
         OBSERVE(_center, UIApplicationWillEnterForegroundNotification, {
             [weakMe _setTransitionState:KSCrashAppTransitionStateForegrounding];
@@ -323,3 +381,5 @@ - (void)stop
 }
 
 @end
+
+