import { NutrientKey, NutrientMeta } from '@/types/nutrition';

export const LEVEL1_NUTRITION_KEYS: NutrientKey[] = [
  'protein_g',
  'carbs_g',
  'fat_g',
  'fiber_g',
  'soluble_fiber_g',
  'insoluble_fiber_g',
  'a_mcg',
  'retinol_mcg',
  'b1_mg',
  'b2_mg',
  'b3_mg',
  'b5_mg',
  'b6_mg',
  'b9_mcg',
  'b12_mcg',
  'c_mg',
  'e_mg',
  'k_mcg',
  'calcium_mg',
  'iron_mg',
  'magnesium_mg',
  'zinc_mg',
  'potassium_mg',
  'phosphorus_mg',
  'selenium_mcg',
  'chromium_mcg',
  'molybdenum_mcg',
  'iodine_mcg',
  'omega_3_mg',
  'ala_mg',
  'epa_mg',
  'dha_mg',
  'biotin_mcg',
  'probiotics_cfu',
  'copper_mg',
  'manganese_mg',
  'choline_mg',
  'omega_6_mg',
  'lycopene_mcg',
  'lutein_zeaxanthin_mcg',
  'beta_carotene_mcg',
  'tryptophan_mg',
  'tyrosine_mg',
  'alpha_carotene_mcg',
  'cryptoxanthin_mcg',
  'anthocyanins_mg',
  'monounsaturated_fat_g',
  'polyunsaturated_fat_g',
  'phytosterols_mg',
  'dpa_mg',
  'valine_g',
  'glutamine_g',
  'betaine_mg',
];

export const LEVEL2_NUTRITION_KEYS: NutrientKey[] = [
  'vitamin_d_mcg',
  'collagen_g',
  'coq10_mg',
  'curcumin_mg',
  'creatine_g',
  'resveratrol_mg',
  'astaxanthin_mcg',
];

export const LEVEL3_NUTRITION_KEYS: NutrientKey[] = ['caffeine_mg', 'theobromine_mg', 'taurine_mg'];

export const LEVEL4_NUTRITION_KEYS: NutrientKey[] = [
  'kcal',
  'alcohol_g',
  'cholesterol_mg',
  'saturated_fat_g',
  'sodium_mg',
  'sugar_g',
  'starch_g',
  'oxalate_mg',
  'chloride_mg',
];

export const USA_GOV_DGA_KEYS: NutrientKey[] = [
  'usa_gov_dga_protein_g',
  'usa_gov_dga_healthy_fats_tsp_eq',
  'usa_gov_dga_vegetables_cup_eq',
  'usa_gov_dga_fruits_cup_eq',
  'usa_gov_dga_dairy_cup_eq',
  'usa_gov_dga_whole_grains_oz_eq',
  'usa_gov_dga_dark_green_veg_cup_eq',
  'usa_gov_dga_red_orange_veg_cup_eq',
  'usa_gov_dga_beans_peas_lentils_veg_cup_eq',
  'usa_gov_dga_starchy_veg_cup_eq',
  'usa_gov_dga_other_veg_cup_eq',
];

export const ALL_NUTRITION_KEYS: NutrientKey[] = [
  ...LEVEL1_NUTRITION_KEYS,
  ...USA_GOV_DGA_KEYS,
  ...LEVEL2_NUTRITION_KEYS,
  ...LEVEL3_NUTRITION_KEYS,
  ...LEVEL4_NUTRITION_KEYS,
];

export const NUTRIENT_META: Record<NutrientKey, NutrientMeta> = {
  // Level 1: must (64)
  protein_g: {
    key: 'protein_g',
    label: 'Protein',
    symbol: 'Pro',
    unit: 'g',
    importance: 1,
    level: 'must',
  },
  carbs_g: {
    key: 'carbs_g',
    label: 'Carbs',
    symbol: 'Carb',
    unit: 'g',
    importance: 2,
    level: 'must',
  },
  fat_g: { key: 'fat_g', label: 'Fat', symbol: 'Fat', unit: 'g', importance: 3, level: 'must' },
  fiber_g: {
    key: 'fiber_g',
    label: 'Fiber',
    symbol: 'Fib',
    unit: 'g',
    importance: 4,
    level: 'must',
  },
  soluble_fiber_g: {
    key: 'soluble_fiber_g',
    label: 'Soluble Fiber',
    symbol: 'SFi',
    unit: 'g',
    importance: 50,
    level: 'must',
  },
  insoluble_fiber_g: {
    key: 'insoluble_fiber_g',
    label: 'Insoluble Fiber',
    symbol: 'IFi',
    unit: 'g',
    importance: 51,
    level: 'must',
  },
  a_mcg: {
    key: 'a_mcg',
    label: 'Vitamin A',
    symbol: 'Vit A',
    unit: 'mcg',
    importance: 20,
    level: 'must',
  },
  retinol_mcg: {
    key: 'retinol_mcg',
    label: 'Retinol',
    symbol: 'Ret',
    unit: 'mcg',
    importance: 52,
    level: 'must',
  },
  b1_mg: {
    key: 'b1_mg',
    label: 'Vitamin B1',
    symbol: 'Vit B1',
    unit: 'mg',
    importance: 25,
    level: 'must',
  },
  b2_mg: {
    key: 'b2_mg',
    label: 'Vitamin B2',
    symbol: 'Vit B2',
    unit: 'mg',
    importance: 26,
    level: 'must',
  },
  b3_mg: {
    key: 'b3_mg',
    label: 'Vitamin B3',
    symbol: 'Vit B3',
    unit: 'mg',
    importance: 27,
    level: 'must',
  },
  b5_mg: {
    key: 'b5_mg',
    label: 'Vitamin B5',
    symbol: 'Vit B5',
    unit: 'mg',
    importance: 28,
    level: 'must',
  },
  b6_mg: {
    key: 'b6_mg',
    label: 'Vitamin B6',
    symbol: 'Vit B6',
    unit: 'mg',
    importance: 29,
    level: 'must',
  },
  b9_mcg: {
    key: 'b9_mcg',
    label: 'Folate',
    symbol: 'Vit B9',
    unit: 'mcg',
    importance: 30,
    level: 'must',
  },
  b12_mcg: {
    key: 'b12_mcg',
    label: 'Vitamin B12',
    symbol: 'Vit B12',
    unit: 'mcg',
    importance: 8,
    level: 'must',
  },
  c_mg: {
    key: 'c_mg',
    label: 'Vitamin C',
    symbol: 'Vit C',
    unit: 'mg',
    importance: 21,
    level: 'must',
  },
  e_mg: {
    key: 'e_mg',
    label: 'Vitamin E',
    symbol: 'Vit E',
    unit: 'mg',
    importance: 22,
    level: 'must',
  },
  k_mcg: {
    key: 'k_mcg',
    label: 'Vitamin K',
    symbol: 'Vit K',
    unit: 'mcg',
    importance: 23,
    level: 'must',
  },
  calcium_mg: {
    key: 'calcium_mg',
    label: 'Calcium',
    symbol: 'Ca',
    unit: 'mg',
    importance: 10,
    level: 'must',
  },
  iron_mg: {
    key: 'iron_mg',
    label: 'Iron',
    symbol: 'Fe',
    unit: 'mg',
    importance: 9,
    level: 'must',
  },
  magnesium_mg: {
    key: 'magnesium_mg',
    label: 'Magnesium',
    symbol: 'Mg',
    unit: 'mg',
    importance: 11,
    level: 'must',
  },
  zinc_mg: {
    key: 'zinc_mg',
    label: 'Zinc',
    symbol: 'Zn',
    unit: 'mg',
    importance: 12,
    level: 'must',
  },
  potassium_mg: {
    key: 'potassium_mg',
    label: 'Potassium',
    symbol: 'K',
    unit: 'mg',
    importance: 13,
    level: 'must',
  },
  phosphorus_mg: {
    key: 'phosphorus_mg',
    label: 'Phosphorus',
    symbol: 'P',
    unit: 'mg',
    importance: 35,
    level: 'must',
  },
  selenium_mcg: {
    key: 'selenium_mcg',
    label: 'Selenium',
    symbol: 'Se',
    unit: 'mcg',
    importance: 36,
    level: 'must',
  },
  chromium_mcg: {
    key: 'chromium_mcg',
    label: 'Chromium',
    symbol: 'Cr',
    unit: 'mcg',
    importance: 60,
    level: 'must',
  },
  molybdenum_mcg: {
    key: 'molybdenum_mcg',
    label: 'Molybdenum',
    symbol: 'Mo',
    unit: 'mcg',
    importance: 61,
    level: 'must',
  },
  iodine_mcg: {
    key: 'iodine_mcg',
    label: 'Iodine',
    symbol: 'I',
    unit: 'mcg',
    importance: 37,
    level: 'must',
  },
  omega_3_mg: {
    key: 'omega_3_mg',
    label: 'Omega-3',
    symbol: 'ω3',
    unit: 'mg',
    importance: 7,
    level: 'must',
  },
  ala_mg: { key: 'ala_mg', label: 'ALA', symbol: 'ALA', unit: 'mg', importance: 53, level: 'must' },
  epa_mg: { key: 'epa_mg', label: 'EPA', symbol: 'EPA', unit: 'mg', importance: 54, level: 'must' },
  dha_mg: { key: 'dha_mg', label: 'DHA', symbol: 'DHA', unit: 'mg', importance: 55, level: 'must' },
  biotin_mcg: {
    key: 'biotin_mcg',
    label: 'Biotin',
    symbol: 'B7',
    unit: 'mcg',
    importance: 38,
    level: 'must',
  },
  probiotics_cfu: {
    key: 'probiotics_cfu',
    label: 'Probiotics',
    symbol: 'Prb',
    unit: 'CFU',
    importance: 71,
    level: 'must',
  },
  copper_mg: {
    key: 'copper_mg',
    label: 'Copper',
    symbol: 'Cu',
    unit: 'mg',
    importance: 39,
    level: 'must',
  },
  manganese_mg: {
    key: 'manganese_mg',
    label: 'Manganese',
    symbol: 'Mn',
    unit: 'mg',
    importance: 40,
    level: 'must',
  },
  choline_mg: {
    key: 'choline_mg',
    label: 'Choline',
    symbol: 'Cho',
    unit: 'mg',
    importance: 14,
    level: 'must',
  },
  usa_gov_dga_protein_g: {
    key: 'usa_gov_dga_protein_g',
    label: 'Protein',
    symbol: 'Pro',
    unit: 'g',
    importance: 1,
    level: 'must',
  },
  usa_gov_dga_healthy_fats_tsp_eq: {
    key: 'usa_gov_dga_healthy_fats_tsp_eq',
    label: 'Healthy Fats',
    symbol: 'H Fat',
    unit: 'tsp',
    importance: 2,
    level: 'must',
  },
  usa_gov_dga_vegetables_cup_eq: {
    key: 'usa_gov_dga_vegetables_cup_eq',
    label: 'Vegetables',
    symbol: 'Veg',
    unit: 'cup',
    importance: 3,
    level: 'must',
  },
  usa_gov_dga_fruits_cup_eq: {
    key: 'usa_gov_dga_fruits_cup_eq',
    label: 'Fruits',
    symbol: 'Fru',
    unit: 'cup',
    importance: 4,
    level: 'must',
  },
  usa_gov_dga_dairy_cup_eq: {
    key: 'usa_gov_dga_dairy_cup_eq',
    label: 'Dairy',
    symbol: 'Dai',
    unit: 'cup',
    importance: 5,
    level: 'must',
  },
  usa_gov_dga_whole_grains_oz_eq: {
    key: 'usa_gov_dga_whole_grains_oz_eq',
    label: 'Whole Grains',
    symbol: 'Grn',
    unit: 'oz',
    importance: 6,
    level: 'must',
  },
  usa_gov_dga_dark_green_veg_cup_eq: {
    key: 'usa_gov_dga_dark_green_veg_cup_eq',
    label: 'Dark Green Veg',
    symbol: 'DGV',
    unit: 'cup',
    importance: 7,
    level: 'must',
  },
  usa_gov_dga_red_orange_veg_cup_eq: {
    key: 'usa_gov_dga_red_orange_veg_cup_eq',
    label: 'Red/Orange Veg',
    symbol: 'ROV',
    unit: 'cup',
    importance: 8,
    level: 'must',
  },
  usa_gov_dga_beans_peas_lentils_veg_cup_eq: {
    key: 'usa_gov_dga_beans_peas_lentils_veg_cup_eq',
    label: 'Legumes',
    symbol: 'Leg',
    unit: 'cup',
    importance: 9,
    level: 'must',
  },
  usa_gov_dga_starchy_veg_cup_eq: {
    key: 'usa_gov_dga_starchy_veg_cup_eq',
    label: 'Starchy Veg',
    symbol: 'StV',
    unit: 'cup',
    importance: 10,
    level: 'must',
  },
  usa_gov_dga_other_veg_cup_eq: {
    key: 'usa_gov_dga_other_veg_cup_eq',
    label: 'Other Veg',
    symbol: 'OtV',
    unit: 'cup',
    importance: 11,
    level: 'must',
  },
  omega_6_mg: {
    key: 'omega_6_mg',
    label: 'Omega-6',
    symbol: 'ω6',
    unit: 'mg',
    importance: 56,
    level: 'must',
  },
  lycopene_mcg: {
    key: 'lycopene_mcg',
    label: 'Lycopene',
    symbol: 'Lyc',
    unit: 'mcg',
    importance: 63,
    level: 'must',
  },
  lutein_zeaxanthin_mcg: {
    key: 'lutein_zeaxanthin_mcg',
    label: 'Lutein',
    symbol: 'Lut',
    unit: 'mcg',
    importance: 64,
    level: 'must',
  },
  beta_carotene_mcg: {
    key: 'beta_carotene_mcg',
    label: 'Beta Carotene',
    symbol: 'βCa',
    unit: 'mcg',
    importance: 65,
    level: 'must',
  },
  tryptophan_mg: {
    key: 'tryptophan_mg',
    label: 'Tryptophan',
    symbol: 'Trp',
    unit: 'mg',
    importance: 73,
    level: 'must',
  },
  tyrosine_mg: {
    key: 'tyrosine_mg',
    label: 'Tyrosine',
    symbol: 'Tyr',
    unit: 'mg',
    importance: 74,
    level: 'must',
  },
  alpha_carotene_mcg: {
    key: 'alpha_carotene_mcg',
    label: 'Alpha Carotene',
    symbol: 'αCa',
    unit: 'mcg',
    importance: 66,
    level: 'must',
  },
  cryptoxanthin_mcg: {
    key: 'cryptoxanthin_mcg',
    label: 'Cryptoxanthin',
    symbol: 'Cry',
    unit: 'mcg',
    importance: 67,
    level: 'must',
  },
  anthocyanins_mg: {
    key: 'anthocyanins_mg',
    label: 'Anthocyanins',
    symbol: 'Ant',
    unit: 'mg',
    importance: 83,
    level: 'must',
  },
  monounsaturated_fat_g: {
    key: 'monounsaturated_fat_g',
    label: 'Mono Fat',
    symbol: 'MFa',
    unit: 'g',
    importance: 58,
    level: 'must',
  },
  polyunsaturated_fat_g: {
    key: 'polyunsaturated_fat_g',
    label: 'Poly Fat',
    symbol: 'PFa',
    unit: 'g',
    importance: 59,
    level: 'must',
  },
  phytosterols_mg: {
    key: 'phytosterols_mg',
    label: 'Phytosterols',
    symbol: 'Phy',
    unit: 'mg',
    importance: 86,
    level: 'must',
  },
  dpa_mg: { key: 'dpa_mg', label: 'DPA', symbol: 'DPA', unit: 'mg', importance: 69, level: 'must' },
  valine_g: {
    key: 'valine_g',
    label: 'Valine',
    symbol: 'Val',
    unit: 'g',
    importance: 78,
    level: 'must',
  },
  glutamine_g: {
    key: 'glutamine_g',
    label: 'Glutamine',
    symbol: 'Gln',
    unit: 'g',
    importance: 79,
    level: 'must',
  },
  betaine_mg: {
    key: 'betaine_mg',
    label: 'Betaine',
    symbol: 'Bet',
    unit: 'mg',
    importance: 88,
    level: 'must',
  },
  // Level 2: cannot_as_food (7)
  vitamin_d_mcg: {
    key: 'vitamin_d_mcg',
    label: 'Vitamin D',
    symbol: 'Vit D',
    unit: 'mcg',
    importance: 6,
    level: 'cannot_as_food',
  },
  collagen_g: {
    key: 'collagen_g',
    label: 'Collagen',
    symbol: 'Col',
    unit: 'g',
    importance: 70,
    level: 'cannot_as_food',
  },
  coq10_mg: {
    key: 'coq10_mg',
    label: 'CoQ10',
    symbol: 'CoQ10',
    unit: 'mg',
    importance: 42,
    level: 'cannot_as_food',
  },
  curcumin_mg: {
    key: 'curcumin_mg',
    label: 'Curcumin',
    symbol: 'Cur',
    unit: 'mg',
    importance: 85,
    level: 'cannot_as_food',
  },
  creatine_g: {
    key: 'creatine_g',
    label: 'Creatine',
    symbol: 'Cre',
    unit: 'g',
    importance: 75,
    level: 'cannot_as_food',
  },
  resveratrol_mg: {
    key: 'resveratrol_mg',
    label: 'Resveratrol',
    symbol: 'Res',
    unit: 'mg',
    importance: 84,
    level: 'cannot_as_food',
  },
  astaxanthin_mcg: {
    key: 'astaxanthin_mcg',
    label: 'Astaxanthin',
    symbol: 'Ast',
    unit: 'mcg',
    importance: 68,
    level: 'cannot_as_food',
  },
  // Level 3: bonus (4)
  water_ml: {
    key: 'water_ml',
    label: 'Water',
    symbol: 'H₂O',
    unit: 'ml',
    importance: 72,
    level: 'bonus',
  },
  caffeine_mg: {
    key: 'caffeine_mg',
    label: 'Caffeine',
    symbol: 'Caf',
    unit: 'mg',
    importance: 80,
    level: 'bonus',
  },
  theobromine_mg: {
    key: 'theobromine_mg',
    label: 'Theobromine',
    symbol: 'Thb',
    unit: 'mg',
    importance: 82,
    level: 'bonus',
  },
  taurine_mg: {
    key: 'taurine_mg',
    label: 'Taurine',
    symbol: 'Tau',
    unit: 'mg',
    importance: 87,
    level: 'bonus',
  },
  // Level 4: hidden (9)
  kcal: {
    key: 'kcal',
    label: 'Calories',
    symbol: 'Cal',
    unit: 'kcal',
    importance: 1,
    level: 'hidden',
  },
  alcohol_g: {
    key: 'alcohol_g',
    label: 'Alcohol',
    symbol: 'Alc',
    unit: 'g',
    importance: 81,
    level: 'hidden',
  },
  cholesterol_mg: {
    key: 'cholesterol_mg',
    label: 'Cholesterol',
    symbol: 'Chl',
    unit: 'mg',
    importance: 41,
    level: 'hidden',
  },
  saturated_fat_g: {
    key: 'saturated_fat_g',
    label: 'Saturated Fat',
    symbol: 'SFa',
    unit: 'g',
    importance: 18,
    level: 'hidden',
  },
  sodium_mg: {
    key: 'sodium_mg',
    label: 'Sodium',
    symbol: 'Na',
    unit: 'mg',
    importance: 15,
    level: 'hidden',
  },
  sugar_g: {
    key: 'sugar_g',
    label: 'Sugar',
    symbol: 'Sug',
    unit: 'g',
    importance: 16,
    level: 'hidden',
  },
  starch_g: {
    key: 'starch_g',
    label: 'Starch',
    symbol: 'Sta',
    unit: 'g',
    importance: 57,
    level: 'hidden',
  },
  oxalate_mg: {
    key: 'oxalate_mg',
    label: 'Oxalate',
    symbol: 'Oxa',
    unit: 'mg',
    importance: 89,
    level: 'hidden',
  },
  chloride_mg: {
    key: 'chloride_mg',
    label: 'Chloride',
    symbol: 'Cl',
    unit: 'mg',
    importance: 62,
    level: 'hidden',
  },
};

export const BASE_DAILY_TARGETS: Record<string, number> = {
  protein_g: 50,
  carbs_g: 275,
  fat_g: 78,
  fiber_g: 28,
  soluble_fiber_g: 10,
  insoluble_fiber_g: 18,
  a_mcg: 900,
  retinol_mcg: 700,
  b1_mg: 1.2,
  b2_mg: 1.3,
  b3_mg: 16,
  b5_mg: 5,
  b6_mg: 1.7,
  b9_mcg: 400,
  b12_mcg: 2.4,
  c_mg: 90,
  e_mg: 15,
  k_mcg: 120,
  vitamin_d_mcg: 20,
  biotin_mcg: 30,
  choline_mg: 550,
  calcium_mg: 1000,
  iron_mg: 18,
  magnesium_mg: 420,
  zinc_mg: 11,
  potassium_mg: 4700,
  phosphorus_mg: 700,
  selenium_mcg: 55,
  chromium_mcg: 35,
  molybdenum_mcg: 45,
  iodine_mcg: 150,
  copper_mg: 0.9,
  manganese_mg: 2.3,
  chloride_mg: 2300,
  omega_3_mg: 1600,
  ala_mg: 1100,
  epa_mg: 250,
  dha_mg: 250,
  dpa_mg: 100,
  omega_6_mg: 11000,
  monounsaturated_fat_g: 20,
  polyunsaturated_fat_g: 17,
  tryptophan_mg: 280,
  tyrosine_mg: 1000,
  valine_g: 2.4,
  glutamine_g: 5,
  lycopene_mcg: 10000,
  lutein_zeaxanthin_mcg: 6000,
  beta_carotene_mcg: 6000,
  alpha_carotene_mcg: 700,
  cryptoxanthin_mcg: 400,
  anthocyanins_mg: 50,
  phytosterols_mg: 2000,
  betaine_mg: 500,
  probiotics_cfu: 1000000000,
  collagen_g: 10,
  coq10_mg: 100,
  curcumin_mg: 500,
  creatine_g: 3,
  resveratrol_mg: 250,
  astaxanthin_mcg: 4000,
  water_ml: 3700,
  caffeine_mg: 400,
  theobromine_mg: 250,
  taurine_mg: 1000,
  kcal: 2000,
  alcohol_g: 14,
  cholesterol_mg: 300,
  saturated_fat_g: 20,
  sodium_mg: 2300,
  sugar_g: 50,
  starch_g: 150,
  oxalate_mg: 200,
  usa_gov_dga_protein_g: 50,
  usa_gov_dga_healthy_fats_tsp_eq: 6,
  usa_gov_dga_vegetables_cup_eq: 2.5,
  usa_gov_dga_fruits_cup_eq: 2,
  usa_gov_dga_dairy_cup_eq: 3,
  usa_gov_dga_whole_grains_oz_eq: 3,
  usa_gov_dga_dark_green_veg_cup_eq: 0.3,
  usa_gov_dga_red_orange_veg_cup_eq: 0.8,
  usa_gov_dga_beans_peas_lentils_veg_cup_eq: 0.3,
  usa_gov_dga_starchy_veg_cup_eq: 0.7,
  usa_gov_dga_other_veg_cup_eq: 0.6,
};