#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "csv_parser.h"

static void print_document(const csv_document_t *doc);
static void print_usage(const char *program_name);

int main(int argc, char *argv[])
{
	csv_parser_t *parser = csv_parser_create();
	if (!parser) {
		fprintf(stderr, "Error: Failed to create CSV parser\n");
		return 1;
	}
	
	csv_document_t *document = NULL;
	csv_error_t result;
	bool has_header = false;
	char *filename = NULL;
	
	/* Parse command line arguments */
	for (int i = 1; i < argc; i++) {
		if (strcmp(argv[i], "--header") == 0 || strcmp(argv[i], "-h") == 0) {
			has_header = true;
		} else if (strcmp(argv[i], "--help") == 0) {
			print_usage(argv[0]);
			csv_parser_destroy(parser);
			return 0;
		} else if (argv[i][0] != '-') {
			filename = argv[i];
		} else {
			fprintf(stderr, "Error: Unknown option '%s'\n", argv[i]);
			print_usage(argv[0]);
			csv_parser_destroy(parser);
			return 1;
		}
	}
	
	if (filename) {
		/* Parse file argument */
		FILE *file = fopen(filename, "r");
		if (!file) {
			fprintf(stderr, "Error: Cannot open file '%s'\n", filename);
			csv_parser_destroy(parser);
			return 1;
		}
		
		printf("Parsing CSV file: %s\n", filename);
		result = csv_parser_parse_file(parser, file, &document);
		fclose(file);
	} else {
		/* Parse from stdin */
		printf("CSV Parser - Enter CSV data (Ctrl+D to end):\n");
		result = csv_parser_parse_file(parser, stdin, &document);
	}
	
	if (result == CSV_SUCCESS) {
		/* If user specified header option, treat first record as header */
		if (has_header && document && document->records) {
			csv_error_t header_result = csv_document_set_first_record_as_header(document);
			if (header_result != CSV_SUCCESS) {
				printf("Warning: Failed to set header: %s\n", csv_error_string(header_result));
			}
		}
		
		printf("Parsing completed successfully.\n");
		print_document(document);
		csv_document_free(document);
	} else {
		printf("Parsing failed: %s\n", csv_error_string(result));
		printf("Error details: %s\n", csv_parser_get_error(parser));
	}
	
	csv_parser_destroy(parser);
	return result == CSV_SUCCESS ? 0 : 1;
}

static void print_document(const csv_document_t *doc)
{
	if (!doc) {
		printf("Document is NULL\n");
		return;
	}
	
	printf("\nDocument summary:\n");
	printf("- Records: %zu\n", doc->record_count);
	printf("- Has header: %s\n", csv_document_has_header(doc) ? "yes" : "no");
	
	/* Print header if present */
	if (csv_document_has_header(doc)) {
		printf("\nHeader:\n");
		csv_field_t *field = csv_record_get_first_field(doc->header);
		int field_num = 1;
		while (field) {
			printf("  Field %d: '%s'\n", field_num++, csv_field_get_content(field));
			field = csv_field_get_next(field);
		}
	}
	
	/* Print data records */
	printf("\nData records:\n");
	csv_record_t *record = csv_document_get_first_record(doc);
	int record_num = 1;
	
	while (record) {
		printf("Record %d (%zu fields):\n", record_num++, record->field_count);
		
		csv_field_t *field = csv_record_get_first_field(record);
		int field_num = 1;
		while (field) {
			const char *content = csv_field_get_content(field);
			printf("  Field %d: '%s'\n", field_num++, content ? content : "");
			field = csv_field_get_next(field);
		}
		
		record = csv_record_get_next(record);
	}
}

static void print_usage(const char *program_name)
{
	printf("Usage: %s [options] [filename]\n", program_name);
	printf("Options:\n");
	printf("  -h, --header    Treat the first record as a header\n");
	printf("  --help          Show this help message\n");
	printf("\n");
	printf("  If filename is provided, parse that file\n");
	printf("  Otherwise, read from standard input\n");
}