from django.shortcuts import render, get_object_or_404
from .models import Recipe, Version
from .forms import VersionForm, IngredientFormSet
from django.contrib.auth.decorators import login_required, permission_required
from django.urls import reverse
from django.http.response import HttpResponseRedirect
from django.conf import settings

@login_required
@permission_required('recipes.view_recipe')
def recipes(request):
	recipes = Recipe.objects.all()
	return render(request, 'recipes.html', context={'recipes': recipes})

@login_required
@permission_required('recipes.view_recipe')
def version(request, slug_recipe, slug_version):
	recipe = get_object_or_404(Recipe, slug=slug_recipe)
	has_multiple_versions = recipe.versions.all().count() > 1 # type: ignore
	version = get_object_or_404(Version, recipe=recipe, slug=slug_version)
	ingredients = version.ingredients.all() # type: ignore

	return render(request, 'version.html', context={'recipe': recipe, 'has_multiple_versions': has_multiple_versions, 'version': version, 'ingredients': ingredients})

@login_required
@permission_required('recipes.view_recipe')
def recipe(request, slug):
	recipe = get_object_or_404(Recipe, slug=slug)
	versions = recipe.versions.all() # type: ignore

	if (1 == versions.count()):
		if settings.ENFORCE_RECIPE_VERSION_URL:
			return HttpResponseRedirect(reverse('version', kwargs={'slug_recipe': slug, 'slug_version': versions.get().slug}))
		else:
			return version(request, slug, versions.get().slug)
	else:
		return render(request, 'recipe.html', context={'recipe': recipe, 'versions': versions})

VERSION_FORM_PREFIX = 'version'
INGREDIENTS_FORMSET_PREFIX = 'ingredient'

@login_required
@permission_required('recipes.add_recipe')
def add_version(request, slug_recipe):
	recipe = Recipe.objects.get(slug=slug_recipe)

	if request.method == 'POST':
		version_form = VersionForm(request.POST, prefix=VERSION_FORM_PREFIX)
		ingredients_formset = IngredientFormSet(request.POST, prefix=INGREDIENTS_FORMSET_PREFIX)
		version_form.recipe_id = recipe.id # type: ignore

		if version_form.is_valid() and ingredients_formset.is_valid():
			version = version_form.save(commit=False)
			if version_form.cleaned_data['use_user_for_author']:
				version.author_user = request.user
				version.author_name = ''
			elif 'author_name' in version_form.cleaned_data:
				version.author_name = version_form.cleaned_data['author_name']
			version.recipe = recipe
			version.save()

			for ingredient_form in ingredients_formset:
				if 'text' in ingredient_form.cleaned_data and ingredient_form.cleaned_data['text']:
					ingredient = ingredient_form.save(commit=False)
					ingredient.version = version
					ingredient.save()

			return HttpResponseRedirect(reverse('version', kwargs={'slug_recipe': version.recipe.slug, 'slug_version': version.slug}))

	else:
		if (recipe.versions.all().count() > 0): # type: ignore
			initial = {'label': '', 'slug': ''}
		else:
			initial = {}
		version_form = VersionForm(prefix=VERSION_FORM_PREFIX, initial=initial)
		ingredients_formset = IngredientFormSet(prefix=INGREDIENTS_FORMSET_PREFIX)

	return render(request, 'add-version.html', {'version_form': version_form, 'ingredients_formset': ingredients_formset})