from django import forms
from django.shortcuts import render, redirect
from notes.models import Note, Attachment, Tag, Version

class NoteForm(forms.Form):
    title = forms.CharField()
    content = forms.CharField(widget=forms.Textarea(attrs={'rows': 20, 
        'style': 'width:100%;height:100%'}))
    shared = forms.BooleanField()
    attachment = forms.FileField(required=False)

def edit(request, note_id=None):

    note = Note(user_id=1)

    if note_id:
        note = Note.objects.get(pk=note_id)

    if request.method == 'POST':
        form = NoteForm(request.POST, request.FILES)

        if not form.is_valid():
            return render(request, 'edit.html', {'form': form})

        note.title = form.cleaned_data.get('title')
        note.save()
        
        version = Version(note=note, user_id=1)
        version.content = form.cleaned_data.get('content')
        version.shared = form.cleaned_data.get('shared')
        version.save()

        return render(request, 'view.html', {'note': note, 'version': version})

    form = NoteForm(initial={'content': note.content, 'shared': note.shared})

    return render(request, 'edit.html', {'form': form})

def index(request, tag_id=None):
    notes = Note.objects.filter(user_id=1)

    if tag_id:
        notes = notes.filter(tags__pk=tag_id)

    tags = Tag.objects.distinct()
    return render(request, 'index.html', {'notes': notes, 'tags': tags})

def view(request, note_id):
    note = Note.objects.get(pk=note_id)
    version = note.version_set.latest()
    return render(request, 'view.html', {'note': note, 'version': version})