Mercurial > hg > cline / file revision

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

suffix_list_t* new_suffix_list_t() {
  suffix_list_t* suffixList = malloc(sizeof(suffix_list_t));
  suffixList->count = 0;
  suffixList->items = NULL;

  return suffixList;
}

void add_suffix(suffix_list_t* list, char* item) {
  char** reallocated_list =
    realloc(list->items, sizeof(char*) * list->count + 1);
  if (reallocated_list != NULL) {
    list->items = reallocated_list;
    list->items[list->count] = item;
    list->count++;
  }
}

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->recursive          = false;
    settings->includeSuffixes    = false;
    settings->matchesOnly        = false;
    settings->suffixList         = new_suffix_list_t();
  }
  
  return settings;
}

void destroy_settings_t(settings_t* settings) {
  free(settings->suffixList);
  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"
    "So each file in the working directory is counted. If you want to count C"
    "\nsource code in your working directory and its subdirectories, type:"
    "\n  %s -rs .c\n";
    
  printf(helpText, prgName, prgName, prgName, prgName);
}

int exit_with_help(char* prgName, settings_t* settings, int code) {
  printHelpText(prgName);
  destroy_settings_t(settings);
  return code;
}

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++;
  }

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

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

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

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

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

  // Find tokens
  char* finder = strtok(suffix, ",");
  while (finder != NULL) {
    add_suffix(settings->suffixList, finder);
    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);
  destroy_settings_t(settings);

  // 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 */

  return 0;
}