#include <dirent.h>

#include "QF/dstring.h"
#include "QF/quakeio.h"
#include "QF/script.h"
#include "QF/sys.h"
#include "QF/va.h"

#include "EntityClass.h"

@implementation EntityClass

static int
parse_vector (script_t * script, vec3_t vec)
{
	int         r;

	if (!Script_GetToken (script, 0))
		return 0;
	if (strcmp (script->token->str, "("))
		return 0;

	r = sscanf (script->p, "%f %f %f)", &vec[0], &vec[1], &vec[2]);
	if (r != 3)
		return 0;

	while (strcmp (script->token->str, ")")) {
		if (!Script_GetToken (script, 0))
			return 0;
	}
	return 1;
}

// the classname, color triple, and bounding box are parsed out of comments
// A ? size means take the exact brush size.
//
// /*QUAKED <classname> (0 0 0) ?
// /*QUAKED <classname> (0 0 0) (-8 -8 -8) (8 8 8)
//
// Flag names can follow the size description:
//
// /*QUAKED func_door (0 .5 .8) ? START_OPEN STONE_SOUND DOOR_DONT_LINK GOLD_KEY SILVER_KEY

-initFromText:(const char *)
text        source:(const char *) filename
{
	const char *t;
	size_t      len;
	int         i;
	script_t   *script;

	[super init];

	text += strlen ("/*QUAKED ");

	script = Script_New ();
	Script_Start (script, filename, text);

	// grab the name
	if (!Script_GetToken (script, 0))
		return 0;
	if (!strcmp (script->token->str, "*/"))
		return 0;
	name = strdup (script->token->str);

	// grab the color
	if (!parse_vector (script, color))
		return 0;

	// get the size 
	if (!strcmp (script->token->str, "(")) {
		Script_UngetToken (script);
		if (!parse_vector (script, mins))
			return 0;
		if (!parse_vector (script, maxs))
			return 0;
		esize = esize_fixed;
	} else if (!strcmp (script->token->str, "?")) {
		// use the brushes
		esize = esize_model;
	} else {
		return 0;
	}

	// get the flags
	// any remaining words on the line are parm flags
	for (i = 0; i < MAX_FLAGS; i++) {
		if (!Script_TokenAvailable (script, 0))
			break;
		Script_GetToken (script, 0);
		flagnames[i] = strdup (script->token->str);
	}
	while (Script_TokenAvailable (script, 0))
		Script_GetToken (script, 0);

// find the length until close comment
	for (t = script->p; t[0] && !(t[0] == '*' && t[1] == '/'); t++);

// copy the comment block out
	len = t - text;
	comments = malloc (len + 1);
	memcpy (comments, text, len);
	comments[len] = 0;

	return self;
}

-(esize_t) esize
{
	return esize;
}

-(char *) classname
{
	return name;
}

-(float *) mins
{
	return mins;
}

-(float *) maxs
{
	return maxs;
}

-(float *) drawColor
{
	return color;
}

-(char *) comments
{
	return comments;
}


-(char *) flagName:(unsigned) flagnum
{
	if (flagnum >= MAX_FLAGS)
		Sys_Error ("EntityClass flagName: bad number");
	return flagnames[flagnum];
}

@end
//===========================================================================

@implementation EntityClassList
/*
=================
insertEC:
=================
*/
- (void) insertEC:ec
{
	char       *name;
	int         i;

	name =[ec classname];
	for (i = 0; i <[self count]; i++) {
		if (strcasecmp (name,[[self objectAtIndex:i] classname]) < 0) {
			[self insertObject: ec atIndex:i];
			return;
		}
	}
	[self addObject:ec];
}


/*
=================
scanFile
=================
*/
-(void) scanFile:(char *) filename
{
	int         size, line;
	char       *data;
	id          cl;
	int         i;
	char        path[1024];
	QFile      *file;

	sprintf (path, "%s/%s", source_path, filename);

	file = Qopen (path, "rt");
	if (!file)
		return;
	size = Qfilesize (file);
	data = malloc (size + 1);
	size = Qread (file, data, size);
	data[size] = 0;
	Qclose (file);

	line = 1;
	for (i = 0; i < size; i++) {
		if (!strncmp (data + i, "/*QUAKED", 8)) {
			cl =[[EntityClass alloc] initFromText:(data + i)
			source:va ("%s:%d", filename, line)];
			if (cl)
				[self insertEC:cl];
		} else if (data[i] == '\n') {
			line++;
		}
	}

	free (data);
}


/*
=================
scanDirectory
=================
*/
-(void) scanDirectory
{
	int         count, i;
	struct dirent **namelist, *ent;

	[self removeAllObjects];

	count = scandir (source_path, &namelist, NULL, NULL);

	for (i = 0; i < count; i++) {
		int         len;

		ent = namelist[i];
		len = strlen (ent->d_name);
		if (len <= 3)
			continue;
		if (!strcmp (ent->d_name + len - 3, ".qc"))
			[self scanFile:ent->d_name];
	}
}


id          entity_classes_i;


-initForSourceDirectory:(char *) path
{
	[super init];

	source_path = path;
	[self scanDirectory];

	entity_classes_i = self;

	nullclass =[[EntityClass alloc]
	initFromText: "/*QUAKED UNKNOWN_CLASS (0 0.5 0) ?*/" source:va ("%s:%d", __FILE__,
					__LINE__ -
					1)];

	return self;
}

-(id) classForName:(char *) name
{
	int         i;
	id          o;

	for (i = 0; i <[self count]; i++) {
		o =[self objectAtIndex:i];
		if (!strcmp (name,[o classname]))
			return o;
	}

	return nullclass;
}


@end