asdl.c

#include <getopt.h>
#include <stdbool.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sysexits.h>
#include <unistd.h>

#include "asdl.h"

extern FILE *yyin;
extern int yyparse(void);

static inline void print_help_and_exit() {
  printf(
      "Usage: asdl [-o output] [-s symfile] [-d def_suffix] [-f fn_suffix] [-k "
      "kind_suffix] [-a arg_suffix] [-p fn_prefix] FILE\n");
  printf("\n");
  printf("Options:\n");
  printf("  -o output      	Specify the output file name\n");
  printf("  -d header file	Specify the symbol declaration file header name\n");
  printf("  -D def_suffix  	Specify the suffix for typedefs\n");
  printf("  -f fn_suffix   	Specify the suffix for functions\n");
  printf("  -k kind_suffix 	Specify the suffix for kind types\n");
  printf("  -a arg_suffix  	Specify the suffix for arguments\n");
  printf("  -p fn_prefix   	Specify the prefix for function signatures\n");
  printf("\n");
  exit(EX_USAGE);
}

void parse_arguments(int argc, char **argv) {
  int c = 0;

  char *def_suffix = "def", *fn_suffix = "fn", *arg_suffix = "arg",
       *kind_suffix = "kind";
  char *fn_prefix = NULL;
  char *outpath = NULL;
  char *sympath = NULL;
  yyin = stdin;

  while ((c = getopt(argc, argv, "o:d:f:a:k:p:h:D")) != -1) {
    switch (c) {
    case 'o':
      outpath = optarg;
      break;
    case 'd':
      sympath = optarg;
      break;
    case 'D':
      def_suffix = optarg;
      break;
    case 'a':
      arg_suffix = optarg;
      break;
    case 'f':
      fn_suffix = optarg;
      break;
    case 'k':
      kind_suffix = optarg;
      break;
    case 'p':
      fn_prefix = optarg;
      break;    
    case 'h':
      print_help_and_exit();
      break;
    default:
      fprintf(stderr, "Error: wrong argument, pass -h to view help\n");
      exit(EXIT_FAILURE);
    }
  }

  if (optind >= argc && isatty(STDIN_FILENO)) {
    fprintf(stderr,
            "Error: no input file given, neither via arguments nor STDIN\n");
    exit(EXIT_FAILURE);
  } else if (optind < argc && isatty(STDIN_FILENO)) {
    if (access(argv[optind], F_OK | R_OK) == 0)
      yyin = fopen(argv[optind], "r");
    else {
      fprintf(stderr, "Error: input file is not readable, or does not exist");
      exit(EXIT_FAILURE);
    }
  } else if (optind < argc && !isatty(STDIN_FILENO)) {
    yyin = stdin;
  }

  assign_prefixes(fn_prefix);
  assign_suffixes(def_suffix, fn_suffix, arg_suffix, kind_suffix);

  init_translator(outpath, sympath);
  init_absyn();
}

void do_at_exit(void) {
  dump_absyn();
  dump_translator();
}

int main(int argc, char **argv) {
  parse_arguments(argc, argv);
  atexit(do_at_exit);

  symtable_init();

  yyparse();

  finalize_absyn();
  finalize_translator();

  if (yyin != stdin)
    fclose(yyin);

  fprintf(stderr, "Successfully tanslated\n");

  return 0;
}