Change red traffic light button when document is edited

INAppStoreWindow/INAppStoreWindow.m

@@ -620,6 +620,11 @@ - (void)setMinSize:(NSSize)size
 	[self _layoutTrafficLightsAndContent];
 }
 
+-(void)setDocumentEdited:(BOOL)flag {
+	[super setDocumentEdited:flag];
+	[[NSNotificationCenter defaultCenter] postNotificationName:@"windowDidSetDocumentEdited" object:self];
+}
+
 - (void)setToolbar:(NSToolbar *)toolbar
 {
 	[self setTitleBarHeight:[self _minimumTitlebarHeight]];

INAppStoreWindow/INWindowButton.h

@@ -23,27 +23,42 @@
 /**
  An image for the normal state.
  */
-@property (nonatomic, strong) NSImage *activeImage;
+@property (nonatomic, strong) NSImage *normalImage;
+
+/**
+ An image used when user hovers receiver with mouse pointer.
+ */
+@property (nonatomic, strong) NSImage *rolloverImage;
+
+/**
+ An image for the pressed state.
+ */
+@property (nonatomic, strong) NSImage *pressedImage;
 
 /**
  An image for the normal state, but displayed when receiver's window in not a key.
  */
-@property (nonatomic, strong) NSImage *activeNotKeyWindowImage;
+@property (nonatomic, strong) NSImage *notKeyImage;
 
 /**
  An image used in disabled state.
  */
-@property (nonatomic, strong) NSImage *inactiveImage;
+@property (nonatomic, strong) NSImage *disabledImage;
 
 /**
- An image used when user hovers receiver with mouse pointer.
+ An image for the normal edited state.
  */
-@property (nonatomic, strong) NSImage *rolloverImage;
+@property (nonatomic, strong) NSImage *normalEditedImage;
 
 /**
- An image for the pressed state.
+ An image for the pressed edited state.
  */
-@property (nonatomic, strong) NSImage *pressedImage;
+@property (nonatomic, strong) NSImage *pressedEditedImage;
+
+/**
+ An image for the edited state, but displayed when receiver's window in not a key.
+ */
+@property (nonatomic, strong) NSImage *notKeyEditedImage;
 
 /**
  @param size Designated size of the button. System size is 14x16 and you are considered to use it.

INAppStoreWindow/INWindowButton.m

@@ -141,7 +141,7 @@ - (INWindowButtonGroup *)group
 
 - (void)windowButtonGroupDidUpdateRolloverStateNotification:(NSNotification *)n
 {
-	[self updateRollOverImage];
+	[self updateImage];
 }
 
 #pragma mark - Tracking Area
@@ -187,6 +187,7 @@ - (void)viewWillMoveToWindow:(NSWindow *)newWindow
 		[nc addObserver:self selector:@selector(windowDidChangeFocus:) name:NSWindowDidBecomeKeyNotification object:newWindow];
 		[nc addObserver:self selector:@selector(windowDidChangeFocus:) name:NSWindowDidResignKeyNotification object:newWindow];
 		[nc addObserver:self selector:@selector(windowDidMiniaturize:) name:NSWindowDidMiniaturizeNotification object:newWindow];
+		[nc addObserver:self selector:@selector(windowDidSetDocumentEdited:) name:@"windowDidSetDocumentEdited" object:newWindow];
 		if ([NSWindow instancesRespondToSelector:@selector(toggleFullScreen:)]) {
 			[nc addObserver:self selector:@selector(windowWillEnterFullScreen:) name:NSWindowWillEnterFullScreenNotification object:newWindow];
 			[nc addObserver:self selector:@selector(windowWillExitFullScreen:) name:NSWindowWillExitFullScreenNotification object:newWindow];
@@ -197,6 +198,7 @@ - (void)viewWillMoveToWindow:(NSWindow *)newWindow
 - (void)windowDidChangeFocus:(NSNotification *)n
 {
 	[self updateImage];
+	[self.group resetMouseCaptures];
 }
 
 - (void)windowWillEnterFullScreen:(NSNotification *)n
@@ -216,6 +218,11 @@ - (void)windowDidMiniaturize:(NSNotification *)notification
 	[self.group resetMouseCaptures];
 }
 
+- (void)windowDidSetDocumentEdited:(NSNotification *)n
+{
+	[self updateImage];
+}
+
 #pragma mark - Event Handling
 
 - (void)viewDidEndLiveResize
@@ -228,63 +235,57 @@ - (void)mouseEntered:(NSEvent *)theEvent
 {
 	[super mouseEntered:theEvent];
 	[self.group didCaptureMousePointer];
-	[self updateRollOverImage];
+	[self updateImage];
 }
 
 - (void)mouseExited:(NSEvent *)theEvent
 {
 	[super mouseExited:theEvent];
 	[self.group didReleaseMousePointer];
-	[self updateRollOverImage];
+	[self updateImage];
 }
 
 #pragma mark - Button Appearance
 
-- (void)setPressedImage:(NSImage *)pressedImage
-{
-	self.alternateImage = pressedImage;
-}
-
-- (NSImage *)pressedImage
-{
-	return self.alternateImage;
-}
-
 - (void)setEnabled:(BOOL)enabled
 {
 	[super setEnabled:enabled];
 	if (enabled) {
-		self.image = self.activeImage;
-	} else {
-		self.image = self.inactiveImage;
-	}
-}
-
-- (void)updateRollOverImage
-{
-	if ([self.group shouldDisplayRollOver] && [self isEnabled]) {
-		self.image = self.rolloverImage;
+		self.image = self.normalImage;
 	} else {
-		[self updateImage];
+		self.image = self.disabledImage;
 	}
 }
 
 - (void)updateImage
 {
-	if ([self.window isKeyWindow]) {
-		[self updateActiveImage];
-	} else {
-		self.image = self.activeNotKeyWindowImage;
-	}
-}
-
-- (void)updateActiveImage
-{
-	if ([self isEnabled]) {
-		self.image = self.activeImage;
+	if ([self.group shouldDisplayRollOver] && [self isEnabled]) {
+		if ([self.window isDocumentEdited] && self.normalEditedImage) {
+			self.image = self.normalEditedImage;
+		} else {
+			self.image = self.rolloverImage;
+		}
+		if ([self.window isDocumentEdited] && self.pressedEditedImage) {
+			self.alternateImage = self.pressedEditedImage;
+		} else {
+			self.alternateImage = self.pressedImage;
+		}
+	} else if ([self.window isKeyWindow]) {
+		if ([self isEnabled]) {
+			if ([self.window isDocumentEdited] && self.normalEditedImage) {
+				self.image = self.normalEditedImage;
+			} else {
+				self.image = self.normalImage;
+			}
+		} else {
+			self.image = self.disabledImage;
+		}
 	} else {
-		self.image = self.inactiveImage;
+		if ([self.window isDocumentEdited] && self.notKeyEditedImage) {
+			self.image = self.notKeyEditedImage;
+		} else {
+			self.image = self.notKeyImage;
+		}
 	}
 }
-
 @end