/// <summary>
/// Harvest a directory.
/// </summary>
/// <param name="argument">The path of the directory.</param>
/// <returns>The harvested directory.</returns>
public override Wix.Fragment[] Harvest(string argument)
{
if (null == argument)
{
throw new ArgumentNullException("argument");
}
Wix.Directory directory = this.HarvestDirectory(argument, "SourceDir\\", true);
Wix.DirectoryRef directoryRef = new Wix.DirectoryRef();
directoryRef.Id = this.rootedDirectoryRef;
if (this.suppressRootDirectory)
{
foreach (Wix.ISchemaElement element in directory.Children)
{
directoryRef.AddChild(element);
}
}
else
{
directoryRef.AddChild(directory);
}
Wix.Fragment fragment = new Wix.Fragment();
fragment.AddChild(directoryRef);
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>
/// Mutate the directories.
/// </summary>
private void MutateDirectories()
{
if (!this.setUniqueIdentifiers)
{
// assign all identifiers before fragmenting (because fragmenting requires them all to be present)
IdentifierGenerator identifierGenerator = new IdentifierGenerator("Directory");
if (TemplateType.Module == this.templateType)
{
identifierGenerator.MaxIdentifierLength = IdentifierGenerator.MaxModuleIdentifierLength;
}
foreach (Wix.Directory directory in this.directories)
{
if (null == directory.Id)
{
directory.Id = identifierGenerator.GetIdentifier(directory.Name);
}
}
}
if (this.createFragments)
{
foreach (Wix.Directory directory in this.directories)
{
if (directory.ParentElement is Wix.Directory)
{
Wix.Directory parentDirectory = (Wix.Directory)directory.ParentElement;
// parent directory must have an identifier to create a reference to it
if (null == parentDirectory.Id)
{
return;
}
// create a new Fragment
Wix.Fragment fragment = new Wix.Fragment();
this.fragments.Add(String.Concat("Directory:", ("TARGETDIR" == directory.Id ? null : (null != directory.Id ? directory.Id : this.fragments.Count.ToString()))), fragment);
// create a new DirectoryRef
Wix.DirectoryRef directoryRef = new Wix.DirectoryRef();
directoryRef.Id = parentDirectory.Id;
fragment.AddChild(directoryRef);
// move the Directory from the parent Directory to DirectoryRef
parentDirectory.RemoveChild(directory);
directoryRef.AddChild(directory);
}
else if (directory.ParentElement is Wix.Fragment)
{
// When creating fragments, remove any top-level Directory elements;
// the fragments should be pulled in by their DirectoryRefs instead.
Wix.Fragment parent = (Wix.Fragment)directory.ParentElement;
parent.RemoveChild(directory);
// Remove the fragment if it is empty.
if (parent.Children.GetEnumerator().Current == null && parent.ParentElement != null)
{
((Wix.IParentElement)parent.ParentElement).RemoveChild(parent);
}
}
else if (directory.ParentElement == this.rootElement)
{
// create a new Fragment
Wix.Fragment fragment = new Wix.Fragment();
this.fragments.Add(String.Concat("Directory:", ("TARGETDIR" == directory.Id ? null : (null != directory.Id ? directory.Id : this.fragments.Count.ToString()))), fragment);
// move the Directory from the root element to the Fragment
this.rootElement.RemoveChild(directory);
fragment.AddChild(directory);
}
}
}
}
/// <summary>
/// Mutate the components.
/// </summary>
private void MutateComponents()
{
IdentifierGenerator identifierGenerator = new IdentifierGenerator("Component");
if (TemplateType.Module == this.templateType)
{
identifierGenerator.MaxIdentifierLength = IdentifierGenerator.MaxModuleIdentifierLength;
}
foreach (Wix.Component component in this.components)
{
if (null == component.Id)
{
string firstFileId = string.Empty;
// attempt to create a possible identifier from the first file identifier in the component
foreach (Wix.File file in component[typeof(Wix.File)])
{
firstFileId = file.Id;
break;
}
if (string.IsNullOrEmpty(firstFileId))
{
firstFileId = GetGuid();
}
component.Id = identifierGenerator.GetIdentifier(firstFileId);
}
if (null == component.Guid)
{
if (this.AutogenerateGuids)
{
component.Guid = "*";
}
else
{
component.Guid = this.GetGuid();
}
}
if (this.createFragments && component.ParentElement is Wix.Directory)
{
Wix.Directory directory = (Wix.Directory)component.ParentElement;
// parent directory must have an identifier to create a reference to it
if (null == directory.Id)
{
break;
}
if (this.rootElement is Wix.Module)
{
// add a ComponentRef for the Component
Wix.ComponentRef componentRef = new Wix.ComponentRef();
componentRef.Id = component.Id;
this.rootElement.AddChild(componentRef);
}
// create a new Fragment
Wix.Fragment fragment = new Wix.Fragment();
this.fragments.Add(String.Concat("Component:", (null != component.Id ? component.Id : this.fragments.Count.ToString())), fragment);
// create a new DirectoryRef
Wix.DirectoryRef directoryRef = new Wix.DirectoryRef();
directoryRef.Id = directory.Id;
fragment.AddChild(directoryRef);
// move the Component from the the Directory to the DirectoryRef
directory.RemoveChild(component);
directoryRef.AddChild(component);
}
}
}