// p_menu.c

#include "g_local.h"

// Note that the pmenu entries are duplicated. This is so that a static set of pmenu entries 
// can be used for multiple clients and changed without interference.
// Note that arg will be freed when the menu is closed; it must be allocated memory.
pmenuhnd_t *PMenu_Open(edict_t *ent, pmenu_t *entries, int cur, int num, void *arg)
{
	pmenuhnd_t	*hnd;
	pmenu_t		*p;
	int			i;

//CW++
	if (ent->isabot)
		return NULL;
//CW--

	if (!ent->client)
		return NULL;

	if (ent->client->menu)
	{
		gi.dprintf("warning, ent already has a menu\n");
		PMenu_Close(ent);
	}

	hnd = gi.TagMalloc(sizeof(*hnd), TAG_LEVEL);													//CW
	hnd->arg = arg;
	hnd->entries = gi.TagMalloc(sizeof(pmenu_t)*num, TAG_LEVEL);									//CW
	memcpy(hnd->entries, entries, sizeof(pmenu_t) * num);
	
	// duplicate the strings since they may be from static memory
	for (i = 0; i < num; i++)
	{
		if (entries[i].text)
			hnd->entries[i].text = strdup(entries[i].text);
	}
	hnd->num = num;

	if ((cur < 0) || !entries[cur].SelectFunc)
	{
		for (i = 0, p = entries; i < num; i++, p++)
		{
			if (p->SelectFunc)
				break;
		}
	}
	else
		i = cur;

	if (i >= num)
		hnd->cur = -1;
	else
		hnd->cur = i;

	ent->client->showscores = true;
	ent->client->inmenu = true;
	ent->client->menu = hnd;

	PMenu_Do_Update(ent);
	gi.unicast(ent, true);

	return hnd;
}

void PMenu_Close(edict_t *ent)
{
	pmenuhnd_t	*hnd;
	int			i;

	if (!ent->client->menu)
		return;

	hnd = ent->client->menu;
	for (i = 0; i < hnd->num; i++)
	{
		if (hnd->entries[i].text)
			free(hnd->entries[i].text);
	}

	gi.TagFree(hnd->entries);																		//CW

	if (hnd->arg)
		gi.TagFree(hnd->arg);																		//CW

	gi.TagFree(hnd);																				//CW

	ent->client->menu = NULL;
	ent->client->showscores = false;
}

// only use on pmenu's that have been called with PMenu_Open
void PMenu_UpdateEntry(pmenu_t *entry, const char *text, int align, SelectFunc_t SelectFunc)
{
	if (entry->text)
		free(entry->text);

	entry->text = strdup(text);
	entry->align = align;
	entry->SelectFunc = SelectFunc;
}

void PMenu_Do_Update(edict_t *ent)
{
	pmenuhnd_t	*hnd;
	pmenu_t		*p;
	char		string[1400];
	char		*t;
	int			x;
	int			i;
	qboolean	alt = false;
	size_t		stringLen;

	if (!ent->client->menu)
	{
		gi.dprintf("warning:  ent has no menu\n");
		return;
	}

	hnd = ent->client->menu;

	Com_strcpy(string, sizeof(string), "xv 32 yv 8 picn inventory ");
	for (i = 0, p = hnd->entries; i < hnd->num; i++, p++)
	{
		if (!p->text || !*(p->text))
			continue; // blank line

		t = p->text;
		if (*t == '*')
		{
			alt = true;
			t++;
		}
	//	sprintf(string + strlen(string), "yv %d ", 32 + i * 8);
		stringLen = strlen(string);
		Com_sprintf(string + stringLen, sizeof(string) - stringLen, "yv %d ", 32 + i * 8);
		if (p->align == PMENU_ALIGN_CENTER)
			x = 162 - ((int)strlen(t) * 4);
		else if (p->align == PMENU_ALIGN_RIGHT)
			x = 260 - ((int)strlen(t) * 8);
		else
			x = 58;

	//	sprintf(string + strlen(string), "xv %d ", x - ((hnd->cur == i) ? 8 : 0));
		stringLen = strlen(string);
		Com_sprintf(string + stringLen, sizeof(string) - stringLen, "xv %d ", x - ((hnd->cur == i) ? 8 : 0));

		if (hnd->cur == i) {
		//	sprintf(string + strlen(string), "string2 \"\x0d%s\" ", t);
			stringLen = strlen(string);
			Com_sprintf(string + stringLen, sizeof(string) - stringLen, "string2 \"\x0d%s\" ", t);
		}
		else if (alt) {
		//	sprintf(string + strlen(string), "string2 \"%s\" ", t);
			stringLen = strlen(string);
			Com_sprintf(string + stringLen, sizeof(string) - stringLen, "string2 \"%s\" ", t);
		}
		else {
		//	sprintf(string + strlen(string), "string \"%s\" ", t);
			stringLen = strlen(string);
			Com_sprintf(string + stringLen, sizeof(string) - stringLen, "string \"%s\" ", t);
		}
		alt = false;
	}

	gi.WriteByte(svc_layout);
	gi.WriteString(string);
}

void PMenu_Update(edict_t *ent)
{
	if (!ent->client->menu)
	{
		gi.dprintf("warning:  ent has no menu\n");
		return;
	}

	if (level.time - ent->client->menutime >= 1.0)
	{
		// been a second or more since last update, update now
		PMenu_Do_Update(ent);
		gi.unicast(ent, true);
		ent->client->menutime = level.time;
		ent->client->menudirty = false;
	}
	ent->client->menutime = level.time + 0.2;
	ent->client->menudirty = true;
}

void PMenu_Next(edict_t *ent)
{
	pmenuhnd_t	*hnd;
	pmenu_t		*p;
	int			i;

	if (!ent->client->menu)
	{
		gi.dprintf("warning:  ent has no menu\n");
		return;
	}

	hnd = ent->client->menu;

	if (hnd->cur < 0)
		return; // no selectable entries

	i = hnd->cur;
	p = hnd->entries + hnd->cur;
	do
	{
		i++, p++;
		if (i == hnd->num)
			i = 0, p = hnd->entries;
		if (p->SelectFunc)
			break;
	} while (i != hnd->cur);

	hnd->cur = i;

	PMenu_Update(ent);
}

void PMenu_Prev(edict_t *ent)
{
	pmenuhnd_t	*hnd;
	pmenu_t		*p;
	int			i;

	if (!ent->client->menu)
	{
		gi.dprintf("warning:  ent has no menu\n");
		return;
	}

	hnd = ent->client->menu;

	if (hnd->cur < 0)
		return; // no selectable entries

	i = hnd->cur;
	p = hnd->entries + hnd->cur;
	do
	{
		if (i == 0)
		{
			i = hnd->num - 1;
			p = hnd->entries + i;
		}
		else
			i--, p--;
		if (p->SelectFunc)
			break;
	} while (i != hnd->cur);

	hnd->cur = i;

	PMenu_Update(ent);
}

void PMenu_Select(edict_t *ent)
{
	pmenuhnd_t	*hnd;
	pmenu_t		*p;

	if (!ent->client->menu)
	{
		gi.dprintf("warning:  ent has no menu\n");
		return;
	}

	hnd = ent->client->menu;

	if (hnd->cur < 0)
		return; // no selectable entries

	p = hnd->entries + hnd->cur;

	if (p->SelectFunc)
		p->SelectFunc(ent, hnd);
}