/*
 * Copyright (c) 2021, Jan de Visser <jan@de-visser.net>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <AK/NonnullRefPtr.h>
#include <LibCore/Object.h>
#include <LibSQL/Database.h>
#include <LibSQL/Result.h>
#include <LibSQL/Type.h>
#include <SQLServer/Forward.h>

namespace SQLServer {

class DatabaseConnection final : public Core::Object {
    C_OBJECT_ABSTRACT(DatabaseConnection)

public:
    static ErrorOr<NonnullRefPtr<DatabaseConnection>> create(StringView database_path, DeprecatedString database_name, int client_id);
    ~DatabaseConnection() override = default;

    static RefPtr<DatabaseConnection> connection_for(SQL::ConnectionID connection_id);
    SQL::ConnectionID connection_id() const { return m_connection_id; }
    int client_id() const { return m_client_id; }
    NonnullRefPtr<SQL::Database> database() { return m_database; }
    StringView database_name() const { return m_database_name; }
    void disconnect();
    SQL::ResultOr<SQL::StatementID> prepare_statement(StringView sql);

private:
    DatabaseConnection(NonnullRefPtr<SQL::Database> database, DeprecatedString database_name, int client_id);

    NonnullRefPtr<SQL::Database> m_database;
    DeprecatedString m_database_name;
    SQL::ConnectionID m_connection_id { 0 };
    int m_client_id { 0 };
};

}