diff --git a/PlayTools/Controls/PTFakeTouch/NSObject+Swizzle.m b/PlayTools/Controls/PTFakeTouch/NSObject+Swizzle.m index 42e6fa5f..0c432746 100644 --- a/PlayTools/Controls/PTFakeTouch/NSObject+Swizzle.m +++ b/PlayTools/Controls/PTFakeTouch/NSObject+Swizzle.m @@ -54,21 +54,9 @@ + (void) load [objc_getClass("FBSSceneSettings") swizzleInstanceMethod:@selector(bounds) withMethod:@selector(hook_bounds)]; [objc_getClass("FBSDisplayMode") swizzleInstanceMethod:@selector(size) withMethod:@selector(hook_size)]; } - - [objc_getClass("_UIMenuBuilder") swizzleInstanceMethod:sel_getUid("initWithRootMenu:") withMethod:@selector(initWithRootMenuHook:)]; [objc_getClass("IOSViewController") swizzleInstanceMethod:@selector(prefersPointerLocked) withMethod:@selector(hook_prefersPointerLocked)]; -} - -bool menuWasCreated = false; -- (id) initWithRootMenuHook:(id)rootMenu { - self = [self initWithRootMenuHook:rootMenu]; - if (!menuWasCreated) { - [PlayCover initMenuWithMenu: self]; - menuWasCreated = TRUE; - } - - return self; + [self swizzleInstanceMethod:@selector(init) withMethod:@selector(hook_init)]; } - (BOOL) hook_prefersPointerLocked { @@ -87,4 +75,16 @@ - (CGSize) hook_size { return [PlayScreen sizeAspectRatio:[self hook_size]]; } +bool menuWasCreated = false; + +-(id) hook_init { + if (!menuWasCreated) { + if ([[self class] isEqual: NSClassFromString(@"_UIMenuBuilder")]) { + [PlayCover initMenuWithMenu: self]; + menuWasCreated = TRUE; + } + } + + return self; +} @end