#include <iostream>
#include <fstream>
#include <string>
#include <optional>
#include <bitset>
#include <sstream>

#include "lexer.hpp"
#include "token.hpp"
#include "ast_printer.hpp"
#include "ast_resolver.hpp"
#include "ast_emitter.hpp"
#include "parser.hpp"

namespace dsp {

bool had_error = false;

}

static void write(std::ostream& os, uint32_t v)
{
  const char out[4] = {
    static_cast<char>((v >> 24) & 0xff),
    static_cast<char>((v >> 16) & 0xff),
    static_cast<char>((v >> 8 ) & 0xff),
    static_cast<char>((v >> 0 ) & 0xff),
  };
  os.write(out, 4);
}

static void run(std::ostream& os, std::string source)
{
  using namespace dsp;

  std::string_view buf(source);
  lexer_t lexer(buf);
  std::vector<token_t> tokens = lexer.lex_tokens();
  parser_t pass1(tokens);
  parser_t pass2(tokens);
  ast::printer_t printer(std::cout);
  ast::pc_t pc;
  ast::variables_t variables;
  ast::resolver_t resolver(pc, variables);
  while (auto stmt_o = pass1.statement()) {
    (*stmt_o)->accept(&printer);
    std::cout << std::endl << std::flush;
    (*stmt_o)->accept(&resolver);
  }
  ast::emitter_t emitter(variables);
  while (auto stmt_o = pass2.statement()) {
    uint32_t output = (*stmt_o)->accept(&emitter);
    if (output != 0xffff'ffff) {
      std::cout << std::bitset<32>(output) << std::endl;
      write(os, output);
    }
  }
}

static void run_prompt()
{
  constexpr auto prompt = "> ";
  std::string line;
  std::cout << prompt << std::flush;
  while (std::getline(std::cin, line)) {
    std::ostringstream os;
    run(os, line);
    std::cout << prompt << std::flush;
  }
}

static int run_file(char const * const input_filename, char const * const output_filename)
{
  std::ifstream is {input_filename, std::ios::binary | std::ios::ate};
  if (!is.is_open()) {
    std::cerr << "failed to open " << input_filename << std::endl;
    return -1;
  }
  const std::streampos size = is.tellg();
  std::string buf(size, '\0');
  is.seekg(0);
  if (!is.read(&buf[0], size)) {
    std::cerr << "read failed" << std::endl;
    return -1;
  }

  std::ostringstream os;
  run(os, buf);

  if (!dsp::had_error) {
    std::ofstream ofs {output_filename, std::ios::binary | std::ios::trunc};
    if (!ofs.is_open()) {
      std::cerr << "failed to open " << output_filename << std::endl;
      return -1;
    }
    ofs << os.str();
  }

  return dsp::had_error;
}

int main(const int argc, char const * const argv[])
{
  switch (argc) {
  case 1: run_prompt(); return dsp::had_error;
  case 3: return run_file(argv[1], argv[2]);
  default:
    std::cerr << "Usage: " << argv[0] << " [input-filename output-filename]" << std::endl;
    return -1;
  }
}