public static StatelessResult PerformOneOff(CommandLineParser args)
{
// one-off command:
// Anolis.Resourcer.exe -op:add -src:"C:\dest.exe" -type:ICONDIR -name:NAME -lang:1033 -file:"C:\foo\icon.ico"
// Anolis.Resourcer.exe -op:upd -src:"C:\dest.exe" -type:ICONDIR -name:NAME [-lang:1033] -file:"C:\foo\icon.ico"
// Anolis.Resourcer.exe -op:ext -src:"C:\dest.exe" -type:ICONDIR -name:NAME -lang:1033 -file:"C:\foo\icon.ico"
// Anolis.Resourcer.exe -op:del -src:"C:\dest.exe" -type:ICONDIR -name:NAME [-lang:1033]
CommandLineFlag
operation = args.GetFlag("op"),
resSource = args.GetFlag("src"),
dataType = args.GetFlag("type"),
dataName = args.GetFlag("name"),
dataLang = args.GetFlag("lang"),
dataFile = args.GetFlag("file");
ushort dataLangN = 0;
///////////////////////////////////////
// Validation
// Individual arguments
if (operation == null || resSource == null || dataType == null || dataName == null)
{
return(new StatelessResult(false, "Syntax error: not all arguments set"));
}
switch (operation.Argument)
{
case "add":
case "upd":
case "ext":
case "del":
break;
default:
return(new StatelessResult(
"Syntax error: invalid operation value, must be one of 'add', 'del', 'upd', or 'ext'."
));
}
// resolve fileNames
if (!string.IsNullOrEmpty(resSource.Argument))
{
resSource.Argument = Path.GetFullPath(resSource.Argument);
}
if (dataFile != null && !string.IsNullOrEmpty(dataFile.Argument))
{
dataFile.Argument = Path.GetFullPath(dataFile.Argument);
}
if (!File.Exists(resSource.Argument))
{
return(new StatelessResult("File not found: " + resSource.Argument));
}
if (dataLang != null)
{
var num = dataLang.Argument;
if (!ushort.TryParse(num, out dataLangN))
{
return(new StatelessResult("Syntax error: langId is not an unsigned integer less than 2^16"));
}
}
// Combinations
//////// Add
if (operation.Argument == "add" && dataLang == null)
{
return(new StatelessResult("LangId not specified for add"));
}
if (operation.Argument == "add" && !File.Exists(dataFile.Argument))
{
return(new StatelessResult("File not found: " + dataFile.Argument));
}
//////// Update
if (operation.Argument == "upd" && !File.Exists(dataFile.Argument))
{
return(new StatelessResult("File not found: " + dataFile.Argument));
}
//////// Extract
//////// Delete
if (operation.Argument != "del" && dataFile == null)
{
return(new StatelessResult("A file must be specified"));
}
///////////////////////////////////////
var typeId = ResourceTypeIdentifier.CreateFromString(dataType.Argument, true);
var nameId = ResourceIdentifier.CreateFromString(dataName.Argument);
var langId = dataLang != null ? dataLangN : ushort.MinValue;
switch (operation.Argument)
{
case "del":
return(OneOffDelete(resSource.Argument, typeId, nameId));
case "add":
return(OneOffAdd(resSource.Argument, typeId, nameId, langId, dataFile.Argument));
case "upd":
return(dataLang != null
? OneOffUpdate(resSource.Argument, typeId, nameId, langId, dataFile.Argument)
: OneOffUpdate(resSource.Argument, typeId, nameId, dataFile.Argument));
case "ext":
return(OneOffExtract(resSource.Argument, typeId, nameId, langId, dataFile.Argument));
}
return(new StatelessResult("No valid operation specified, must be one of 'add', 'del', 'upd', or 'ext'."));
}
protected override Boolean PatchFile(String fileName)
{
List <PatchResource> patchResources = new List <PatchResource>();
foreach (PatchResourceSet set in _resourceSets)
{
if (EvaluatePatchResourceSet(set, fileName))
{
// HACK: This just adds them together into a massive list. If the same name is mentioned it'll be overwritten several times
// fortunately it isnt' very expensive as only the last "final" one counts, but could do with filtering at this stage maybe?
patchResources.AddRange(set.Resources);
}
else
{
Package.Log.Add(LogSeverity.Info, "Expression evaluation non-one: " + set.Condition.ExpressionString + ", did not process " + set.Resources.Count + " resources");
}
}
if (patchResources.Count == 0)
{
Package.Log.Add(LogSeverity.Warning, "No resources to patch: " + fileName);
return(false);
}
try {
// for now, use lazy-load under all circumstances. In future analyse the Resources list to see if it's necessary or not
// but the performance impact is minimal and it's the safest option, so keep it as it is
using (ResourceSource source = ResourceSource.Open(fileName, false, ResourceSourceLoadMode.LazyLoadData)) {
List <String> tempFiles = new List <String>();
foreach (PatchResource res in patchResources)
{
if (res.Source.StartsWith("comp:", StringComparison.OrdinalIgnoreCase))
{
CompositedImage comp = new CompositedImage(res.Source, Package.RootDirectory);
DirectoryInfo packageTempDirectory = new DirectoryInfo(P.Combine(Package.RootDirectory.FullName, "Temp"));
if (!packageTempDirectory.Exists)
{
packageTempDirectory.Create();
}
// I think not using the *.bmp extension messes up Bitmap import
String tempFileName = PackageUtility.GetUnusedFileName(P.Combine(packageTempDirectory.FullName, P.GetFileName(Path) + res.Name) + ".bmp");
comp.Save(tempFileName, System.Drawing.Imaging.ImageFormat.Bmp);
res.File = tempFileName;
tempFiles.Add(tempFileName);
}
else
{
res.File = res.Source;
}
if (!File.Exists(res.File))
{
Package.Log.Add(LogSeverity.Error, "Data File not found: " + res.File);
continue;
}
ResourceTypeIdentifier typeId = ResourceTypeIdentifier.CreateFromString(res.Type, true);
ResourceIdentifier nameId = ResourceIdentifier.CreateFromString(res.Name);
UInt16 langId = String.IsNullOrEmpty(res.Lang) ? UInt16.MaxValue : UInt16.Parse(res.Lang, System.Globalization.NumberStyles.Integer, System.Globalization.CultureInfo.InvariantCulture);
if (langId == UInt16.MaxValue) // if the lang="" attribute was not specified
{
ResourceName name = source.GetName(typeId, nameId);
if (name == null) // if the specified name does not exist
{
if (res.Add)
{
UInt16 sysLang = (UInt16)CultureInfo.InvariantCulture.LCID;
ResourceData data = ResourceData.FromFileToAdd(res.File, sysLang, source);
source.Add(typeId, nameId, sysLang, data);
}
else
{
// Error
String sourcePath = source.Name;
Anolis.Core.Source.FileResourceSource frs = source as Anolis.Core.Source.FileResourceSource;
if (frs != null)
{
sourcePath = frs.FileInfo.FullName;
}
Package.Log.Add(LogSeverity.Warning, "Resource name not found: " + sourcePath + '\\' + typeId.ToString() + '\\' + nameId.FriendlyName);
}
}
else
{
foreach (ResourceLang lang in name.Langs)
{
ResourceData data = ResourceData.FromFileToUpdate(res.File, lang);
lang.SwapData(data);
}
}
}
else // if the lang="" attribute was specified
{
ResourceLang lang = source.GetLang(typeId, nameId, langId);
if (lang == null)
{
ResourceData data = ResourceData.FromFileToAdd(res.File, langId, source);
source.Add(typeId, nameId, langId, data);
}
else
{
ResourceData data = ResourceData.FromFileToUpdate(res.File, lang);
lang.SwapData(data);
}
}
} //foreach
// note that Win32ResourceSource now recomptues the PE checksum by itself
source.CommitChanges();
foreach (String tempFile in tempFiles)
{
File.Delete(tempFile);
}
return(true);
} //using source
} catch (AnolisException aex) {
Package.Log.Add(LogSeverity.Error, "Patch Exception: " + aex.Message);
if (File.Exists(fileName))
{
File.Delete(fileName);
}
throw;
}
}