// A key value key to the database that is mostly used for atproto session storage and state storage during the oauth session creation
// The "stores" are divided up by a where on the store type so it can share the same interface just with that

import { db } from './db';
import { keyValueStore } from './db/schema';
import { and, eq } from 'drizzle-orm';

export const SESSION_STORE = 'sessions';
export const STATE_STORE = 'states';

export class Cache {

    db: typeof db;
    cacheName: string;

    constructor(database: typeof db, cacheName: string) {
        this.db = database;
        this.cacheName = cacheName;
    }

    async get(key: string) {
        const result = await this.db.select().from(keyValueStore).where(and(
            eq(keyValueStore.key, key),
            eq(keyValueStore.storeName, this.cacheName)
        )).limit(1);
        if(result.length > 0){
            return result[0].value;
        }
        return null;
    }

    async set(key: string, value: string) {
        return this.db.insert(keyValueStore)
            .values({ key, value, storeName: this.cacheName, createdAt: new Date() })
            .onConflictDoUpdate({
                target: keyValueStore.key,
                set: { value, createdAt: new Date() }
            });
    }

    async delete(key: string) {
        return this.db.delete(keyValueStore).where(eq(keyValueStore.key, key));
    }

}