#include <iostream>
#include "simdjson.h"
using namespace std;
using namespace simdjson;

void document_parse_error_code() {
  cout << __func__ << endl;

  string json("[ 1, 2, 3 ]");
  auto [doc, error] = document::parse(json);
  if (error) { cerr << "Error: " << error << endl; exit(1); }
  if (!doc.print_json(cout)) { exit(1); }
  cout << endl;
}

void document_parse_exception() {
  cout << __func__ << endl;

  string json("[ 1, 2, 3 ]");
  document doc = document::parse(json);
  if (!doc.print_json(cout)) { exit(1); }
  cout << endl;
}

void document_parse_padded_string() {
  cout << __func__ << endl;

  padded_string json(string("[ 1, 2, 3 ]"));
  document doc = document::parse(json);
  if (!doc.print_json(cout)) { exit(1); }
  cout << endl;
}

void document_parse_get_corpus() {
  cout << __func__ << endl;

  auto json(get_corpus("jsonexamples/small/demo.json"));
  document doc = document::parse(json);
  if (!doc.print_json(cout)) { exit(1); }
  cout << endl;
}

void document_load() {
  cout << __func__ << endl;

  document doc = document::load("jsonexamples/small/demo.json");
  if (!doc.print_json(cout)) { exit(1); }
  cout << endl;
}

void parser_parse() {
  cout << __func__ << endl;

  // Allocate a parser big enough for all files
  document::parser parser;
  simdjson::error_code capacity_error = parser.set_capacity(1024*1024);
  if (capacity_error) { cerr << "Error setting capacity: " << capacity_error << endl; exit(1); }

  // Read files with the parser, one by one
  for (padded_string json : { string("[1, 2, 3]"), string("true"), string("[ true, false ]") }) {
    cout << "Parsing " << json.data() << " ..." << endl;
    auto [doc, error] = parser.parse(json);
    if (error) { cerr << "Error: " << error << endl; exit(1); }
    if (!doc.print_json(cout)) { cerr << "print failed!\n"; exit(1); }
    cout << endl;
  }
}

void parser_parse_many_error_code() {
  cout << __func__ << endl;

  // Read files with the parser
  padded_string json = string("[1, 2, 3] true [ true, false ]");
  cout << "Parsing " << json.data() << " ..." << endl;
  document::parser parser;
  for (auto [doc, error] : parser.parse_many(json)) {
    if (error) { cerr << "Error: " << error << endl; exit(1); }
    if (!doc.print_json(cout)) { exit(1); }
    cout << endl;
  }
}

void parser_parse_many_exception() {
  cout << __func__ << endl;

  // Read files with the parser
  padded_string json = string("[1, 2, 3] true [ true, false ]");
  cout << "Parsing " << json.data() << " ..." << endl;
  document::parser parser;
  for (const document &doc : parser.parse_many(json)) {
    if (!doc.print_json(cout)) { exit(1); }
    cout << endl;
  }
}

void parser_parse_max_capacity() {
  int argc = 2;
  padded_string argv[] { string("[1,2,3]"), string("true") };
  document::parser parser(1024*1024); // Set max capacity to 1MB
  for (int i=0;i<argc;i++) {
    auto [doc, error] = parser.parse(argv[i]);
    if (error == CAPACITY) { cerr << "JSON files larger than 1MB are not supported!" << endl; exit(1); }
    if (error) { cerr << error << endl; exit(1); }
    doc.print_json(cout);
  }
}

void parser_parse_fixed_capacity() {
  int argc = 2;
  padded_string argv[] { string("[1,2,3]"), string("true") };
  document::parser parser(0); // This parser is not allowed to auto-allocate
  auto alloc_error = parser.set_capacity(1024*1024);
  if (alloc_error) { cerr << alloc_error << endl; exit(1); }; // Set initial capacity to 1MB
  for (int i=0;i<argc;i++) {
    auto [doc, error] = parser.parse(argv[i]);
    if (error == CAPACITY) { cerr << "JSON files larger than 1MB are not supported!" << endl; exit(1); }
    if (error) { cerr << error << endl; exit(1); }
    doc.print_json(cout);
  }
}

int main() {
  cout << "Running examples." << endl;
  document_parse_error_code();
  document_parse_exception();
  document_parse_padded_string();
  document_parse_get_corpus();
  document_load();
  parser_parse();
  parser_parse_many_error_code();
  parser_parse_many_exception();
  parser_parse_max_capacity();
  parser_parse_fixed_capacity();
  cout << "Ran to completion!" << endl;
  return 0;
}