1 files changed, 80 insertions, 0 deletions

diff --git a/src/WixToolset.Core.WindowsInstaller/Bind/UpdateControlTextCommand.cs b/src/WixToolset.Core.WindowsInstaller/Bind/UpdateControlTextCommand.cs
new file mode 100644
index 00000000..7da32206
--- /dev/null
+++ b/src/WixToolset.Core.WindowsInstaller/Bind/UpdateControlTextCommand.cs
@@ -0,0 +1,80 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+namespace WixToolset.Core.WindowsInstaller.Databases
+{
+    using System;
+    using System.IO;
+    using WixToolset.Data;
+    using WixToolset.Data.Rows;
+
+    internal class UpdateControlTextCommand
+    {
+        public Table BBControlTable { private get; set; }
+
+        public Table WixBBControlTable { private get; set; }
+
+        public Table ControlTable { private get; set; }
+
+        public Table WixControlTable { private get; set; }
+
+        public void Execute()
+        {
+            if (null != this.WixBBControlTable)
+            {
+                RowDictionary<BBControlRow> bbControlRows = new RowDictionary<BBControlRow>(this.BBControlTable);
+                foreach (Row wixRow in this.WixBBControlTable.Rows)
+                {
+                    BBControlRow bbControlRow = bbControlRows.Get(wixRow.GetPrimaryKey());
+                    bbControlRow.Text = this.ReadTextFile(bbControlRow.SourceLineNumbers, wixRow.FieldAsString(2));
+                }
+            }
+
+            if (null != this.WixControlTable)
+            {
+                RowDictionary<ControlRow> controlRows = new RowDictionary<ControlRow>(this.ControlTable);
+                foreach (Row wixRow in this.WixControlTable.Rows)
+                {
+                    ControlRow controlRow = controlRows.Get(wixRow.GetPrimaryKey());
+                    controlRow.Text = this.ReadTextFile(controlRow.SourceLineNumbers, wixRow.FieldAsString(2));
+                }
+            }
+        }
+
+        /// <summary>
+        /// Reads a text file and returns the contents.
+        /// </summary>
+        /// <param name="sourceLineNumbers">Source line numbers for row from source.</param>
+        /// <param name="source">Source path to file to read.</param>
+        /// <returns>Text string read from file.</returns>
+        private string ReadTextFile(SourceLineNumber sourceLineNumbers, string source)
+        {
+            string text = null;
+
+            try
+            {
+                using (StreamReader reader = new StreamReader(source))
+                {
+                    text = reader.ReadToEnd();
+                }
+            }
+            catch (DirectoryNotFoundException e)
+            {
+                Messaging.Instance.OnMessage(WixErrors.BinderFileManagerMissingFile(sourceLineNumbers, e.Message));
+            }
+            catch (FileNotFoundException e)
+            {
+                Messaging.Instance.OnMessage(WixErrors.BinderFileManagerMissingFile(sourceLineNumbers, e.Message));
+            }
+            catch (IOException e)
+            {
+                Messaging.Instance.OnMessage(WixErrors.BinderFileManagerMissingFile(sourceLineNumbers, e.Message));
+            }
+            catch (NotSupportedException)
+            {
+                Messaging.Instance.OnMessage(WixErrors.FileNotFound(sourceLineNumbers, source));
+            }
+
+            return text;
+        }
+    }
+}