Mercurial > hg > cline / file revision

#include "cline.h"
#include "functions.h"

settings_t* new_settings_t() {
  settings_t *settings = malloc(sizeof(settings_t*));
  if (settings != NULL) {
  #ifdef _WIN32
    settings->fileSeparator      = '\\';
  #else
    settings->fileSeparator      = '/';
  #endif /* _WIN32 */
    settings->suffixc            = 1;
    settings->recursive          = false;
    settings->includeSuffixes    = false;
    settings->matchesOnly        = false;
  }
  
  return settings;
}

void destroy_settings_t(settings_t* settings) {
  if (settings->suffixv != NULL) {
    free(settings->suffixv);
  }
  free(settings);
}

void printHelpText(const char* prgName) {
  // Help text
  const char* helpText = 
    "\nUsage:"
    "\n      %s [-hrm][-s suffix][<directory>]"
    "\n      %s [-hrm][-S suffix][<directory>]"
    "\n\nCounts the line terminator characters (\\n) within all"
    " files in the specified\ndirectory."
    "\n\nOptions:"
    "\n  -h, --help          - this help text"
    "\n  -m                  - print information about matching files only"
    "\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, -R              - includes subdirectories"
    "\n\n"
    "The default call without any options is:"    
    "\n  %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 -rs .c\n";
    
  printf(helpText, prgName, prgName, prgName, prgName);
}

int main(int argc, char** argv) {

  // Settings
  settings_t *settings = new_settings_t();
  if (settings == NULL) {
    fprintf(stderr, "Memory allocation failed.\n");
    return 1;
  }

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

  // Defaults
  char* _suffix = " ";
  char _directory[3];
  _directory[0] = '.';
  _directory[1] = settings->fileSeparator;
  _directory[2] = 0;

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

  for (int t = 1 ; t < argc ; t++) {

    int argflags = checkArgument(argv[t], "hsSrRm");

    // s
    if ((argflags & 2) > 0) {
      if ((checked & 1) > 0) {
        printHelpText(prgName);
        destroy_settings_t(settings);
        return 1;
      }
      settings->includeSuffixes = true;
      t++;
      if (t >= argc) {
        printHelpText(prgName);
        destroy_settings_t(settings);
        return 1;
      }
      suffix = argv[t]; 
      checked |= 1;
    }
    // S
    if ((argflags & 4) > 0) {
      if ((checked & 1) > 0) {
        printHelpText(prgName);
        destroy_settings_t(settings);
        return 1;
      }
      settings->includeSuffixes = false;
      t++;
      if (t >= argc) {
        printHelpText(prgName);
        destroy_settings_t(settings);
        return 1;
      }
      suffix = argv[t];
      checked |= 1;
    }
    // h
    if ((argflags & 1) > 0 || strcmp(argv[t], "--help") == 0) {
      if ((checked & 2) > 0) {
        printHelpText(prgName);
        destroy_settings_t(settings);
        return 1;
      }
      checked |= 2;
      showHelp = true;
    }
    // r, R
    if ((argflags & 24) > 0) {
      if ((checked & 4) > 0) {
        printHelpText(prgName);
        destroy_settings_t(settings);
        return 1;
      }
      checked |= 4;
      settings->recursive = true;
    }
    if ((argflags & 32) > 0) {
      if ((checked & 32) > 0) {
        printHelpText(prgName);
        destroy_settings_t(settings);
        return 1;
      }
      checked |= 32;
      settings->matchesOnly = true;
    }
    // other
    if (argflags == 0) {
      if ((checked & 8) > 0) {
        printHelpText(prgName);
        destroy_settings_t(settings);
        return 1;
      }
      checked |= 8;
      directory = argv[t];
    }
  }

  // Show help and quit
  if (showHelp) {
    printHelpText(prgName);
    destroy_settings_t(settings);
    return 0;
  }

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

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

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

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

  // Print double line and line count
  #ifdef _WIN32
    const int columns = 79;
  #else
    const int columns = 80;
  #endif /* _WIN32 */

  for (int t = 0 ; t < columns ; t++) {
    printf("=");
  }
  #ifdef _WIN32
    printf("\n%73d lines\n", lines);
  #else
    printf("\n%74d lines\n", lines);
  #endif /* _WIN32 */

  destroy_settings_t(settings);
  return 0;
}