#!/usr/bin/perl
#use strict;
#use warnings;
use Inline C;

if( scalar(@ARGV) < 2 ){
	print "Usage: $0 [directory] [file extension]\n";
	exit 1;
}
my $count = count_file_extension_matches(@ARGV);
print $count,"\n";


__END__
__C__

/*
 * recursively search for a file extension
 * within the directory given.
 *
 */

int count_file_extension_matches(const char* dirname,const char* extension){
	DIR *tmp = opendir(dirname); 
	struct dirent *file;
	struct stat file_stat;
	int total_count = 0;
	char* result;

	if( tmp != NULL ){

		if( chdir(dirname) == -1 ){
			printf("chdir failed\n");
			return 0;
		}

		while( (file = readdir(tmp)) ){

			/* make sure we don't have . or .. */
			if( (strcmp(file->d_name,"..")) == 0 || (strcmp(file->d_name,".") == 0) )
				continue;

			/* setup file_stat struct */
			if( (stat(file->d_name,&file_stat)) == -1)
				continue;

			/* if its a directory, recurse */
			if( S_ISDIR(file_stat.st_mode) ){

				total_count += count_file_extension_matches(file->d_name,extension);

				if( chdir("..") == -1 ){
					printf("chdir .. failed\n");
					return 0;
				}

				continue;
			}

			/* only look at regular files */
			if( S_ISREG(file_stat.st_mode) ){

				/* continue unless our extension is found within the filename */
				if( ( result = strstr(file->d_name,extension) ) == NULL )
					continue;

				/* result should match extension, continue otherwise */
				if( strcmp(extension,result) != 0 )
					continue;

				total_count += 1;
			}

		}
		closedir(tmp);

	}else{
		printf("Error opening directory: %s\n",dirname);
		exit(1);
	}

	return total_count;
}