Re-add titleBarContainer when setStyleMask: causes the themeFrameView…

… to change
Kapeli · Oct 5, 2013 · 1a8a887 · 1a8a887
1 parent 6c41376
commit 1a8a887
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/INAppStoreWindow.m b/INAppStoreWindow.m
@@ -453,6 +453,7 @@ - (void)dealloc
     [_minimizeButton release];
     [_zoomButton release];
     [_fullScreenButton release];
+    [_titleBarContainer release];
     [super dealloc];
     #endif
 }
@@ -721,6 +722,12 @@ - (void)setStyleMask:(NSUInteger)styleMask
 	_preventWindowFrameChange = YES;
 	[super setStyleMask:styleMask];
 	_preventWindowFrameChange = NO;
+	if(_titleBarContainer && [_titleBarContainer superview] != [self themeFrameView])
+    	{
+        	NSView *firstSubview = [[[self themeFrameView] subviews] objectAtIndex:0];
+        	[self _recalculateFrameForTitleBarContainer];
+        	[[self themeFrameView] addSubview:_titleBarContainer positioned:NSWindowBelow relativeTo:firstSubview];
+    	}
 }
 
 - (void)setFrame:(NSRect)frameRect display:(BOOL)flag
@@ -963,11 +970,10 @@ - (void)_createTitlebarView
     NSView *firstSubview = [[[self themeFrameView] subviews] objectAtIndex:0];
     [self _recalculateFrameForTitleBarContainer];
     [[self themeFrameView] addSubview:container positioned:NSWindowBelow relativeTo:firstSubview];
-    #if __has_feature(objc_arc)
     _titleBarContainer = container;
+    #if __has_feature(objc_arc)
     self.titleBarView = [[INTitlebarView alloc] initWithFrame:NSZeroRect];
     #else
-    _titleBarContainer = container;
     self.titleBarView = [[[INTitlebarView alloc] initWithFrame:NSZeroRect] autorelease];
     #endif
 }