import type { ReadableFamilyToken, ReadableToken, ViewOf } from "atom.io"
import type { Canonical } from "atom.io/foundations/canonical"

import { type Store, withdraw } from "../store/index.ts"
import { ensureState } from "./ensure-state.ts"
import { getFallback } from "./get-fallback.ts"
import { readOrComputeValue } from "./read-or-compute-value.ts"

export function getFromStore<T, E = never>(
	store: Store,
	token: ReadableToken<T, any, E>,
): ViewOf<E | T>

export function getFromStore<T, K extends Canonical, E = never>(
	store: Store,
	token: ReadableFamilyToken<T, K, E>,
	key: NoInfer<K>,
): ViewOf<E | T>

export function getFromStore<T, K extends Canonical, E = never>(
	store: Store,
	...params:
		| [token: ReadableFamilyToken<T, K, E>, key: NoInfer<K>]
		| [token: ReadableToken<T, any, E>]
): ViewOf<E | T>

export function getFromStore<T, K extends Canonical, E>(
	store: Store,
	...params:
		| [token: ReadableFamilyToken<T, K, E>, key: NoInfer<K>]
		| [token: ReadableToken<T, any, E>]
): ViewOf<E | T> {
	const { token, family, subKey } = ensureState(store, ...params)

	if (`counterfeit` in token && family && subKey) {
		return getFallback(store, token, family, subKey)
	}
	const state = withdraw(store, token)

	return readOrComputeValue(store, state)
}