#ifndef ARBUTILS_EXCEPTION_HPP
#define ARBUTILS_EXCEPTION_HPP

#include <exception>
#include <sstream>
#include <string>
#include <utility>
#if !WINDOWS
#define BACKWARD_HAS_DW 1
#include "../extern/backward.hpp"
#endif

namespace ArbUt {
    class Exception : std::logic_error {
#if !WINDOWS
        backward::StackTrace _stack;
#endif

    public:
        explicit Exception(const std::string& msg) : std::logic_error(msg) {
#if !WINDOWS
            _stack.load_here(32);
#endif
        }

        Exception(const Exception& e) noexcept
            : std::logic_error(e.what())
#if !WINDOWS
              ,
              _stack(e._stack)
#endif
        {
        }

        const char* what() const _GLIBCXX_TXN_SAFE_DYN _GLIBCXX_NOTHROW override { return logic_error::what(); }

        [[nodiscard]] std::string GetStacktrace([[maybe_unused]] size_t depth = 6,
                                                [[maybe_unused]] bool include_addr = true) const {
#if !WINDOWS
            if (_stack.size() == 0) {
                return "No stack trace could be retrieved.";
            }
            backward::TraceResolver tr;
            tr.load_stacktrace(_stack);
            std::stringstream ss;
            ss << "Stacktrace with depth " << depth << ": " << std::endl;
            bool foundExceptionClass = false;
            size_t framesAppended = -1;
            for (size_t i = 0; i < _stack.size(); ++i) {
                backward::ResolvedTrace trace = tr.resolve(_stack[i]);
                if (trace.source.filename.empty()) {
                    AppendNoSourceStack(ss, trace, include_addr);
                } else {
                    AppendSourceStack(ss, trace, foundExceptionClass);
                    if (foundExceptionClass){
                        framesAppended++;
                        if (framesAppended >= depth)
                            break;
                    }
                }
            }
            return ss.str();
#else
            return "Stack trace not available on Windows.";
#endif
        }

#if !WINDOWS
    private:
        static void AppendNoSourceStack(std::stringstream& ss, const backward::ResolvedTrace& trace,
                                        bool include_addr) {
            auto objectName =
                (strrchr(trace.object_filename.c_str(), '/') ? strrchr(trace.object_filename.c_str(), '/') + 1
                                                             : trace.object_filename.c_str());
            auto function = trace.object_function;
            if (function.length() > 70) {
                function = function.substr(0, 67);
                function += "...";
            }
            ss << objectName;
            if (include_addr) {
                ss << "[" << trace.addr << "]";
            }
            ss << " " << function << std::endl;
        }
        static void AppendSourceStack(std::stringstream& ss, const backward::ResolvedTrace& trace, bool& foundExceptionClass) {
            auto fileName =
                (strrchr(trace.source.filename.c_str(), '/') ? strrchr(trace.source.filename.c_str(), '/') + 1
                                                             : trace.source.filename.c_str());
            if (strcmp(fileName, "Exception.hpp") == 0) {
                foundExceptionClass = true;
                return;
            } else if (!foundExceptionClass) {
                return;
            }

            auto function = trace.object_function;
            if (function.length() > 70) {
                function = function.substr(0, 67);
                function += "...";
            }
            ss << fileName << "[" << trace.source.line << "] " << function << std::endl;
        }
    };
#endif
}

#endif // ARBUTILS_EXCEPTION_HPP