redis-orm-node/src/internalStorage.ts

`redis-orm` is a lightweight Object-Relational Mapping (ORM) library for Node.js.

import { EntityDefinition, PropertyMetadata, PropertyOption } from "./types";
import { EntityOptions } from "./types";

class InternalStorage {
    entityMetadata = new WeakMap<object, EntityOptions>()
    propertiesMetadata: PropertyMetadata[] = []
    entityDefinition = new WeakMap<object, EntityDefinition>()

    pushEntityOptions(target: object, options: EntityOptions) {
        this.entityMetadata.set(target, options);
    }

    pushColumnOptions(object: object, propertyName: string, options: PropertyOption) {
        this.propertiesMetadata.push({
            object, propertyName, options
        });
    }

    getEntityWorkspace(target: object) {
        return {
            entityOptions: this.entityMetadata.get(target),
            propertyMetadata: this.propertiesMetadata.filter(property => target == property.object.constructor),
        }
    }

    pushEntityDefinition(object: object, definition: EntityDefinition) {
        this.entityDefinition.set(object, definition)
    }

    getEntityDefinition(object: object) {
        const EntityDefinition = this.entityDefinition.get(object)
        if (!EntityDefinition)
            throw new Error(`Entity definition not found`)
        return EntityDefinition
    }
}

export const internalStorage = new InternalStorage