main.m

//
//  QuickTimeFixer
//
//  This code was written by Jonathan Alland.
//


#import <Foundation/Foundation.h>
#import <Cocoa/Cocoa.h>
#import <AVFoundation/AVFoundation.h>
#import <objc/runtime.h>
#import "ZKSwizzle/ZKSwizzle.h"

#define EMPTY_SWIZZLE_INTERFACE(CLASS_NAME, SUPERCLASS) @interface CLASS_NAME : SUPERCLASS @end

@interface NSWindow (quickTimeFixer)
- (id)_canBecomeFullScreen;
@end




@interface QTFixer_AVPlayerItem : AVPlayerItem
- (id)_trackWithTrackID:(int)arg1;
@end

@implementation QTFixer_AVPlayerItem
//Apple removed these methods from AVFoundation, but QuickTime needs them!

- (int)selectedTrackIDInTrackGroup:(id)trackGroup {
	NSArray *trackIds = [trackGroup trackIDs];
	for (int i = 0; i < [trackIds count]; i++) {
		int currTrackID = [[trackIds objectAtIndex:i]intValue];
		AVAssetTrack *currentTrack = [self _trackWithTrackID: currTrackID];
		if ([currentTrack isEnabled]) {
			return currTrackID;
		}
	}
	return -1;
}

- (void)selectTrackWithID:(int)trackID inTrackGroup:(id)trackGroup {
	NSArray *trackIds = [trackGroup trackIDs];
	for (int i = 0; i < [trackIds count]; i++) {
		int currTrackID = [[trackIds objectAtIndex:i]intValue];
		if (currTrackID == trackID) {
			[[self _trackWithTrackID: currTrackID] setEnabled: true];
		} else {
			[[self _trackWithTrackID: currTrackID] setEnabled: false];
		}
	}
}

@end




EMPTY_SWIZZLE_INTERFACE(QTFixer_AVAssetExportSession, AVAssetExportSession);
@implementation QTFixer_AVAssetExportSession

//Apple removed this method from AVFoundation, but all we need is the stub.
- (void)setUsesHardwareVideoEncoderIfAvailable:(BOOL)arg1 {}

@end




EMPTY_SWIZZLE_INTERFACE(QTFixer_MGDocumentViewController, NSViewController);
@implementation QTFixer_MGDocumentViewController

- (void)loadView {
	ZKOrig(void);
	if (
		[self isKindOfClass:NSClassFromString(@"MGAudioPlaybackViewController")] || 
		[self isKindOfClass:NSClassFromString(@"MGVideoPlaybackViewController")]
	) {
		[self runUserScript:@"userFileOpenedScript"];
	}
}

- (void)close {
	if (
		[self isKindOfClass:NSClassFromString(@"MGAudioPlaybackViewController")] || 
		[self isKindOfClass:NSClassFromString(@"MGVideoPlaybackViewController")]
	) {
		[self runUserScript:@"userFileClosedScript"];
	}
	ZKOrig(void);
}

- (void)runUserScript:(NSString*)scriptName {	
	NSString* path = [[NSBundle mainBundle] pathForResource:scriptName ofType:@"scpt"];
	if (path != nil) {
		NSString *scriptSource = [
			NSString stringWithFormat:@"run script (load script POSIX file \"%@\") with parameters {\"%@\"}",
			path,
			[[self valueForKey:@"document"]displayName]
		];
		[[[NSAppleScript alloc] initWithSource:scriptSource] executeAndReturnError:nil];
	}
}

@end




static const char kNeedsSetHasAutoCanDrawSubviewsIntoLayerKey;
static const char kNeedsCheckWindowButtonsKey;
@interface QTFixer_MGCinematicFrameView : NSView
- (void) _setHasAutoCanDrawSubviewsIntoLayer:(bool)arg1;
@end

@implementation QTFixer_MGCinematicFrameView
/*In this class, we correct graphical issues using fixes discovered via trial and error.
 The timing of when these fixes are needed is very specific!*/

- (BOOL)needsCheckWindowButtons {
	return [objc_getAssociatedObject(self, &kNeedsCheckWindowButtonsKey) boolValue];
}

- (void)setNeedsCheckWindowButtons:(BOOL)value {
	objc_setAssociatedObject(self, &kNeedsCheckWindowButtonsKey, @(value), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (BOOL)needsSetHasAutoCanDrawSubviewsIntoLayer {
	return [objc_getAssociatedObject(self, &kNeedsSetHasAutoCanDrawSubviewsIntoLayerKey) boolValue];
}

- (void)setNeedsSetHasAutoCanDrawSubviewsIntoLayer:(BOOL)value {
	objc_setAssociatedObject(self, &kNeedsSetHasAutoCanDrawSubviewsIntoLayerKey, @(value), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (void)setTitle:(id)arg1 {
	[self setLayerContentsRedrawPolicy:NSViewLayerContentsRedrawDuringViewResize];
	
	// Why can't we just setCanDrawSubviewsIntoLayer here? Because it will break screen recording documents.
	[self setNeedsSetHasAutoCanDrawSubviewsIntoLayer:YES];
	ZKOrig(void, arg1);
}

- (void)displayIfNeeded {
	//Fix non-video documents displaying without a background.
	
	//This is very misleading. ZKHookIvar will return a set of nine (!) bits from MGCinematicFrameView.
	//The fifth of these bits represents _entireBackBufferIsDirty.
	unsigned int *Ivars = &ZKHookIvar(self, unsigned int, "_entireBackBufferIsDirty");
	
	//Set the _entireBackBufferIsDirty bit to 1
	*Ivars |= 1UL << 4;
	
	// Disabling screen updates here prevents a brief flash of glitchiness.
	NSDisableScreenUpdates();
	[super displayIfNeeded];
	ZKOrig(void);
	NSEnableScreenUpdates();
	
	// I think I might finally semi-understand why this works. Decompilers show that [self displayIfNeeded]
	// calls [super displayIfNeeded] at the end. I think [super displayIfNeeded] re-breaks whatever
	// [self displayIfNeeded] fixes. However, [super displayIfNeeded] won't do anything if it's not "needed"!
	// By calling the super implementation first, we prevent it from being necessary later, and so the breakage
	// happen before the fix instead of the other way around. If we also disable screen updates during the
	// brief moment of breakage, it never becomes visible to the user.

	if ([self needsSetHasAutoCanDrawSubviewsIntoLayer] && [[self window] _canBecomeFullScreen] != NULL) {
		//Fix FullScreen animation glitch
		[self setCanDrawSubviewsIntoLayer:true];
		[self setNeedsSetHasAutoCanDrawSubviewsIntoLayer:NO];
	}
	ZKOrig(void);
}

- (void)_windowChangedKeyState {
	[self setNeedsCheckWindowButtons:YES];
	[self unstickWindowButtonHoverState];
	
	ZKOrig(void);
}

- (void)setFrameSize:(struct CGSize)arg1 {
	[self setNeedsCheckWindowButtons:YES];
	[self performSelector:@selector(unstickWindowButtonHoverState) withObject:nil afterDelay:0.7];
	ZKOrig(void, arg1);
}

- (void)unstickWindowButtonHoverState {
	if ([self needsCheckWindowButtons]) {
		//This won't always work; it depends on the location of the user's mouse at the time this code is run.
		//(This bug exists in Mountain Lion too, btw.)
		[self setNeedsCheckWindowButtons:NO];
		[[self subviews][1] viewDidEndLiveResize];
	}
}

@end




EMPTY_SWIZZLE_INTERFACE(QTFixer_MGCinematicWindow, NSWindow);
@implementation QTFixer_MGCinematicWindow

- (void)_windowTransformAnimationDidEnd:(id)arg1 {
	ZKOrig(void, arg1);
	for (int i = 0; i <= 100; i++) {
		[self performSelector:@selector(invalidateShadow) withObject:nil afterDelay:(i * 0.01)];
	}
	// Just in case
	[self performSelector:@selector(invalidateShadow) withObject:nil afterDelay:2];
	[self performSelector:@selector(invalidateShadow) withObject:nil afterDelay:3];
	[self performSelector:@selector(invalidateShadow) withObject:nil afterDelay:5];
}

@end




EMPTY_SWIZZLE_INTERFACE(QTFixer_MGScrollEventHandlingHUDSlider, NSObject);
@implementation QTFixer_MGScrollEventHandlingHUDSlider
//Prevent an audio glitch.
- (void)beginGestureWithEvent:(id)arg1 {}
- (void)scrollWheel:(id)arg1 {}
@end




EMPTY_SWIZZLE_INTERFACE(QTFixer_MGPlayerController, NSController);
@implementation QTFixer_MGPlayerController
//Prevent an audio glitch.
- (void)increaseVolume:(id)arg1 {}
- (void)decreaseVolume:(id)arg1 {}
@end




EMPTY_SWIZZLE_INTERFACE(QTFixer_QTHUDButton, NSControl);
@implementation QTFixer_QTHUDButton
//Tabbing between QTHUDButtons can cause QuickTime to crash. This behavior is annoying anyway.
- (BOOL)becomeFirstResponder {
	return false;
}
@end




EMPTY_SWIZZLE_INTERFACE(QTFixer_NSWindow, NSWindow);
@implementation QTFixer_NSWindow
//Continuation of above: Tabbing between QTHUDButtons can cause QuickTime to crash.
- (void)selectKeyViewFollowingView:(id)arg1 {
	if (strcmp(object_getClassName(arg1), "NSView") != 0 && strcmp(object_getClassName(arg1), "MGPlayPauseShuttleControllerView") != 0) {
		ZKOrig(void, arg1);
	}
}
@end




static const char kIsSettingMainViewControllerKey;

@interface QTFixer_MGDocumentWindowController : NSWindowController
- (void)toggleFloating:(id)arg1;
@property (nonatomic, strong) id currentMainViewController;
@end
@implementation QTFixer_MGDocumentWindowController

- (BOOL)isSettingMainViewController {
	return [objc_getAssociatedObject(self, &kIsSettingMainViewControllerKey) boolValue];
}

- (void)setIsSettingMainViewController:(BOOL)value {
	objc_setAssociatedObject(self, &kIsSettingMainViewControllerKey, @(value), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (void)setCurrentMainViewController:(id)controller {
	if ([self isSettingMainViewController]) {
		ZKOrig(void, controller);
		return;
	}

	[self setIsSettingMainViewController:YES];

	id oldValue = [self valueForKey:@"currentMainViewController"];

	if (oldValue != controller) {
		[self willChangeValueForKey:@"currentMainViewController"];
		ZKOrig(void, controller);
		[self didChangeValueForKey:@"currentMainViewController"];
	} else {
		ZKOrig(void, controller);
	}

	[self setIsSettingMainViewController:NO];
}

+ (BOOL)automaticallyNotifiesObserversOfCurrentMainViewController {
	return NO;
}

- (id)customWindowsToEnterFullScreenForWindow:(id)arg1 {
	if (ZKHookIvar(self, int, "_isFloating")) {
		[self toggleFloating:nil];
	}

	return ZKOrig(id, arg1);
}
@end




EMPTY_SWIZZLE_INTERFACE(QTFixer_MGDocumentController, NSDocumentController);
@implementation QTFixer_MGDocumentController

- (NSString *)typeForContentsOfURL:(NSURL *)url error:(NSError **)outError {

	// Disabling AVFoundation appears to:
	//	+ Allows QuickTime Framework based audio decoders to work.
	//		• With AVFoundation enabled, QuickTime does seemingly try to use these decoders, but fails.
	//		• Documents will open but not play. Console messages:
	//			• `>audiocomp> AudioComponentPlugin.cpp:75: NewInstance: error -3000 returned from Open`
	//			• `>aq> AudioQueueObject.cpp:1590: Prime: failed (-9405); will stop (66150/0 frames)`
	//	+ Allows third-party QuickTime Framework based AVI importers to work.
	//		• Presumably because with AVFoundation enabled, Apple's built-in AVI support takes priority.
	//		• (Apple's built-in AVI support fails to open many/most files.)
	//	- Disables features:
	//		• All video editing functionality
	//		• Sharing
	//	- Breaks Apple's native AVFoundation decoders.
	//	- Breaks modern Audio Component decoders.
	
	NSArray *extensionsThatForceQTKit = @[@"avi", @"flv", @"ogg", @"ogv"];
	
	BOOL shouldForceQTKit = [extensionsThatForceQTKit containsObject:[[url pathExtension] lowercaseString]];
	if ([NSEvent modifierFlags] & NSAlternateKeyMask) shouldForceQTKit = !shouldForceQTKit; // alt inverts behavior
	
	if (shouldForceQTKit) {
		[[NSUserDefaults standardUserDefaults] setBool:NO forKey:@"MGEnableAVFoundation"];
		[[NSUserDefaults standardUserDefaults] synchronize];
	} else {
		[[NSUserDefaults standardUserDefaults] setBool:YES forKey:@"MGEnableAVFoundation"];
		[[NSUserDefaults standardUserDefaults] synchronize];
	}
	
	return ZKOrig(NSString *, url, outError);
}

@end




EMPTY_SWIZZLE_INTERFACE(QTFixer_MGAssetLoader, NSObject);
@implementation QTFixer_MGAssetLoader

- (void)loadingDidFailWithError:(id)arg1 {
	// While not strictly necessary, this is very useful information to have in the console.
	NSLog(@"QuickTime failed to load document due to error: %@", arg1);
	ZKOrig(void, arg1);
}

@end




@implementation NSObject (main)

+ (void)load {
	ZKSwizzle(QTFixer_AVPlayerItem, AVPlayerItem);
	ZKSwizzle(QTFixer_AVAssetExportSession, AVAssetExportSession);
	ZKSwizzle(QTFixer_MGDocumentViewController, MGDocumentViewController);
	ZKSwizzle(QTFixer_MGCinematicFrameView, MGCinematicFrameView);
	ZKSwizzle(QTFixer_MGCinematicWindow, MGCinematicWindow);
	ZKSwizzle(QTFixer_MGScrollEventHandlingHUDSlider, MGScrollEventHandlingHUDSlider);
	ZKSwizzle(QTFixer_MGPlayerController, MGPlayerController);
	ZKSwizzle(QTFixer_QTHUDButton, QTHUDButton);
	ZKSwizzle(QTFixer_NSWindow, NSWindow);
	ZKSwizzle(QTFixer_MGDocumentWindowController, MGDocumentWindowController);
	ZKSwizzle(QTFixer_MGDocumentController, MGDocumentController);
	ZKSwizzle(QTFixer_MGAssetLoader, MGAssetLoader);
	
	
	//Fix menu bar not switching to QuickTime.
	[[[NSAppleScript alloc] initWithSource:@"tell application (path to frontmost application as text) to activate"] executeAndReturnError:nil];
}

@end




int main() {}