denoland-deno/libdeno/snapshot_creator.cc

// Copyright 2018-2019 the Deno authors. All rights reserved. MIT license.
// Hint: --trace_serializer is a useful debugging flag.
#include <fstream>
#include <iostream>
#include "deno.h"
#include "file_util.h"
#include "internal.h"
#include "third_party/v8/include/v8.h"
#include "third_party/v8/src/base/logging.h"

namespace deno {}  // namespace deno

int main(int argc, char** argv) {
  const char* snapshot_out_bin = argv[1];
  const char* js_fn = argv[2];

  v8::V8::SetFlagsFromCommandLine(&argc, argv, true);

  CHECK_NE(js_fn, nullptr);
  CHECK_NE(snapshot_out_bin, nullptr);

  std::string js_source;
  CHECK(deno::ReadFileToString(js_fn, &js_source));

  deno_init();
  deno_config config = {1, deno::empty_buf, deno::empty_buf, nullptr, nullptr};
  Deno* d = deno_new(config);

  int r = deno_execute(d, nullptr, js_fn, js_source.c_str());
  if (!r) {
    std::cerr << "Snapshot Exception " << std::endl;
    std::cerr << deno_last_exception(d) << std::endl;
    deno_delete(d);
    return 1;
  }

  auto snapshot = deno_get_snapshot(d);

  std::ofstream file_(snapshot_out_bin, std::ios::binary);
  file_.write(reinterpret_cast<char*>(snapshot.data_ptr), snapshot.data_len);
  file_.close();

  delete[] snapshot.data_ptr;
  deno_delete(d);

  return file_.bad();
}