import { invariant } from '@roenlie/mimic-core/validation';
import type { PropertyValues } from 'lit';

import { getCurrentRef } from '../core/component.js';


type UseUpdate = (
	func: (changedProps: PropertyValues) => void,
	deps?: string[],
) => void;


export const useUpdate = ((
	func: (changedProps: PropertyValues) => void,
	deps?: string[],
) => {
	const cls = getCurrentRef();
	invariant(cls, 'Could not get component instance.');

	cls.__updateHooks.push((props) => {
		if (deps?.some(dep => props.has(dep)) ?? true)
			func(props);
	});
}) satisfies UseUpdate;