ladybird/Services/RequestServer/Cache/CacheIndex.h

/*
 * Copyright (c) 2025, Tim Flynn <trflynn89@ladybird.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <AK/Error.h>
#include <AK/HashMap.h>
#include <AK/Time.h>
#include <AK/Types.h>
#include <LibDatabase/Database.h>
#include <LibHTTP/HeaderMap.h>
#include <LibRequests/CacheSizes.h>

namespace RequestServer {

// The cache index is a SQL database containing metadata about each cache entry. An entry in the index is created once
// the entire cache entry has been successfully written to disk.
class CacheIndex {
    struct Entry {
        u64 cache_key { 0 };

        String url;
        HTTP::HeaderMap response_headers;
        u64 data_size { 0 };

        UnixDateTime request_time;
        UnixDateTime response_time;
        UnixDateTime last_access_time;
    };

public:
    static ErrorOr<CacheIndex> create(Database::Database&);

    void create_entry(u64 cache_key, String url, HTTP::HeaderMap, u64 data_size, UnixDateTime request_time, UnixDateTime response_time);
    void remove_entry(u64 cache_key);
    void remove_entries_accessed_since(UnixDateTime, Function<void(u64 cache_key)> on_entry_removed);

    Optional<Entry&> find_entry(u64 cache_key);

    void update_response_headers(u64 cache_key, HTTP::HeaderMap);
    void update_last_access_time(u64 cache_key);

    Requests::CacheSizes estimate_cache_size_accessed_since(UnixDateTime since) const;

private:
    struct Statements {
        Database::StatementID insert_entry { 0 };
        Database::StatementID remove_entry { 0 };
        Database::StatementID remove_entries_accessed_since { 0 };
        Database::StatementID select_entry { 0 };
        Database::StatementID update_response_headers { 0 };
        Database::StatementID update_last_access_time { 0 };
        Database::StatementID estimate_cache_size_accessed_since { 0 };
    };

    CacheIndex(Database::Database&, Statements);

    Database::Database& m_database;
    Statements m_statements;

    HashMap<u32, Entry> m_entries;
};

}