variant.ts

import {css} from "styled-components";

export function variant<R extends { variant?: keyof T },
  T extends {
    [variantName: string]: ReturnType<typeof css>;
  }>(variantMap: T) {
  return function ({variant: variantProp}: R) {
    return variantMap[variantProp];
  };
}