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
|
// 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.Bind.Bundles
{
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using WixToolset.Data;
using WixToolset.Data.Rows;
internal class AutomaticallySlipstreamPatchesCommand : ICommand
{
public IEnumerable<PackageFacade> PackageFacades { private get; set; }
public Table WixBundlePatchTargetCodeTable { private get; set; }
public Table SlipstreamMspTable { private get; set; }
public void Execute()
{
List<WixBundleMsiPackageRow> msiPackages = new List<WixBundleMsiPackageRow>();
Dictionary<string, List<WixBundlePatchTargetCodeRow>> targetsProductCode = new Dictionary<string, List<WixBundlePatchTargetCodeRow>>();
Dictionary<string, List<WixBundlePatchTargetCodeRow>> targetsUpgradeCode = new Dictionary<string, List<WixBundlePatchTargetCodeRow>>();
foreach (PackageFacade facade in this.PackageFacades)
{
if (WixBundlePackageType.Msi == facade.Package.Type)
{
// Keep track of all MSI packages.
msiPackages.Add(facade.MsiPackage);
}
else if (WixBundlePackageType.Msp == facade.Package.Type && facade.MspPackage.Slipstream)
{
IEnumerable<WixBundlePatchTargetCodeRow> patchTargetCodeRows = this.WixBundlePatchTargetCodeTable.RowsAs<WixBundlePatchTargetCodeRow>().Where(r => r.MspPackageId == facade.Package.WixChainItemId);
// Index target ProductCodes and UpgradeCodes for slipstreamed MSPs.
foreach (WixBundlePatchTargetCodeRow row in patchTargetCodeRows)
{
if (row.TargetsProductCode)
{
List<WixBundlePatchTargetCodeRow> rows;
if (!targetsProductCode.TryGetValue(row.TargetCode, out rows))
{
rows = new List<WixBundlePatchTargetCodeRow>();
targetsProductCode.Add(row.TargetCode, rows);
}
rows.Add(row);
}
else if (row.TargetsUpgradeCode)
{
List<WixBundlePatchTargetCodeRow> rows;
if (!targetsUpgradeCode.TryGetValue(row.TargetCode, out rows))
{
rows = new List<WixBundlePatchTargetCodeRow>();
targetsUpgradeCode.Add(row.TargetCode, rows);
}
}
}
}
}
RowIndexedList<Row> slipstreamMspRows = new RowIndexedList<Row>(SlipstreamMspTable);
// Loop through the MSI and slipstream patches targeting it.
foreach (WixBundleMsiPackageRow msi in msiPackages)
{
List<WixBundlePatchTargetCodeRow> rows;
if (targetsProductCode.TryGetValue(msi.ProductCode, out rows))
{
foreach (WixBundlePatchTargetCodeRow row in rows)
{
Debug.Assert(row.TargetsProductCode);
Debug.Assert(!row.TargetsUpgradeCode);
Row slipstreamMspRow = SlipstreamMspTable.CreateRow(row.SourceLineNumbers, false);
slipstreamMspRow[0] = msi.ChainPackageId;
slipstreamMspRow[1] = row.MspPackageId;
if (slipstreamMspRows.TryAdd(slipstreamMspRow))
{
SlipstreamMspTable.Rows.Add(slipstreamMspRow);
}
}
rows = null;
}
if (!String.IsNullOrEmpty(msi.UpgradeCode) && targetsUpgradeCode.TryGetValue(msi.UpgradeCode, out rows))
{
foreach (WixBundlePatchTargetCodeRow row in rows)
{
Debug.Assert(!row.TargetsProductCode);
Debug.Assert(row.TargetsUpgradeCode);
Row slipstreamMspRow = SlipstreamMspTable.CreateRow(row.SourceLineNumbers, false);
slipstreamMspRow[0] = msi.ChainPackageId;
slipstreamMspRow[1] = row.MspPackageId;
if (slipstreamMspRows.TryAdd(slipstreamMspRow))
{
SlipstreamMspTable.Rows.Add(slipstreamMspRow);
}
}
rows = null;
}
}
}
}
}
|
