public void Execute() { // Add the _Validation rows. if (!this.SuppressAddingValidationRows) { var validationTable = this.Output.EnsureTable(this.TableDefinitions["_Validation"]); foreach (var table in this.Output.Tables) { if (!table.Definition.Unreal) { // Add the validation rows for this table. foreach (ColumnDefinition columnDef in table.Definition.Columns) { var row = validationTable.CreateRow(null); row[0] = table.Name; row[1] = columnDef.Name; if (columnDef.Nullable) { row[2] = "Y"; } else { row[2] = "N"; } if (columnDef.MinValue.HasValue) { row[3] = columnDef.MinValue.Value; } if (columnDef.MaxValue.HasValue) { row[4] = columnDef.MaxValue.Value; } row[5] = columnDef.KeyTable; if (columnDef.KeyColumn.HasValue) { row[6] = columnDef.KeyColumn.Value; } if (ColumnCategory.Unknown != columnDef.Category) { row[7] = columnDef.Category.ToString(); } row[8] = columnDef.Possibilities; row[9] = columnDef.Description; } } } } // Set the base directory. var baseDirectory = this.IntermediateFolder; if (this.UseSubDirectory) { string filename = Path.GetFileNameWithoutExtension(this.OutputPath); baseDirectory = Path.Combine(baseDirectory, filename); // make sure the directory exists Directory.CreateDirectory(baseDirectory); } var idtDirectory = Path.Combine(baseDirectory, "_idts"); Directory.CreateDirectory(idtDirectory); try { OpenDatabase type = OpenDatabase.CreateDirect; // set special flag for patch files if (OutputType.Patch == this.Output.Type) { type |= OpenDatabase.OpenPatchFile; } #if DEBUG Console.WriteLine("Opening database at: {0}", this.OutputPath); #endif // Localize the codepage if a value was specified directly. if (-1 != this.Codepage) { this.Output.Codepage = this.Codepage; } Directory.CreateDirectory(Path.GetDirectoryName(this.OutputPath)); using (Database db = new Database(this.OutputPath, type)) { // if we're not using the default codepage, import a new one into our // database before we add any tables (or the tables would be added // with the wrong codepage). if (0 != this.Output.Codepage) { this.SetDatabaseCodepage(db, this.Output.Codepage, idtDirectory); } foreach (Table table in this.Output.Tables) { Table importTable = table; bool hasBinaryColumn = false; // Skip all unreal tables other than _Streams. if (table.Definition.Unreal && "_Streams" != table.Name) { continue; } // Do not put the _Validation table in patches, it is not needed. if (OutputType.Patch == this.Output.Type && "_Validation" == table.Name) { continue; } // The only way to import binary data is to copy it to a local subdirectory first. // To avoid this extra copying and perf hit, import an empty table with the same // definition and later import the binary data from source using records. foreach (ColumnDefinition columnDefinition in table.Definition.Columns) { if (ColumnType.Object == columnDefinition.Type) { importTable = new Table(table.Definition); hasBinaryColumn = true; break; } } // Create the table via IDT import. if ("_Streams" != importTable.Name) { try { var command = new CreateIdtFileCommand(this.Messaging, importTable, this.Output.Codepage, idtDirectory, this.KeepAddedColumns); command.Execute(); var buildOutput = this.BackendHelper.TrackFile(command.IdtPath, TrackedFileType.Temporary); this.GeneratedTemporaryFiles.Add(buildOutput); db.Import(command.IdtPath); } catch (WixInvalidIdtException) { // If ValidateRows finds anything it doesn't like, it throws importTable.ValidateRows(); // Otherwise we rethrow the InvalidIdt throw; } } // insert the rows via SQL query if this table contains object fields if (hasBinaryColumn) { StringBuilder query = new StringBuilder("SELECT "); // Build the query for the view. bool firstColumn = true; foreach (ColumnDefinition columnDefinition in table.Definition.Columns) { if (!firstColumn) { query.Append(","); } query.AppendFormat(" `{0}`", columnDefinition.Name); firstColumn = false; } query.AppendFormat(" FROM `{0}`", table.Name); using (View tableView = db.OpenExecuteView(query.ToString())) { // Import each row containing a stream foreach (Row row in table.Rows) { using (Record record = new Record(table.Definition.Columns.Length)) { StringBuilder streamName = new StringBuilder(); bool needStream = false; // the _Streams table doesn't prepend the table name (or a period) if ("_Streams" != table.Name) { streamName.Append(table.Name); } for (int i = 0; i < table.Definition.Columns.Length; i++) { ColumnDefinition columnDefinition = table.Definition.Columns[i]; switch (columnDefinition.Type) { case ColumnType.Localized: case ColumnType.Preserved: case ColumnType.String: if (columnDefinition.PrimaryKey) { if (0 < streamName.Length) { streamName.Append("."); } streamName.Append((string)row[i]); } record.SetString(i + 1, (string)row[i]); break; case ColumnType.Number: record.SetInteger(i + 1, Convert.ToInt32(row[i], CultureInfo.InvariantCulture)); break; case ColumnType.Object: if (null != row[i]) { needStream = true; try { record.SetStream(i + 1, (string)row[i]); } catch (Win32Exception e) { if (0xA1 == e.NativeErrorCode) // ERROR_BAD_PATHNAME { throw new WixException(ErrorMessages.FileNotFound(row.SourceLineNumbers, (string)row[i])); } else { throw new WixException(ErrorMessages.Win32Exception(e.NativeErrorCode, e.Message)); } } } break; } } // stream names are created by concatenating the name of the table with the values // of the primary key (delimited by periods) // check for a stream name that is more than 62 characters long (the maximum allowed length) if (needStream && MsiInterop.MsiMaxStreamNameLength < streamName.Length) { this.Messaging.Write(ErrorMessages.StreamNameTooLong(row.SourceLineNumbers, table.Name, streamName.ToString(), streamName.Length)); } else // add the row to the database { tableView.Modify(ModifyView.Assign, record); } } } } // Remove rows from the _Streams table for wixpdbs. if ("_Streams" == table.Name) { table.Rows.Clear(); } } } // Insert substorages (usually transforms inside a patch or instance transforms in a package). if (0 < this.Output.SubStorages.Count) { using (View storagesView = new View(db, "SELECT `Name`, `Data` FROM `_Storages`")) { foreach (SubStorage subStorage in this.Output.SubStorages) { string transformFile = Path.Combine(this.IntermediateFolder, String.Concat(subStorage.Name, ".mst")); // Bind the transform. this.BindTransform(subStorage.Data, transformFile); if (this.Messaging.EncounteredError) { continue; } // add the storage using (Record record = new Record(2)) { record.SetString(1, subStorage.Name); record.SetStream(2, transformFile); storagesView.Modify(ModifyView.Assign, record); } } } } // We're good, commit the changes to the new database. db.Commit(); } } catch (IOException e) { // TODO: this error message doesn't seem specific enough throw new WixException(ErrorMessages.FileNotFound(new SourceLineNumber(this.OutputPath), this.OutputPath), e); } }
private void ImportTables(Database db, string idtDirectory) { foreach (var table in this.Data.Tables) { var importTable = table; var hasBinaryColumn = false; // Skip all unreal tables other than _Streams. if (table.Definition.Unreal && "_Streams" != table.Name) { continue; } // Do not put the _Validation table in patches, it is not needed. if (OutputType.Patch == this.Data.Type && "_Validation" == table.Name) { continue; } // The only way to import binary data is to copy it to a local subdirectory first. // To avoid this extra copying and perf hit, import an empty table with the same // definition and later import the binary data from source using records. foreach (var columnDefinition in table.Definition.Columns) { if (ColumnType.Object == columnDefinition.Type) { importTable = new Table(table.Definition); hasBinaryColumn = true; break; } } // Create the table via IDT import. if ("_Streams" != importTable.Name) { try { var command = new CreateIdtFileCommand(this.Messaging, importTable, this.Data.Codepage, idtDirectory, this.KeepAddedColumns); command.Execute(); var trackIdt = this.BackendHelper.TrackFile(command.IdtPath, TrackedFileType.Temporary); this.GeneratedTemporaryFiles.Add(trackIdt); db.Import(command.IdtPath); } catch (WixInvalidIdtException) { // If ValidateRows finds anything it doesn't like, it throws importTable.ValidateRows(); // Otherwise we rethrow the InvalidIdt throw; } } // insert the rows via SQL query if this table contains object fields if (hasBinaryColumn) { var query = new StringBuilder("SELECT "); // Build the query for the view. var firstColumn = true; foreach (var columnDefinition in table.Definition.Columns) { if (columnDefinition.Unreal) { continue; } if (!firstColumn) { query.Append(","); } query.AppendFormat(" `{0}`", columnDefinition.Name); firstColumn = false; } query.AppendFormat(" FROM `{0}`", table.Name); using (var tableView = db.OpenExecuteView(query.ToString())) { // Import each row containing a stream foreach (var row in table.Rows) { using (var record = new Record(table.Definition.Columns.Length)) { // Stream names are created by concatenating the name of the table with the values // of the primary key (delimited by periods). var streamName = new StringBuilder(); // the _Streams table doesn't prepend the table name (or a period) if ("_Streams" != table.Name) { streamName.Append(table.Name); } var needStream = false; for (var i = 0; i < table.Definition.Columns.Length; i++) { var columnDefinition = table.Definition.Columns[i]; if (columnDefinition.Unreal) { continue; } switch (columnDefinition.Type) { case ColumnType.Localized: case ColumnType.Preserved: case ColumnType.String: var str = row.FieldAsString(i); if (columnDefinition.PrimaryKey) { if (0 < streamName.Length) { streamName.Append("."); } streamName.Append(str); } record.SetString(i + 1, str); break; case ColumnType.Number: record.SetInteger(i + 1, row.FieldAsInteger(i)); break; case ColumnType.Object: var path = row.FieldAsString(i); if (null != path) { needStream = true; try { record.SetStream(i + 1, path); } catch (Win32Exception e) { if (0xA1 == e.NativeErrorCode) // ERROR_BAD_PATHNAME { throw new WixException(ErrorMessages.FileNotFound(row.SourceLineNumbers, path)); } else { throw new WixException(ErrorMessages.Win32Exception(e.NativeErrorCode, e.Message)); } } } break; } } // check for a stream name that is more than 62 characters long (the maximum allowed length) if (needStream && Database.MsiMaxStreamNameLength < streamName.Length) { this.Messaging.Write(ErrorMessages.StreamNameTooLong(row.SourceLineNumbers, table.Name, streamName.ToString(), streamName.Length)); } else // add the row to the database { tableView.Modify(ModifyView.Assign, record); } } } } // Remove rows from the _Streams table for wixpdbs. if ("_Streams" == table.Name) { table.Rows.Clear(); } } } }