mumble-django/pyweb/mumble/views.py

# -*- coding: utf-8 -*-

"""
 *  Copyright (C) 2009, Michael "Svedrin" Ziegler <diese-addy@funzt-halt.net>
 *
 *  Mumble-Django is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This package is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
"""

from StringIO				import StringIO

from django.shortcuts			import render_to_response, get_object_or_404, get_list_or_404
from django.template			import RequestContext
from django.http			import Http404, HttpResponse, HttpResponseRedirect
from django.core.urlresolvers		import reverse
from django.contrib.auth.decorators	import login_required

from models				import Mumble, MumbleUser
from forms				import *
from mmobjects				import mmServer, mmChannel

# Handler class for all Server specific views

class Storage( object ):
	s = list();
	r = None;


def mumbles( request ):
	"Displays a list of all configured Mumble servers."
	return render_to_response(
		'mumble/list.htm',
		{ 'MumbleObjects': get_list_or_404( Mumble ),
		  'MumbleActive':  True,
		},
		context_instance = RequestContext(request)
		);


def show( request, server ):
	"Displays the channel list for the given Server ID."
	srv, o = createChannelList( server );
	
	isAdmin = srv.isUserAdmin( request.user );
	
	# The tab to start on.
	displayTab = 0;
	
	if isAdmin:
		if request.method == 'POST' and "mode" in request.POST and request.POST['mode'] == 'admin':
			adminform = MumbleForm( request.POST, instance=srv );
			# In case we redisplay the page, it was displayed with errors on the admin form, so tell
			# Ext to show the admin form tab first.
			displayTab = 1;
			if adminform.is_valid():
				adminform.save();
				return HttpResponseRedirect( '/mumble/%d' % int(server) );
		else:
			adminform = MumbleForm( instance=srv );
	else:
		adminform = None;
	
	registered = False;
	
	if request.user.is_authenticated():
		if request.method == 'POST' and 'mode' in request.POST and request.POST['mode'] == 'reg':
			try:
				user    = MumbleUser.objects.get( server=srv, owner=request.user );
			except MumbleUser.DoesNotExist:
				regform = MumbleUserForm( request.POST );
				if regform.is_valid():
					model = regform.save( commit=False );
					model.isAdmin = False;
					model.server  = srv;
					model.owner   = request.user;
					model.save();
					return HttpResponseRedirect( '/mumble/%d' % int(server) );
			else:
				regform = MumbleUserForm( request.POST, instance=user );
				if regform.is_valid():
					regform.save();
					return HttpResponseRedirect( '/mumble/%d' % int(server) );
		else:
			try:
				user  = MumbleUser.objects.get( server=srv, owner=request.user );
			except MumbleUser.DoesNotExist:
				regform = MumbleUserForm();
			else:
				regform = MumbleUserForm( instance=user );
				registered = True;
		
		if request.method == 'POST' and 'mode' in request.POST and request.POST['mode'] == 'texture' and registered:
			textureform = MumbleTextureForm( request.POST, request.FILES );
			if textureform.is_valid():
				user.setTexture( request.FILES['texturefile'] );
				return HttpResponseRedirect( '/mumble/%d' % int(server) );
		else:
			textureform = MumbleTextureForm();

	else:
		regform = None;
		textureform = None;
	
	return render_to_response(
		'mumble/mumble.htm',
		{
			'DBaseObject':  srv,
			'ServerObject': o,
			'ChannelTable': Storage.s,
			"CurrentUserIsAdmin": isAdmin,
			"AdminForm":    adminform,
			"RegForm":      regform,
			"TextureForm":  textureform,
			"Registered":   registered,
			"DisplayTab":   displayTab,
			'MumbleActive':  True,
		},
		context_instance = RequestContext(request)
		);


def showTexture( request, server ):
	if request.user.is_authenticated():
		srv  = Mumble.objects.get( id=int(server) );
		user = MumbleUser.objects.get( server=srv, owner=request.user );
		try:
			img  = user.getTexture();
		except ValueError:
			raise Http404();
		else:
			buffer = StringIO();
			img.save( buffer, "PNG" );
			return HttpResponse( buffer.getvalue(), "image/png" );
	raise Http404();

def showContent( server, user = None ):
	"Renders and returns the channel list for the given Server ID."
	from django.template import Context, loader
	
	srv, o = createChannelList( server );
	
	mumbleAcc = None;
	if user.is_authenticated():
		mmUsers = MumbleUser.objects.filter( owner = user );
		if mmUsers:
			mumbleAcc = mmUsers[0];
	
	t_content = loader.get_template( 'mumble/content.htm' );
	c_content = Context( {
		'DBaseObject': srv,
		'ServerObject': o,
		'ChannelTable': Storage.s,
		'user': user,
		'mumbleAccount': mumbleAcc,
		"CurrentUserIsAdmin": srv.isUserAdmin( request.user ),
		'MumbleActive':  True,
		} );
	r_content = t_content.render( c_content );
	
	return r_content;


def createChannelList( server ):
	"Renders the channel list."
	srv = get_object_or_404( Mumble, id=server );
	
	o = srv.getServerObject();
	
	Storage.s = list();
	Storage.r = o.channels[0];
	o.channels[0].visit( renderListItem, 0 );
	
	return srv, o;


def renderListItem( item, level ):
	"Stores a line in the channel list."
	if item == Storage.r:
		return;
	
	# Filter channels that don't have players in them and are not a subchannel of root
	if level > 1 and item.playerCount == 0:
		# I used to test if item is an instance of mmChannel here. For some reason, that doesn't work. Dunno why.
		return;
	
	if isinstance( item, mmChannel ):
		Storage.s.append( ( level, item, item.parentChannels() ) );
	else:
		Storage.s.append( ( level, item ) );