from django.shortcuts import render, redirect
from django.contrib import messages
from django.contrib.auth import authenticate, login
from django.contrib.auth.decorators import login_required
from django.contrib.auth.forms import AuthenticationForm
from .models import Issue, IssueCategory
from .forms import CitizenRegistrationForm, IssueForm

def home(request):
    return render(request, 'core/index.html')


def home(request):
    if request.user.is_authenticated and request.user.is_citizen:
        return redirect('citizen_dashboard')
    return render(request, 'core/index.html')


def home(request):
    if request.user.is_authenticated and request.user.is_citizen:
        return redirect('citizen_dashboard')
    return render(request, 'core/index.html')

@login_required
def citizen_dashboard(request):
    if not request.user.is_citizen:
        messages.error(request, 'Access denied. Citizen role required.')
        return redirect('home')
    
    # Get only basic data for now
    all_user_issues = Issue.objects.filter(reporter=request.user)
    user_issues_display = all_user_issues.order_by('-created_at')[:5]
    resolved_count = all_user_issues.filter(status='resolved').count()
    categories = IssueCategory.objects.all()
    
    context = {
        'user_issues': user_issues_display, 
        'resolved_count': resolved_count,    
        'categories': categories,
        'issue_form': IssueForm(),
    }
    return render(request, 'core/citizen_dashboard.html', context)

@login_required
def report_issue(request):
    if not request.user.is_citizen:
        messages.error(request, 'Access denied. Citizen role required.')
        return redirect('home')
    
    if request.method == 'POST':
        form = IssueForm(request.POST, request.FILES)
        if form.is_valid():
            issue = form.save(commit=False)
            issue.reporter = request.user
            issue.save()
            messages.success(request, 'Issue reported successfully!')
            return redirect('citizen_dashboard')
        else:
            messages.error(request, 'Please correct the errors below.')
    else:
        form = IssueForm()
    
    return render(request, 'core/report_issue.html', {'form': form})

def register(request):
    if request.method == 'POST':
        form = CitizenRegistrationForm(request.POST)
        if form.is_valid():
            user = form.save()
            messages.success(request, 'Registration successful! Please login.')
            return redirect('login')
    else:
        form = CitizenRegistrationForm()
    
    return render(request, 'core/register.html', {'form': form})

def custom_login(request):
    if request.method == 'POST':
        form = AuthenticationForm(request, data=request.POST)
        if form.is_valid():
            username = form.cleaned_data.get('username')
            password = form.cleaned_data.get('password')
            user = authenticate(username=username, password=password)
            if user is not None:
                login(request, user)
                messages.success(request, f'Welcome back, {username}!')
                return redirect('home')
            else:
                messages.error(request, 'Invalid username or password.')
        else:
            messages.error(request, 'Invalid username or password.')
    else:
        form = AuthenticationForm()
    return render(request, 'core/login.html', {'form': form})