#pragma once

#include <types.h>

class CPU {
public:
  struct features_t {
    bool fpu : 1 = false;   // onboard x87 fpu
    bool vme : 1 = false;   // virtual 8086 mode
    bool de : 1 = false;    // debugging extensions
    bool pse : 1 = false;   // page size extension
    bool tsc : 1 = false;   // time stamp counter
    bool msr : 1 = false;   // model specific registers
    bool pae : 1 = false;   // physical address extension
    bool mce : 1 = false;   // machine check exception
    bool cx8 : 1 = false;   // compare and swap instruction
    bool apic : 1 = false;  // onboard apic
    bool __r0 : 1 = false;
    bool sep : 1 = false;    // sysenter and sysexit
    bool mtrr : 1 = false;   // memory type range registers
    bool pge : 1 = false;    // page global enable bit
    bool mca : 1 = false;    // machine check architecture
    bool cmov : 1 = false;   // conditional move
    bool pat : 1 = false;    // page attribute table
    bool pse36 : 1 = false;  // 36-bit page size extension
    bool psn : 1 = false;    // processor serial number
    bool clfsh : 1 = false;  // clflish
    bool __r1 : 1 = false;
    bool ds : 1 = false;    // debug store
    bool acpi : 1 = false;  // onboard thermal control for acpi
    bool mmx : 1 = false;   // mmx extension
    bool fxsr : 1 = false;  // fxsave, fxrestor
    bool sse : 1 = false;   // sse extension
    bool sse2 : 1 = false;  // sse2 extension
    bool ss : 1 = false;    // cpu cache implements self-snoop
    bool htt : 1 = false;   // hyper-threading
    bool tm : 1 = false;    // thermal monitor auto-limits temperature
    bool ia64 : 1 = false;  // ia64 processor emulating x86
    bool pbe : 1 = false;   // pending break enable

    bool sse3 : 1 = false;       // sse3 extension
    bool pclmulqdq : 1 = false;  // pclmulqdq
    bool dtes64 : 1 = false;     // 64-bit debug store
    bool monitor : 1 = false;    // monitor and mwait instructions (sse3)
    bool ds_cpl : 1 = false;     // CPL qualified debug store
    bool vmx : 1 = false;        // virtual machine extensions
    bool smx : 1 = false;        // safer mode extensions
    bool est : 1 = false;        // enahnced speedstep
    bool tm2 : 1 = false;        // thermal monitor 2
    bool ssse3 : 1 = false;      // supplemental sse3
    bool cnxt_id : 1 = false;    // L1 context ID
    bool sdbg : 1 = false;       // Silicon Debug interface
    bool fma : 1 = false;        // fused multiply-add
    bool cx16 : 1 = false;       // cmpxchg16b instruction
    bool xtpr : 1 = false;       // can disable sending task priority messages
    bool pdcm : 1 = false;       // perfom and debug capability
    bool __r2 : 1 = false;
    bool pcid : 1 = false;          // process context identifiers
    bool dca : 1 = false;           // direct cache access for DMA writes
    bool sse41 : 1 = false;         // sse 4.1
    bool sse42 : 1 = false;         // sse 4.2
    bool x2apic : 1 = false;        // x2APIX
    bool movbe : 1 = false;         // movbe instruction
    bool popcnt : 1 = false;        // popcnt instruction
    bool tsc_deadline : 1 = false;  // apic implements one-shot operation using tsc deadline
    bool aes : 1 = false;           // aes instruction set
    bool xsave : 1 = false;         // xsave, xrestor, xsetbv, xgetbv
    bool oxsave : 1 = false;        // xsave enabled by OS
    bool avx : 1 = false;           // advanced vector extensions
    bool f16c : 1 = false;          // f16c fp feature
    bool rdrnd : 1 = false;         // rdrand
    bool hypervisor : 1 = false;    // hypervisor present
  } __attribute__((packed));

  CPU();
  ~CPU() = default;

  [[nodiscard]] const char* manufacturer() const { return m_manufacturer; }
  [[nodiscard]] uint8_t family() const { return m_info.family == 15 ? m_info.family + m_info.family_ex : m_info.family; }
  [[nodiscard]] uint8_t model() const {
    return (m_info.model == 6 || m_info.model == 15) ? (static_cast<uint8_t>(m_info.model_ex << 4) + m_info.model)
                                                     : m_info.model;
  }
  [[nodiscard]] uint8_t stepping() const { return m_info.stepping; }
  [[nodiscard]] auto features() const { return m_features; }

private:
  char m_manufacturer[13] = {'\0'};
  struct version_t {
    uint8_t stepping : 4 = 0;
    uint8_t model : 4 = 0;
    uint8_t family : 4 = 0;
    uint8_t type : 2 = 0;
    [[maybe_unused]] uint8_t unused_1 : 2 = 0;
    uint8_t model_ex : 4 = 0;
    uint8_t family_ex : 8 = 0;
    [[maybe_unused]] uint8_t unused_2 : 4 = 0;
  } __attribute__((packed)) m_info;

  features_t m_features;
};