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

#include "../shared/sockets.h"
#include "../shared/debug.h"

#include "screen.h"
#include "widget.h"

char *types[] = { "none",
	"string",
	"hbar",
	"vbar",
	"icon",
	"title",
	"scroller",
	"frame",
	"num",
	//"",
	NULL,
};

static widget *widget_finder (LL * list, char *id);
static int widget_remover (LL * list, widget * w);

widget *
widget_create ()
{
	widget *w;

	debug ("widget_create()\n");

	w = malloc (sizeof (widget));
	if (!w) {
		fprintf (stderr, "widget_create: Error allocating widget\n");
		return NULL;
	}

	w->id = NULL;
	w->type = 0;
	w->x = 1;
	w->y = 1;
	w->wid = 0;
	w->hgt = 0;
	w->left = 1;
	w->top = 1;
	w->right = 0;
	w->bottom = 0;
	w->length = 1;
	w->speed = 1;
	w->text = NULL;
	w->kids = NULL;

	return w;
}

int
widget_destroy (widget * w)
{
	LL *list;
	widget *foo;

	if (!w)
		return -1;

	debug ("widget_destroy(%s)\n", w->id);

	if (w->id)
		free (w->id);
	//debug("widget_destroy: id...\n");
	if (w->text)
		free (w->text);
	//debug("widget_destroy: text...\n");

	// TODO: Free kids!
	if (w->kids) {
		list = w->kids;
		LL_Rewind (list);
		do {
			foo = LL_Get (list);
			if (foo)
				widget_destroy (foo);
		} while (0 == LL_Next (list));

		LL_Destroy (list);
		w->kids = NULL;
	}

	free (w);
	//debug("widget_destroy: widget...\n");

	return 0;
}

widget *
widget_find (screen * s, char *id)
{
	//widget *w, *err;

	if (!s)
		return NULL;
	if (!id)
		return NULL;

	debug ("widget_find(%s)\n", id);

	return widget_finder (s->widgets, id);
/*
   LL_Rewind(s->widgets);
   do {
      w = LL_Get(s->widgets);
      if( (w)  &&  (0 == strcmp(w->id, id)))
      {
	 debug("widget_find:  Found %s\n", id);
	 return w;
      }
      // TODO:  Search kids too!
      if( w->type == WID_FRAME )
      {
	 err = widget_finder(w->kids, id);
	 if(err) return err;
      }
      
   } while(LL_Next(s->widgets) == 0);

   return NULL;
*/
}

widget *
widget_finder (LL * list, char *id)
{
	widget *w, *err;

	if (!list)
		return NULL;
	if (!id)
		return NULL;

	debug ("widget_finder(%s)\n", id);

	LL_Rewind (list);
	do {
		//debug("widget_finder: Iteration\n");
		w = LL_Get (list);
		if (w) {
			//debug("widget_finder: ...\n");
			if (0 == strcmp (w->id, id)) {
				debug ("widget_finder:  Found %s\n", id);
				return w;
			}
			// Search kids recursively
			//debug("widget_finder: ...\n");
			if (w->type == WID_FRAME) {
				err = widget_finder (w->kids, id);
				if (err)
					return err;
			}
			//debug("widget_finder: ...\n");
		}

	} while (LL_Next (list) == 0);

	return NULL;
}

int
widget_add (screen * s, char *id, char *type, char *in, int sock)
{
	int i;
	int valid = 0;
	int wid_type = 0;
	widget *w, *parent;
	LL *list;

	debug ("widget_add(%s, %s, %s)\n", id, type, in);

	if (!s)
		return -1;
	if (!id)
		return -1;

	list = s->widgets;

	if (0 == strcmp (id, "heartbeat")) {
		s->heartbeat = 1;
		return 0;
	}
	// Make sure this screen doesn't already exist...
	w = widget_find (s, id);
	if (w) {
		// already exists
		sock_send_string (sock, "huh? You already have a widget with that id#\n");
		return 1;
	}
	// Make sure the container, if any, is real
	if (in) {
		parent = widget_find (s, in);
		if (!parent) {
			// no frame to use as parent
			sock_send_string (sock, "huh? Frame doesn't exist\n");
			return 3;
		} else {
			if (parent->type == WID_FRAME) {
				list = parent->kids;
				if (!list)
					fprintf (stderr, "widget_add: Parent has no kids\n");
			} else {
				// no frame to use as parent
				sock_send_string (sock, "huh? Not a frame\n");
				return 4;
			}
		}
	}
	// Make sure it's a valid widget type
	for (i = 1; types[i]; i++) {
		if (0 == strcmp (types[i], type)) {
			valid = 1;
			wid_type = i;
		}
	}
	if (!valid) {
		// invalid widget type
		sock_send_string (sock, "huh? Invalid widget type\n");
		return 2;
	}

	debug ("widget_add: making widget\n");

	// Now, make it...
	w = widget_create ();
	if (!w) {
		fprintf (stderr, "widget_add:  Error creating widget\n");
		return -1;
	}

	w->id = strdup (id);
	if (!w->id) {
		fprintf (stderr, "widget_add:  Error allocating id\n");
		return -1;
	}

	w->type = wid_type;

	// Set up the container's widget list...
	if (w->type == WID_FRAME) {
		if (!w->kids) {
			w->kids = LL_new ();
			if (!w->kids) {
				fprintf (stderr, "widget_add: error allocating kids for frame\n");
				return -1;
			}
		}
	}
	// TODO:  Check for errors here?
	LL_Push (list, (void *) w);

	return 0;
}

int
widget_remove (screen * s, char *id, int sock)
{
	widget *w;
	LL *list;

	debug ("widget_remove(%s)\n", id);

	if (!s)
		return -1;
	if (!id)
		return -1;

	list = s->widgets;

	if (0 == strcmp (id, "heartbeat")) {
		s->heartbeat = 0;
		return 0;
	}
	// Make sure this screen *does* exist...
	w = widget_find (s, id);
	if (!w) {
		sock_send_string (sock, "huh? You don't have a widget with that id#\n");
		debug ("widget_remove:  Error finding widget %s\n", id);
		return 1;
	}

/*
   // TODO:  Check for errors here?
   // TODO:  Make this work with frames...
//   LL_Remove(list, (void *)w);

   
   // TODO:  Check for errors here?
//   widget_destroy(w);
*/

	return widget_remover (list, w);

//   return 0;
}

int
widget_remover (LL * list, widget * w)
{
	widget *foo, *bar;
	int err;

	debug ("widget_remover\n");

	if (!list)
		return 0;
	if (!w)
		return 0;

	// Search through the list...
	LL_Rewind (list);
	do {
		// Test each item
		foo = LL_Get (list);
		if (foo) {
			// Frames require recursion to search and/or destroy
			if (foo->type == WID_FRAME) {
				// If removing a frame, kill all its kids, too...
				if (foo == w) {
					if (!foo->kids) {
						fprintf (stderr, "widget_remover: frame has no kids\n");
					} else			  // Kill the kids...
					{
						LL_Rewind (foo->kids);
						do {
							bar = LL_Get (foo->kids);
							err = widget_remover (foo->kids, bar);
						} while (0 == LL_Next (foo->kids));

						// Then kill the parent...
						LL_Remove (list, (void *) w);
						widget_destroy (w);
					}

				} else				  // Otherwise, search the frame recursively...
				{
					if (!foo->kids) {
						fprintf (stderr, "widget_remover: frame has no kids\n");
					} else
						err = widget_remover (foo->kids, w);

				}
			} else					  // If not a frame, remove it if it matches...
			{
				if (foo == w) {
					LL_Remove (list, (void *) w);
					widget_destroy (w);
				}
			}
		}
	} while (0 == LL_Next (list));

	return 0;
}