/*
 * Copyright (C) Contributors to the Suwayomi project
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

export class CustomCache {
    private keyToResponseMap = new Map<string, unknown>();

    private keyToFetchTimestampMap = new Map<string, number>();

    public readonly createKeyFn = (endpoint: string, data: unknown): string => `${endpoint}_${JSON.stringify(data)}`;

    constructor(createKeyFn?: (endpoint: string, data: unknown) => string) {
        this.createKeyFn = createKeyFn ?? this.createKeyFn;
    }

    public getKeyFor(key: string, data: unknown): string {
        return this.createKeyFn(key, data);
    }

    public cacheResponse(endpoint: string, data: unknown, response: unknown) {
        const createdKey = this.getKeyFor(endpoint, data);
        this.keyToFetchTimestampMap.set(createdKey, Date.now());
        this.keyToResponseMap.set(createdKey, response);
    }

    public getFetchTimestampFor(endpoint: string, data: unknown): number | undefined {
        const key = this.getKeyFor(endpoint, data);
        return this.keyToFetchTimestampMap.get(key);
    }

    public getResponseFor<Response = any>(endpoint: string, data: unknown): Response | undefined {
        const key = this.getKeyFor(endpoint, data);
        return this.keyToResponseMap.get(key) as Response;
    }
}