Mercurial > hg > cline / file revision

/*
 * cline.c
 *
 *  Created on: 23.05.2011
 *      Author: beckermi
 */

#include "cline.h"
#include "scanner.h"
#include "settings.h"
#include "arguments.h"
#include "stream.h"

void printHelpText() {
  const char* helpText = 
    "\nUsage:"
    "\n      cline [-hrm][-s suffix][<directory>]"
    "\n      cline [-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  -v, --version       - print out version information"
    "\n  -V                  - turn verbose output off, print the result only"
    "\n\n"
    "The default call without any options is:"    
    "\n  cline ./\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  cline -rs .c\n";
    
  printf(helpText);
}

int exit_with_version(settings_t* settings) {
  printf("cline - Revision: %s", VERSION);
  destroy_settings_t(settings);
  return 0;
}

int exit_with_help(settings_t* settings, int code) {
  printHelpText();
  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;
  }

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

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

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

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

  // Configure output
  if (!settings->verbose) {
    close_stdout();
  }

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

  if (!settings->verbose) {
    reopen_stdout();
    printf("%d", lines);
  }

  fflush(stdout);
  return 0;
}