import { SetStoreFunction, createStore } from "solid-js/store";

import { Hydrators, hydrate } from "../hydration";

/**
 * Wrapper around Solid.js store
 */
export class ObjectStorage<T> {
  private store: Record<string, T>;
  readonly set: SetStoreFunction<Record<string, T>>;

  /**
   * Create new object storage
   */
  constructor() {
    const [store, setStore] = createStore({});
    this.store = store as never;
    this.set = setStore;
  }

  /**
   * Get object by ID
   * @param id ID
   * @returns Object
   */
  get(id: string) {
    return this.store[id];
  }

  /**
   * Hydrate some data into storage
   * @param id ID
   * @param type Hydration type
   * @param data Input Data
   */
  hydrate(id: string, type: keyof Hydrators, data?: unknown) {
    if (data) {
      this.set(id, hydrate(type, data as never) as T);
    }
  }
}