es.cc

#include <iostream>
#include <string>
#include <fstream>
#include <vector>
#include <codecvt>
#include <locale>

#include <es/parser/parser.h>
#include <es/types/property_descriptor_object_conversion.h>
#include <es/enter_code.h>
#include <es/eval.h>
#include <es/utils/helper.h>
#include <es/gc/heap.h>
#include <es/impl.h>
#ifdef PERF
#include <gperftools/profiler.h>
#endif

using namespace es;

// NOTE(zhuzilin) There are some copy and paste from stackoverflow...
std::u16string ReadUTF8FileToUTF16String(std::string filename) {
  // https://stackoverflow.com/a/2602258/5163915
  std::ifstream file(filename);
  std::stringstream buffer;
  buffer << file.rdbuf();
  std::string utf8_file_content = buffer.str();

  // https://stackoverflow.com/a/7235204/5163915
  std::wstring_convert<std::codecvt_utf8_utf16<char16_t>, char16_t> convert;
  std::u16string utf16_file_content = convert.from_bytes(utf8_file_content);

  return utf16_file_content;
}

int main(int argc, char* argv[]) {
  if (argc < 2) {
    std::cout << "no filename presented" << "\n";
    return 0;
  }
  std::string filename(argv[1]);
  std::u16string source = ReadUTF8FileToUTF16String(filename);

#ifdef PERF
  ProfilerStart("parse_program.profile");
#endif
  Parser parser(source);
  AST* ast = parser.ParseProgram();
#ifdef PERF
  ProfilerStop();
#endif
  if (ast->IsIllegal()) {
    size_t start = ast->start();
    for (size_t i = 0; i < 10; i++) {
      if (start == 0 || character::IsLineTerminator(source[start - 1]))
        break;
      start--;
    }
    size_t end = ast->end();
    for (size_t i = 0; i < 10; i++) {
      if (end == source.size() || character::IsLineTerminator(source[end]))
        break;
      end++;
    }
    std::cout << "\033[1;31m" << "ParserError: " << "\033[0m"
              << log::ToString(source.substr(start, ast->start() - start))
              << "\033[1;31m" << log::ToString(ast->source()) << "\033[0m"
              << log::ToString(source.substr(ast->end(), end - ast->end())) << "\n";
    return 0;
  }
#ifdef PERF
  ProfilerStart("optimize_ast.profile");
#endif
  ast = Optimize(ast);
#ifdef PERF
  ProfilerStop();
#endif

#ifdef PARSER_ONLY
  std::cout << "finish parsing" << std::endl;
  return 0;
#endif

  Init();
  Handle<Error> e = Error::Ok();
#ifdef PERF
  ProfilerStart("eval_program.profile");
#endif
  EnterGlobalCode(e, ast);
  if (unlikely(!e.val()->IsOk())) {
    std::cout << "\033[1;31m" << "\033[2menter\033[0m global failed: " << e.ToString() << "\033[0m" << "\n";
    return 0;
  }
  Completion res = EvalProgram(ast);
#ifdef PERF
  ProfilerStop();
#endif
  switch (res.type()) {
    case Completion::THROW: {
      std::cout << "\033[1;31m" << "Uncaught ";
      Handle<JSValue> val;
      if (res.value().val()->IsError()) {
        Handle<Error> error = static_cast<Handle<Error>>(res.value());
        val = error.val()->value();
        switch (error.val()->error_type()) {
          case Error::E_EVAL:
            std::cout << "EvalError: ";
            break;
          case Error::E_RANGE:
            std::cout << "RangeError: ";
            break;
          case Error::E_REFERENCE:
            std::cout << "ReferenceError: ";
            break;
          case Error::E_SYNTAX:
            std::cout << "SyntaxError: ";
            break;
          case Error::E_TYPE:
            std::cout << "TypeError: ";
            break;
          case Error::E_URI:
            std::cout << "URIError: ";
            break;
          default:
            break;
        }
      } else {
        val = res.value();
      }
      log::Debugger::TurnOff();
      Handle<Error> e = Error::Ok();
      Handle<String> msg = ToString(e, val);
      if (unlikely(!e.val()->IsOk()))
        std::cout << val.ToString() << "\033[0m" << "\n";
      else
        std::cout << log::ToString(msg.val()->data()) << "\033[0m" << "\n";
      break;
    }
    default:
      break;
  }
#ifdef STATS
  std::cout << "Heap stats in the end:" << std::endl;
  Stats();
#endif
}