// Copyright (c) Facebook, Inc. and its affiliates. (http://www.facebook.com)
#include "builtins.h"
#include "bytes-builtins.h"
#include "frame.h"
#include "globals.h"
#include "marshal.h"
#include "module-builtins.h"
#include "modules.h"
#include "objects.h"
#include "runtime.h"
#include "thread.h"

namespace py {

void FUNC(marshal, __init_module__)(Thread* thread, const Module& module,
                                    View<byte> bytecode) {
  HandleScope scope(thread);

  // Add module variables
  {
    Object magic_number(&scope, SmallInt::fromWord(kPycMagic));
    moduleAtPutById(thread, module, ID(magic_number), magic_number);
  }

  executeFrozenModule(thread, module, bytecode);
}

RawObject FUNC(marshal, loads)(Thread* thread, Arguments args) {
  HandleScope scope(thread);
  Object bytes_obj(&scope, args.get(0));
  Runtime* runtime = thread->runtime();
  if (!runtime->isInstanceOfBytes(*bytes_obj)) {
    // TODO(T38902048): Load from buffer protocol objects
    UNIMPLEMENTED("marshal.loads with non-bytes objects");
  }
  Bytes bytes(&scope, bytesUnderlying(*bytes_obj));
  word length = bytes.length();
  // TODO(T38902583): Use updated Marshal reader to read from Bytes object
  // directly
  std::unique_ptr<byte[]> buffer(new byte[length]);
  bytes.copyTo(buffer.get(), length);
  Marshal::Reader reader(&scope, thread, View<byte>(buffer.get(), length));
  return reader.readObject();
}

}  // namespace py