Redesign key management pages.

[people/jschlag/pbs.git] / web / handlers_mirrors.py

#!/usr/bin/python

import tornado.web

import backend

from handlers_base import BaseHandler

class MirrorListHandler(BaseHandler):
	def get(self):
		mirrors = self.pakfire.mirrors.get_all()
		mirrors_nearby = self.pakfire.mirrors.get_for_location(self.remote_address)

		mirrors_worldwide = []
		for mirror in mirrors:
			if mirror in mirrors_nearby:
				continue

			mirrors_worldwide.append(mirror)

		kwargs = {
			"mirrors" : mirrors,
			"mirrors_nearby" : mirrors_nearby,
			"mirrors_worldwide" : mirrors_worldwide,
		}

		# Get recent log messages.
		kwargs["log"] = self.pakfire.mirrors.get_history(limit=10)

		self.render("mirrors-list.html", **kwargs)


class MirrorDetailHandler(BaseHandler):
	def get(self, hostname):
		mirror = self.pakfire.mirrors.get_by_hostname(hostname)
		if not mirror:
			raise tornado.web.HTTPError(404, "Could not find mirror: %s" % hostname)

		self.render("mirrors-detail.html", mirror=mirror)


class MirrorActionHandler(BaseHandler):
	"""
		A handler that makes sure if the user has got sufficent rights to
		do actions.
	"""
	def prepare(self):
		# Check if the user has sufficient rights to create a new mirror.
		if not self.current_user.has_perm("manage_mirrors"):
			raise tornado.web.HTTPError(403)


class MirrorNewHandler(MirrorActionHandler):
	@tornado.web.authenticated
	def get(self, hostname="", path="", hostname_missing=False, path_invalid=False):
		self.render("mirrors-new.html", _hostname=hostname, path=path,
			hostname_missing=hostname_missing, path_invalid=path_invalid)

	@tornado.web.authenticated
	def post(self):
		errors = {}

		hostname = self.get_argument("name", None)
		if not hostname:
			errors["hostname_missing"] = True

		path = self.get_argument("path", "")
		if path is None:
			errors["path_invalid"] = True

		if errors:
			errors.update({
				"hostname" : hostname,
				"path" : path,
			})
			return self.get(**errors)

		print hostname, path

		mirror = backend.mirrors.Mirror.create(self.pakfire, hostname, path,
			user=self.current_user)
		assert mirror

		self.redirect("/mirror/%s" % mirror.hostname)


class MirrorEditHandler(MirrorActionHandler):
	@tornado.web.authenticated
	def get(self, hostname):
		mirror = self.pakfire.mirrors.get_by_hostname(hostname)
		if not mirror:
			raise tornado.web.HTTPError(404, "Could not find mirror: %s" % hostname)

		self.render("mirrors-edit.html", mirror=mirror)

	@tornado.web.authenticated
	def post(self, hostname):
		mirror = self.pakfire.mirrors.get_by_hostname(hostname)
		if not mirror:
			raise tornado.web.HTTPError(404, "Could not find mirror: %s" % hostname)

		hostname = self.get_argument("name")
		path     = self.get_argument("path", "")
		owner    = self.get_argument("owner", None)
		contact  = self.get_argument("contact", None)
		enabled  = self.get_argument("enabled", None)

		if enabled:
			mirror.set_status("enabled")
		else:
			mirror.set_status("disabled")

		mirror.hostname = hostname
		mirror.path     = path
		mirror.owner    = owner
		mirror.contact  = contact

		self.redirect("/mirror/%s" % mirror.hostname)


class MirrorDeleteHandler(MirrorActionHandler):
	@tornado.web.authenticated
	def get(self, hostname):
		mirror = self.pakfire.mirrors.get_by_hostname(hostname)
		if not mirror:
			raise tornado.web.HTTPError(404, "Could not find mirror: %s" % hostname)

		confirmed = self.get_argument("confirmed", None)	
		if confirmed:
			mirror.set_status("deleted", user=self.current_user)

			self.redirect("/mirrors")
			return

		self.render("mirrors-delete.html", mirror=mirror)