// This file is part of CAF, the C++ Actor Framework. See the file LICENSE in
// the main distribution directory for license terms and copyright or visit
// https://github.com/actor-framework/actor-framework/blob/master/LICENSE.

#include <algorithm>
#include <cstring>
#include <fstream>
#include <iostream>
#include <string>
#include <vector>

using std::cerr;
using std::find;
using std::find_if;
using std::string;
using std::vector;

void split(const string& str, const string& separator, vector<string>& dest) {
  size_t current, previous = 0;
  current = str.find(separator);
  while (current != std::string::npos) {
    dest.emplace_back(str.substr(previous, current - previous));
    previous = current + separator.size();
    current = str.find(separator, previous);
  }
  dest.push_back(str.substr(previous, current - previous));
}

void trim(string& str) {
  auto not_space = [](char c) { return isspace(c) == 0; };
  str.erase(str.begin(), find_if(str.begin(), str.end(), not_space));
  str.erase(find_if(str.rbegin(), str.rend(), not_space).base(), str.end());
}

template <size_t N>
bool starts_with(const string& str, const char (&prefix)[N]) {
  return str.compare(0, N - 1, prefix) == 0;
}

template <size_t N>
void drop_prefix(string& str, const char (&prefix)[N]) {
  if (str.compare(0, N - 1, prefix) == 0)
    str.erase(str.begin(), str.begin() + (N - 1));
}

void keep_alnum(string& str) {
  auto not_alnum = [](char c) { return isalnum(c) == 0 && c != '_'; };
  str.erase(find_if(str.begin(), str.end(), not_alnum), str.end());
}

int main(int argc, char** argv) {
  if (argc != 3) {
    cerr << "wrong number of arguments.\n"
         << "usage: " << argv[0] << " input-file output-file\n";
    return EXIT_FAILURE;
  }
  std::ifstream in{argv[1]};
  if (!in) {
    cerr << "unable to open input file: " << argv[1] << '\n';
    return EXIT_FAILURE;
  }
  vector<string> namespaces;
  string enum_name;
  string line;
  bool is_enum_class = false;
  // Locate the beginning of the enum.
  for (;;) {
    if (!getline(in, line)) {
      cerr << "unable to locate enum in file: " << argv[1] << '\n';
      return EXIT_FAILURE;
    }
    trim(line);
    if (starts_with(line, "enum ")) {
      drop_prefix(line, "enum ");
      if (starts_with(line, "class ")) {
        is_enum_class = true;
        drop_prefix(line, "class ");
      }
      trim(line);
      keep_alnum(line);
      enum_name = line;
      break;
    }
    if (starts_with(line, "namespace ")) {
      if (line.back() == '{')
        line.pop_back();
      line.erase(line.begin(), find(line.begin(), line.end(), ' '));
      trim(line);
      split(line, "::", namespaces);
    }
  }
  // Sanity checking.
  if (namespaces.empty()) {
    cerr << "enum found outside of a namespace\n";
    return EXIT_FAILURE;
  }
  if (enum_name.empty()) {
    cerr << "empty enum name found\n";
    return EXIT_FAILURE;
  }
  std::string case_label_prefix;
  if (is_enum_class)
    case_label_prefix = enum_name + "::";
  // Read until hitting the closing '}'.
  std::vector<std::string> enum_values;
  for (;;) {
    if (!getline(in, line)) {
      cerr << "unable to read enum values\n";
      return EXIT_FAILURE;
    }
    trim(line);
    if (line.empty())
      continue;
    if (line[0] == '}')
      break;
    if (line[0] != '/') {
      keep_alnum(line);
      enum_values.emplace_back(line);
    }
  }
  // Generate output file.
  std::ofstream out{argv[2]};
  if (!out) {
    cerr << "unable to open output file: " << argv[2] << '\n';
    return EXIT_FAILURE;
  }
  // Print file header.
  out << "// clang-format off\n"
      << "// DO NOT EDIT: "
         "this file is auto-generated by caf-generate-enum-strings.\n"
         "// Run the target update-enum-strings if this file is out of sync.\n"
         "#include \"caf/config.hpp\"\n"
         "#include \"caf/string_view.hpp\"\n\n"
         "CAF_PUSH_DEPRECATED_WARNING\n\n"
      << "#include \"" << namespaces[0];
  for (size_t i = 1; i < namespaces.size(); ++i)
    out << '/' << namespaces[i];
  out << '/' << enum_name << ".hpp\"\n\n"
      << "#include <string>\n\n"
      << "namespace " << namespaces[0] << " {\n";
  for (size_t i = 1; i < namespaces.size(); ++i)
    out << "namespace " << namespaces[i] << " {\n";
  out << '\n';
  // Generate to_string implementation.
  out << "std::string to_string(" << enum_name << " x) {\n"
      << "  switch(x) {\n"
      << "    default:\n"
      << "      return \"???\";\n";
  for (auto& val : enum_values)
    out << "    case " << case_label_prefix << val << ":\n"
        << "      return \"" << val << "\";\n";
  out << "  };\n"
      << "}\n\n";
  // Generate from_string implementation.
  out << "bool from_string(string_view in, " << enum_name << "& out) {\n  ";
  for (auto& val : enum_values)
    out << "if (in == \"" << val << "\") {\n"
        << "    out = " << case_label_prefix << val << ";\n"
        << "    return true;\n"
        << "  } else ";
  out << "{\n"
      << "    return false;\n"
      << "  }\n"
      << "}\n\n";
  // Generate from_integer implementation.
  out << "bool from_integer(std::underlying_type_t<" << enum_name << "> in,\n"
      << "                  " << enum_name << "& out) {\n"
      << "  auto result = static_cast<" << enum_name << ">(in);\n"
      << "  switch(result) {\n"
      << "    default:\n"
      << "      return false;\n";
  for (auto& val : enum_values)
    out << "    case " << case_label_prefix << val << ":\n";
  out << "      out = result;\n"
      << "      return true;\n"
      << "  };\n"
      << "}\n\n";
  // Done. Print file footer and exit.
  for (auto i = namespaces.rbegin(); i != namespaces.rend(); ++i)
    out << "} // namespace " << *i << '\n';
  out << "\nCAF_POP_WARNINGS\n";
  return EXIT_SUCCESS;
}
