// Copyright 2016 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <stdio.h>

#include <map>
#include <numeric>
#include <string>

#include "abigen_generator.h"
#include "syscall_parser.h"
#include "types.h"

using std::string;

constexpr Dispatch<AbigenGenerator> abigen_table[] = {
    // comments start with '#' and terminate at the end of line. this also handles #! reqs.
    {"#", nullptr, process_comment},
    // sycalls start with 'syscall' and terminate with ';'.
    {"syscall", ";", process_syscall},
    // table terminator.
    {nullptr, nullptr, nullptr}};

// =================================== driver ====================================================

int main(int argc, char* argv[]) {
  string output_prefix = "generated";
  bool verbose = false;
  bool generate_all = false;
  std::map<string, string> type_to_filename;

  argc--;
  argv++;
  while (argc > 0) {
    const string command(argv[0]);
    if (command[0] != '-')
      break;
    else if (get_type_to_generator().find(command.substr(1)) != get_type_to_generator().end()) {
      string type = command.substr(1);
      type_to_filename[type] = string(argv[1]);
      argc--;
      argv++;
    } else if (command == "-a") {
      generate_all = true;
    } else if (command == "-v") {
      verbose = true;
    } else if (command == "-o") {
      if (argc < 2) {
        fprintf(stderr, "no output prefix given\n");
        return -1;
      }
      output_prefix.assign(argv[1]);
      argc--;
      argv++;
    } else if (command == "-h") {
      fprintf(stderr,
              "usage: abigen [-a] [-v] [-o output_prefix] "
              "[-<type> filename] file1 ... fileN\n");
      const string delimiter = ", ";
      const string valid_types = std::accumulate(
          get_type_to_default_suffix().begin(), get_type_to_default_suffix().end(), std::string(),
          [delimiter](const std::string& s, const std::pair<const std::string, std::string>& p) {
            return s + (s.empty() ? std::string() : delimiter) + p.first;
          });
      fprintf(stderr, "\n       Valid <type>s: %s\n", valid_types.c_str());
      return 0;
    } else {
      fprintf(stderr, "unknown option: %s\n", command.c_str());
      return -1;
    }
    argc--;
    argv++;
  }
  if (argc < 1) {
    fprintf(stderr, "no syscall-spec input given\n");
    return -1;
  }

  // Use defaults for anything not specified.
  if (generate_all) {
    for (auto& entry : get_type_to_default_suffix()) {
      if (type_to_filename.find(entry.first) == type_to_filename.end()) {
        type_to_filename[entry.first] = output_prefix + entry.second;
      }
    }
  }

  AbigenGenerator generator(verbose);

  for (int ix = 0; ix < argc; ix++) {
    if (!run_parser(&generator, abigen_table, argv[ix], verbose))
      return 1;
  }
  return generator.Generate(type_to_filename) ? 0 : 1;
}