from .recipe import V60, ESPRESSO, Recipe

class RecipeManager:
    def __init__(self):
        self.recipes = [
            V60,
            ESPRESSO
        ]
        self.tmp_recipe = None
    
    def add_recipe(self, recipe):
        self.recipes.append(recipe)
    
    def update_recipe(self, recipe_id, new_recipe):
        self.recipes[recipe_id] = new_recipe

    def remove_recipe(self, recipe):
        self.recipes.remove(recipe)
    
    def get_recipe_id(self, recipe):
        return self.recipes.index(recipe)
    
    def get_recipe(self, recipe_id) -> Recipe:
        if recipe_id == None:
            return self.tmp_recipe
        return self.recipes[recipe_id]