import type { JoinToken } from "atom.io"
import type { Junction } from "atom.io/foundations/junction"

import { newest } from "../lineage.ts"
import type { Store } from "../store/index.ts"
import { isChildStore } from "../transaction/index.ts"
import { getJoin } from "./get-join.ts"

export function editRelationsInStore<
	AName extends string,
	A extends string,
	BName extends string,
	B extends string,
	Cardinality extends `1:1` | `1:n` | `n:n`,
>(
	store: Store,
	token: JoinToken<AName, A, BName, B, Cardinality>,
	change: (relations: Junction<AName, A, BName, B>) => void,
): void {
	const myJoin = getJoin(store, token)
	const target = newest(store)
	if (isChildStore(target)) {
		const { toolkit } = target.transactionMeta
		myJoin.transact(toolkit, ({ relations }) => {
			change(relations)
		})
	} else {
		change(myJoin.relations)
	}
}