From ad1f7c1f76fa83390b3f580aa41948e44b5ec727 Mon Sep 17 00:00:00 2001
From: Ryan O'Hara <ryan@silverstripe.com>
Date: Fri, 4 May 2012 11:26:40 +1200
Subject: [PATCH] ENHANCEMENT: Add Watermark for preview states (TRAC-7214)

---
 code/controllers/SilverStripeNavigator.php | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/code/controllers/SilverStripeNavigator.php b/code/controllers/SilverStripeNavigator.php
index b3ba5802..40e734d7 100644
--- a/code/controllers/SilverStripeNavigator.php
+++ b/code/controllers/SilverStripeNavigator.php
@@ -119,6 +119,12 @@ class SilverStripeNavigatorItem extends ViewableData {
 	 * For example, a "future state" item might show a date selector.
 	 */
 	function getHTML() {}
+
+	/**
+	* @return String
+	* Text displayed in watermark
+	*/
+	function getWatermarkHTML() {}
 	
 	/**
 	 * Optional link to a specific view of this record.
@@ -214,6 +220,10 @@ class SilverStripeNavigatorItem_StageLink extends SilverStripeNavigatorItem {
 			return "<a href=\"$this->recordLink\">". _t('ContentController.DRAFTSITE', 'Draft Site') ."</a>";
 		}
 	}
+
+	function getWatermarkHTML() {
+		return "Draft Site";
+	}
 	
 	function getMessage() {
 		return "<div id=\"SilverStripeNavigatorMessage\" title=\"". _t('ContentControl.NOTEWONTBESHOWN', 'Note: this message will not be shown to your visitors') ."\">".  _t('ContentController.DRAFTSITE', 'Draft Site') ."</div>";
@@ -258,6 +268,10 @@ class SilverStripeNavigatorItem_LiveLink extends SilverStripeNavigatorItem {
 			return "<a href=\"$this->recordLink\">". _t('ContentController.PUBLISHEDSITE', 'Published Site') ."</a>";
 		}
 	}
+
+	function getWatermarkHTML() {
+		return "Published Site";
+	}
 	
 	function getMessage() {
 		return "<div id=\"SilverStripeNavigatorMessage\" title=\"". _t('ContentControl.NOTEWONTBESHOWN', 'Note: this message will not be shown to your visitors') ."\">".  _t('ContentController.PUBLISHEDSITE', 'Published Site') ."</div>";