import { accessWith } from "@kobalte/utils";
import { type Accessor, createMemo, createSignal, untrack } from "solid-js";

export interface CreateControllableSignalProps<T> {
	/** The value to be used, in controlled mode. */
	value?: Accessor<T | undefined>;

	/** The initial value to be used, in uncontrolled mode. */
	defaultValue?: Accessor<T | undefined>;

	/** A function that will be called when the value changes. */
	onChange?: (value: T) => void;
}

/**
 * Creates a simple reactive state with a getter and setter,
 * that can be controlled with `value` and `onChange` props.
 */
export function createControllableSignal<T>(
	props: CreateControllableSignalProps<T>,
) {
	// Internal uncontrolled value
	const [_value, _setValue] = createSignal(props.defaultValue?.());

	const isControlled = createMemo(() => props.value?.() !== undefined);

	const value = createMemo(() => (isControlled() ? props.value?.() : _value()));

	const setValue = (next: Exclude<T, Function> | ((prev: T) => T)) => {
		untrack(() => {
			const nextValue = accessWith(next, value() as T);

			if (!Object.is(nextValue, value())) {
				if (!isControlled()) {
					_setValue(nextValue as Exclude<T, Function>);
				}

				props.onChange?.(nextValue);
			}

			return nextValue;
		});
	};

	return [value, setValue] as const;
}

/**
 * Creates a simple reactive Boolean state with a getter, setter and a fallback value of `false`,
 * that can be controlled with `value` and `onChange` props.
 */
export function createControllableBooleanSignal(
	props: CreateControllableSignalProps<boolean>,
) {
	const [_value, setValue] = createControllableSignal(props);

	const value: Accessor<boolean> = () => _value() ?? false;

	return [value, setValue] as const;
}

/**
 * Creates a simple reactive Array state with a getter, setter and a fallback value of `[]`,
 * that can be controlled with `value` and `onChange` props.
 */
export function createControllableArraySignal<T>(
	props: CreateControllableSignalProps<Array<T>>,
) {
	const [_value, setValue] = createControllableSignal(props);

	const value: Accessor<Array<T>> = () => _value() ?? [];

	return [value, setValue] as const;
}

/**
 * Creates a simple reactive Set state with a getter, setter and a fallback value of `Set()`,
 * that can be controlled with `value` and `onChange` props.
 */
export function createControllableSetSignal<T>(
	props: CreateControllableSignalProps<Set<T>>,
) {
	const [_value, setValue] = createControllableSignal(props);

	const value: Accessor<Set<T>> = () => _value() ?? new Set();

	return [value, setValue] as const;
}