123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302 |
- #include <iostream>
- #include <fstream>
- #include <ctime>
- #include <algorithm>
- #ifdef _WIN32
- #include <direct.h>
- #define mkdir(dir, mode) _mkdir(dir)
- #endif
- #include "datfile.h"
- using namespace LOTRO_DAT;
- // Change these variables to true if you want export category to files.
- bool exportTextsToFiles = false;
- bool exportImagesToFiles = false;
- bool exportFontsToFiles = false;
- bool exportSoundsToFiles = false;
- bool exportTexturesToFiles = false;
- bool exportUnknownToFiles = false;
- bool exportTextsToDb = false;
- bool exportImagesToDb = false;
- bool exportFontsToDb = false;
- bool exportSoundsToDb = false;
- bool exportTexturesToDb = false;
- bool exportUnknownToDb = false;
- void InitializeFileLoop(DatFile& file) {
- while (!file.Initialized()) {
- std::cout << "Please, tell, where the .dat file is\n";
- std::cout << "Enter path to file (including filename): ";
- std::string filename;
- std::getline(std::cin, filename);
- // std::cout << "Enter file id: ";
- // std::string file_id_str;
- // std::getline(std::cin, file_id_str);
- // int file_id = 0;
- // try {
- // file_id = stoi(file_id_str);
- // } catch (std::invalid_argument) {
- // std::cout << "Invalid dat file id entered!\n\n";
- // continue;
- // }
- std::cout << "Opening file " << filename << std::endl;
- bool result = file.Init(filename); // TODO: INIT WITH DAT FILE ID
- if (!result) {
- std::cout << "Dat initialisation failed.\nPlease enter path to .dat file once more\n";
- }
- }
- }
- int main() {
- std::cout.precision(1);
- std::cout << std::fixed;
- std::cout << "Endevir's LotRO .dat extractor ver. 1.0.0" << std::endl;
- std::cout << "Hello! I'm a basic shell version of .dat file extractor. I can open .dat file directly, "
- "if you write path to it (with name of file) in file \"dat_file_path.txt\"\n";
-
- DatFile file(0);
- std::ifstream in_stream("dat_file_path.txt");
- if (!in_stream.fail()) {
- std::string filename;
- std::getline(in_stream, filename);
- // std::string file_id_str;
- // std::getline(in_stream, file_id_str);
- // int file_id = 0;
- // try {
- // file_id = stoi(file_id_str);
- // } catch (std::invalid_argument) {
- // file_id = -1;
- // std::cout << "Invalid file_id on second line of file dat_file_path.txt!\n\n";
- // }
- // if (file_id != -1) {
- std::cout << "Using .dat file from dat_file_path.txt...\n";
- std::cout << "Opening file " << filename << std::endl;
- bool result = file.Init(filename);
- if (!result) {
- std::cout << "Dat initialisation failed!\n";
- }
- // }
- }
- InitializeFileLoop(file);
- std::cout << "Great! File initialised successfully!\n";
- while (true) {
- std::cout << "Please, choose, what should I do. I can extract .dat file to files (press 1), "
- "open another .dat file (press 2), or configure, what should I extract. Choose, what to do or exit (press -1)\n";
- std::cout << "Enter number of command (1-3): ";
- std::string command;
- std::getline(std::cin, command);
- int cmd = -10;
- try {
- cmd = stoi(command);
- } catch (std::invalid_argument) {
- std::cout << "Invalid command entered!\n\n";
- continue;
- }
- if (cmd == -1) {
- std::cout << "Exiting. Thanks for using me!\n";
- break;
- }
- if (cmd == 1) {
- const clock_t begin_time = clock();
- mkdir("Extracted data", 744);
- std::time_t result = std::time(nullptr);
- char *ttime = std::asctime(std::localtime(&result));
- auto *out_time = new char[25];
- memcpy(out_time, ttime, 24);
- out_time[24] = '\0';
- std::string filename = file.GetFilename();
- std::cout << "FILENAME = " << filename << std::endl;
- std::string output_dir = std::string("Extracted data\\") + filename.substr(std::max(filename.length() - 16, 0u), 16) + " " + std::string(out_time) + "\\";
- std::replace(output_dir.begin(), output_dir.end(), ':', '-');
- mkdir(output_dir.c_str(), 744);
-
- std::cout << "Beginning unpacking... Please, wait for some minutes."
- "\nMaybe it's a good idea to have a cup of tea, while unpacker is working...\n" << std::flush;
- Database output_db;
- if (exportImagesToDb) {
- output_db.InitDatabase(output_dir + std::string("Images.db"), true);
- int extracted_jpg_files_num = file.ExportFilesByType(JPG, output_db);
- std::cout << "Extracted " << extracted_jpg_files_num << " .jpg files to Images.db" << std::endl << std::flush;
- output_db.CloseDatabase();
- }
- if (exportSoundsToDb) {
- output_db.InitDatabase(output_dir + std::string("Sounds.db"), true);
- int extracted_wav_files_num = file.ExportFilesByType(WAV, output_db);
- int extracted_ogg_files_num = file.ExportFilesByType(OGG, output_db);
- std::cout << "Extracted " << extracted_wav_files_num << " .wav files to Sounds.db" << std::endl << std::flush;
- std::cout << "Extracted " << extracted_ogg_files_num << " .ogg files to Sounds.db" << std::endl << std::flush;
- output_db.CloseDatabase();
- }
- if (exportTextsToDb) {
- output_db.InitDatabase(output_dir + std::string("Texts.db"), true);
- int extracted_text_files_num = file.ExportFilesByType(TEXT, output_db);
- std::cout << "Extracted " << extracted_text_files_num << " text files to Texts.db" << std::endl << std::flush;
- output_db.CloseDatabase();
- }
- if (exportFontsToDb) {
- output_db.InitDatabase(output_dir + std::string("Fonts.db"), true);
- int extracted_font_files_num = file.ExportFilesByType(FONT, output_db);
- std::cout << "Extracted " << extracted_font_files_num << " font files to Fonts.db" << std::endl << std::flush;
- output_db.CloseDatabase();
- }
- if (exportTexturesToDb) {
- output_db.InitDatabase(output_dir + std::string("Textures.db"), true);
- int extracted_dds_files_num = file.ExportFilesByType(DDS, output_db);
- std::cout << "Extracted " << extracted_dds_files_num << " .dds files to Textures.db" << std::endl << std::flush;
- output_db.CloseDatabase();
- }
- if (exportImagesToFiles) {
- mkdir((output_dir + "jpg").c_str(), 744);
- int extracted_jpg_files_num = file.ExportFilesByType(JPG, output_dir + "jpg\\");
- std::cout << "Extracted " << extracted_jpg_files_num << " .jpg files to directory" << std::endl << std::flush;
- }
- if (exportTexturesToFiles) {
- mkdir((output_dir + "dds").c_str(), 744);
- int extracted_dds_files_num = file.ExportFilesByType(DDS, output_dir + "dds\\");
- std::cout << "Extracted " << extracted_dds_files_num << " .dds files to directory" << std::endl << std::flush;
- }
- if (exportSoundsToFiles) {
- mkdir((output_dir + "wav").c_str(), 744);
- int extracted_wav_files_num = file.ExportFilesByType(WAV, output_dir + "wav\\");
- std::cout << "Extracted " << extracted_wav_files_num << " .wav files to directory" << std::endl << std::flush;
- mkdir((output_dir + "ogg").c_str(), 744);
- int extracted_ogg_files_num = file.ExportFilesByType(OGG, output_dir + "ogg\\");
- std::cout << "Extracted " << extracted_ogg_files_num << " .ogg files to directory" << std::endl << std::flush;
- }
- if (exportFontsToFiles) {
- mkdir((output_dir + "fonts").c_str(), 744);
- int extracted_font_files_num = file.ExportFilesByType(FONT, output_dir + "fonts\\");
- std::cout << "Extracted " << extracted_font_files_num << " font files to directory" << std::endl << std::flush;
- }
- if (exportUnknownToFiles) {
- mkdir((output_dir + "unknown").c_str(), 744);
- int extracted_unknown_files_num = file.ExportFilesByType(UNKNOWN, output_dir + "unknown\\");
- std::cout << "Extracted " << extracted_unknown_files_num << " unknown files to directory" << std::endl << std::flush;
- }
- if (exportTextsToFiles) {
- mkdir((output_dir + "texts").c_str(), 744);
- int extracted_text_files_num = file.ExportFilesByType(TEXT, output_dir + "texts\\");
- std::cout << "Extracted " << extracted_text_files_num << " text files to directory" << std::endl << std::flush;
- }
- fprintf(stdout, "Spent %f seconds on running unpacker! Thank you for your patience!\n",
- float(clock() - begin_time) / CLOCKS_PER_SEC);
- }
- if (cmd == 2) {
- std::cout << "Closing file...\n";
- file.Deinit();
- InitializeFileLoop(file);
- }
- if (cmd == 3) {
- std::cout << "Current parameters:\n";
- std::cout << "(01) Export texts to files - " << (exportTextsToFiles ? "true\n" : "false\n");
- std::cout << "(02) Export texts to database - " << (exportTextsToDb ? "true\n" : "false\n");
- std::cout << "(11) Export images to files - " << (exportImagesToFiles ? "true\n" : "false\n");
- std::cout << "(12) Export images to database - " << (exportImagesToDb ? "true\n" : "false\n");
- std::cout << "(21) Export fonts to files - " << (exportFontsToFiles ? "true\n" : "false\n");
- std::cout << "(22) Export fonts to database - " << (exportFontsToDb ? "true\n" : "false\n");
- std::cout << "(31) Export sounds to files - " << (exportSoundsToFiles ? "true\n" : "false\n");
- std::cout << "(32) Export sounds to database - " << (exportSoundsToDb ? "true\n" : "false\n");
- std::cout << "(41) Export textures to files - " << (exportTexturesToFiles ? "true\n" : "false\n");
- std::cout << "(42) Export textures to database - " << (exportTexturesToDb ? "true\n" : "false\n");
- std::cout << "(51) Export unknown files to files - " << (exportUnknownToFiles ? "true\n" : "false\n");
- std::cout << "(52) Export unknown files to database - " << (exportUnknownToDb ? "true\n" : "false\n");
- std::cout << "Enter number of parameter to change (or -1 to exit this menu): ";
- int number = 0;
- std::cin >> number;
- std::string tmp;
- std::getline(std::cin, tmp);
- if (number == -1) {
- std::cout << "Returning to main menu..\n";
- cmd = 0;
- } else {
- switch (number) {
- case 01:
- exportTextsToFiles = !exportTextsToFiles;
- break;
- case 02:
- exportTextsToDb = !exportTextsToDb;
- break;
- case 11:
- exportImagesToFiles = !exportImagesToFiles;
- break;
- case 12:
- exportImagesToDb = !exportImagesToDb;
- break;
- case 21:
- exportFontsToFiles = !exportFontsToFiles;
- break;
- case 22:
- exportFontsToDb = !exportFontsToDb;
- break;
- case 31:
- exportSoundsToFiles = !exportSoundsToFiles;
- break;
- case 32:
- exportSoundsToDb = !exportSoundsToDb;
- break;
- case 41:
- exportTexturesToFiles = !exportTexturesToFiles;
- break;
- case 42:
- exportTexturesToDb = !exportTexturesToDb;
- break;
- case 51:
- exportUnknownToFiles = !exportTexturesToFiles;
- break;
- case 52:
- exportUnknownToDb = !exportTexturesToDb;
- break;
- default:
- std::cout << "Incorrect number. Returning to main menu..\n";
- }
- }
- }
- }
- file.Deinit();
- system("pause");
- return 0;
- }
|