import {
  ResolvableModeLevelAlias,
  ResolvableTopLevelAlias,
  ResolvableValueLevelAlias,
  SpecifyDesignTokenTypeName,
} from '@specifyapp/specify-design-token-format';

export type AllResolvableAlias<
  Type extends SpecifyDesignTokenTypeName = SpecifyDesignTokenTypeName,
> =
  | ResolvableValueLevelAlias<Type>
  | ResolvableModeLevelAlias<Type>
  | ResolvableTopLevelAlias<Type>;

export type ResolvableAliasStrategy<Return, Composites extends SpecifyDesignTokenTypeName> = <
  Alias extends AllResolvableAlias = AllResolvableAlias,
>(
  alias: Alias,
) => Alias extends ResolvableTopLevelAlias
  ? {
      [mode: string]: Alias extends AllResolvableAlias<Composites>
        ? Record<string, Return>
        : Return;
    }
  : Alias extends AllResolvableAlias<Composites>
  ? Record<string, Return>
  : Return;