/*
Copyright (C) 2003-2006 Andrey Nazarov

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  

See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

*/

#include "ui_local.h"

/*
=======================================================================

DEMOS MENU

=======================================================================
*/

#define MAX_MENU_DEMOS	1024

#define DEMO_EXTENSIONS ".dm2;.dm2.gz;.mvd2;.mvd2.gz"

#define FFILE_UP	1
#define FFILE_FOLDER	2
#define FFILE_DEMO	3

#define ID_LIST			105

typedef struct {
    int type;
    int size;
    time_t mtime;
} demoEntry_t;

typedef struct m_demos_s {
	menuFrameWork_t		menu;
	menuList_t		list;
	menuStatic_t	banner;

	int			count;
	char		*names[MAX_MENU_DEMOS+1];
	demoEntry_t	entries[MAX_MENU_DEMOS];
} m_demos_t;

static m_demos_t	m_demos;

static void Demos_BuildName( const char *path, fsFileInfo_t *info ) {
    char buffer[MAX_OSPATH];
    demoInfo_t demo;

#if 1
	Q_concat( buffer, sizeof( buffer ), path, "/", info->name, NULL );
	if( !client.GetDemoInfo( buffer, &demo ) ) {
        return;
    }
#else
        strcpy( demo.map, "???" );
        strcpy( demo.pov, "???" );
#endif

    if( info->size >= 1000000 ) {
        sprintf( buffer, "%2.1fM", ( float )info->size / 1000000 );
    } else if( info->size >= 1000 ) {
        sprintf( buffer, "%3dK", info->size / 1000 );
    } else {
        sprintf( buffer, "%3db", info->size );
    }

    m_demos.names[m_demos.count] = UI_FormatColumns( 4,
        info->name, buffer, demo.map, demo.pov );
    m_demos.entries[m_demos.count].type = FFILE_DEMO;
    m_demos.entries[m_demos.count].size = info->size;
    m_demos.entries[m_demos.count].mtime = info->mtime;
    m_demos.count++;
}

static void Demos_WriteCache( const char *path ) {
    char buffer[MAX_OSPATH];
    fileHandle_t f;
    int i;
    char *name, *map, *pov;
    demoEntry_t *e;

	if( *path == '/' ) {
		path++;
	}

	Q_concat( buffer, sizeof( buffer ), path, "/.democache", NULL );
    fs.FOpenFile( buffer, &f, FS_MODE_WRITE );
    if( !f ) {
        return;
    }

	for( i = 0; i < m_demos.count; i++ ) {
        e = &m_demos.entries[i];
        if( e->type != FFILE_DEMO ) {
            continue;
        }
        name = m_demos.names[i];
        map = name + strlen( name ) + 1;
        map = map + strlen( map ) + 1;
        pov = map + strlen( map ) + 1;
        fs.FPrintf( f, "%s %d %d %s %s\n", name, e->mtime, e->size, map, pov );
    }
    fs.FCloseFile( f );
}

static void Demos_BuildList( const char *path ) {
	int numFiles;
	void **list;
	int i, pos;

	if( *path == '/' ) {
		path++;
	}
	
	if( *path ) {
		m_demos.names[m_demos.count] = UI_FormatColumns( 1, ".." );
		m_demos.entries[m_demos.count].type = FFILE_UP;
		m_demos.count++;
	}

	// list directories first
	list = fs.ListFiles( path, NULL, FS_PATH_GAME |
        FS_SEARCHDIRS_ONLY, &numFiles );
	if( list ) {
        if( m_demos.count + numFiles > MAX_MENU_DEMOS ) {
            numFiles = MAX_MENU_DEMOS - m_demos.count;
        }
		for( i = 0; i < numFiles; i++ ) {
			m_demos.names[m_demos.count] = UI_FormatColumns( 1, list[i] );
			m_demos.entries[m_demos.count].type = FFILE_FOLDER;
			m_demos.count++;
		}
		fs.FreeList( list );
	}	

	pos = m_demos.count;

    // list demos
	list = fs.ListFiles( path, DEMO_EXTENSIONS, FS_PATH_GAME |
        FS_SEARCH_EXTRAINFO, &numFiles );
	if( list ) {
        if( m_demos.count + numFiles > MAX_MENU_DEMOS ) {
            numFiles = MAX_MENU_DEMOS - m_demos.count;
        }
        for( i = 0; i < numFiles; i++ ) {
            Demos_BuildName( path, list[i] );
        }

        // sort them
        qsort( m_demos.names + pos, m_demos.count - pos,
            sizeof( m_demos.names[0] ), SortStrcmp );

        fs.FreeList( list );
    }

    // terminate the list
	m_demos.names[m_demos.count] = NULL;
}

static void Demos_Free( void ) {
	int i;

    Demos_WriteCache( uis.m_demos_browse );

	for( i = 0; i < m_demos.count; i++ ) {
		com.Free( m_demos.names[i] );
		m_demos.names[i] = NULL;
        memset( &m_demos.entries[i], 0, sizeof( m_demos.entries[0] ) );
	}

	m_demos.count = 0;
	m_demos.list.curvalue = 0;
}

static void Demos_LeaveDirectory( void ) {
	char buffer[MAX_OSPATH];
	char *s;
	int i;

	s = strrchr( uis.m_demos_browse, '/' );
	if( s ) {
		*s = 0;
		strcpy( buffer, s + 1 );
	} else {
		buffer[0] = 0;
	}

	// rebuild list
	Demos_Free();
	Demos_BuildList( uis.m_demos_browse );
	MenuList_Init( &m_demos.list );

	if( s == uis.m_demos_browse ) {
		uis.m_demos_browse[0] = '/';
		uis.m_demos_browse[1] = 0;
	}

	// move cursor to the previous directory
	if( buffer[0] ) {
		for( i = 0; i < m_demos.count; i++ ) {
			if( !strcmp( m_demos.names[i], buffer ) ) {
				MenuList_SetValue( &m_demos.list, i );
				return;
			}
		}
	}

	MenuList_SetValue( &m_demos.list, 0 );
}

static int Demos_Action( void ) {
	char buffer[MAX_OSPATH];
	int length, baseLength;
    char *s = m_demos.names[m_demos.list.curvalue];
    demoEntry_t *e = &m_demos.entries[m_demos.list.curvalue];

	switch( e->type ) {
	case FFILE_UP:
		Demos_LeaveDirectory();
		return QMS_OUT;

	case FFILE_FOLDER:
		baseLength = strlen( uis.m_demos_browse );
		length = strlen( s );
		if( baseLength + length > sizeof( uis.m_demos_browse ) - 2 ) {
			return QMS_BEEP;
		}
		if( uis.m_demos_browse[ baseLength - 1 ] != '/' ) {
			uis.m_demos_browse[ baseLength++ ] = '/';
		}

		strcpy( uis.m_demos_browse + baseLength, s );
		
		// rebuild list
		Demos_Free();
		Demos_BuildList( uis.m_demos_browse );
		MenuList_Init( &m_demos.list );
		return QMS_IN;

	case FFILE_DEMO:
		Com_sprintf( buffer, sizeof( buffer ), "%s \"%s/%s\"\n",
            /*m_demos.demo.mvd ? "mvdplay" : */"demo",
			uis.m_demos_browse, s );
		cmd.ExecuteText( EXEC_APPEND, buffer );
		return QMS_SILENT;
	}

	return QMS_NOTHANDLED;
}

static int Demos_MenuCallback( int id, int msg, int param ) {
	switch( msg ) {
	case QM_ACTIVATE:
		switch( id ) {
		case ID_LIST:
			return Demos_Action();
		}
		return QMS_IN;

	case QM_CHANGE:
		switch( id ) {
		case ID_LIST:
            if( m_demos.entries[m_demos.list.curvalue].type == FFILE_DEMO ) {
                m_demos.menu.statusbar = "Press Enter to play";
            } else {
                m_demos.menu.statusbar = NULL;
            }
			break;
		default:
			break;
		}
		return QMS_MOVE;

	case QM_KEY:
		switch( id ) {
		case ID_LIST:
			switch( param ) {
			case K_BACKSPACE:
				Demos_LeaveDirectory();
				return QMS_OUT;
			default:
				break;
			}
		default:
			break;
		}
		break;

	case QM_DESTROY:
		uis.m_demos_selection = m_demos.list.curvalue; // save previous position
		Demos_Free();
		break;

	default:
		break;
	}

	return QMS_NOTHANDLED;
}

static void Demos_MenuInit( void ) {
	memset( &m_demos, 0, sizeof( m_demos ) );

	m_demos.menu.callback = Demos_MenuCallback;

	Demos_BuildList( uis.m_demos_browse );

	m_demos.list.generic.type	= MTYPE_LIST;
	m_demos.list.generic.id		= ID_LIST;
	m_demos.list.generic.flags  = QMF_HASFOCUS;
	m_demos.list.generic.x		= 10;
	m_demos.list.generic.y		= 32;
	m_demos.list.generic.width	= 0;
	m_demos.list.generic.height	= uis.height - 64;
	m_demos.list.generic.name	= NULL;
	m_demos.list.itemnames		= ( const char ** )m_demos.names;
	m_demos.list.drawNames		= qtrue;
	m_demos.list.numcolumns     = 4;

	m_demos.list.columns[0].width = uis.width - 30 - 196;
	m_demos.list.columns[0].name = uis.m_demos_browse;
	m_demos.list.columns[0].uiFlags = UI_LEFT;

	m_demos.list.columns[1].width = 40;
	m_demos.list.columns[1].name = "Size";
	m_demos.list.columns[1].uiFlags = UI_CENTER;

	m_demos.list.columns[2].width = 60;
	m_demos.list.columns[2].name = "Map";
	m_demos.list.columns[2].uiFlags = UI_CENTER;

	m_demos.list.columns[3].width = 96;
	m_demos.list.columns[3].name = "POV";
	m_demos.list.columns[3].uiFlags = UI_CENTER;

	UI_SetupDefaultBanner( &m_demos.banner, "Demo Browser" );

	Menu_AddItem( &m_demos.menu, (void *)&m_demos.list );
	Menu_AddItem( &m_demos.menu, (void *)&m_demos.banner );

	// move cursor to previous position
	MenuList_SetValue( &m_demos.list, uis.m_demos_selection );
}

void M_Menu_Demos_f( void ) {
	Demos_MenuInit();
	UI_PushMenu( &m_demos.menu );
}