/*
 * dir.c
 *
 * Developed by Ondrej Jombik <nepto@php.net>
 * Copyright (c) 2003 Platon SDG, http://www.platon.sk/
 * Licensed under terms of GNU General Public License.
 * All rights reserved.
 *
 * Changelog:
 * 19/05/2003 - created
 *
 */

/* $Platon: games/3do-view/dir.c,v 1.2 2003/05/20 15:25:27 nepto Exp $ */

#include <string.h>
#include <stdlib.h>
#include <system.h>

#if PLATON_SYSTEM_LINUX
#  include <unistd.h>
#  include <dirent.h>
#  include <sys/types.h>
#  include <sys/stat.h>
#  define	DIR_DIRECTORY_CONDITION		(S_ISDIR(st.st_mode))
#  define	DIR_FILENAME				(dp->d_name)
#  define	DIR_INIT_CONDITION			(! ((dirp = opendir(".")) == NULL \
					&& !chdir("/") && (dirp = opendir(".")) == NULL))
	/* If current directory cannot be opened, fallback to root directory. */
#endif

#if PLATON_SYSTEM_MSDOS
#  include <dos.h>
#  include <dir.h>
#  define	DIR_DIRECTORY_CONDITION		(ffblk.ff_attrib & FA_DIREC)
#  define	DIR_FILENAME				(ffblk.ff_name)
#  define	DIR_INIT_CONDITION			(! (findfirst("*.*", &ffblk, \
					FA_DIREC + FA_HIDDEN + FA_RDONLY + FA_SYSTEM)))
#endif

#define DIR_REALLOC_STEP			(10)

void get_directories_and_files(char ***directories, char ***files) /* {{{ */
{
#if PLATON_SYSTEM_LINUX
	DIR *dirp;
	struct dirent *dp;
	struct stat st;
#endif
#if PLATON_SYSTEM_MSDOS
	struct ffblk ffblk;
#endif
	register char ***ptr;
	int directories_size, files_size, *size;

	if (directories != NULL) *directories = NULL;
	if (files       != NULL) *files       = NULL;

	if (! DIR_INIT_CONDITION)
		return;

	directories_size = 0;
	files_size       = 0;

#if PLATON_SYSTEM_LINUX
	for (dp = readdir(dirp); dp != NULL; dp = readdir(dirp)) {
		if (stat(DIR_FILENAME, &st))
			continue;
#endif
#if PLATON_SYSTEM_MSDOS
	do {
#endif

		if (DIR_DIRECTORY_CONDITION) {
			if (directories == NULL || ! strcmp(DIR_FILENAME, "."))
				continue;
			ptr  = directories;
			size = &directories_size;
		} else {
			if (files == NULL)
				continue;
			ptr  = files;
			size = &files_size;
		}
		if (! *size || !((*size + 1) % DIR_REALLOC_STEP)) {
			register char **new_ptr;
			new_ptr = (char **) realloc((void *) *ptr,
					sizeof(char *) * (*size + 1 + DIR_REALLOC_STEP));
			if (new_ptr == NULL)
				break;
			*ptr = new_ptr;
		}
		if (((*ptr)[*size] = strdup(DIR_FILENAME)) == NULL)
			break;
		(*size)++;
		(*ptr)[*size] = NULL;
#if PLATON_SYSTEM_LINUX
	}
	closedir(dirp);
#endif
#if PLATON_SYSTEM_MSDOS
	} while (! findnext(&ffblk));
#endif
} /* }}} */

void free_directories_or_files(char **ar) /* {{{ */
{
	register int i;
	for (i = 0; ar[i] != NULL; i++)
		free(ar[i]);
	free(ar);
} /* }}} */

#if defined(SELF) || defined(SELFTEST) || defined(SELF_DIR)

#include <stdio.h>

int main(int argc, char **argv) /* {{{ */
{
	register int k;
	char **dirs, **files;

	get_directories_and_files(&dirs, &files);

	puts("Directories:");
	for (k = 0; dirs != NULL && dirs[k] != NULL; k++)
		printf("%03d: %s\n", k, dirs[k]);

	puts("Files:");
	for (k = 0; files != NULL && files[k] != NULL; k++)
		printf("%03d: %s\n", k, files[k]);

	return 0;
} /* }}} */

#endif /* #if defined(SELF) || defined(SELFTEST) || defined(SELF_DIR) */

/* Modeline for ViM {{{
 * vim: set ts=4:
 * vim600: fdm=marker fdl=0 fdc=0:
 * }}} */