/**
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#pragma once

#include <string>
#include <unordered_map>

#include <jsi/jsi.h>

#include "JSCallInvoker.h"

namespace facebook {
namespace react {

/**
 * For now, support the same set of return types as existing impl.
 * This can be improved to support richer typed objects.
 */
enum TurboModuleMethodValueKind {
  VoidKind,
  BooleanKind,
  NumberKind,
  StringKind,
  ObjectKind,
  ArrayKind,
  FunctionKind,
  PromiseKind,
};

/**
 * Base HostObject class for every module to be exposed to JS
 */
class JSI_EXPORT TurboModule : public facebook::jsi::HostObject {
public:
  TurboModule(const std::string &name, std::shared_ptr<JSCallInvoker> jsInvoker);
  virtual ~TurboModule();

  /**
   * Instruct this module to invalidate itself.
   */
  virtual void invalidate();

  virtual facebook::jsi::Value get(facebook::jsi::Runtime& runtime, const facebook::jsi::PropNameID& propName) override;

  /**
   * General method invocation mechanism.
   * Each subclass decides how the invocation should be, and whether it should be platform-specific.
   */
  virtual jsi::Value invokeMethod(
      jsi::Runtime &runtime,
      TurboModuleMethodValueKind valueKind,
      const std::string &methodName,
      const jsi::Value *args,
      size_t count);

  const std::string name_;
  std::shared_ptr<JSCallInvoker> jsInvoker_;

protected:
  struct MethodMetadata {
    size_t argCount;
    facebook::jsi::Value (*invoker)(
        facebook::jsi::Runtime& rt,
        TurboModule &turboModule,
        const facebook::jsi::Value* args,
        size_t count);
  };

  std::unordered_map<std::string, MethodMetadata> methodMap_;
};

/**
 * An app/platform-specific provider function to get an instance of a module given a name.
 */
using TurboModuleProviderFunctionType = std::function<std::shared_ptr<TurboModule>(
    const std::string &name)>;

} // namespace react
} // namespace facebook