#include <v8.h>
#include <v8-profiler.h>
#include <node.h>
#include "heap-snapshot.h"

using namespace v8;
using namespace node;

static Handle<Value> DeleteAllSnapshots(const Arguments& args) {
  HandleScope scope;

  HeapProfiler::DeleteAllSnapshots();

  return Undefined();
}

static Handle<Value> FindSnapshot(const Arguments& args) {
  HandleScope scope;

  if (args.Length() < 1) {
    return ThrowException(Exception::Error(String::New("No uid specified")));
  }
  uint32_t uid = args[0]->Uint32Value();
  const HeapSnapshot* snapshot = HeapProfiler::FindSnapshot(uid);

  return scope.Close(Snapshot::New(snapshot));
}

static Handle<Value> GetSnapshot(const Arguments& args) {
  HandleScope scope;

  if (args.Length() < 1) {
    return ThrowException(Exception::Error(String::New("No index specified")));
  } else if (!args[0]->IsInt32()) {
    return ThrowException(Exception::TypeError(String::New("Argument must be an integer")));
  }
  int32_t index = args[0]->Int32Value();
  const HeapSnapshot* snapshot = HeapProfiler::GetSnapshot(index);

  return scope.Close(Snapshot::New(snapshot));
}

static Handle<Value> GetSnapshotsCount(const Arguments& args) {
  HandleScope scope;

  return scope.Close(Integer::New(HeapProfiler::GetSnapshotsCount()));
}

static Handle<Value> TakeSnapshot(const Arguments& args) {
  HandleScope scope;
  Local<String> title = String::New("");

  int32_t len = args.Length();
  HeapSnapshot::Type mode = HeapSnapshot::kFull;
  if (len > 0) {
    title = args[0]->ToString();
  }
  if (len > 1 && args[1]->IsInt32()) {
    mode = static_cast<HeapSnapshot::Type>(args[1]->Int32Value());
  }
  const HeapSnapshot* snapshot = HeapProfiler::TakeSnapshot(title, mode);

  return scope.Close(Snapshot::New(snapshot));
}

extern "C" void init(Handle<Object> target) {
  HandleScope scope;

  NODE_SET_METHOD(target, "deleteAllSnapshots", DeleteAllSnapshots);
  NODE_SET_METHOD(target, "findSnapshot",       FindSnapshot);
  NODE_SET_METHOD(target, "getSnapshot",        GetSnapshot);
  NODE_SET_METHOD(target, "snapshotCount",      GetSnapshotsCount);
  NODE_SET_METHOD(target, "takeSnapshot",       TakeSnapshot);
}