/*
 * Copyright 2008, Axel Dörfler, axeld@pinc-software.de.
 * This file may be used under the terms of the MIT License.
 */


#include <dirent.h>
#include <unistd.h>
#include <stdio.h>
#include <string.h>

#include <Entry.h>
#include <File.h>
#include <fs_attr.h>
#include <TypeConstants.h>


const char* kTempFile = "/tmp/bfs_attribute_iterator_test";

bool gVerbose;


bool
is_marker(const char* name)
{
	return strstr(name, ":marker") != NULL;
}


int32
attribute_index(const char* name)
{
	int32 number;
	if (sscanf(name, "test:%ld", &number) == 1)
		return number;

	return 0;
}


void
add_attributes(BFile& file, int32 start, int32 count)
{
	for (int32 index = start; index < start + count; index++) {
		char name[B_ATTR_NAME_LENGTH];
		snprintf(name, sizeof(name), "test:%ld", index);
		file.WriteAttr(name, B_INT32_TYPE, 0, &index, sizeof(int32));
	}
}


void
remove_attributes(BFile& file, int32 start, int32 count)
{
	for (int32 index = start; index < start + count; index++) {
		char name[B_ATTR_NAME_LENGTH];
		snprintf(name, sizeof(name), "test:%ld", index);
		file.RemoveAttr(name);
	}
}


void
add_marker_attribute(BFile& file, int32 index)
{
	char name[B_ATTR_NAME_LENGTH];
	snprintf(name, sizeof(name), "test:%ld:marker", index);
	file.WriteAttr(name, B_INT32_TYPE, 0, &index, sizeof(int32));
}


void
remove_marker_attribute(BFile& file, int32 index)
{
	char name[B_ATTR_NAME_LENGTH];
	snprintf(name, sizeof(name), "test:%ld:marker", index);
	file.RemoveAttr(name);
}


void
test_remove_attributes(BFile& file, int32 start, int32 count, int32 removeAt,
	int32 removeIndex1, int32 removeIndex2)
{
	int fd = file.Dup();
	if (fd < 0)
		return;

	int32 index = 0;
	bool seen[4096] = {0};

	if (gVerbose)
		printf("test removeAt %ld\n", removeAt);

	DIR* dir = fs_fopen_attr_dir(fd);
	while (struct dirent* entry = fs_read_attr_dir(dir)) {
		if (gVerbose)
			printf("  %ld. %s\n", index, entry->d_name);
		if (index == removeAt) {
			if (removeIndex1 > 0)
				remove_marker_attribute(file, removeIndex1);
			if (removeIndex2 > 0)
				remove_marker_attribute(file, removeIndex2);
		}
		index++;

		if (is_marker(entry->d_name))
			continue;

		int32 attributeIndex = attribute_index(entry->d_name);
		if (attributeIndex > 0) {
			if (seen[attributeIndex]) {
				printf("attribute index %ld already listed!\n", attributeIndex);
				exit(1);
			} else
				seen[attributeIndex] = true;
		}
	}

	fs_close_attr_dir(dir);
	close(fd);

	for (int32 i = start; i < start + count; i++) {
		if (!seen[i]) {
			printf("attribute index %ld not listed, saw only %ld/%ld!\n", i,
				index, count);
			exit(1);
		}
	}
}


void
test_add_attributes(BFile& file, int32 start, int32 count, int32 addAt,
	int32 addIndex1, int32 addIndex2)
{
	int fd = file.Dup();
	if (fd < 0)
		return;

	int32 index = 0;
	bool seen[4096] = {0};

	if (gVerbose)
		printf("test addAt %ld\n", addAt);

	DIR* dir = fs_fopen_attr_dir(fd);
	while (struct dirent* entry = fs_read_attr_dir(dir)) {
		if (gVerbose)
			printf("  %ld. %s\n", index, entry->d_name);
		if (index == addAt) {
			if (addIndex1 > 0)
				add_marker_attribute(file, addIndex1);
			if (addIndex2 > 0)
				add_marker_attribute(file, addIndex2);
		}
		index++;

		if (is_marker(entry->d_name))
			continue;

		int32 attributeIndex = attribute_index(entry->d_name);
		if (attributeIndex > 0) {
			if (seen[attributeIndex]) {
				printf("attribute index %ld already listed!\n", attributeIndex);
				exit(1);
			} else
				seen[attributeIndex] = true;
		}
	}

	fs_close_attr_dir(dir);
	close(fd);

	for (int32 i = start; i < start + count; i++) {
		if (!seen[i]) {
			printf("attribute index %ld not listed, saw only %ld/%ld!\n", i,
				index, count);
			exit(1);
		}
	}
}


int
main(int argc, char** argv)
{
	if (argc > 1 && !strcmp(argv[1], "-v"))
		gVerbose = true;

	BFile file;
	status_t status = file.SetTo(kTempFile, B_CREATE_FILE | B_READ_WRITE);
	if (status != B_OK) {
		fprintf(stderr, "Could not create temporary file: %s\n",
			strerror(status));
		return 1;
	}

	puts("--------- Remove Tests ----------");

	// remove test, many attributes

	puts("Test 1...");

	for (int32 count = 5; count <= 100; count += 5) {
		add_attributes(file, 1, count);
		add_marker_attribute(file, count);
		add_attributes(file, count + 1, count);

		test_remove_attributes(file, 1, count,
			count & 1 ? count - 1 : count + 1, count, -1);

		remove_attributes(file, 1, count * 2);
	}

	// remove test, all positions

	puts("Test 2...");

	for (int32 i = 1; i < 100; i++) {
		add_attributes(file, 1, 50);
		add_marker_attribute(file, 51);
		add_attributes(file, 51, 50);

		test_remove_attributes(file, 1, 100, i, 51, -1);

		remove_attributes(file, 1, 100);
	}

	// remove test, all positions, remove many

	puts("Test 3...");

	for (int32 i = 1; i < 100; i++) {
		add_attributes(file, 1, 33);
		add_marker_attribute(file, 33);
		add_attributes(file, 34, 34);
		add_marker_attribute(file, 67);
		add_attributes(file, 68, 33);

		test_remove_attributes(file, 1, 100, i, 33, 67);

		remove_attributes(file, 1, 100);
	}

	puts("--------- Add Tests ----------");

	// add test, many attributes

	puts("Test 4...");

	for (int32 count = 10; count <= 200; count += 10) {
		add_attributes(file, 1, count);

		int32 half = count / 2;

		test_add_attributes(file, 1, count,
			half & 1 ? half - 1 : half + 1, half - 2, half + 2);

		remove_attributes(file, 1, count);
	}

	// add test, all iterator positions

	puts("Test 5...");

	for (int32 i = 1; i < 100; i++) {
		add_attributes(file, 1, 100);

		test_add_attributes(file, 1, 100, i, 50, -1);

		remove_attributes(file, 1, 100);
	}

	// add test, all attribute positions

	puts("Test 6...");

	for (int32 i = 1; i < 100; i++) {
		add_attributes(file, 1, 100);

		test_add_attributes(file, 1, 100, 50, i, -1);

		remove_attributes(file, 1, 100);
	}

	// add test, many attributes

	puts("Test 7...");

	for (int32 i = 1; i < 100; i++) {
		add_attributes(file, 1, 100);

		test_add_attributes(file, 1, 100, i, 33, 67);

		remove_attributes(file, 1, 100);
	}

	// add test, many attributes

	puts("Test 8...");

	for (int32 i = 1; i < 100; i++) {
		add_attributes(file, 1, 100);

		test_add_attributes(file, 1, 100, 50, i - 1, i + 1);

		remove_attributes(file, 1, 100);
	}

	BEntry entry;
	if (entry.SetTo(kTempFile) == B_OK)
		entry.Remove();

	return 0;
}