import type { ReadableToken } from "atom.io"

import { newest } from "../lineage.ts"
import type { Store } from "../store/index.ts"
import { traceRootSelectorAtoms } from "./trace-selector-atoms.ts"

export function updateSelectorAtoms(
	store: Store,
	selectorType:
		| `readonly_held_selector`
		| `readonly_pure_selector`
		| `writable_held_selector`
		| `writable_pure_selector`,
	selectorKey: string,
	dependency: ReadableToken<unknown, any, unknown>,
	covered: Set<string>,
): void {
	const target = newest(store)
	const { type: dependencyType, key: dependencyKey } = dependency
	if (dependencyType === `atom` || dependencyType === `mutable_atom`) {
		target.selectorAtoms.set({
			selectorKey,
			atomKey: dependencyKey,
		})
		store.logger.info(
			`🔍`,
			selectorType,
			selectorKey,
			`discovers root atom "${dependencyKey}"`,
		)
	} else {
		const rootKeys = traceRootSelectorAtoms(store, dependencyKey, covered)
		store.logger.info(
			`🔍`,
			selectorType,
			selectorKey,
			`discovers root atoms: [ ${[...rootKeys.values()]
				.map((root) => `"${root.key}"`)
				.join(`, `)} ]`,
		)
		for (const { key: atomKey } of rootKeys.values()) {
			target.selectorAtoms = target.selectorAtoms.set({
				selectorKey,
				atomKey,
			})
		}
	}
	covered.add(dependencyKey)
}