blob: 0767adb02e510a608a50dbb42a1902e75440a611 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
// 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.Bind
{
using System.Collections.Generic;
using System.Linq;
using WixToolset.Core.Bind;
using WixToolset.Data;
using WixToolset.Data.Tuples;
using WixToolset.Data.WindowsInstaller;
using WixToolset.Data.WindowsInstaller.Rows;
internal class UpdateMediaSequencesCommand
{
public UpdateMediaSequencesCommand(Output output, List<FileFacade> fileFacades, Dictionary<int, MediaTuple> assignedMediaRows)
{
this.Output = output;
this.FileFacades = fileFacades;
}
private Output Output { get; }
private List<FileFacade> FileFacades { get; }
public void Execute()
{
var fileRows = new RowDictionary<FileRow>(this.Output.Tables["File"]);
var mediaRows = new RowDictionary<MediaRow>(this.Output.Tables["Media"]);
// Calculate sequence numbers and media disk id layout for all file media information objects.
if (OutputType.Module == this.Output.Type)
{
var lastSequence = 0;
// Order by Component to group the files by directory.
var optimized = this.OptimizedFileFacades();
foreach (var fileId in optimized.Select(f => f.File.File))
{
var fileRow = fileRows.Get(fileId);
fileRow.Sequence = ++lastSequence;
}
}
else
{
int lastSequence = 0;
MediaRow mediaRow = null;
Dictionary<int, List<FileFacade>> patchGroups = new Dictionary<int, List<FileFacade>>();
// sequence the non-patch-added files
var optimized = this.OptimizedFileFacades();
foreach (FileFacade facade in optimized)
{
if (null == mediaRow)
{
mediaRow = mediaRows.Get(facade.WixFile.DiskId);
if (OutputType.Patch == this.Output.Type)
{
// patch Media cannot start at zero
lastSequence = mediaRow.LastSequence;
}
}
else if (mediaRow.DiskId != facade.WixFile.DiskId)
{
mediaRow.LastSequence = lastSequence;
mediaRow = mediaRows.Get(facade.WixFile.DiskId);
}
if (0 < facade.WixFile.PatchGroup)
{
if (patchGroups.TryGetValue(facade.WixFile.PatchGroup, out var patchGroup))
{
patchGroup = new List<FileFacade>();
patchGroups.Add(facade.WixFile.PatchGroup, patchGroup);
}
patchGroup.Add(facade);
}
else
{
var fileRow = fileRows.Get(facade.File.File);
fileRow.Sequence = ++lastSequence;
}
}
if (null != mediaRow)
{
mediaRow.LastSequence = lastSequence;
mediaRow = null;
}
// sequence the patch-added files
foreach (var patchGroup in patchGroups.Values)
{
foreach (var facade in patchGroup)
{
if (null == mediaRow)
{
mediaRow = mediaRows.Get(facade.WixFile.DiskId);
}
else if (mediaRow.DiskId != facade.WixFile.DiskId)
{
mediaRow.LastSequence = lastSequence;
mediaRow = mediaRows.Get(facade.WixFile.DiskId);
}
var fileRow = fileRows.Get(facade.File.File);
fileRow.Sequence = ++lastSequence;
}
}
if (null != mediaRow)
{
mediaRow.LastSequence = lastSequence;
}
}
}
private IEnumerable<FileFacade> OptimizedFileFacades()
{
// TODO: Sort these facades even smarter by directory path and component id
// and maybe file size or file extension and other creative ideas to
// get optimal install speed out of MSI.
return this.FileFacades.OrderBy(f => f.File.Component_);
}
}
}
|
