Mercurial > hg > cline / file revision

#include <stdio.h>
#include <string.h>
#include <stdbool.h>
#include <stdlib.h>
#include <dirent.h>

static int suffixc;
static char** suffixv;
static bool recursive;
static bool includeSuffixes;

bool testSuffix(char* filename)
{
  bool ret = false;
  int tokenlen, fnamelen = strlen(filename);
  for (int t = 0 ; t < suffixc ; t++)
  {
    tokenlen = strlen(suffixv[t]);
    if (fnamelen >= tokenlen)
    {
      if (strncmp(filename+fnamelen-tokenlen, suffixv[t], tokenlen) == 0)
      {
        ret = true;
        break;
      }
    }
  }
  return ret ^ !includeSuffixes;
}

int scanDirectory(DIR *dir, const int spaces, char* currdir)
{
  DIR *subdir;
  char* subdirname;
  struct dirent *entry;
  int lines, digits, a;
  int lineSum = 0;

  while ((entry = readdir(dir)) != NULL)
  {
    if (strcmp(entry->d_name, ".") != 0 && strcmp(entry->d_name, "..") != 0)
    {
      // Print occurence
      char entryname[strlen(entry->d_name)+spaces];
      for (int t = 0 ; t < spaces ; t++)
      {
        entryname[t]=' ';
      }
      entryname[spaces] = 0;
      strcat(entryname, entry->d_name);
      printf("%-60s", entryname);
  
      // Check for subdirectory
      char subdirname[(1+strlen(currdir)+strlen(entry->d_name))];
      strcpy(subdirname, currdir);
      strcat(subdirname, "/");
      strcat(subdirname, entry->d_name);
      if ((subdir = opendir(subdirname)) != NULL)
      {
        printf("\n");
        if (recursive)
        {
          lineSum += scanDirectory(subdir, spaces+1, subdirname);
        }
        closedir(subdir);
        continue;
      }

      // Count lines
      lines = 0;
      char filename[(1+strlen(currdir)+strlen(entry->d_name))];
      strcpy(filename, currdir);
      strcat(filename, "/");
      strcat(filename, entry->d_name);
      if (testSuffix(filename))
      {
        FILE *file = fopen(filename, "r");
        if (file == NULL)
        {
          perror("  File acces failed");
          continue;
        }

        do
        {
          a = fgetc(file);

          if (a == 10)
          {
            lines++;
          }
        } while (a != EOF);
        fclose(file);

        // Print line count
        printf("%14d lines\n", lines);

        lineSum += lines;
      }
      else
      {
        // Print hint
        printf("%20s\n", "no match");
      }
    }
  }
  return lineSum;
}

int main(int argc, char** argv)
{
  // Help text
  const char* helpText = 
    "\nUsage:\n%s [-h|--help|(-s=<suffix>|-S=<suffix>)|(-r|-R)|<directory>]"
    "\n\nCounts the line terminator characters (\\n) within all"
    " files in the specified\ndirectory."
    "\n\nOptions:"
    "\n  -h, --help          - this help text"
    "\n  -s=<suffixes>       - only count files with these suffixes (separated"
    "\n                        by commas)"
    "\n  -S=<suffixes>       - count any file except those with these suffixes"
    "\n                        (separated by commas)"
    "\n  -R                  - excludes subdirectories"
    "\n  -r                  - includes subdirecotires"
    "\n\n"
    "The default call without any options is:"
    "\n  %s -r ./ -S=\n"
    "That means each file in each subdirectory is counted. If you want to count"
    "\nC source code in your working directory and its subdirectories, type:"
    "\n  %s -s=.c\n";

  // Program name
  char* prgName = strrchr(argv[0], '/');
  if (prgName == NULL)
  {
    prgName = argv[0];
  }
  else
  {
    prgName++;
  }

  // Defaults
  char* _suffix = "";
  char* _directory = "./";

  // Get arguments
  char* directory;
  char* suffix;
  bool showHelp = false;
  recursive = true;
  includeSuffixes = false;
  char checked = 0;

  for (int t = 1 ; t < argc ; t++)
  {
    if (strncmp(argv[t], "-s=", 3) == 0)
    {
      if ((checked & 1) > 0)
      {
        printf(helpText, prgName, prgName, prgName);
        return -1;
      }
      includeSuffixes = true;
      suffix = argv[t]+3; 
      checked |= 1;
    }
    else if (strncmp(argv[t], "-S=", 3) == 0)
    {
      if ((checked & 1) > 0)
      {
        printf(helpText, prgName, prgName, prgName);
        return -1;
      }
      includeSuffixes = false;
      suffix = argv[t]+3;
      checked |= 1;
    }
    else if (strcmp(argv[t], "-h") == 0 || strcmp(argv[t], "--help") == 0)
    {
      if ((checked & 2) > 0)
      {
        printf(helpText, prgName, prgName, prgName);
        return -1;
      }
      checked |= 2;
      showHelp = true;
    }
    else if (strcmp(argv[t], "-r") == 0)
    {
      if ((checked & 4) > 0)
      {
        printf(helpText, prgName, prgName, prgName);
        return -1;
      }
      checked |= 4;
      recursive = true;
    }
    else if (strcmp(argv[t], "-R") == 0)
    {
      if ((checked & 4) > 0)
      {
        printf(helpText, prgName, prgName, prgName);
        return -1;
      }
      checked |= 4;
      recursive = false;
    }
    else
    {
      if ((checked & 8) > 0)
      {
        printf(helpText, prgName, prgName, prgName);
        return -1;
      }
      checked |= 8;      
      directory = argv[t];
    }
  }

  // Show help and quit
  if (showHelp)
  {
    printf(helpText, prgName, prgName, prgName);
    return 0;
  }

  // Default values
  if ((checked & 1) == 0)
  {
    suffix = _suffix;
  }

  if ((checked & 8) == 0)
  {
    directory = _directory;
  }

  // Find tokens
  char* finder;
  suffixc = 1;
  finder = strchr(suffix, ',');
  while (finder != NULL)
  {
    suffixc++;
    finder = strchr(finder+1, ',');
  }
  suffixv = (char**) malloc(sizeof(suffixv)*suffixc);
  if (suffixv == NULL)
  {
    fprintf(stderr, "Memory allocation failed.\n");
    return 1;
  }
  finder = strtok(suffix, ",");
  int c = 0;
  while (finder != NULL)
  {
    suffixv[c] = finder;
    c++;
    finder = strtok(NULL, ",");
  }

  // Open directory
  DIR *dir = opendir(directory);
  if (dir == NULL)
  {
    perror("Operation failed");
    free(suffixv);
    return 1;
  }
  
  // Scan directory
  int lines = scanDirectory(dir, 0, directory);

  // Print double line and line count
  for (int t = 0 ; t < 80 ; t++)
  {
    printf("=");
  }
  printf("\n%74d lines\n", lines);

  closedir(dir);
  free(suffixv);
  return 0;
}