Fix button in Xcode3CodesignTroubleshootingVC

Author: neonichu

FixCode/FixCode.m

@@ -68,6 +68,16 @@ - (void)doItRight {
     [self.currentCodeSigningWindowController close];
 }
 
+- (void)findAndReplaceFixIssueButtonInView:(NSView*)view {
+    NSButton* fixIssueButton = [self findFirstButtonRecursive:view];
+    fixIssueButton.title = @"Do it right ✨🚀✨";
+
+    fixIssueButton.action = @selector(doItRight);
+    fixIssueButton.target = self;
+
+    [fixIssueButton sizeToFit];
+}
+
 - (NSButton*)findFirstButtonRecursive:(NSView*)view {
     if ([view isKindOfClass:NSButton.class]) {
         return (NSButton*)view;
@@ -108,11 +118,7 @@ - (void)swizzleCodeSigningResolution
         self.currentCodeSigningWindowController = info.instance;
         NSView* contentView = [[self.currentCodeSigningWindowController window] contentView];
 
-        NSButton* fixIssueButton = [self findFirstButtonRecursive:contentView];
-        fixIssueButton.title = @"Do it right ✨🚀✨";
-
-        fixIssueButton.action = @selector(doItRight);
-        fixIssueButton.target = self;
+        [self findAndReplaceFixIssueButtonInView:contentView];
 
         NSTextField* field = [self findLastTextFieldRecursive:contentView];
         [field setStringValue:[field.stringValue componentsSeparatedByString:@"\n"][0]];
@@ -121,6 +127,15 @@ - (void)swizzleCodeSigningResolution
     if (error) {
         NSLog(@"Error: %@", error);
     }
+
+    [objc_getClass("Xcode3CodesignTroubleshootingViewController") aspect_hookSelector:@selector(viewDidAppear) withOptions:AspectPositionAfter usingBlock:^(id<AspectInfo> info) {
+        NSView* view = [info.instance view];
+        [self findAndReplaceFixIssueButtonInView:view];
+    } error:&error];
+
+    if (error) {
+        NSLog(@"Error: %@", error);
+    }
 }
 
 @end