import {useOnce} from "./use-once.js"
import {hooks} from "./plumbing/hooks.js"

export function useMount(fn: () => () => void) {
	const {scope} = hooks.increment()
	return useOnce(() => scope.mounts.mount(fn))
}

export class Mounts {
	#mounters: (() => () => void)[] = []
	#unmounters: (() => void)[] = []

	mount(mount: () => () => void) {
		this.#mounters.push(mount)
		this.#unmounters.push(mount())
	}

	unmountAll() {
		for (const unmount of this.#unmounters)
			unmount()
		this.#unmounters = []
	}

	remountAll() {
		for (const mount of this.#mounters)
			this.#unmounters.push(mount())
	}
}