import type { MembershipType } from "@/lib/validation/schemas";

export interface MembershipFeatures {
  recommendationsPerMonth: number;
  hydrationTracking: boolean;
  nutritionTracking: boolean;
  advancedStatistics: boolean;
}

export const MEMBERSHIP_FEATURES: Record<MembershipType, MembershipFeatures> = {
  basic: {
    recommendationsPerMonth: 1,
    hydrationTracking: false,
    nutritionTracking: false,
    advancedStatistics: false,
  },
  premium: {
    recommendationsPerMonth: -1,
    hydrationTracking: true,
    nutritionTracking: true,
    advancedStatistics: true,
  },
  vip: {
    recommendationsPerMonth: -1,
    hydrationTracking: true,
    nutritionTracking: true,
    advancedStatistics: true,
  },
};

export function getMembershipFeatures(
  membershipType: MembershipType,
): MembershipFeatures {
  return MEMBERSHIP_FEATURES[membershipType];
}