#pragma once

#include <memory>
#include "Test.hpp"

namespace smart_pointer {
// `exception` class definition
class exception : std::exception {
    using base_class = std::exception;
    using base_class::base_class;
};

// `SmartPointer` class declaration
template<
        typename T,
        typename Allocator
>
class SmartPointer {
    // don't remove this macro
    ENABLE_CLASS_TESTS;

 public:
    using value_type = T;

    explicit SmartPointer(value_type *p = nullptr) {
        if (p == nullptr) {
            core = nullptr;
        } else {
            core = new Core;
            core->m_obj = p;
            core->count_ptr = 1;
        }
    }

    // copy constructor
    SmartPointer(const SmartPointer &p) {
        if (core != nullptr) {
            if (core->count_ptr > 1) {
                core->count_ptr -= 1;
            } else {
                delete core->m_obj;
                delete core;
                core = new Core;
            }
        }
        core = p.core;
        if (core != nullptr)
            core->count_ptr += 1;
    }

    // move constructor
    SmartPointer(SmartPointer &&p) {
        if (core != nullptr) {
            if (core->count_ptr > 1) {
                core->count_ptr -= 1;
            } else {
                delete core->m_obj;
                delete core;
                core = new Core;
            }
        }
        core = p.core;
        if (p.core != nullptr)
            p.core = nullptr;
    }

    // copy assigment
    SmartPointer &operator=(const SmartPointer &p) {
        if (core != nullptr) {
            if (core->count_ptr > 1) {
                core->count_ptr -= 1;
            } else {
                delete this->core->m_obj;
                delete this->core;
                core = nullptr;
                // core = new Core;
            }
        }
        core = p.core;
        if (core != nullptr)
            core->count_ptr += 1;
        return *this;
    }

    // move assigment
    SmartPointer &operator=(SmartPointer &&p) {
        if (core != nullptr) {
            if (core->count_ptr > 1) {
                core->count_ptr -= 1;
            } else {
                delete core->m_obj;
                delete core;
            }
        }
        core = p.core;
        if (core != nullptr) {
            p.core = nullptr;
        }
        return *this;
    }

    //
    SmartPointer &operator=(value_type *p) {
        if (core != nullptr) {
            if (core->count_ptr > 1) {
                core->count_ptr -= 1;
            } else {
                delete core->m_obj;
                delete core;
                core = new Core;
            }
        } else {
            core = new Core;
        }
        if (p != nullptr) {
            core->m_obj = p;
        } else {
            delete core;
            core = nullptr;
        }
        if (core != nullptr)
        core->count_ptr = 1;
        return *this;
    }

    ~SmartPointer() {
        if (core == nullptr) {
            delete core;
        } else if (core->count_ptr > 1) {
            core->count_ptr -= 1;
        } else {
            if (core->m_obj != nullptr) {
                delete core->m_obj;
            }
            delete core;
        }
    }

    // return reference to the object of class/type T
    // if SmartPointer contains nullptr throw `SmartPointer::exception`
    value_type &operator*() {
        if (get() == nullptr) {
            throw exception();
        } else {
            return *core->m_obj;
        }
    }

    const value_type &operator*() const {
        if (get() == nullptr) {
            throw exception();
        } else {
            return *core->m_obj;
        }
    }

    // return pointer to the object of class/type T
    value_type *operator->() const {
        return core ? core->m_obj : nullptr;
    }

    value_type *get() const {
        return core ? core->m_obj : nullptr;
    }

    // if pointer == nullptr => return false
    operator bool() const {
        return core != nullptr;
    }

    // if pointers points to the same address or both null => true
    template<typename U, typename AnotherAllocator>
    bool operator==(const SmartPointer<U, AnotherAllocator> &p) const {
        return (p.get() == nullptr && get() == nullptr) ||
               (p.get() != nullptr && get() != nullptr &&
                *p.get() == *get() &&
                static_cast<void *>(p.get()) == static_cast<void *>(get()));
    }

    // if pointers points to the same address or both null => false
    template<typename U, typename AnotherAllocator>
    bool operator!=(const SmartPointer<U, AnotherAllocator> &p) const {
        return !((p.get() == nullptr && get() == nullptr) ||
                 (p.get() != nullptr && get() != nullptr &&
                  *p.get() == *get() &&
                  static_cast<void *>(p.get()) ==
                  static_cast<void *>(get())));
    }

    // if smart pointer contains non-nullptr => return count owners
    // if smart pointer contains nullptr => return 0
    [[nodiscard]] std::size_t count_owners() const {
        if (core != nullptr)
            return core->count_ptr;
        return 0;
    }

 private:
    class Core {
     public:
        value_type *m_obj;
        std::size_t count_ptr;
    };

    Core *core = nullptr;
};
}  // namespace smart_pointer