123 lines
3.4 KiB
C#
123 lines
3.4 KiB
C#
/***************************************************************************
|
|
|
|
Copyright (c) Microsoft Corporation 2011.
|
|
|
|
This code is licensed using the Microsoft Public License (Ms-PL). The text of the license can be found here:
|
|
|
|
http://www.microsoft.com/resources/sharedsource/licensingbasics/publiclicense.mspx
|
|
|
|
***************************************************************************/
|
|
|
|
using System;
|
|
using System.Linq;
|
|
using System.Xml.Linq;
|
|
using System.Management.Automation;
|
|
|
|
namespace OpenXmlPowerTools.Commands
|
|
{
|
|
/// <summary>
|
|
/// Set the header files of a word document
|
|
/// </summary>
|
|
[Cmdlet(VerbsCommon.Set, "OpenXmlHeader", SupportsShouldProcess = true)]
|
|
[OutputType("OpenXmlPowerToolsDocument")]
|
|
public class SetOpenXmlHeaderCmdlet : PowerToolsModifierCmdlet
|
|
{
|
|
|
|
#region Parameters
|
|
|
|
private string headerPath;
|
|
private XDocument header;
|
|
private HeaderType kind;
|
|
|
|
[Parameter(Position = 2,
|
|
Mandatory = false,
|
|
HelpMessage = "The path of the header to add in the document")
|
|
]
|
|
[ValidateNotNullOrEmpty]
|
|
public string HeaderPath
|
|
{
|
|
get
|
|
{
|
|
return headerPath;
|
|
}
|
|
set
|
|
{
|
|
headerPath = SessionState.Path.GetResolvedPSPathFromPSPath(value).First().Path;
|
|
}
|
|
}
|
|
|
|
[Parameter(Position = 3,
|
|
Mandatory = false,
|
|
HelpMessage = "Specify the kind of the header to extract")
|
|
]
|
|
public HeaderType HeaderType
|
|
{
|
|
get
|
|
{
|
|
return kind;
|
|
}
|
|
set
|
|
{
|
|
kind = value;
|
|
}
|
|
}
|
|
|
|
[Parameter(Position = 4,
|
|
Mandatory = false,
|
|
HelpMessage = "Number of section to modify")
|
|
]
|
|
public int Section = 1;
|
|
|
|
[Parameter(
|
|
Mandatory = false,
|
|
ValueFromPipeline = true,
|
|
ValueFromPipelineByPropertyName = true,
|
|
HelpMessage = "XDocument of the header to add")
|
|
]
|
|
[ValidateNotNullOrEmpty]
|
|
public XDocument Header
|
|
{
|
|
get
|
|
{
|
|
return header;
|
|
}
|
|
set
|
|
{
|
|
header = value;
|
|
}
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region Cmdlet Overrides
|
|
|
|
/// <summary>
|
|
/// Entry point for PowerShell cmdlets
|
|
/// </summary>
|
|
protected override void ProcessRecord()
|
|
{
|
|
if (headerPath != null)
|
|
header = XDocument.Load(headerPath);
|
|
if (header == null)
|
|
{
|
|
WriteError(new ErrorRecord(new ArgumentException("No footer was specified."), "OpenXmlPowerTools", ErrorCategory.InvalidArgument, null));
|
|
return;
|
|
}
|
|
foreach (var document in AllDocuments("Set-OpenXmlHeader"))
|
|
{
|
|
try
|
|
{
|
|
if (!(document is WmlDocument))
|
|
throw new PowerToolsDocumentException("Not a wordprocessing document.");
|
|
OutputDocument(HeaderAccessor.SetHeader((WmlDocument)document, header, kind, Section - 1));
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
WriteError(PowerToolsExceptionHandling.GetExceptionErrorRecord(e, document));
|
|
}
|
|
}
|
|
}
|
|
#endregion
|
|
}
|
|
}
|