public IEnumerable <double> IExtract(string outputFolder)
{
yield return(0);
cpkMaker.AnalyzeCpkFile(FilePath);
cpkMaker.StartToExtract(outputFolder);
while (cpkMaker.GetProgress() < 100)
{
Status = cpkMaker.Execute();
double progress;
Progress = progress = cpkMaker.GetProgress();
yield return(progress);
}
yield return(100);
}
static void Main(string[] args)
{
if (args.Length == 0)
{
PrintUsage(NO_ARGS);
return;
}
bool bList = false; // Old method, TODO update
bool bVerbose = false;
bool bUseCsv = false;
bool doExtract = false;
bool doPack = false;
bool doReplace = false;
bool _doList = false; // Special case
string inFileName = null;
string outFileName = null;
string workingDir = Directory.GetCurrentDirectory();
string packCompressCodec = null;
string packDataAlign = null;
string extractWhat = null;
string replaceWhat = null;
string replaceWith = null;
string csvFileName = null;
// STEP 1 - PARSE THE ARGUMENTS
byte usageHelpFlags = 0x00;
bool isUsageSimple = (args[0][0] != '-');
if (!isUsageSimple) // Indicates technical usage vs simple usage; first argument must determine this
{
for (int i = 0; i < args.Length; i++)
{
string option = args[i];
if (option[0] == '-')
{
try
{
switch (option[1])
{
case 'X':
doExtract = true;
try { extractWhat = args[i + 1]; } catch { extractWhat = null; }
break;
case 'P': doPack = true; break;
case 'R':
doReplace = true;
try
{
replaceWhat = args[i + 1];
replaceWith = args[i + 2];
}
catch
{
replaceWhat = null;
replaceWith = null;
usageHelpFlags |= WRONG_USAGE;
}
break;
case 'L': bList = true; break;
case 'i': inFileName = args[i + 1]; break;
case 'o': outFileName = args[i + 1]; break;
case 'd': workingDir = args[i + 1]; break;
case 'v': bVerbose = true; break;
// Extra options
case '-':
switch (option.Substring(2))
{
// These are packing options
case "codec": packCompressCodec = args[i + 1]; break;
case "align": packDataAlign = args[i + 1]; break;
// For each packing, extracting, and listing
case "csv":
bUseCsv = true;
try { csvFileName = args[i + 1]; }
catch { csvFileName = null; }
break;
}
break;
case 'h': usageHelpFlags |= HELP_INFO; break;
}
}
catch
{
usageHelpFlags |= WRONG_USAGE;
break;
}
}
}
}
else // Simple usage
{
inFileName = args[0];
try { outFileName = args[1]; }
catch { outFileName = null; }
}
// Set if we only wanted to list the contents
_doList = bList && !(doExtract || doPack || doReplace) && !(isUsageSimple);
// If we had any reason to stop, now's the time. After this if-statement, it's assumed the user wanted to perform an operation.
if ((usageHelpFlags > 0) || (inFileName == null) || (!(doExtract || doPack || doReplace || _doList) && !isUsageSimple))
{
PrintUsage(usageHelpFlags);
return;
}
// STEP 2 - VERIFY THAT VALID OPTIONS WERE PASSED (1) input file - Granted the above conditions, inFileName cannot be null at this point
if (isUsageSimple || doExtract || doPack || doReplace || _doList)
{
if (File.Exists(inFileName)) // Applies to drag-n-drop uses
{
if (isUsageSimple)
{
doExtract = true;
inFileName = (new FileInfo(inFileName)).FullName;
}
}
else if (File.Exists(workingDir + "\\" + inFileName))
{
if (isUsageSimple)
{
doExtract = true;
}
inFileName = workingDir + "\\" + inFileName;
}
else if (Directory.Exists(inFileName)) // Applies to drag-n-drop uses
{
if (isUsageSimple)
{
doPack = true;
inFileName = Path.GetFullPath(inFileName);
}
}
else if (Directory.Exists(workingDir + "\\" + inFileName))
{
if (isUsageSimple)
{
doPack = true;
}
inFileName = workingDir + "\\" + inFileName;
}
else
{
Console.WriteLine("Error: Could not find the specified input file or folder.");
return;
}
}
// (2) output path - Preparing output directories
if (outFileName != null)
{
if (!ValidateFilePathString(workingDir, ref outFileName))
{
Console.WriteLine("Error: Invalid output path specified. Exiting process.");
return;
}
}
// (3) packing options - compression codec, data alignment
EnumCompressCodec compressCodec = EnumCompressCodec.CodecDpk;
uint dataAlign = 2048;
if (doPack)
{
if (packCompressCodec != null)
{
if (String.Equals(packCompressCodec, "none", StringComparison.CurrentCultureIgnoreCase))
{
compressCodec = EnumCompressCodec.CodecDpk;
}
else if (String.Equals(packCompressCodec, "layla", StringComparison.CurrentCultureIgnoreCase))
{
compressCodec = EnumCompressCodec.CodecLayla;
}/* Not implemented, according to CpkMaker
* else if (String.Equals(packCompressCodec, "lzma", StringComparison.CurrentCultureIgnoreCase))
* {
* compressCodec = EnumCompressCodec.CodecLZMA;
* }
* else if (String.Equals(packCompressCodec, "relc", StringComparison.CurrentCultureIgnoreCase))
* {
* compressCodec = EnumCompressCodec.CodecRELC;
* }*/
else
{
Console.WriteLine("Error: Invalid packing codec specified: \"" + packCompressCodec + "\"");
return;
}
}
if (packDataAlign != null)
{
try
{
Int32.TryParse(packDataAlign, out int tmp_da);
if ((tmp_da > 0) && ((tmp_da & (tmp_da - 1)) == 0))
{
dataAlign = (uint)tmp_da;
}
else
{
Console.WriteLine("Error: The specified data alignment is not a power of two!");
return;
}
}
catch { Console.WriteLine("Error: A non-number was specified for the data alignment!"); return; }
}
}
// (4) Validate the path for a CSV file
if (bUseCsv)
{
if ((csvFileName == null) || (csvFileName[0] == '-'))
{
csvFileName = Path.GetFileNameWithoutExtension(inFileName);
}
if (!(ValidateFilePathString(workingDir, ref csvFileName)))
{
Console.WriteLine("Error: Unable to prepare a file path for the CSV file!");
return;
}
if (!String.Equals(Path.GetExtension(csvFileName), ".csv", StringComparison.OrdinalIgnoreCase))
{
csvFileName += ".csv";
}
}
// STEP 3 - NOW HERE'S WHERE THE PROCESS GETS STARTED
CpkMaker cpkMaker = new CpkMaker();
CFileData cpkFileData = null;
CAsyncFile cManager = new CAsyncFile();
Status status;
if (doExtract || doReplace || _doList)
{
if (!cpkMaker.AnalyzeCpkFile(inFileName))
{
Console.WriteLine("Error: AnalyzeCpkFile returned false!");
return;
}
cpkFileData = cpkMaker.FileData;
}
if (doExtract)
{
//Console.WriteLine("DEBUG: doExtract!");
if (outFileName == null)
{
outFileName = Path.GetDirectoryName(inFileName).Replace('/', '\\') + "\\" + Path.GetFileNameWithoutExtension(inFileName);
}
if ((extractWhat != null) && (extractWhat[0] != '-')) // If user wanted a specific file
{
CFileInfo cFileInfo = cpkFileData.GetFileData(extractWhat);
if (cFileInfo != null)
{
outFileName += "\\" + extractWhat.Replace('/', '\\');
try { Directory.CreateDirectory(Path.GetDirectoryName(outFileName)); } catch { }
CAsyncFile cpkReader = new CAsyncFile();
cpkReader.ReadOpen(inFileName);
cpkReader.WaitForComplete();
CAsyncFile extWriter = new CAsyncFile();
extWriter.WriteOpen(outFileName, false);
extWriter.WaitForComplete();
cManager.Copy(cpkReader, extWriter, cFileInfo.Offset, cFileInfo.Filesize, CAsyncFile.CopyMode.ReadDecompress, cFileInfo.Extractsize);
cManager.WaitForComplete();
Console.WriteLine("Successfully extracted the file!");
}
else
{
Console.WriteLine("Error: Unable to locate the specified file in the CPK \"" + extractWhat + "\"");
return;
}
}
else // Just extract everything
{
try { Directory.CreateDirectory(outFileName); } catch { }
cpkMaker.StartToExtract(outFileName); // Continues at STEP 4
}
if (bUseCsv)
{
if (!ExportCsv(csvFileName, Path.GetFileNameWithoutExtension(outFileName), ref cpkFileData))
{
Console.WriteLine("Error: Something went wrong exporting CSV file!");
return;
}
}
}
else if (doPack)
{
//Console.WriteLine("DEBUG: doPack!");
if (bUseCsv)
{
if (AnalyzeCsv(csvFileName, dataAlign, out CpkMaker csvCpkMaker))
{
cpkMaker = csvCpkMaker;
}
else
{
Console.WriteLine("Error: AnalyzeCsv() returned false! Invalid CSV entry?"); return;
}
//if (true) return; // iz debugging PokeSlow
}
else
{
cpkMaker.CpkFileMode = CpkMaker.EnumCpkFileMode.ModeFilename;
cpkMaker.CompressCodec = compressCodec;
cpkMaker.DataAlign = dataAlign;
uint i = 0;
string[] files = Directory.GetFiles(inFileName, "*", SearchOption.AllDirectories);
foreach (string file in files)
{
if (File.Exists(file))
{
string localpath = file.Replace(inFileName, "");
localpath = localpath.Replace('\\', '/');
if (localpath[0] == '/')
{
localpath = localpath.Substring(1);
}
//Console.WriteLine("Local path = \"" + localpath + "\"");
cpkMaker.AddFile(file, localpath, i++, (((int)compressCodec == 1) ? false : true), "", "", dataAlign);
}
}
}
if (outFileName == null)
{
outFileName = inFileName.Replace('/', '\\') + ".cpk";
}
File.Create(outFileName).Close();
cpkMaker.StartToBuild(outFileName); // Continues at STEP 4
}
else if (doReplace)
{
//Console.WriteLine("DEBUG: doReplace!");
CFileInfo cFileInfo = cpkFileData.GetFileData(replaceWhat);
if (cFileInfo != null)
{
if (!(ValidateFilePathString(workingDir, ref replaceWith)))
{
Console.WriteLine("Error: Unable to locate the specified file to inject.");
return;
}
if (outFileName == null)
{
outFileName = (workingDir.Replace('/', '\\') + "\\new_" + Path.GetFileName(inFileName));
}
if (!(String.Equals(Path.GetExtension(outFileName), ".cpk", StringComparison.OrdinalIgnoreCase)))
{
outFileName += ".cpk";
}
cpkMaker.DeleteFile(replaceWhat);
cpkMaker.AddFile(replaceWith, replaceWhat, cFileInfo.FileId, cFileInfo.IsCompressed,
cFileInfo.GroupString, cFileInfo.AttributeString, cFileInfo.DataAlign);
cpkMaker.FileData.UpdateFileInfoPackingOrder();
Console.WriteLine("Preparing new CPK...");
File.Create(outFileName).Close();
cpkMaker.StartToBuild(outFileName);
cpkMaker.WaitForComplete();
CAsyncFile currOldFile = new CAsyncFile(2);
currOldFile.ReadOpen(inFileName);
currOldFile.WaitForComplete();
CAsyncFile patchedFile = new CAsyncFile(2);
patchedFile.WriteOpen(outFileName, true);
patchedFile.WaitForComplete();
Console.WriteLine("Patching in files...");
for (int i = 0; i < cpkMaker.FileData.FileInfos.Count; i++)
{
// I feel like I'd rather just compare CFileInfo.InfoIndex, but I'm a bit paranoid about that...
// At least directly comparing the content path is assured without a shadow of a doubt to tell us what we need to know
// I want to avoid unnecessary instance invocations of objects and complex conditionals if possible
// Please, for the sake of this loop, never invoke CpkMaker.FileData.UpdateFileInfoIndex()
//if(String.Equals(cpkFileData.FileInfos[i].ContentFilePath, cFileInfo.ContentFilePath)) { continue; }
CFileInfo currNewFileInfo = cpkMaker.FileData.FileInfos[i];
CFileInfo currOldFileInfo = cpkFileData.GetFileData(currNewFileInfo.ContentFilePath);
bool wasThisPatched = (currNewFileInfo.InfoIndex == cFileInfo.InfoIndex);
if (!wasThisPatched) // Eh, I'll try it.
{
patchedFile.Position = currNewFileInfo.Offset + currOldFileInfo.DataAlign;
//Console.WriteLine("Current position = 0x" + patchedFile.Position.ToString("X8"));
currOldFile.ReadAlloc(currOldFileInfo.Offset + currOldFileInfo.DataAlign, currOldFileInfo.Filesize);
currOldFile.WaitForComplete();
unsafe
{
patchedFile.Write(currOldFile.ReadBuffer, currOldFileInfo.Filesize, CAsyncFile.WriteMode.Normal);
patchedFile.WaitForComplete();
}
currOldFile.ReleaseBuffer();
}
if (bVerbose)
{
Console.WriteLine("[" + currNewFileInfo.InfoIndex.ToString().PadLeft(5) + "] " +
(wasThisPatched ? "PATCHED " : " ") + currNewFileInfo.ContentFilePath + " ...");
}
//if (true) return; // Used with debugging
}
Console.WriteLine("Patch complete!");
}
else
{
Console.WriteLine("Error: Unable to locate the specified file in the CPK \"" + replaceWhat + "\"");
return;
}
}
else if (_doList)
{
//Console.WriteLine("DEBUG: doList! - TODO");
if (bVerbose)
{
cpkMaker.DebugPrintInternalInfo();
}
else
{
Console.WriteLine(cpkMaker.GetCpkInformationString(true, false));
}
if (bUseCsv)
{
if (!ExportCsv(csvFileName, Path.GetFileNameWithoutExtension(inFileName), ref cpkFileData))
{
Console.WriteLine("Error: Something went wrong exporting CSV file!");
return;
}
}
}
else
{
Console.WriteLine("Error: Did nothing?");
return;
}
// STEP 4 - PROGRESS LOOP FOR WHERE APPLICABLE (after I complete the above versions)
if (doExtract || doPack)
{
int last_p = -1;
int percent = 0;
status = cpkMaker.Execute();
while ((status > Status.Stop) && (percent < 100))
{
percent = (int)Math.Floor(cpkMaker.GetProgress());
if (percent > last_p)
{
Console.CursorLeft = 0;
Console.Write(percent.ToString() + "% " + (doExtract ? "extracted" : "packed") + "...");
last_p = percent;
}
status = cpkMaker.Execute();
}
Console.WriteLine("");
Console.WriteLine("Status = " + status.ToString());
}
Console.WriteLine("\nProcess finished (hopefully) without issues!");
}