FDBDatabase.ts

import FDBTransaction from "./FDBTransaction.js";
import Database from "./lib/Database.js";
import {
    ConstraintError,
    InvalidAccessError,
    InvalidStateError,
    NotFoundError,
    TransactionInactiveError,
} from "./lib/errors.js";
import FakeDOMStringList from "./lib/FakeDOMStringList.js";
import FakeEventTarget from "./lib/FakeEventTarget.js";
import ObjectStore from "./lib/ObjectStore.js";
import { queueTask } from "./lib/scheduling.js";
import { KeyPath, TransactionMode } from "./lib/types.js";
import validateKeyPath from "./lib/validateKeyPath.js";

const confirmActiveVersionchangeTransaction = (database: FDBDatabase) => {
    if (!database._runningVersionchangeTransaction) {
        throw new InvalidStateError();
    }

    // Find the latest versionchange transaction
    const transactions = database._rawDatabase.transactions.filter((tx) => {
        return tx.mode === "versionchange";
    });
    const transaction = transactions[transactions.length - 1];

    if (!transaction || transaction._state === "finished") {
        throw new InvalidStateError();
    }

    if (transaction._state !== "active") {
        throw new TransactionInactiveError();
    }

    return transaction;
};

// http://www.w3.org/TR/2015/REC-IndexedDB-20150108/#database-closing-steps
const closeConnection = (connection: FDBDatabase) => {
    connection._closePending = true;

    const transactionsComplete = connection._rawDatabase.transactions.every(
        (transaction) => {
            return transaction._state === "finished";
        },
    );

    if (transactionsComplete) {
        connection._closed = true;
        connection._rawDatabase.connections =
            connection._rawDatabase.connections.filter((otherConnection) => {
                return connection !== otherConnection;
            });
    } else {
        queueTask(() => {
            closeConnection(connection);
        });
    }
};

// http://www.w3.org/TR/2015/REC-IndexedDB-20150108/#database-interface
class FDBDatabase extends FakeEventTarget {
    public _closePending = false;
    public _closed = false;
    public _runningVersionchangeTransaction = false;
    public _rawDatabase: Database;

    public readonly name: string;
    public objectStoreNames: FakeDOMStringList;

    constructor(rawDatabase: Database) {
        super();

        this._rawDatabase = rawDatabase;
        this._rawDatabase.connections.push(this);

        this.name = rawDatabase.name;
        this.objectStoreNames = new FakeDOMStringList(
            ...Array.from(rawDatabase.rawObjectStores.keys()).sort(),
        );
    }

    public get version() : number {
        return this._rawDatabase.version
    }

    // http://w3c.github.io/IndexedDB/#dom-idbdatabase-createobjectstore
    public createObjectStore(
        name: string,
        options: { autoIncrement?: boolean; keyPath?: KeyPath } | null = {},
    ) {
        if (name === undefined) {
            throw new TypeError();
        }
        const transaction = confirmActiveVersionchangeTransaction(this);

        const keyPath =
            options !== null && options.keyPath !== undefined
                ? options.keyPath
                : null;
        const autoIncrement =
            options !== null && options.autoIncrement !== undefined
                ? options.autoIncrement
                : false;

        if (keyPath !== null) {
            validateKeyPath(keyPath);
        }

        if (this._rawDatabase.rawObjectStores.has(name)) {
            throw new ConstraintError();
        }

        if (autoIncrement && (keyPath === "" || Array.isArray(keyPath))) {
            throw new InvalidAccessError();
        }

        const objectStoreNames = [...this.objectStoreNames];
        transaction._rollbackLog.push(() => {
            const objectStore = this._rawDatabase.rawObjectStores.get(name);
            if (objectStore) {
                objectStore.deleted = true;
            }

            this.objectStoreNames = new FakeDOMStringList(...objectStoreNames);
            transaction._scope.delete(name);
            this._rawDatabase.rawObjectStores = new Map(Array.from(this._rawDatabase.rawObjectStores.entries()).filter(([it]) => it !== name));
        });

        const rawObjectStore = new ObjectStore(
            this._rawDatabase,
            name,
            keyPath,
            autoIncrement,
        );
        this.objectStoreNames._push(name);
        this.objectStoreNames._sort();
        transaction._scope.add(name);

        this._rawDatabase.rawObjectStores = new Map([...this._rawDatabase.rawObjectStores.entries(), [name, rawObjectStore]]);
        transaction.objectStoreNames = new FakeDOMStringList(
            ...this.objectStoreNames,
        );
        return transaction.objectStore(name);
    }

    public deleteObjectStore(name: string) {
        if (name === undefined) {
            throw new TypeError();
        }
        const transaction = confirmActiveVersionchangeTransaction(this);

        const store = this._rawDatabase.rawObjectStores.get(name);
        if (store === undefined) {
            throw new NotFoundError();
        }

        this.objectStoreNames = new FakeDOMStringList(
            ...Array.from(this.objectStoreNames).filter((objectStoreName) => {
                return objectStoreName !== name;
            }),
        );
        transaction.objectStoreNames = new FakeDOMStringList(
            ...this.objectStoreNames,
        );

        transaction._rollbackLog.push(() => {
            store.deleted = false;
            this._rawDatabase.rawObjectStores = new Map([...this._rawDatabase.rawObjectStores.entries(), [name, store]]);
            this.objectStoreNames._push(name);
            this.objectStoreNames._sort();
        });

        store.deleted = true;
        store.clear(transaction._rollbackLog);
        this._rawDatabase.rawObjectStores = new Map(Array.from(this._rawDatabase.rawObjectStores.entries()).filter(([it]) => it !== name));
        transaction._objectStoresCache.delete(name);
    }

    public transaction(storeNames: string | string[], mode?: TransactionMode) {
        mode = mode !== undefined ? mode : "readonly";
        if (
            mode !== "readonly" &&
            mode !== "readwrite" &&
            mode !== "versionchange"
        ) {
            throw new TypeError("Invalid mode: " + mode);
        }

        const hasActiveVersionchange = this._rawDatabase.transactions.some(
            (transaction) => {
                return (
                    transaction._state === "active" &&
                    transaction.mode === "versionchange" &&
                    transaction.db === this
                );
            },
        );
        if (hasActiveVersionchange) {
            throw new InvalidStateError();
        }

        if (this._closePending) {
            throw new InvalidStateError();
        }

        if (!Array.isArray(storeNames)) {
            storeNames = [storeNames];
        }
        if (storeNames.length === 0 && mode !== "versionchange") {
            throw new InvalidAccessError();
        }
        for (const storeName of storeNames) {
            if (!this.objectStoreNames.contains(storeName)) {
                throw new NotFoundError(
                    "No objectStore named " + storeName + " in this database",
                );
            }
        }

        const tx = new FDBTransaction(storeNames, mode, this);
        this._rawDatabase.transactions.push(tx);
        this._rawDatabase.processTransactions(); // See if can start right away (async)

        return tx;
    }

    public close() {
        closeConnection(this);
    }

    public toString() {
        return "[object IDBDatabase]";
    }
}

export default FDBDatabase;