import os
import json
from uuid import uuid4

from seed_profile import SeedProfile, GrowthStage
from trait_normalizer import expressed_rgb_from_alleles

PROFILE_FILE = os.path.join(os.path.dirname(__file__), "starter_profiles.json")


class StarterSeedLoader:
    def __init__(self, catalog, name_generator):
        self.catalog = catalog
        self.name_generator = name_generator

    def load_starter_seeds(self):
        with open(PROFILE_FILE, "r") as f:
            starter_profiles = json.load(f)

        for profile in starter_profiles:
            alleles = profile["alleles"]
            leaf_rgb = expressed_rgb_from_alleles(alleles["LeafColor"])
            bud_rgb = expressed_rgb_from_alleles(alleles["BudColor"])
            base = profile["traits_base"]

            seed = SeedProfile(
                strain_name=profile["name"],
                type=profile["type"],
                alleles=alleles,
                stabilities=profile["stabilities"],
                thc=base["THC"],
                cbd=base["CBD"],
                yield_=base["Yield"],
                grow_time=base["GrowTime"],
                bud_color=bud_rgb,
                leaf_color=leaf_rgb,
                base_image_name=profile["base_image_name"],
                growth_stage=GrowthStage.SEED,
                lineage=(None, None),
                description=profile.get("description", ""),
                is_starter=True,
            )
            self.catalog.add_seed(seed)