Mercurial > hg > cline / file revision

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

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

void printHelpText() {
  const char* helpText = 
    "\nUsage:"
    "\n      cline [-hrmvV][-s suffix][-b level][<directory>]"
    "\n      cline [-hrmvV][-S suffix][-b level][<directory>]"
    "\n\nCounts the line terminator characters (\\n) within all"
    " files in the specified\ndirectory."
    "\n\nOptions:"
    "\n  -b <level>          - binary file heuristics level (default medium)"
    "\n                        One of: ignore low medium high"
    "\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], "hsSrRmvVb");

    /* 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;
    }
    /* b */
    if ((argflags & 256) > 0) {
      if (registerArgument(&checked, 256)) {
        return exit_with_help(settings, 1);
      }
      t++;
      if (t >= argc) {
        return exit_with_help(settings, 1);
      }
      if (stricmp(argv[t], "ignore") == 0) {
        settings->bfileHeuristics->level = BFILE_IGNORE;
      } else if (stricmp(argv[t], "low") == 0) {
        settings->bfileHeuristics->level = BFILE_LOW_ACCURACY;
      } else if (stricmp(argv[t], "medium") == 0) {
        settings->bfileHeuristics->level = BFILE_MEDIUM_ACCURACY;
      } else if (stricmp(argv[t], "high") == 0) {
        settings->bfileHeuristics->level = BFILE_HIGH_ACCURACY;
      } else {
        return exit_with_help(settings, 1);
      }
    }
    /* 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_string(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 */
  for (int t = 0 ; t < 79 ; t++) {
    printf("=");
  }
  printf("\n%73d lines\n", lines);

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

  fflush(stdout);
  return 0;
}