import { Subject, BehaviorSubject } from "rxjs"
import { map } from "rxjs/operators"
import assign from "lodash/assign"
import clone from "lodash/clone"

export type Dispatchers<StoreType> = {
  [key: string]: (currentVal: StoreType, payload: any) => Partial<StoreType>
}

type Dispatch<
  StoreType,
  DispatchersType extends Dispatchers<StoreType>,
  K extends keyof DispatchersType
> = {
  dispatcher: K & string
  payload: any
}

export default class DispatchingStore<
  StoreType,
  DispatchersType extends Dispatchers<StoreType>
> {
  #state$: BehaviorSubject<StoreType>
  #dispatchers: Dispatchers<StoreType>
  #dispatches$: Subject<
    Dispatch<StoreType, DispatchersType, keyof DispatchersType>
  > = new Subject()

  constructor(initialValue: StoreType, dispatchers: DispatchersType) {
    this.#state$ = new BehaviorSubject(initialValue)
    this.#dispatchers = dispatchers

    this.#dispatches$
      .pipe(
        map(({ dispatcher, payload }) =>
          this.#dispatchers[dispatcher](this.value, payload)
        )
      )
      .subscribe((val) => {
        const data = clone(this.value)
        assign(data, val)

        this.#state$.next(data)
      })
  }

  get subject$() {
    return this.#state$
  }

  get value() {
    return this.subject$.value
  }

  get dispatches$() {
    return this.#dispatches$
  }

  dispatch({
    dispatcher,
    payload,
  }: Dispatch<StoreType, DispatchersType, keyof DispatchersType>) {
    if (!this.#dispatchers[dispatcher])
      throw new Error(`Undefined dispatch type '${dispatcher}'`)

    this.#dispatches$.next({ dispatcher, payload })
  }
}