From 5174ac839521f87a6ad2d7cc9d5f338134b75099 Mon Sep 17 00:00:00 2001
From: loudej <none@none>
Date: Thu, 10 Dec 2009 09:02:54 +0000
Subject: [PATCH] adding an AutomaticPartDriver, and support for setting zone
 and position with Location

--HG--
extra : convert_revision : svn%3A5ff7c347-ad56-4c35-b696-ccb81de16e03/trunk%4043674
---
 .../Models/PartDriverHandlerTests.cs          |  9 ++--
 src/Orchard/Models/PartDriver.cs              | 45 ++++++++++++++++---
 2 files changed, 46 insertions(+), 8 deletions(-)

diff --git a/src/Orchard.Tests/Models/PartDriverHandlerTests.cs b/src/Orchard.Tests/Models/PartDriverHandlerTests.cs
index 907680c1e..007d367af 100644
--- a/src/Orchard.Tests/Models/PartDriverHandlerTests.cs
+++ b/src/Orchard.Tests/Models/PartDriverHandlerTests.cs
@@ -73,19 +73,22 @@ namespace Orchard.Tests.Models {
 
             protected override DriverResult Display(StubPart part, string groupName, string displayType) {
                 var viewModel = new StubViewModel { Foo = string.Join(",", part.Foo) };
-                return PartialView(viewModel);
+                if (displayType.StartsWith("Summary"))
+                    return PartialView(viewModel, "StubViewModelTerse").Location("topmeta");
+
+                return PartialView(viewModel).Location("topmeta");
             }
 
             protected override DriverResult Editor(StubPart part, string groupName) {
                 var viewModel = new StubViewModel { Foo = string.Join(",", part.Foo) };
-                return PartialView(viewModel);
+                return PartialView(viewModel).Location("last", "10");
             }
 
             protected override DriverResult Editor(StubPart part, string groupName, IUpdateModel updater) {
                 var viewModel = new StubViewModel { Foo = string.Join(",", part.Foo) };
                 updater.TryUpdateModel(viewModel, Prefix, null, null);
                 part.Foo = viewModel.Foo.Split(new[] { ',' }).Select(x => x.Trim()).ToArray();
-                return PartialView(viewModel);
+                return PartialView(viewModel).Location("last", "10");
             }
         }
 
diff --git a/src/Orchard/Models/PartDriver.cs b/src/Orchard/Models/PartDriver.cs
index 970f47a27..5a47efe84 100644
--- a/src/Orchard/Models/PartDriver.cs
+++ b/src/Orchard/Models/PartDriver.cs
@@ -43,15 +43,29 @@ namespace Orchard.Models {
         }
 
         protected virtual string Prefix { get { return ""; } }
+        protected virtual string Zone { get { return "body"; } }
 
         public TemplateResult PartialView(object model) {
-            return new TemplateResult(model, null, Prefix);
+            return new TemplateResult(model, null, Prefix).Location(Zone);
         }
         public TemplateResult PartialView(object model, string template) {
-            return new TemplateResult(model, template, Prefix);
+            return new TemplateResult(model, template, Prefix).Location(Zone);
         }
         public TemplateResult PartialView(object model, string template, string prefix) {
-            return new TemplateResult(model, template, prefix);
+            return new TemplateResult(model, template, prefix).Location(Zone);
+        }
+    }
+
+    public abstract class AutomaticPartDriver<TPart> : PartDriver<TPart> where TPart : class, IContent {
+        protected override DriverResult Display(TPart part, string groupName, string displayType) {
+            return PartialView(part);
+        }
+        protected override DriverResult Editor(TPart part, string groupName) {
+            return PartialView(part);
+        }
+        protected override DriverResult Editor(TPart part, string groupName, IUpdateModel updater) {
+            updater.TryUpdateModel(part, Prefix, null, null);
+            return PartialView(part);
         }
     }
 
@@ -64,6 +78,8 @@ namespace Orchard.Models {
         public object Model { get; set; }
         public string TemplateName { get; set; }
         public string Prefix { get; set; }
+        public string Zone { get; set; }
+        public string Position { get; set; }
 
         public TemplateResult(object model, string templateName, string prefix) {
             Model = model;
@@ -72,11 +88,30 @@ namespace Orchard.Models {
         }
 
         public override void Apply(BuildDisplayModelContext context) {
-            context.AddDisplay(new TemplateViewModel(Model, Prefix) { TemplateName = TemplateName });
+            context.AddDisplay(new TemplateViewModel(Model, Prefix) {
+                TemplateName = TemplateName,
+                ZoneName = Zone,
+                Position = Position
+            });
         }
 
         public override void Apply(BuildEditorModelContext context) {
-            context.AddEditor(new TemplateViewModel(Model, Prefix) { TemplateName = TemplateName });
+            context.AddEditor(new TemplateViewModel(Model, Prefix) {
+                TemplateName = TemplateName,
+                ZoneName = Zone,
+                Position = Position
+            });
+        }
+
+        public TemplateResult Location(string zone) {
+            Zone = zone;
+            return this;
+        }
+
+        public TemplateResult Location(string zone, string position) {
+            Zone = zone;
+            Position = position;
+            return this;
         }
     }