Titivillus/quest/views.py

#!/usr/bin/env python3
"""
Quest and quest accessory views.
"""
from datetime import timedelta, datetime, timezone

import bleach
from user_messages import api
from django.views.decorators.http import require_POST
from django.contrib import messages
from django.shortcuts import render, redirect
from django.db.models import F
from django.conf import settings
from django.urls import reverse

from .models import Quest, DiceRoll, PollOption, PollVote, Page, Post
from .forms import EditQuestForm, QuestForm, PostForm
from user.models import User
from .tools import download_img, send_to_websocket

def index(request):
	"""The quest page index."""
	search_terms = request.GET.get('search_terms')
	search_type = request.GET.get('search_type')
	tags = request.GET.get('tags')
	order_by = request.GET.get('order_by')
	order_way = request.GET.get('order_way', 'desc')

	order_keys = {
		'ceation_date': 'id',
		'last_post': 'latest_post_date',
		'next_live': 'live_time',
	}
	order_by = order_keys.get(order_by, 'id')

	results = Quest.objects.all()
	if search_terms:
		if search_type == 'title':
			results = results.filter(title__unaccent__icontains=search_terms)
		elif search_type == 'author':
			results = results.filter(
				owner__username__unaccent__icontains=search_terms)
	if tags:
		results = results.filter(tags__name__in=tags.split()).distinct()

	if order_way == 'asc':
		results = results.order_by(F(order_by).asc(nulls_last=True))
	elif order_way == 'desc':
		results = results.order_by(F(order_by).desc(nulls_last=True))

	context = locals()
	return render(request, 'quest/index.html', context)


def quest(request, quest_id, page_num='0'):
	"""
	Arbituary quest page view.
	"""
	# TODO: 404 quest not found
	quest = Quest.objects.get(id=quest_id)
	pages = Page.objects.filter(
		quest=quest, appendix=False).order_by('page_num')
	appendices = Page.objects.filter(
		quest=quest, appendix=True).order_by('title')
	chat_messages = reversed(quest.message_set.all().order_by('-id')[:50])
	try:
		page = Page.objects.get(quest=quest, page_num=page_num)
	except Page.DoesNotExist:
		messages.error(request, "Page not found, redirecting you.")
		return redirect('quest:quest', quest_id=quest.id, page_num='0')
	posts = quest.post_set.filter(page=page)
	# TODO: filter by page_num as well
	dice_rolls = DiceRoll.objects.filter(dicecall__post__quest=quest)
	poll_options = PollOption.objects.filter(poll__post__quest=quest)
	poll_votes = PollVote.objects.filter(option__poll__post__quest=quest)
	ip_address = request.META['REMOTE_ADDR']
	server_user_id = settings.SERVER_USER_ID
	context = locals()
	if page_num == '0':
		return render(request, 'quest/quest_homepage.html', context)
	else:
		return render(request, 'quest/quest.html', context)


def edit_quest(request, quest_id, page_num='0'):
	"""
	Edit quest page. Only available to the QM.
	"""
	quest = Quest.objects.get(id=quest_id)
	prev_live = quest.live
	if quest.owner != request.user:
		return redirect('quest:quest', quest_id=quest_id, page_num=page_num)
	if request.method == 'POST':
		form = EditQuestForm(request.POST)
		if form.is_valid():
			quest.title = form.cleaned_data['quest_title']
			quest.anon_name = form.cleaned_data['anon_name']
			quest.live = form.cleaned_data['live']
			quest.description = form.cleaned_data['description']
			quest.banner_url = download_img(form.cleaned_data['banner_url'])
			if not quest.banner_url.startswith('http'):
				quest.banner_url = None

			live_date = form.cleaned_data['live_date']
			live_time = form.cleaned_data['live_time']
			if live_date and live_time:
				live_datetime = datetime.combine(
					live_date,
					live_time,
					timezone.utc
				)
				tz_delta = timedelta(minutes=form.cleaned_data['timezone'])
				live_datetime = live_datetime + tz_delta
				quest.live_time = live_datetime
			else:
				quest.live_time = None
			quest.save()
			data = {
				'live': quest.live,
				'live_time': quest.live_time,
			}
			if data['live_time']:
				data['live_time'] =data['live_time'].strftime('%Y-%m-%d %H:%M')
			send_to_websocket('live', quest_id, data)
			if prev_live == False and quest.live == True:
				subscribers = quest.user_set.all()
				for subscriber in subscribers:
					api.info(
						subscriber,
						f"{quest.title} has gone live!",
						extra_tags='subscription',
						meta={
							'url': reverse(
								'quest:quest',
								args=(quest.id, 0),
							),
						},
						deliver_once=False,
					)
			return redirect('quest:quest',quest_id=quest.id, page_num=page_num)
		else:
			messages.error(request, "Error")
	else:
		pass
	context = locals()
	return render(request, 'quest/edit_quest.html', context)


def new_quest(request):
	"""
	The page for creating new quests.
	"""
	if not request.user.is_authenticated:
		return redirect('login:index')
	if request.method == 'POST':
		# TODO: clean the post body
		quest_form = QuestForm(request.POST)
		if quest_form.is_valid():
			quest = quest_form.save(commit=False)
			quest.owner = request.user
			quest.banner_url = download_img(quest.banner_url)
			if not quest.banner_url.startswith('http'):
				quest.banner_url = None
			quest.save()
			quest_form.save_m2m()
			print(quest.tags.names())
			page = Page(
				quest=quest,
				page_num=0,
				title="Homepage",
				appendix=False,
			)
			page.save()
			return redirect('quest:quest', quest_id=quest.id)
	else:
		quest_form = QuestForm()
	context = locals()
	return render(request, 'quest/new_quest.html', context)


@require_POST
def new_tag(request, quest_id):
	"""Endpoint for adding new tags to a quest."""
	if not request.user.is_authenticated:
		return redirect('login:index')
	tag = request.POST.get('tag', '').strip()
	if not tag:
		return redirect('quest:quest', quest_id=quest_id, page_num='0')

	quest = Quest.objects.get(id=quest_id)
	tag = bleach.clean(tag)
	if tag in quest.tags.names():
		return redirect('quest:quest', quest_id=quest_id, page_num='0')

	quest.tags.add(tag)
	return redirect('quest:quest', quest_id=quest_id, page_num='0')