1 files changed, 11 insertions, 41 deletions

diff --git a/src/WixToolset.Core/Linker.cs b/src/WixToolset.Core/Linker.cs
index 7faf69ba..c893a01d 100644
--- a/src/WixToolset.Core/Linker.cs
+++ b/src/WixToolset.Core/Linker.cs
@@ -56,32 +56,6 @@ namespace WixToolset.Core
         //internal IBindVariableResolver WixVariableResolver { get; set; }
 
         /// <summary>
-        /// Adds an extension.
-        /// </summary>
-        /// <param name="extension">The extension to add.</param>
-        //public void AddExtensionData(IExtensionData extension)
-        //{
-        //    if (null != extension.TableDefinitions)
-        //    {
-        //        foreach (TableDefinition tableDefinition in extension.TableDefinitions)
-        //        {
-        //            if (!this.tableDefinitions.Contains(tableDefinition.Name))
-        //            {
-        //                this.tableDefinitions.Add(tableDefinition);
-        //            }
-        //            else
-        //            {
-        //                throw new WixException(WixErrors.DuplicateExtensionTable(extension.GetType().ToString(), tableDefinition.Name));
-        //            }
-        //        }
-        //    }
-
-        //    // keep track of extension data so the libraries can be loaded from these later once all the table definitions
-        //    // are loaded; this will allow extensions to have cross table definition dependencies
-        //    this.extensionData.Add(extension);
-        //}
-
-        /// <summary>
         /// Links a collection of sections into an output.
         /// </summary>
         /// <param name="inputs">The collection of sections to link together.</param>
@@ -91,10 +65,19 @@ namespace WixToolset.Core
         {
             this.Context = context ?? throw new ArgumentNullException(nameof(context));
 
-            //IEnumerable<Section> inputs, OutputType expectedOutputType
-
             var sections = this.Context.Intermediates.SelectMany(i => i.Sections).ToList();
 
+            // Add sections from the extensions with data.
+            foreach (var data in context.ExtensionData)
+            {
+                var library = data.GetLibrary(context.TupleDefinitionCreator);
+
+                if (library != null)
+                {
+                    sections.AddRange(library.Sections);
+                }
+            }
+
 #if MOVE_TO_BACKEND
                 bool containsModuleSubstitution = false;
                 bool containsModuleConfiguration = false;
@@ -144,19 +127,6 @@ namespace WixToolset.Core
                 }
 #endif
 
-#if TODO
-                // Add sections from the extensions with data.
-                foreach (IExtensionData data in this.extensionData)
-                {
-                    Library library = data.GetLibrary(this.tableDefinitions);
-
-                    if (null != library)
-                    {
-                        sections.AddRange(library.Sections);
-                    }
-                }
-#endif
-
             // First find the entry section and while processing all sections load all the symbols from all of the sections.
             // sections.FindEntrySectionAndLoadSymbols(false, this, expectedOutputType, out entrySection, out allSymbols);
             var find = new FindEntrySectionAndLoadSymbolsCommand(sections);