textout/textoutpc/Tags/Link.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from .__base__ import *
from html import escape as _htmlescape

__all__ = ["TextoutLinkTag", "TextoutProfileTag"]

class TextoutLinkTag(TextoutRawInlineTag):
	""" The main link tag.
		Example uses:
		
		[url=https://example.org/hi]Go to example.org[/url]!
		[url=/Fr/index.php][/url]
		[url]https://random.org/randomize.php[/url]
	"""

	aliases = ('[url]',)

	def _validate(self):
		for prefix in ('http://', 'https://', 'ftp://', '/', '#'):
			if self.url.startswith(prefix):
				break
		else:
			raise Exception("No allowed prefix!")

	def prepare(self, name, value):
		self.url = None

		# If there is no value, wait until we have a content to
		# decide if we are valid or not.
		if value == None:
			return

		# Otherwise, get the URL and validate.
		self.url = value
		self._validate()

	def preprocess(self, cin):
		content = cin.read()

		if not self.url:
			self.url = content
			self._validate()
		if not content:
			content = self.url

		return content

	def end_text(self):
		return ' (voir "{}")'.format(self.value)

	def begin_html(self):
		return '<a href="{}" target="_blank" rel="noopener">' \
			.format(_htmlescape(self.url))

	def end_html(self):
		return '</a>'

class TextoutProfileTag(TextoutLinkTag):
	""" A special link tag for Planète Casio's profiles.
		Adds the prefix to the content, and sets the value.
		Example uses:
		
		[profil]Cakeisalie5[/profil]
	"""

	aliases = ('[profil]', '[profile]')

	def prepare(self, name, value):
		pass

	def preprocess(self, cin):
		# FIXME: check the username content!
		username = cin.read()
		self.url = 'https://www.planet-casio.com/Fr/compte/voir_profil.php' \
			'?membre={}'.format(content)
		self._validate()

# End of file.