/* ========================================

  Copyright (C) 2024-2025 Amlal El Mahrouss, Licensed under the Apache 2.0 license

======================================== */

#pragma once

#include <CompilerKit/Config.h>
#include <CompilerKit/Macros.h>
#include <CompilerKit/StringKit.h>

#define CK_ASSEMBLY_INTERFACE : public ::CompilerKit::AssemblyInterface
#define CK_ENCODER : public ::CompilerKit::EncoderInterface

namespace CompilerKit {
class AssemblyFactory;
class AssemblyInterface;

/// =========================================================== ///
/// @brief Simple assembly factory
/// =========================================================== ///
class AssemblyFactory final {
 public:
  explicit AssemblyFactory() = default;
  ~AssemblyFactory()         = default;

  NECTI_COPY_DEFAULT(AssemblyFactory);

 public:
  enum {
    kArchInvalid = 0,
    kArchAMD64   = 100,
    kArch32x0,
    kArch64x0,
    kArchRISCV,
    kArchPowerPC,
    kArchAARCH64,
    kArchUnknown,
    kArchCount = kArchUnknown - kArchAMD64,
  };

  Int32 Compile(std::string sourceFile, const Int32& arch) noexcept;

  void               Mount(AssemblyInterface* mountPtr) noexcept;
  AssemblyInterface* Unmount() noexcept;

 private:
  AssemblyInterface* fMounted{nullptr};
};

/// =========================================================== ///
///	@brief Assembly to binary generator class.
/// @note This interface creates according to the CPU target of the child class.
/// =========================================================== ///
class AssemblyInterface {
 public:
  explicit AssemblyInterface() = default;
  virtual ~AssemblyInterface() = default;

  NECTI_COPY_DEFAULT(AssemblyInterface);

  virtual UInt32 Arch() noexcept { return AssemblyFactory::kArchAMD64; }

  /// =========================================================== ///
  /// @brief compile to object file.
  /// @note Example C++ -> MASM -> AE object.
  /// =========================================================== ///
  virtual Int32 CompileToFormat(std::string src, Int32 arch) = 0;
};

/// =========================================================== ///
/// @brief Number casting unions for different sizes.
/// =========================================================== ///
union NumberCastBase {
  NumberCastBase()  = default;
  ~NumberCastBase() = default;
};

union NumberCast64 final {
  NumberCast64() = default;
  explicit NumberCast64(UInt64 raw) : raw(raw) {}

  ~NumberCast64() { raw = 0; }

  Char   number[8];
  UInt64 raw;
};

union NumberCast32 final {
  NumberCast32() = default;
  explicit NumberCast32(UInt32 raw) : raw(raw) {}

  ~NumberCast32() { raw = 0; }

  Char   number[4];
  UInt32 raw;
};

union NumberCast16 final {
  NumberCast16() = default;
  explicit NumberCast16(UInt16 raw) : raw(raw) {}

  ~NumberCast16() { raw = 0; }

  Char   number[2];
  UInt16 raw;
};

union NumberCast8 final {
  NumberCast8() = default;
  explicit NumberCast8(UInt8 raw) : raw(raw) {}

  ~NumberCast8() { raw = 0; }

  Char  number;
  UInt8 raw;
};

/// =========================================================== ///
/// @brief Assembly encoder interface.
/// =========================================================== ///
class EncoderInterface {
 public:
  explicit EncoderInterface() = default;
  virtual ~EncoderInterface() = default;

  NECTI_COPY_DEFAULT(EncoderInterface);

  virtual std::string CheckLine(std::string line, std::string file)               = 0;
  virtual bool        WriteLine(std::string line, std::string file)               = 0;
  virtual bool        WriteNumber(const std::size_t& pos, std::string& from_what) = 0;
};

/// =========================================================== ///
/// @brief Different architecture encoders.
/// =========================================================== ///

#ifdef __ASM_NEED_AMD64__

class EncoderAMD64 final : public EncoderInterface {
 public:
  explicit EncoderAMD64()  = default;
  ~EncoderAMD64() override = default;

  NECTI_COPY_DEFAULT(EncoderAMD64);

  virtual std::string CheckLine(std::string line, std::string file) override;
  virtual bool        WriteLine(std::string line, std::string file) override;
  virtual bool        WriteNumber(const std::size_t& pos, std::string& from_what) override;

  virtual bool WriteNumber16(const std::size_t& pos, std::string& from_what);
  virtual bool WriteNumber32(const std::size_t& pos, std::string& from_what);
  virtual bool WriteNumber8(const std::size_t& pos, std::string& from_what);
};

#endif  // __ASM_NEED_AMD64__

#ifdef __ASM_NEED_ARM64__

class EncoderARM64 final : public EncoderInterface {
 public:
  explicit EncoderARM64()  = default;
  ~EncoderARM64() override = default;

  NECTI_COPY_DEFAULT(EncoderARM64);

  virtual std::string CheckLine(std::string line, std::string file) override;
  virtual bool        WriteLine(std::string line, std::string file) override;
  virtual bool        WriteNumber(const std::size_t& pos, std::string& from_what) override;
};

#endif  // __ASM_NEED_ARM64__

#ifdef __ASM_NEED_64x0__

class Encoder64x0 final : public EncoderInterface {
 public:
  explicit Encoder64x0()  = default;
  ~Encoder64x0() override = default;

  NECTI_COPY_DEFAULT(Encoder64x0);

  virtual std::string CheckLine(std::string line, std::string file) override;
  virtual bool        WriteLine(std::string line, std::string file) override;
  virtual bool        WriteNumber(const std::size_t& pos, std::string& from_what) override;
};

#endif  // __ASM_NEED_64x0__

#ifdef __ASM_NEED_32x0__

class Encoder32x0 final : public EncoderInterface {
 public:
  explicit Encoder32x0()  = default;
  ~Encoder32x0() override = default;

  NECTI_COPY_DEFAULT(Encoder32x0);

  virtual std::string CheckLine(std::string line, std::string file) override;
  virtual bool        WriteLine(std::string line, std::string file) override;
  virtual bool        WriteNumber(const std::size_t& pos, std::string& from_what) override;
};

#endif  // __ASM_NEED_32x0__

#ifdef __ASM_NEED_PPC__

class EncoderPowerPC final : public EncoderInterface {
 public:
  explicit EncoderPowerPC()  = default;
  ~EncoderPowerPC() override = default;

  NECTI_COPY_DEFAULT(EncoderPowerPC);

  virtual std::string CheckLine(std::string line, std::string file) override;
  virtual bool        WriteLine(std::string line, std::string file) override;
  virtual bool        WriteNumber(const std::size_t& pos, std::string& from_what) override;
};

#endif  // __ASM_NEED_32x0__
}  // namespace CompilerKit