From fd95b7284f6301c591b3b2c4376e68526b78860b Mon Sep 17 00:00:00 2001
From: Nathan Heskew <nathan.heskew@microsoft.com>
Date: Wed, 7 Jul 2010 12:29:30 -0700
Subject: [PATCH] Getting at available parts (to hook on to content types)
 differently to include appropriate code- and user-defined parts

--HG--
branch : dev
---
 .../Services/ContentDefinitionService.cs            | 13 +++++++++++--
 .../Views/Admin/AddPartsTo.ascx                     |  2 +-
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/src/Orchard.Web/Modules/Orchard.ContentTypes/Services/ContentDefinitionService.cs b/src/Orchard.Web/Modules/Orchard.ContentTypes/Services/ContentDefinitionService.cs
index 07fded3a2..2667690bd 100644
--- a/src/Orchard.Web/Modules/Orchard.ContentTypes/Services/ContentDefinitionService.cs
+++ b/src/Orchard.Web/Modules/Orchard.ContentTypes/Services/ContentDefinitionService.cs
@@ -12,16 +12,20 @@ using Orchard.Localization;
 namespace Orchard.ContentTypes.Services {
     public class ContentDefinitionService : IContentDefinitionService {
         private readonly IContentDefinitionManager _contentDefinitionManager;
+        private readonly IEnumerable<IContentPartDriver> _contentPartDrivers;
         private readonly IEnumerable<IContentFieldDriver> _contentFieldDrivers;
         private readonly IContentDefinitionEditorEvents _contentDefinitionEditorEvents;
 
         public ContentDefinitionService(
                 IOrchardServices services,
                 IContentDefinitionManager contentDefinitionManager,
+                IEnumerable<IContentPartDriver> contentPartDrivers,
                 IEnumerable<IContentFieldDriver> contentFieldDrivers,
-                IContentDefinitionEditorEvents contentDefinitionEditorEvents) {
+                IContentDefinitionEditorEvents contentDefinitionEditorEvents)
+        {
             Services = services;
             _contentDefinitionManager = contentDefinitionManager;
+            _contentPartDrivers = contentPartDrivers;
             _contentFieldDrivers = contentFieldDrivers;
             _contentDefinitionEditorEvents = contentDefinitionEditorEvents;
             T = NullLocalizer.Instance;
@@ -141,7 +145,12 @@ namespace Orchard.ContentTypes.Services {
 
         public IEnumerable<EditPartViewModel> GetParts() {
             var typeNames = GetTypes().Select(ctd => ctd.Name);
-            return _contentDefinitionManager.ListPartDefinitions().Where(cpd => !typeNames.Contains(cpd.Name)).Select(cpd => new EditPartViewModel(cpd));
+            // code-defined parts
+            var codeDefinedParts = _contentPartDrivers.SelectMany(d => d.GetPartInfo().Select(cpi => new EditPartViewModel {Name = cpi.PartName}));
+            // user-defined parts
+            var contentParts = _contentDefinitionManager.ListPartDefinitions().Where(cpd => !codeDefinedParts.Any(m => m.Name == cpd.Name)).Select(cpd => new EditPartViewModel(cpd));
+            // all together now, except for those parts with the same name as a type (implicit type's part or a mistake)
+            return contentParts.Where(m => !typeNames.Contains(m.Name)).Union(codeDefinedParts).OrderBy(m => m.Name);
         }
 
         public EditPartViewModel GetPart(string name) {
diff --git a/src/Orchard.Web/Modules/Orchard.ContentTypes/Views/Admin/AddPartsTo.ascx b/src/Orchard.Web/Modules/Orchard.ContentTypes/Views/Admin/AddPartsTo.ascx
index 7f70491e7..b9055c5fd 100644
--- a/src/Orchard.Web/Modules/Orchard.ContentTypes/Views/Admin/AddPartsTo.ascx
+++ b/src/Orchard.Web/Modules/Orchard.ContentTypes/Views/Admin/AddPartsTo.ascx
@@ -13,7 +13,7 @@ using (Html.BeginFormAntiForgeryPost()) { %>
                     string.Format(
                         "{0} <label for=\"{1}\" class=\"forcheckbox\">{2}</label>{3}",
                         Html.CheckBox(fieldNameStart + "IsSelected"),
-                        fieldNameStart + "IsSelected",
+                        ViewData.TemplateInfo.GetFullHtmlFieldId(fieldNameStart + "IsSelected"),
                         partSelection.PartName,
                         Html.Hidden(fieldNameStart + "PartName", partSelection.PartName)));
             },