import {
	type ApiBalanceInput,
	apiBalanceToAllowed,
	type FeatureInput,
} from "../convert/apiBalanceToAllowed";

type CheckFeatureInput = FeatureInput & {
	id: string;
};

type CheckEntityInput = {
	balances?: Record<string, ApiBalanceInput | undefined>;
};

export const getFeatureToUseForCheck = ({
	creditSystems,
	feature,
	apiEntity,
	requiredBalance,
}: {
	creditSystems: CheckFeatureInput[];
	feature: CheckFeatureInput;
	apiEntity: CheckEntityInput;
	requiredBalance: number;
}) => {
	// 1. If there's a credit system & cusEnts for that credit system -> return credit system
	// 2. If there's cusEnts for the feature -> return feature
	// 3. Otherwise, feature to use is credit system if exists, otherwise return feature
	if (creditSystems.length === 0) return feature;

	const mainBalance = apiEntity?.balances?.[feature.id];

	if (
		mainBalance &&
		apiBalanceToAllowed({
			apiBalance: mainBalance,
			feature,
			requiredBalance,
		})
	) {
		return feature;
	}

	for (const creditSystem of creditSystems) {
		const apiBalance = apiEntity?.balances?.[creditSystem.id];
		if (!apiBalance) continue;

		if (
			apiBalanceToAllowed({
				apiBalance,
				feature: creditSystem,
				requiredBalance,
			})
		) {
			return creditSystem;
		}
	}

	return creditSystems[0];
};