/// <summary>
/// Harvest a WiX document.
/// </summary>
/// <param name="argument">The argument for harvesting.</param>
/// <returns>The harvested Fragment.</returns>
public override Wix.Fragment[] Harvest(string argument)
{
DirectoryHarvester directoryHarvester = new DirectoryHarvester();
directoryHarvester.Core = this.Core;
directoryHarvester.KeepEmptyDirectories = true;
IIsWebSiteHarvester iisWebSiteHarvester = new IIsWebSiteHarvester();
iisWebSiteHarvester.Core = this.Core;
IIs.WebSite webSite = iisWebSiteHarvester.HarvestWebSite(argument);
Wix.Component component = new Wix.Component();
component.AddChild(new Wix.CreateFolder());
component.AddChild(webSite);
this.Core.RootDirectory = webSite.Directory;
Wix.Directory directory = directoryHarvester.HarvestDirectory(webSite.Directory, true);
directory.AddChild(component);
Wix.Fragment fragment = new Wix.Fragment();
fragment.AddChild(directory);
return(new Wix.Fragment[] { fragment });
}
/// <summary>
/// Harvest a file.
/// </summary>
/// <param name="argument">The path of the file.</param>
/// <returns>A harvested file.</returns>
public override Wix.Fragment[] Harvest(string argument)
{
if (null == argument)
{
throw new ArgumentNullException("argument");
}
if (null == this.rootedDirectoryRef)
{
this.rootedDirectoryRef = "TARGETDIR";
}
string fullPath = Path.GetFullPath(argument);
Wix.DirectoryRef directoryRef = new Wix.DirectoryRef();
directoryRef.Id = this.rootedDirectoryRef;
Wix.File file = this.HarvestFile(fullPath);
if (!this.suppressRootDirectory)
{
file.Source = String.Concat("SourceDir\\", Path.GetFileName(Path.GetDirectoryName(fullPath)), "\\", Path.GetFileName(fullPath));
}
Wix.Component component = new Wix.Component();
component.AddChild(file);
Wix.Directory directory = new Wix.Directory();
if (this.suppressRootDirectory)
{
directoryRef.AddChild(component);
}
else
{
string directoryPath = Path.GetDirectoryName(Path.GetFullPath(argument));
directory.Name = Path.GetFileName(directoryPath);
if (this.setUniqueIdentifiers)
{
directory.Id = this.Core.GenerateIdentifier(DirectoryPrefix, directoryRef.Id, directory.Name);
}
directory.AddChild(component);
directoryRef.AddChild(directory);
}
if (this.setUniqueIdentifiers)
{
file.Id = this.Core.GenerateIdentifier(FilePrefix, (this.suppressRootDirectory) ? directoryRef.Id : directory.Id, Path.GetFileName(file.Source));
component.Id = this.Core.GenerateIdentifier(ComponentPrefix, (this.suppressRootDirectory) ? directoryRef.Id : directory.Id, file.Id);
}
Wix.Fragment fragment = new Wix.Fragment();
fragment.AddChild(directoryRef);
return(new Wix.Fragment[] { fragment });
}
/// <summary>
/// Converts the registry key to a WiX component element.
/// </summary>
/// <param name="sr">The registry file stream.</param>
/// <param name="directory">A WiX directory reference.</param>
/// <param name="root">The root key.</param>
/// <param name="line">The current line.</param>
private void ConvertKey(StreamReader sr, ref Wix.Directory directory, Wix.RegistryRootType root, string line)
{
Wix.Component component = new Wix.Component();
component.Id = this.Core.GenerateIdentifier(ComponentPrefix, line);
component.KeyPath = Wix.YesNoType.yes;
this.ConvertValues(sr, ref component, root, line);
directory.AddChild(component);
}
/// <summary>
/// Mutate the WebFilter elements.
/// </summary>
private void MutateWebFilters()
{
IdentifierGenerator identifierGenerator = null;
if (this.setUniqueIdentifiers)
{
identifierGenerator = new IdentifierGenerator("WebFilter", this.Core);
// index all the existing identifiers and names
foreach (IIs.WebFilter webFilter in this.webFilters)
{
if (null != webFilter.Id)
{
identifierGenerator.IndexExistingIdentifier(webFilter.Id);
}
else
{
identifierGenerator.IndexName(webFilter.Name);
}
}
}
foreach (IIs.WebFilter webFilter in this.webFilters)
{
if (this.setUniqueIdentifiers && null == webFilter.Id)
{
webFilter.Id = identifierGenerator.GetIdentifier(webFilter.Name);
}
// harvest the file for this WebFilter
Wix.Directory directory = this.HarvestUniqueDirectory(Path.GetDirectoryName(webFilter.Path), false);
Wix.Component component = new Wix.Component();
directory.AddChild(component);
Wix.File file = this.fileHarvester.HarvestFile(webFilter.Path);
component.AddChild(file);
}
}
/// <summary>
/// Harvest a performance category.
/// </summary>
/// <param name="argument">The name of the performance category.</param>
/// <returns>A harvested performance category.</returns>
public override Wix.Fragment[] Harvest(string argument)
{
if (null == argument)
{
throw new ArgumentNullException("argument");
}
Util.PerformanceCategory perf = this.HarvestPerformanceCategory(argument);
Wix.Component component = new Wix.Component();
component.Id = this.Core.CreateIdentifierFromFilename(argument);
component.KeyPath = Wix.YesNoType.yes;
component.AddChild(perf);
Wix.Directory directory = new Wix.Directory();
directory.Id = "TARGETDIR";
//directory.Name = directory.Id;
directory.AddChild(component);
Wix.Fragment fragment = new Wix.Fragment();
fragment.AddChild(directory);
return(new Wix.Fragment[] { fragment });
}
/// <summary>
/// Mutate the Component elements.
/// </summary>
private void MutateComponents()
{
if (this.setUniqueIdentifiers)
{
IdentifierGenerator identifierGenerator = new IdentifierGenerator("Component", this.Core);
// index all the existing identifiers
foreach (Wix.Component component in this.components)
{
if (null != component.Id)
{
identifierGenerator.IndexExistingIdentifier(component.Id);
}
}
// index all the web site identifiers
foreach (IIs.WebSite webSite in this.webSites)
{
if (webSite.ParentElement is Wix.Component)
{
identifierGenerator.IndexName(webSite.Id);
}
}
// create an identifier for each component based on its child web site identifier
foreach (IIs.WebSite webSite in this.webSites)
{
Wix.Component component = webSite.ParentElement as Wix.Component;
if (null != component)
{
component.Id = identifierGenerator.GetIdentifier(webSite.Id);
}
}
}
}
/// <summary>
/// Converts the registry values to WiX regisry key element.
/// </summary>
/// <param name="sr">The registry file stream.</param>
/// <param name="component">A WiX component reference.</param>
/// <param name="root">The root key.</param>
/// <param name="line">The current line.</param>
private void ConvertValues(StreamReader sr, ref Wix.Component component, Wix.RegistryRootType root, string line)
{
string name = null;
string value = null;
Wix.RegistryValue.TypeType type;
Wix.RegistryKey registryKey = new Wix.RegistryKey();
registryKey.Root = root;
registryKey.Key = line;
while (this.GetValue(sr, ref name, ref value, out type))
{
Wix.RegistryValue registryValue = new Wix.RegistryValue();
ArrayList charArray;
// Don't specifiy name for default attribute
if (!string.IsNullOrEmpty(name))
{
registryValue.Name = name;
}
registryValue.Type = type;
switch (type)
{
case Wix.RegistryValue.TypeType.binary:
registryValue.Value = value.Replace(",", string.Empty).ToUpper();
break;
case Wix.RegistryValue.TypeType.integer:
registryValue.Value = Int32.Parse(value, NumberStyles.HexNumber).ToString();
break;
case Wix.RegistryValue.TypeType.expandable:
charArray = this.ConvertCharList(value);
value = string.Empty;
// create the string, remove the terminating null
for (int i = 0; i < charArray.Count; i++)
{
if ('\0' != (char)charArray[i])
{
value += charArray[i];
}
}
registryValue.Value = value;
break;
case Wix.RegistryValue.TypeType.multiString:
charArray = this.ConvertCharList(value);
value = string.Empty;
// Convert the character array to a string so we can simply split it at the nulls, ignore the final null null.
for (int i = 0; i < (charArray.Count - 2); i++)
{
value += charArray[i];
}
// Although the value can use [~] the preffered way is to use MultiStringValue
string[] parts = value.Split("\0".ToCharArray());
foreach (string part in parts)
{
Wix.MultiStringValue multiStringValue = new Wix.MultiStringValue();
multiStringValue.Content = part;
registryValue.AddChild(multiStringValue);
}
break;
case Wix.RegistryValue.TypeType.@string:
// Remove \\ and \"
value = value.ToString().Replace("\\\"", "\"");
value = value.ToString().Replace(@"\\", @"\");
// Escape [ and ]
value = value.ToString().Replace(@"[", @"[\[]");
value = value.ToString().Replace(@"]", @"[\]]");
// This undoes the duplicate escaping caused by the second replace
value = value.ToString().Replace(@"[\[[\]]", @"[\[]");
// Escape $
value = value.ToString().Replace(@"$", @"$$");
registryValue.Value = value;
break;
default:
throw new ApplicationException(String.Format("Did not recognize the type of reg value on line {0}", this.currentLineNumber));
}
registryKey.AddChild(registryValue);
}
// Make sure empty keys are created
if (null == value)
{
registryKey.ForceCreateOnInstall = Wix.YesNoType.yes;
}
component.AddChild(registryKey);
}
/// <summary>
/// Harvest a directory.
/// </summary>
/// <param name="path">The path of the directory.</param>
/// <param name="relativePath">The relative path that will be used when harvesting.</param>
/// <param name="harvestParent">The directory for this path.</param>
/// <param name="generateType"></param>
/// <returns>The number of files harvested.</returns>
private int HarvestDirectory(string path, string relativePath, Wix.IParentElement harvestParent, GenerateType generateType)
{
int fileCount = 0;
Wix.Directory directory = generateType != GenerateType.PayloadGroup ? (Wix.Directory)harvestParent : null;
// harvest the child directories
foreach (string childDirectoryPath in Directory.GetDirectories(path))
{
var childDirectoryName = Path.GetFileName(childDirectoryPath);
Wix.IParentElement newParent;
Wix.Directory childDirectory = null;
if (generateType == GenerateType.PayloadGroup)
{
newParent = harvestParent;
}
else
{
childDirectory = new Wix.Directory();
newParent = childDirectory;
childDirectory.Name = childDirectoryName;
childDirectory.FileSource = childDirectoryPath;
if (this.SetUniqueIdentifiers)
{
childDirectory.Id = this.Core.GenerateIdentifier(DirectoryPrefix, directory.Id, childDirectory.Name);
}
}
int childFileCount = this.HarvestDirectory(childDirectoryPath, String.Concat(relativePath, childDirectoryName, "\\"), newParent, generateType);
if (generateType != GenerateType.PayloadGroup)
{
// keep the directory if it contained any files (or empty directories are being kept)
if (0 < childFileCount || this.KeepEmptyDirectories)
{
directory.AddChild(childDirectory);
}
}
fileCount += childFileCount;
}
// harvest the files
string[] files = Directory.GetFiles(path);
if (0 < files.Length)
{
foreach (string filePath in Directory.GetFiles(path))
{
string fileName = Path.GetFileName(filePath);
string source = String.Concat(relativePath, fileName);
Wix.ISchemaElement newChild;
if (generateType == GenerateType.PayloadGroup)
{
Wix.Payload payload = new Wix.Payload();
newChild = payload;
payload.SourceFile = source;
}
else
{
Wix.Component component = new Wix.Component();
newChild = component;
Wix.File file = this.fileHarvester.HarvestFile(filePath);
file.Source = source;
if (this.SetUniqueIdentifiers)
{
file.Id = this.Core.GenerateIdentifier(FilePrefix, directory.Id, fileName);
component.Id = this.Core.GenerateIdentifier(ComponentPrefix, directory.Id, file.Id);
}
component.AddChild(file);
}
harvestParent.AddChild(newChild);
}
}
else if (generateType != GenerateType.PayloadGroup && 0 == fileCount && this.KeepEmptyDirectories)
{
Wix.Component component = new Wix.Component();
component.KeyPath = Wix.YesNoType.yes;
if (this.SetUniqueIdentifiers)
{
component.Id = this.Core.GenerateIdentifier(ComponentPrefix, directory.Id);
}
Wix.CreateFolder createFolder = new Wix.CreateFolder();
component.AddChild(createFolder);
directory.AddChild(component);
}
return(fileCount + files.Length);
}
/// <summary>
/// Creates a component group with a given name.
/// </summary>
/// <param name="wix">The Wix document element.</param>
private void CreateComponentGroup(Wix.Wix wix)
{
Wix.ComponentGroup componentGroup = new Wix.ComponentGroup();
componentGroup.Id = this.componentGroupName;
this.componentGroups.Add(componentGroup);
Wix.Fragment cgFragment = new Wix.Fragment();
cgFragment.AddChild(componentGroup);
wix.AddChild(cgFragment);
int componentCount = 0;
for (; componentCount < this.components.Count; componentCount++)
{
Wix.Component c = this.components[componentCount] as Wix.Component;
if (this.createFragments)
{
if (c.ParentElement is Wix.Directory)
{
Wix.Directory parentDirectory = c.ParentElement as Wix.Directory;
componentGroup.AddChild(c);
c.Directory = parentDirectory.Id;
parentDirectory.RemoveChild(c);
}
else if (c.ParentElement is Wix.DirectoryRef)
{
Wix.DirectoryRef parentDirectory = c.ParentElement as Wix.DirectoryRef;
componentGroup.AddChild(c);
c.Directory = parentDirectory.Id;
parentDirectory.RemoveChild(c);
// Remove whole fragment if moving the component to the component group just leaves an empty DirectoryRef
if (0 < this.fragments.Count && parentDirectory.ParentElement is Wix.Fragment)
{
Wix.Fragment parentFragment = parentDirectory.ParentElement as Wix.Fragment;
int childCount = 0;
foreach (Wix.ISchemaElement element in parentFragment.Children)
{
childCount++;
}
// Component should always have an Id but the SortedList creation allows for null and bases the name on the fragment count which we cannot reverse engineer here.
if (1 == childCount && !String.IsNullOrEmpty(c.Id))
{
int removeIndex = this.fragments.IndexOfKey(String.Concat("Component:", c.Id));
if (0 <= removeIndex)
{
this.fragments.RemoveAt(removeIndex);
}
}
}
}
}
else
{
Wix.ComponentRef componentRef = new Wix.ComponentRef();
componentRef.Id = c.Id;
componentGroup.AddChild(componentRef);
}
}
}
