import { get, set } from '@wener/utils';
import { createStore, type StoreApi } from 'zustand';
import { immer } from 'zustand/middleware/immer';
import type { ModuleStore } from './types';

export class DynamicStore implements ModuleStore {
  constructor(
    readonly store: StoreApi<any> = createStore(
      immer(() => {
        return {};
      }),
    ),
  ) {}

  get value() {
    return this.store.getState();
  }

  add(type: string, payload: any) {
    this.store.setState((s: any) => {
      let last = get(s, type, []);
      if (!Array.isArray(last)) {
        last = [last];
      }
      Array.isArray(payload) ? last.push(...payload) : last.push(payload);
      set(s, type, last, false);
    });
  }

  collect(type: string): any {
    let found = get(this.value, type) ?? [];
    if (!Array.isArray(found)) {
      found = [found];
    }
    return found;
  }

  set(key: string, value: any, { merge }: { merge?: boolean } = {}) {
    this.store.setState((s: any) => {
      set(s, key, value, merge);
    });
  }

  get(key: string): any {
    return get(this.value, key);
  }

  as() {
    return this as any;
  }
}