From 51fe8fecb94ea9bac11d159b0b9ad0e1d192e0d8 Mon Sep 17 00:00:00 2001
From: Sipke Schoorstra <sipke@ideliverable.com>
Date: Mon, 12 Oct 2015 21:37:04 +0200
Subject: [PATCH] Added ImportCompleted event to field drivers too.

---
 .../ContentManagement/Drivers/ContentFieldDriver.cs        | 5 +++++
 .../Drivers/Coordinators/ContentFieldDriverCoordinator.cs  | 7 +++++++
 .../ContentManagement/Drivers/IContentFieldDriver.cs       | 1 +
 3 files changed, 13 insertions(+)

diff --git a/src/Orchard/ContentManagement/Drivers/ContentFieldDriver.cs b/src/Orchard/ContentManagement/Drivers/ContentFieldDriver.cs
index 14f778549..73be31461 100644
--- a/src/Orchard/ContentManagement/Drivers/ContentFieldDriver.cs
+++ b/src/Orchard/ContentManagement/Drivers/ContentFieldDriver.cs
@@ -63,6 +63,10 @@ namespace Orchard.ContentManagement.Drivers {
             Process(context.ContentItem, (part, field) => Imported(part, field, context), context.Logger);
         }
 
+        void IContentFieldDriver.ImportCompleted(ImportContentContext context) {
+            Process(context.ContentItem, (part, field) => ImportCompleted(part, field, context), context.Logger);
+        }
+
         void IContentFieldDriver.Exporting(ExportContentContext context) {
             Process(context.ContentItem, (part, field) => Exporting(part, field, context), context.Logger);
         }
@@ -110,6 +114,7 @@ namespace Orchard.ContentManagement.Drivers {
         
         protected virtual void Importing(ContentPart part, TField field, ImportContentContext context) { }
         protected virtual void Imported(ContentPart part, TField field, ImportContentContext context) { }
+        protected virtual void ImportCompleted(ContentPart part, TField field, ImportContentContext context) { }
         protected virtual void Exporting(ContentPart part, TField field, ExportContentContext context) { }
         protected virtual void Exported(ContentPart part, TField field, ExportContentContext context) { }
 
diff --git a/src/Orchard/ContentManagement/Drivers/Coordinators/ContentFieldDriverCoordinator.cs b/src/Orchard/ContentManagement/Drivers/Coordinators/ContentFieldDriverCoordinator.cs
index 8ffb8ec62..375efd779 100644
--- a/src/Orchard/ContentManagement/Drivers/Coordinators/ContentFieldDriverCoordinator.cs
+++ b/src/Orchard/ContentManagement/Drivers/Coordinators/ContentFieldDriverCoordinator.cs
@@ -89,6 +89,13 @@ namespace Orchard.ContentManagement.Drivers.Coordinators {
             }
         }
 
+        public override void ImportCompleted(ImportContentContext context) {
+            context.Logger = Logger;
+            foreach (var contentFieldDriver in _drivers) {
+                contentFieldDriver.ImportCompleted(context);
+            }
+        }
+
         public override void Exporting(ExportContentContext context) {
             context.Logger = Logger;
             foreach (var contentFieldDriver in _drivers.OrderBy(x => x.GetFieldInfo().First().FieldTypeName)) {
diff --git a/src/Orchard/ContentManagement/Drivers/IContentFieldDriver.cs b/src/Orchard/ContentManagement/Drivers/IContentFieldDriver.cs
index a5fecef78..138c412ce 100644
--- a/src/Orchard/ContentManagement/Drivers/IContentFieldDriver.cs
+++ b/src/Orchard/ContentManagement/Drivers/IContentFieldDriver.cs
@@ -9,6 +9,7 @@ namespace Orchard.ContentManagement.Drivers {
         DriverResult UpdateEditorShape(UpdateEditorContext context);
         void Importing(ImportContentContext context);
         void Imported(ImportContentContext context);
+        void ImportCompleted(ImportContentContext context);
         void Exporting(ExportContentContext context);
         void Exported(ExportContentContext context);
         void Describe(DescribeMembersContext context);