Merge branch 'master' of https://github.com/martmists/pokediamond

1 files changed, 108 insertions, 0 deletions

diff --git a/tools/knarc/Source.cpp b/tools/knarc/Source.cpp
new file mode 100644
index 00000000..51bb5320
--- /dev/null
+++ b/tools/knarc/Source.cpp
@@ -0,0 +1,108 @@
+#include <cstring>
+#include <iostream>
+#include <string>
+
+#include "Narc.h"
+
+using namespace std;
+
+void PrintError(NarcError error)
+{
+	switch (error)
+	{
+		case NarcError::None:								cout << "ERROR: No error???" << endl;										break;
+		case NarcError::InvalidInputFile:					cout << "ERROR: Invalid input file" << endl;								break;
+		case NarcError::InvalidHeaderId:					cout << "ERROR: Invalid header ID" << endl;									break;
+		case NarcError::InvalidByteOrderMark:				cout << "ERROR: Invalid byte order mark" << endl;							break;
+		case NarcError::InvalidVersion:						cout << "ERROR: Invalid NARC version" << endl;								break;
+		case NarcError::InvalidHeaderSize:					cout << "ERROR: Invalid header size" << endl;								break;
+		case NarcError::InvalidChunkCount:					cout << "ERROR: Invalid chunk count" << endl;								break;
+		case NarcError::InvalidFileAllocationTableId:		cout << "ERROR: Invalid file allocation table ID" << endl;					break;
+		case NarcError::InvalidFileAllocationTableReserved:	cout << "ERROR: Invalid file allocation table reserved section" << endl;	break;
+		case NarcError::InvalidFileNameTableId:				cout << "ERROR: Invalid file name table ID" << endl;						break;
+		case NarcError::InvalidFileNameTableEntryId:		cout << "ERROR: Invalid file name table entry ID" << endl;					break;
+		case NarcError::InvalidFileImagesId:				cout << "ERROR: Invalid file images ID" << endl;							break;
+		case NarcError::InvalidOutputFile:					cout << "ERROR: Invalid output file" << endl;								break;
+		default:											cout << "ERROR: Unknown error???" << endl;									break;
+	}
+}
+
+int main(int argc, char* argv[])
+{
+	if (argc != 5)
+	{
+		cout << "OVERVIEW: Knarc" << endl << endl;
+		cout << "USAGE: knarc [options] <inputs>" << endl << endl;
+		cout << "OPTIONS:" << endl;
+		cout << "\t-d\tDirectory to pack from/unpack to" << endl;
+		cout << "\t-p\tPack" << endl;
+		cout << "\t-u\tUnpack" << endl;
+
+		return 1;
+	}
+
+	string directory = "";
+	string fileName = "";
+	bool pack = false;
+
+	for (int i = 1; i < argc; ++i)
+	{
+		if (!strcmp(argv[i], "-d"))
+		{
+			if (i == (argc - 1))
+			{
+				cerr << "ERROR: No directory specified" << endl;
+
+				return 1;
+			}
+
+			directory = argv[++i];
+		}
+		else if (!strcmp(argv[i], "-p"))
+		{
+			if (i == (argc - 1))
+			{
+				cerr << "ERROR: No NARC specified to pack to" << endl;
+
+				return 1;
+			}
+
+			fileName = argv[++i];
+			pack = true;
+		}
+		else if (!strcmp(argv[i], "-u"))
+		{
+			if (i == (argc - 1))
+			{
+				cerr << "ERROR: No NARC specified to unpack from" << endl;
+
+				return 1;
+			}
+
+			fileName = argv[++i];
+		}
+	}
+
+	Narc narc;
+
+	if (pack)
+	{
+		if (!narc.Pack(fileName, directory))
+		{
+			PrintError(narc.GetError());
+
+			return 1;
+		}
+	}
+	else
+	{
+		if (!narc.Unpack(fileName, directory))
+		{
+			PrintError(narc.GetError());
+
+			return 1;
+		}
+	}
+
+	return 0;
+}