/*
 ================================================================================
 Name        : Texture-Extractor.c
 Author      : JackCarterSmith
 License     : GPL-v3.0
 Description : DAT textures extractor to PNG format with enhanced function in C
 ================================================================================
 */

#include "Texture-Extractor.h"


int _options;	// Global options settings variable

int main(int argc, char *argv[]) {
	// Init buffer vars
	HMT_FILE *hmt_fdatas = NULL;
	int file_index;

	printf("\n*** RogueSquadron Extractor (RSE) - TEXTURE module - v%s ***\n", VERSION);

	// Check if filenames arguments exist
	if (argc < 2) {
		printf("\n[ERR] No input file/commands specified!\n");
		dispHelp();
		return EXIT_FAILURE;		//TODO: implement own error codes system
	}
	_options = checkArgs(argv, argc);		// Analyse program arguments
	if (_options == -1) return EXIT_SUCCESS;

	// Do the work
	for (file_index=(_options >> 8) & 0xFF; file_index<argc; file_index++) {		// Manage multiple inputs files
		hmt_fdatas = extractDatasFromHMT(argv[file_index]);
		if (hmt_fdatas == NULL) return EXIT_FAILURE;
		if (exportTextures(hmt_fdatas, argv[file_index]) == EXIT_FAILURE) return EXIT_FAILURE;
		purgeHMTFromMemory(hmt_fdatas);		// Clean up memory (because I'm a good boy)
	}

	return EXIT_SUCCESS;
}

int checkArgs(char *args[], int arg_nbr) {
	int _o = 0x0002;		// Default options parameters
	char test[256];
	int i;

	if (arg_nbr > 1) {
		for (i=1; i<arg_nbr; i++) {
			strcpy(test, args[i]);
			if (args[i][0] != '-') break;
			if (strcmp(args[i], "-h") == 0) {
				dispHelp();
				return -1;
			} else if (strcmp(args[i], "-v") == 0) {
				_o |= VERBOSE_ENABLED;
				printf("[INFO] Verbose enabled.\n");
			} else if (strcmp(args[i], "-no-subdir") == 0) {
				_o &= ~OUTPUT_DIR;
				printf("[INFO] Extract to current directory.\n");
			} else {
				printf("[INFO] Unknown option: %s\n", args[i]);
			}
		}
		_o = (i << 8) | (_o & 0x00FF);
	}

	return _o;
}

void createSubDir(char *dirName) {
	if (dirName == NULL) return;
	char _dir[50];
	strcpy(_dir, dirName);
	strcat(_dir, "-out");

	#ifdef _WIN32
		CreateDirectory(_dir, NULL);
	#else
		mkdir(_dir, 0755);
	#endif
}

HMT_FILE *extractDatasFromHMT(char *hmt_filename) {
	FILE *_hmtFile = NULL;
	HMT_FILE *hmt_fdatas = NULL;

	_hmtFile = fopen(hmt_filename, "rb");
	if (_hmtFile != NULL) {
		printf("\n[INFO] - Parsing file: %s\n", hmt_filename);
		hmt_fdatas = parseHMTFile(_hmtFile);
		if (hmt_fdatas == NULL) printf("[ERR] Failed to parse datas from %s\n", hmt_filename);
	} else {
		printf("[ERR] Input file %s not found!\n", hmt_filename);
	}
	fclose(_hmtFile);

	return hmt_fdatas;
}

int exportTextures(HMT_FILE *hmt_f, char *filename) {
	int i;

	if(hmt_f->texture_count > 0) {
		if (_options & OUTPUT_DIR) createSubDir(filename);
		for (i=0; i<hmt_f->texture_count; i++) {
			switch (hmt_f->textures_list[i].image.type_) {
			case 0:
			case 1:
			case 3:
			case 4:
			case 5:
				if (saveToPNG(&(hmt_f->textures_list[i].image), hmt_f->textures_list[i].name, filename)) return EXIT_FAILURE;
				break;
			default:
				printf("[INFO] Image type %d not currently supported!\n", hmt_f->textures_list[i].image.type_);
			}

		}
	}

	return EXIT_SUCCESS;
}

void dispHelp() {
	printf("\n");
	printf("Options:\n	-h  Print this message\n	-v  Activate verbose console output\n	-no-subdir  Extract textures inside current folder\n");
	printf("\n");
	printf("Usage: RSE-Texture_%s [options] <hmt files...>\n", VERSION);
	printf("\n");
}