open-gsio/packages/server/storage/BunSqliteKVNamespace.ts

import type {
    KVNamespace,
    KVNamespaceGetOptions,
    KVNamespaceListOptions,
    KVNamespaceListResult, KVNamespacePutOptions
} from "@cloudflare/workers-types";
import {BunSqliteKeyValue} from "bun-sqlite-key-value";
import {OPEN_GSIO_DATA_DIR} from "../constants";

interface BaseKV extends KVNamespace {

}


interface Options {
    namespace: string;
    path: string
}

const defaultOptions = {
    namespace: "open-gsio",
    path: OPEN_GSIO_DATA_DIR
};

export class BunSqliteKVNamespace implements BaseKV {
    private db: any;

    constructor(options?: { namespace?: string, path?: string }) {
        const merged = {...defaultOptions, ...options};
        const {namespace, path} = merged;

        this.db = new BunSqliteKeyValue(`${path}/${namespace}`);
    }

    async delete(key: string): Promise<void> {
        await this.db.delete(key);
    }

    async get(key: string | Array<string>, options?: Partial<KVNamespaceGetOptions<undefined>> | "text" | "json" | "arrayBuffer" | "stream" | KVNamespaceGetOptions<"text"> | KVNamespaceGetOptions<"json"> | KVNamespaceGetOptions<"arrayBuffer"> | KVNamespaceGetOptions<"stream"> | "text" | "json"): Promise<any> {
        if (Array.isArray(key)) {
            const result = new Map();
            for (const k of key) {
                const value = await this.db.get(k);
                result.set(k, value);
            }
            return result;
        }

        const value = await this.db.get(key);
        if (value === undefined) return null;

        if (!options || options === "text" || (typeof options === "object" && options.type === "text")) {
            return value;
        }
        if (options === "json" || (typeof options === "object" && options.type === "json")) {
            return JSON.parse(value);
        }
        if (options === "arrayBuffer" || (typeof options === "object" && options.type === "arrayBuffer")) {
            return new TextEncoder().encode(value).buffer;
        }
        if (options === "stream" || (typeof options === "object" && options.type === "stream")) {
            return new ReadableStream({
                start(controller) {
                    controller.enqueue(new TextEncoder().encode(value));
                    controller.close();
                }
            });
        }
        return value;
    }

    getWithMetadata(key: string | Array<string>, options?: any): any {
        return null;
    }

    async list<Metadata = unknown>(options?: KVNamespaceListOptions): Promise<KVNamespaceListResult<Metadata, string>> {
        const keys = await this.db.keys();
        return {
            keys: keys.map(key => ({name: key})),
            list_complete: true,
            cursor: ''
        };
    }

    async put(key: string, value: string | ArrayBuffer | ArrayBufferView | ReadableStream, options?: KVNamespacePutOptions): Promise<void> {
        if (value instanceof ArrayBuffer || ArrayBuffer.isView(value)) {
            value = new TextDecoder().decode(value);
        } else if (value instanceof ReadableStream) {
            const reader = value.getReader();
            const chunks = [];
            while (true) {
                const {done, value} = await reader.read();
                if (done) break;
                chunks.push(value);
            }
            value = new TextDecoder().decode(new Uint8Array(Buffer.concat(chunks)));
        }
        await this.db.set(key, value);
    }
}