#pragma once

#include <exception>
#include <iostream>
#include <optional>
#include <string>

const std::string NITRO_SQLITE_EXCEPTION_PREFIX = "[NativeNitroSQLiteException]";

enum NitroSQLiteExceptionType {
  UnknownError,
  DatabaseCannotBeOpened,
  DatabaseNotOpen,
  UnableToAttachToDatabase,
  SqlExecutionError,
  CouldNotLoadFile,
  NoBatchCommandsProvided
};

inline std::unordered_map<NitroSQLiteExceptionType, std::string> exceptionTypeStrings = {
    {UnknownError, "UnknownError"},
    {DatabaseCannotBeOpened, "DatabaseCannotBeOpened"},
    {DatabaseNotOpen, "DatabaseNotOpen"},
    {UnableToAttachToDatabase, "UnableToAttachToDatabase"},
    {SqlExecutionError, "SqlExecutionError"},
    {CouldNotLoadFile, "CouldNotLoadFile"},
    {NoBatchCommandsProvided, "NoBatchCommandsProvided"}};

inline std::string typeToString(NitroSQLiteExceptionType type) {
  return exceptionTypeStrings[type];
}

class NitroSQLiteException : public std::exception {
public:
  explicit NitroSQLiteException(const char* message) : NitroSQLiteException(NitroSQLiteExceptionType::UnknownError, message) {}
  explicit NitroSQLiteException(const std::string& message) : NitroSQLiteException(NitroSQLiteExceptionType::UnknownError, message) {}
  NitroSQLiteException(const NitroSQLiteExceptionType& type, const char* message) : NitroSQLiteException(type, std::string(message)) {}
  NitroSQLiteException(const NitroSQLiteExceptionType& type, const std::string& message)
      : _exceptionString(NITRO_SQLITE_EXCEPTION_PREFIX + "[" + typeToString(type) + "] " + message) {}

private:
  const std::string _exceptionString;

public:
  [[nodiscard]] const char* what() const noexcept override {
    return this->_exceptionString.c_str();
  }

  static NitroSQLiteException DatabaseNotOpen(const std::string& dbName) {
    return NitroSQLiteException(NitroSQLiteExceptionType::UnableToAttachToDatabase, dbName + " is not open");
  }

  static NitroSQLiteException SqlExecution(const std::string& errorMessage) {
    return NitroSQLiteException(NitroSQLiteExceptionType::SqlExecutionError, errorMessage);
  }

  static NitroSQLiteException DatabaseFileNotFound(const std::string& dbFilePath) {
    return NitroSQLiteException(NitroSQLiteExceptionType::SqlExecutionError, "Database file not found: " + dbFilePath);
  }

  static NitroSQLiteException CouldNotLoadFile(const std::string& fileLocation, std::optional<std::string> additionalLine = std::nullopt) {
    const auto message = "Could not load file: " + fileLocation + (additionalLine ? (". " + *additionalLine) : "");
    return NitroSQLiteException(NitroSQLiteExceptionType::CouldNotLoadFile, message);
  }
};