From d42fba7906435b7a946a2d70cf38f9c984fd2028 Mon Sep 17 00:00:00 2001
From: Will Rossiter <will.rossiter@gmail.com>
Date: Fri, 18 Jan 2013 10:49:53 +1300
Subject: [PATCH] Add extension hook points.

---
 code/controllers/GoogleSitemapController.php | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/code/controllers/GoogleSitemapController.php b/code/controllers/GoogleSitemapController.php
index 5143ba6..b3ac6ac 100644
--- a/code/controllers/GoogleSitemapController.php
+++ b/code/controllers/GoogleSitemapController.php
@@ -34,8 +34,11 @@ class GoogleSitemapController extends Controller {
 			
 			$this->getResponse()->addHeader('Content-Type', 'application/xml; charset="utf-8"');
 
+			$sitemaps = GoogleSitemap::get_sitemaps();
+			$this->extend('updateGoogleSitemaps', $sitemaps);
+
 			return array(
-				'Sitemaps' => GoogleSitemap::get_sitemaps()
+				'Sitemaps' => $sitemaps
 			);
 		} else {
 			return new SS_HTTPResponse('Page not found', 404);
@@ -57,9 +60,11 @@ class GoogleSitemapController extends Controller {
 			
 			$this->getResponse()->addHeader('Content-Type', 'application/xml; charset="utf-8"');
 
-			// But we want to still render.
+			$items = GoogleSitemap::get_items($class, $page);
+			$this->extend('updateGoogleSitemapItems', $items, $class, $page);
+
 			return array(
-				'Items' => GoogleSitemap::get_items($class, $page)
+				'Items' => $items
 			);
 		} else {
 			return new SS_HTTPResponse('Page not found', 404);