/*
* Copyright (c) 2005-2009 Nokia Corporation and/or its subsidiary(-ies).
* All rights reserved.
* This component and the accompanying materials are made available
* under the terms of "Eclipse Public License v1.0"
* which accompanies this distribution, and is available
* at the URL "http://www.eclipse.org/legal/epl-v10.html".
*
* Initial Contributors:
* Nokia Corporation - initial contribution.
*
* Contributors:
*
* Description:  
* System Include
*
*/



#include <stdio.h>
#include <windows.h>
#include <direct.h>
#include <assert.h>

/*******************************************************************************
 *
 * Local Include
 *
 ******************************************************************************/
#include "cleantree.h"

/*******************************************************************************
 *
 * Definitions
 *
 ******************************************************************************/
#define MAXPATHLEN	1024

/*******************************************************************************
 *
 * Types
 *
 ******************************************************************************/
typedef void	(*TProcessFunction)(char*,int); 

/*******************************************************************************
 *
 * Prototypes
 *
 ******************************************************************************/
int ProcessDirectoryTree( char *path, int depth, TProcessFunction func );
int UpdateWorkingDirectory( char *previous, char *next, char *path, int op );
void PrintPath( char *str, int depth );
void ProcessFile( char *filename, int depth );

/*******************************************************************************
 *
 * Filescope statics
 *
 ******************************************************************************/
static int debug_flag = 0;
static char *dirname = NULL;
static int deleted_file_count;
static int deleted_directory_count;

/*******************************************************************************
 *
 * Main
 *
 ******************************************************************************/
int main( int argc, char *argv[] )
{
	char fname[1024];
	int slen;

	// check params
	if( argc < 3 ) {
		fprintf( stderr, "usage: cleantree path dirname\n" );
		return -1;
	}
	dirname = argv[2];

	// get the name into the correct format
	strcpy( fname, argv[1] );
	slen = strlen(fname);
	if( (fname[slen-1] != '*') || (fname[slen-2] != '\\') ) {
		fprintf( stderr, "ERROR: pathname must end in '\\*'.\n" );
		return -1;
	}

	// process the directory tree 
	ProcessDirectoryTree( argv[1], 0, ProcessFile );

	// print stats
	printf( "%d files deleted from %d directories\n", deleted_file_count, deleted_directory_count );

	// done
	fflush( stdout );
	return 0;
}

/*******************************************************************************
 *
 * ProcessFile
 *
 ******************************************************************************/
void ProcessFile( char *filename, int depth )
{
	if( debug_flag > 0 ) {
		unlink( filename );
		deleted_file_count++;
	}
}

/*******************************************************************************
 *
 * ProcessDirectoryTree
 *
 ******************************************************************************/
int ProcessDirectoryTree( char *path, int depth, TProcessFunction func )
{
	WIN32_FIND_DATA finfo;
	HANDLE list;
	int plen;
	int err = 1;
	int match;
	char previous_working_directory[MAXPATHLEN];
	char new_working_directory[MAXPATHLEN];
	char fullname[MAXPATHLEN];

	// ignore . and ..
	if( path[0] == '.' ) {
		return 0;
	}

	// ignore any dir beginning with _
	if( path[0] == '_' ) {
		return 0;
	}

	// print the path
//	PrintPath( path, (depth - 1));

	// check if we are in a debug directory
	plen = strlen( path );
	assert( path[plen-1] == '*' );
	assert( path[plen-2] == '\\' );
	path[plen-2] = 0;
	match = strcmp( path, dirname );
	if( match == 0 ) {
		debug_flag++;
	}
	path[plen-2] = '\\';
	
	// get the first file in the directory
	list = FindFirstFile( path, &finfo );
	if( list == INVALID_HANDLE_VALUE ) {
		return 0;
	}

	// update the current working directory
	UpdateWorkingDirectory( previous_working_directory, new_working_directory, path, ((depth == 0) ? 1 : 0) );

	// loop through the rest of the files
	for( err = 1; err != 0; err ) {

		// process the file
		if( finfo.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ) {
			strcat( finfo.cFileName, "\\*" );
			ProcessDirectoryTree( finfo.cFileName, depth + 1, func );
		} else {
			strcpy( fullname, new_working_directory );
			strcat( fullname, "\\" );
			strcat( fullname, finfo.cFileName );
			func( fullname, depth );
		}

		// get the info for the next file
		err = FindNextFile( list, &finfo );
	}

	// update the working directory and close the directory
	UpdateWorkingDirectory( previous_working_directory, new_working_directory, path, 2 );
	FindClose( list );

	// now delete the directory
	if( match == 0 ) {
		err = RemoveDirectory( new_working_directory );
		if( err == 0 ) {
			fprintf( stderr, "ERROR: failed to remove directory (%d)\n", GetLastError() );
		}
		deleted_directory_count++;
		debug_flag--;
	}

	return 0;
}


/*******************************************************************************
 *
 * UpdateWorkingDirectory
 *
 ******************************************************************************/
int UpdateWorkingDirectory( char *previous, char *next, char *path, int op )
{
	int len;

	// op zero means normal PUSH of the working directory
	if( op == 0 ) {
		getcwd( previous, MAXPATHLEN );
		getcwd( next, MAXPATHLEN );
		len = strlen( next );
		next[len] = '\\';
		next[len+1] = 0;
		strcat( next, path );
		len = strlen(next);
		next[len-2] = 0;
		chdir( next );
	} 

	// op one is initial set of the dirs
	if( op == 1 ) {
		getcwd( previous, MAXPATHLEN );
		strcpy( next, path );
		len = strlen(next);
		next[len-1] = 0;
		chdir( next );
	}

	// op two is pop
	if( op == 2 ) {
		chdir( previous );
	}

	// done
	return 0;
}


/*******************************************************************************
 *
 * Print Path
 *
 ******************************************************************************/
void PrintPath( char *str, int depth )
{
	for( int i = 0; i < (depth - 1); i++ )
		printf( "\t" );
	printf( "%s\n", str );
}