Adds public methods to start / stop recording programmatically (witho…

…ut tapping button)

HeapInspector/HINSPDebug.h

@@ -17,6 +17,12 @@ NS_ASSUME_NONNULL_BEGIN
 /// Stops the HeapInspector and removes the inspector's view
 + (void)stop;
 
+// Shows HeapInspector (if not visible yet) and starts the record immediately
++ (void)startRecord;
+
+// Stops record - but does not hide the HeapInspector
++ (void)stopRecord;
+
 /// Add some (or only one) class prefix like `UI` OR `MK` to record classes that match the prefix only.
 /// It's highly recommended to record a specific class or prefix -
 /// otherwise all Cocoa classes will be recorded, which slows down the performance.

HeapInspector/HINSPDebug.m

@@ -118,6 +118,7 @@ - (HINSPHeapStackTableViewController *)heapStackControllerWithHeapStack:(NSArray
 
 - (void)stopRecord
 {
+    _window.recordButton.isRecording = NO;
     _recordedHeap = [HINSPHeapStackInspector recordedHeap];
     [self showInfoLabel];
     [NSObject endSnapshot];
@@ -128,6 +129,7 @@ - (void)beginRecord
     _recordedHeap = nil;
     [self resetInfoLabel];
     [NSObject beginSnapshot];
+    _window.recordButton.isRecording = YES;
     [HINSPHeapStackInspector performHeapShot];
 }
 
@@ -138,6 +140,14 @@ + (void)start
     twDebug = [[HINSPDebug alloc] init];
 }
 
++ (void)startRecord
+{
+    if (!twDebug) {
+        [self start];
+    }
+    [twDebug beginRecord];
+}
+
 + (void)stop
 {
     [NSObject endSnapshot];
@@ -146,6 +156,10 @@ + (void)stop
     twDebug = nil;
 }
 
++ (void)stopRecord {
+    [twDebug stopRecord];
+}
+
 + (void)addClassPrefixesToRecord:(NSArray <NSString *> *)classPrefixes
 {
     if (classPrefixes) {

HeapInspector/HINSPRecordButton.h

@@ -9,5 +9,5 @@
 #import <UIKit/UIKit.h>
 
 @interface HINSPRecordButton : UIControl
-
+@property (nonatomic) BOOL isRecording;
 @end

HeapInspector/HINSPRecordButton.m

@@ -10,8 +10,6 @@
 
 
 @interface HINSPRecordButton ()
-
-@property (nonatomic) BOOL isRecording;
 @property (nonatomic, weak) CAShapeLayer *shapeLayer;
 @end
 
@@ -31,13 +29,19 @@ - (instancetype)initWithFrame:(CGRect)frame
 - (void)tapped:(id)sender
 {
     self.isRecording = !self.isRecording;
-    UIColor *color = nil;
-    if (self.isRecording) {
-        color = [self recordingColor];
-    } else {
-        color = [self defaultColor];
+}
+
+- (void)setIsRecording:(BOOL)isRecording {
+    if (_isRecording != isRecording) {
+        _isRecording = isRecording;
+        UIColor *color = nil;
+        if (isRecording) {
+            color = [self recordingColor];
+        } else {
+            color = [self defaultColor];
+        }
+        _shapeLayer.fillColor = color.CGColor;
     }
-    _shapeLayer.fillColor = color.CGColor;
 }
 
 #pragma mark - Setter