// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef SRC_LIB_FIDL_CODEC_WIRE_TYPES_H_
#define SRC_LIB_FIDL_CODEC_WIRE_TYPES_H_

#include <lib/fit/function.h>
#include <zircon/types.h>

#include <string>

#include <rapidjson/document.h>

#include "src/lib/fidl_codec/message_decoder.h"
#include "src/lib/fidl_codec/wire_object.h"

namespace fidl_codec {

// Transaction header size in bytes.
constexpr uint32_t kTransactionHeaderSize = 16;

class HandleType;
class LibraryLoader;
class StructType;
class TableType;
class TypeVisitor;
class UnionType;
class Uint32Type;

// A FIDL type.  Provides methods for generating instances of this type.
class Type {
  friend class Library;

 public:
  Type() = default;
  virtual ~Type() = default;
  Type& operator=(const Type& other) = default;
  Type(const Type& other) = default;

  // Returns a detailed representation of the type.
  std::string ToString(bool expand = false) const;

  virtual Uint32Type* AsUint32Type() { return nullptr; }
  virtual const UnionType* AsUnionType() const { return nullptr; }
  virtual const StructType* AsStructType() const { return nullptr; }
  virtual const TableType* AsTableType() const { return nullptr; }
  virtual const HandleType* AsHandleType() const { return nullptr; }

  // Returns true if the type is a ArrayType.
  virtual bool IsArray() const { return false; }

  // Is this type valid?
  virtual bool IsValid() const { return true; }

  // Returns a readable representation of the type.
  virtual std::string Name() const = 0;

  // Returns C++ name of the type.
  virtual std::string CppName() const { return this->Name(); }

  // Prints a colorized representation of the type.
  virtual void PrettyPrint(PrettyPrinter& printer) const {
    printer << Green << Name() << ResetColor;
  }

  // Returns the size of this type when embedded in another object.
  virtual size_t InlineSize(WireVersion version) const = 0;

  // Whether this is a nullable type.
  virtual bool Nullable() const { return false; }

  // For vectors and arrays, give the type of the components (members).
  virtual const Type* GetComponentType() const { return nullptr; }

  // Decodes the type's inline part. It generates a Value and, eventually,
  // registers the field for further decoding (secondary objects).
  virtual std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const = 0;

  // Pretty prints the value for this type. This is used to print numerical values.
  virtual void PrettyPrint(const Value* value, PrettyPrinter& printer) const;

  // Use a visitor on this value;
  virtual void Visit(TypeVisitor* visitor) const = 0;

  // Gets a Type object representing the |type_name|.  |type| is a string that
  // represents a scalar type (e.g., "float64", "uint32").
  static std::unique_ptr<Type> ScalarTypeFromName(const std::string& type_name);

  // Gets a Type object representing the |type|.  |type| is a JSON object with a
  // "subtype" field that represents a scalar type (e.g., "float64", "uint32")
  static std::unique_ptr<Type> TypeFromPrimitive(const rapidjson::Value& type);

  // Gets a Type object representing |type|, where |type| is a JSON object
  // representing an "internal" fidl type, with the "subtype" field specifying
  // what internal type is being represented.
  static std::unique_ptr<Type> TypeFromInternal(const rapidjson::Value& type);

  // Gets a Type object representing the |type|.  |type| is a JSON object a
  // field "kind" that states the type.  "kind" is an identifier
  // (e.g.,"foo.bar/Baz").  |loader| is the set of libraries to use to lookup
  // that identifier.
  static std::unique_ptr<Type> TypeFromIdentifier(LibraryLoader* loader,
                                                  const rapidjson::Value& type);

  // Gets a Type object representing the |type|.  |type| is a JSON object a
  // field "kind" that states the type (e.g., "array", "vector", "foo.bar/Baz").
  // |loader| is the set of libraries to use to find types that need to be given
  // by identifier (e.g., "foo.bar/Baz").
  static std::unique_ptr<Type> GetType(LibraryLoader* loader, const rapidjson::Value& type);
};

// An instance of this class is created when the system can't determine the real
// class (e.g., in cases of corrupted metadata).
class InvalidType : public Type {
 public:
  std::string Name() const override { return "unknown"; }
  size_t InlineSize(WireVersion version) const override { return 0; }
  bool IsValid() const override { return false; }
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;
  void Visit(TypeVisitor* visitor) const override;
};

// The type of an empty request or response.
// This has zero size, unlike an empty struct which is a single 0x00 byte.
class EmptyPayloadType : public Type {
 public:
  std::string Name() const override { return "empty-payload"; }
  size_t InlineSize(WireVersion version) const override { return 0; }
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;
  void Visit(TypeVisitor* visitor) const override;
};

class BoolType : public Type {
 public:
  std::string Name() const override { return "bool"; }
  size_t InlineSize(WireVersion version) const override { return 1; }
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;
  void Visit(TypeVisitor* visitor) const override;
};

// A generic type that can be used for any integer numeric value that corresponds to a
// C++ integral value.
template <typename T>
class IntegralType : public Type {
  static_assert(std::is_integral_v<T> && !std::is_same_v<T, bool>,
                "IntegralType can only be used for integers");

 public:
  size_t InlineSize(WireVersion version) const override { return sizeof(T); }

  std::string CppName() const override { return this->Name() + "_t"; }

  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override {
    auto got = decoder->GetAddress(offset, sizeof(T));
    if (got == nullptr) {
      return std::make_unique<InvalidValue>();
    }
    T value = *reinterpret_cast<const T*>(got);
    if (value < 0) {
      // Sign-extend to 64 bits so the cast to uint64_t is a no-op.
      int64_t tmp = value;
      // Cast to uint64 before the negation to avoid a int64 overflow (-min_int64 can't be
      // represented with a int64).
      return std::make_unique<IntegerValue>(-static_cast<uint64_t>(tmp), true);
    }
    return std::make_unique<IntegerValue>(value, false);
  }
};

class Int8Type : public IntegralType<int8_t> {
 public:
  enum class Kind : uint8_t { kChar, kDecimal };
  explicit Int8Type(Kind kind = Kind::kDecimal) : kind_(kind) {}

  std::string Name() const override;
  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  Kind kind_;
};

class Int16Type : public IntegralType<int16_t> {
 public:
  enum class Kind : uint8_t { kDecimal };
  explicit Int16Type(Kind kind = Kind::kDecimal) : kind_(kind) {}

  std::string Name() const override { return "int16"; }
  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  Kind kind_;
};

class Int32Type : public IntegralType<int32_t> {
 public:
  enum class Kind : uint8_t { kDecimal, kFutex };
  explicit Int32Type(Kind kind = Kind::kDecimal) : kind_(kind) {}

  std::string Name() const override;
  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  Kind kind_;
};

class Int64Type : public IntegralType<int64_t> {
 public:
  enum class Kind : uint8_t { kDecimal, kDuration, kTime, kMonotonicTime };
  explicit Int64Type(Kind kind = Kind::kDecimal) : kind_(kind) {}

  std::string Name() const override;
  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  Kind kind_;
};

class Uint8Type : public IntegralType<uint8_t> {
 public:
  enum class Kind : uint8_t { kDecimal, kHexaDecimal, kPacketGuestVcpuType };
  explicit Uint8Type(Kind kind = Kind::kDecimal) : kind_(kind) {}

  std::string Name() const override;
  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  Kind kind_;
};

class Uint16Type : public IntegralType<uint16_t> {
 public:
  enum class Kind : uint8_t { kDecimal, kHexaDecimal, kPacketPageRequestCommand };
  explicit Uint16Type(Kind kind = Kind::kDecimal) : kind_(kind) {}

  std::string Name() const override;
  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  Kind kind_;
};

class Uint32Type : public IntegralType<uint32_t> {
 public:
  enum class Kind : uint8_t {
    kBtiPerm,
    kCachePolicy,
    kChannelOption,
    kClock,
    kDecimal,
    kExceptionChannelType,
    kExceptionState,
    kFeatureKind,
    kGuestTrap,
    kHexaDecimal,
    kInfoMapsType,
    kInterruptFlags,
    kIommuType,
    kKtraceControlAction,
    kObjectInfoTopic,
    kObjType,
    kPciBarType,
    kPolicyAction,
    kPolicyCondition,
    kPolicyTopic,
    kPortPacketType,
    kProfileInfoFlags,
    kPropType,
    kRights,
    kRsrcKind,
    kSignals,
    kSocketCreateOptions,
    kSocketReadOptions,
    kSocketDisposition,
    kStatus,
    kSystemEventType,
    kSystemPowerctl,
    kThreadState,
    kThreadStateTopic,
    kTimerOption,
    kVcpu,
    kVmOption,
    kVmoCreationOption,
    kVmoOp,
    kVmoOption,
    kVmoType
  };

  explicit Uint32Type(Kind kind = Kind::kDecimal) : kind_(kind) {}

  Kind kind() const { return kind_; }
  void set_kind(Kind kind) { kind_ = kind; }

  Uint32Type* AsUint32Type() override { return this; }

  std::string Name() const override;
  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  Kind kind_;
};

class Uint64Type : public IntegralType<uint64_t> {
 public:
  enum class Kind : uint8_t {
    kDecimal,
    kHexaDecimal,
    kVaddr,
    kSize,
    kPaddr,
    kGpAddr,
    kUintptr,
    kKoid
  };
  explicit Uint64Type(Kind kind = Kind::kDecimal) : kind_(kind) {}

  std::string Name() const override;
  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  Kind kind_;
};

// A type that can be used for having actual and requested values.
class ActualAndRequestedType : public Type {
 public:
  size_t InlineSize(WireVersion version) const override { return 2 * sizeof(size_t); }
  std::string Name() const override { return "size"; }
  std::string CppName() const override { return "size_t"; }

  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override {
    FX_LOGS(FATAL) << "Not reachable";
    return nullptr;
  }

  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override;
  void Visit(TypeVisitor* visitor) const override;
};

// A generic type that can be used for any numeric value that corresponds to a
// C++ arithmetic value.
template <typename T>
class NumericType : public Type {
  static_assert(std::is_arithmetic_v<T> && !std::is_same_v<T, bool>,
                "NumericType can only be used for numerics");

 public:
  size_t InlineSize(WireVersion version) const override { return sizeof(T); }

  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override {
    auto got = decoder->GetAddress(offset, sizeof(T));
    if (got == nullptr) {
      return std::make_unique<InvalidValue>();
    }
    return std::make_unique<DoubleValue>(*reinterpret_cast<const T*>(got));
  }
};

class Float32Type : public NumericType<float> {
 public:
  std::string Name() const override { return "float32"; }
  std::string CppName() const override { return "float"; }

  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override {
    double result;
    if (!value->GetDoubleValue(&result)) {
      printer << Red << "invalid" << ResetColor;
    } else {
      printer << Blue << std::to_string(static_cast<float>(result)) << ResetColor;
    }
  }

  void Visit(TypeVisitor* visitor) const override;
};

class Float64Type : public NumericType<double> {
 public:
  std::string Name() const override { return "float64"; }
  std::string CppName() const override { return "double"; }

  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override {
    double result;
    if (!value->GetDoubleValue(&result)) {
      printer << Red << "invalid" << ResetColor;
    } else {
      printer << Blue << std::to_string(result) << ResetColor;
    }
  }

  void Visit(TypeVisitor* visitor) const override;
};

class StringType : public Type {
 public:
  std::string Name() const override { return "string"; }
  std::string CppName() const override { return "std::string"; }
  size_t InlineSize(WireVersion version) const override { return 16; }
  bool Nullable() const override { return true; }
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;
  void Visit(TypeVisitor* visitor) const override;
};

class HandleType : public Type {
 public:
  explicit HandleType(std::optional<zx_rights_t> rights = std::nullopt,
                      std::optional<zx_obj_type_t> obj_type = std::nullopt, bool nullable = false)
      : rights_(rights), obj_type_(obj_type), nullable_(nullable) {}
  std::string Name() const override { return "handle"; }
  std::string CppName() const override { return "zx::handle"; }
  size_t InlineSize(WireVersion version) const override { return sizeof(zx_handle_t); }
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;
  void Visit(TypeVisitor* visitor) const override;
  bool Nullable() const override { return nullable_; }
  zx_obj_type_t ObjectType() const { return obj_type_.value_or(ZX_OBJ_TYPE_NONE); }
  zx_rights_t Rights() const { return rights_.value_or(ZX_RIGHT_SAME_RIGHTS); }
  const HandleType* AsHandleType() const override { return this; }

 private:
  std::optional<zx_rights_t> rights_;
  std::optional<zx_obj_type_t> obj_type_;
  bool nullable_;
};

class EnumType : public Type {
 public:
  explicit EnumType(const Enum& enum_definition) : enum_definition_(enum_definition) {}

  const Enum& enum_definition() const { return enum_definition_; }

  std::string Name() const override;
  std::string CppName() const override;
  size_t InlineSize(WireVersion version) const override;
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;
  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  const Enum& enum_definition_;
};

class BitsType : public Type {
 public:
  explicit BitsType(const Bits& bits_definition) : bits_definition_(bits_definition) {}

  const Bits& bits_definition() const { return bits_definition_; }

  std::string Name() const override;
  std::string CppName() const override;
  size_t InlineSize(WireVersion version) const override;
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;
  void PrettyPrint(const Value* value, PrettyPrinter& printer) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  const Bits& bits_definition_;
};

class UnionType : public Type {
 public:
  UnionType(const Union& union_definition, bool nullable)
      : union_definition_(union_definition), nullable_(nullable) {}

  const Union& union_definition() const { return union_definition_; }

  const UnionType* AsUnionType() const override { return this; }
  std::string Name() const override;
  std::string CppName() const override;
  size_t InlineSize(WireVersion version) const override;
  bool Nullable() const override { return nullable_; }
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;

  void Visit(TypeVisitor* visitor) const override;

 private:
  const Union& union_definition_;
  const bool nullable_;
};

class StructType : public Type {
 public:
  StructType(const Struct& struct_definition, bool nullable)
      : struct_definition_(struct_definition), nullable_(nullable) {}

  const Struct& struct_definition() const { return struct_definition_; }

  const StructType* AsStructType() const override { return this; }
  std::string Name() const override;
  std::string CppName() const override;
  size_t InlineSize(WireVersion version) const override;
  bool Nullable() const override { return nullable_; }
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  const Struct& struct_definition_;
  const bool nullable_;
};

class ElementSequenceType : public Type {
 public:
  explicit ElementSequenceType(std::unique_ptr<Type> component_type)
      : component_type_(std::move(component_type)) {
    FX_DCHECK(component_type_.get() != nullptr);
  }

  const Type* component_type() const { return component_type_.get(); }

  const Type* GetComponentType() const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  std::unique_ptr<Type> component_type_;
};

class ArrayType : public ElementSequenceType {
 public:
  ArrayType(std::unique_ptr<Type> component_type, uint32_t count)
      : ElementSequenceType(std::move(component_type)), count_(count) {}

  uint32_t count() const { return count_; }

  bool IsArray() const override;
  std::string Name() const override;
  std::string CppName() const override;
  void PrettyPrint(PrettyPrinter& printer) const override;
  size_t InlineSize(WireVersion version) const override;
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  uint32_t count_;
};

class VectorType : public ElementSequenceType {
 public:
  explicit VectorType(std::unique_ptr<Type> component_type)
      : ElementSequenceType(std::move(component_type)) {}

  std::string Name() const override;
  std::string CppName() const override;
  void PrettyPrint(PrettyPrinter& printer) const override;
  size_t InlineSize(WireVersion version) const override { return 16; }
  bool Nullable() const override { return true; }
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;
  void Visit(TypeVisitor* visitor) const override;
};

class TableType : public Type {
 public:
  explicit TableType(const Table& table_definition) : table_definition_(table_definition) {}

  const Table& table_definition() const { return table_definition_; }

  const TableType* AsTableType() const override { return this; }
  std::string Name() const override;
  std::string CppName() const override;
  size_t InlineSize(WireVersion version) const override { return 16; }
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;
  void Visit(TypeVisitor* visitor) const override;

 private:
  const Table& table_definition_;
};

class FidlMessageType : public Type {
 public:
  std::string Name() const override { return "fidl-message"; }
  size_t InlineSize(WireVersion version) const override { return 0; }
  std::unique_ptr<Value> Decode(MessageDecoder* decoder, uint64_t offset) const override;
  void Visit(TypeVisitor* visitor) const override;
};

}  // namespace fidl_codec

#endif  // SRC_LIB_FIDL_CODEC_WIRE_TYPES_H_