FC - FONCTION (NUX ONLY)

/*******************************************************************************************
* Autheur : kikuro
* Date : 20/10/2004
* Mail : bramyc@gmail.com
* Thanks : Merci K-J158 pour le temps que tu perd à m'expliquer des trucs tout con ^^
* Contact : bastia.zefunchat.net #prog  #programmation
**************************************************************************************************/
#include <stdio.h>
#include <sys/types.h>
#include <dirent.h>
#include <string.h>
#include <sys/stat.h>

int analyse(char folder[], char type[]);
int arganalyse(char arg[], int nbarg);
void argError(char error[]);

int main(int argc, char *argv[])
{
	int etat, nbFile;

	/*   Si aucun argument n'est fournit on affiche une erreur.   */
	if(argc < 2)
	{
		puts("Argument(s) manquant(s) \"-h\" ou \"--help\" pour de l'aide");
		return -1;
	}
	
	/*   Sinon on analyse les arguments.   */
	etat = arganalyse(argv[1], argc);

	/*   En fonction de l'analyse on lance des fonctions.   */
	switch(etat)
	{
		case 0 : nbFile = 0; break;
		case 1 : nbFile = analyse(argv[1], "df"); return 0; break;
		case 2 : nbFile = analyse(argv[2], "f"); return 0; break;
		case 3 : nbFile = analyse(argv[2], "d"); return 0; break;
		case 4 : nbFile = analyse(argv[2], "df"); return 0; break;
		case -1 : nbFile = 0; argError("Mauvais argument, \"-h\" ou \"--help\" pour de l'aide"); return -1; break;
	}
	return nbFile;
}

int analyse(char folder[], char type[])
{
	DIR *fold;
	struct dirent *read;
	struct stat infos;
	int total = 0, nbFiles = 0, nbDir = 0;
	/*   Si on doit afficher et le nombre de fichier et le nombre de dossier.   */
	if(!strcmp(type, "df"))
	{
		fold = opendir(folder);
		while(read = readdir(fold))
		{
			stat(read->d_name, &infos);
			if(S_ISDIR(infos.st_mode))
			{
				nbDir++;
			}
			if(S_ISREG(infos.st_mode))
			{
				nbFiles++;
			}
			total++;
		}
		printf("Fichiers : %d\nDossiers : %d\nTotal : %d\n", nbFiles, nbDir, total);
	}
	if(!strcmp(type, "d"))
	{
		fold = opendir(folder);
		while(read = readdir(fold))
		{
			stat(read->d_name, &infos);
			if(S_ISDIR(infos.st_mode))
			{
				total++;
			}
		}
		printf("Total : %d\n", total);
	}
	if(!strcmp(type, "f"))
	{
		fold = opendir(folder);
		while(read = readdir(fold))
		{
			stat(read->d_name, &infos);
			if(S_ISREG(infos.st_mode))
			{
				total++;
			}
		}
		printf("Total : %d\n", total);
	}
	return total;
}

int arganalyse(char arg[], int nbarg)
{
	/*   Si le nombre d'argument est egal a deux.   */
	if (nbarg == 2)
	{
		/*   Si le second argument est soit -h ou bien --help on affiche l'aide.   */
		if(!strcmp(arg, "-h") || !strcmp(arg, "--help"))
		{
			char help[] = "fc compte le nombre de fichiers et/ou de dossiers que contient un dossier.\n	arguments :	-f  : Uniquement les fichiers.\n			-d  : Uniquement les dossiers.\n			-df : Equivalent d'aucun arguments ( fichier et dossiers ).\nSynthaxe : fc [-df] <chemin vers dossier>";
			printf("%s\n", help);
			return 0;
		}
		/*   Sinon on dit que le second argument joue le role de dossier.   */
		else
		{
			return 1;
		}
	}
	/*   Si le nombre d'argument est superieur à 2.   */
	if(nbarg > 2)
	{
		/*   Si l'argument est -f alors on indique qu'on compte que les fichiers.   */
		if(!strcmp(arg, "-f"))
		{
			return 2;
		}
		/*   Si l'argument est -d alors on indique qu'on compte que les dossiers.   */
		if(!strcmp(arg, "-d"))
		{
			return 3;
		}
		/*   Si 'largument est -df alors on indique qu'on compte et les fichiers et les dossiers.   */
		if(!strcmp(arg, "-df"))
		{
			return 4;
		}
		/*   Si l'argument est -fd alors on indique qu'on compte et les fichiers et les dossiers.   */
		if(!strcmp(arg, "-fd"))
		{
			return 4;
		}
		/*   Sinon on renvoit une erreur d'argument.   */
		return -1;
	}
	/*   Sinon on renvoit une erreur d'argument (sécurité).   */
	return -1;	
}

void argError(char error[])
{
	printf("Erreur : %s.\n", error);
}