#!/usr/bin/env python
"""Usage: bupa <page>

Options:
  -h, --help     Show this screen.
  -v, --version  Show the version information.
"""
from jinja2 import Environment, FileSystemLoader
import glob, yaml, docopt, datetime
from docutils import core
from pyatom import AtomFeed
from os import path

class Page(object):
	def __init__(self, page_dict):
		self.id = None
		self.title = page_dict['title']
		self.header = page_dict['header']
		self.date = page_dict['date']
		self.author = page_dict['author']
		if 'author_link' in page_dict:
			self.author_link = page_dict['author_link']
		if 'description' in page_dict:
			self.meta_desc = page_dict['description']
		if 'keywords' in page_dict:
			self.meta_keywords = page_dict['keywords']
		self.body = page_dict['body'][0]

def reST_to_html(s, id_prefix='id'):
	formated = core.publish_parts(s, writer_name='html4css1',
	                              settings_overrides={'footnote_references': 'superscript',
	                                                  'auto_id_prefix': str(id_prefix) + '-',
	                                                  'syntax_highlight': 'short'})
	return formated['fragment']

def filename_to_id(file_name):
	filename = path.split(file_name)[1]
	file_id  = filename.split('-')[0].split('.')[0]
	return int(file_id)

def filename_to_extless(filename):
	filename = path.split(filename)[1]
	extless = path.splitext(filename)[0]
	return extless

def parse(filename):
	with open(filename, 'rb') as file:
		data = file.read().decode('utf-8')

	frontmatter, *body = data.split('\n\n', 1)

	frontmatter = frontmatter.partition('\n')[2]
	frontmatter = yaml.load(frontmatter)

	page_dict = frontmatter.copy()
	page_dict['body'] = body
	if not 'header' in page_dict:
		page_dict['header'] = page_dict['title']

	page = Page(page_dict)
	return page

def build_entry(filename, atom_feed):
	entry = parse(filename)
	entry.id = filename_to_id(filename)
	entry.page = filename_to_extless(filename) + '.html'
	entry.srcpage = filename_to_extless(filename) + '.rst'
	entry.body = reST_to_html(entry.body, entry.id)
	atom_feed.add(
		title=entry.title,
		content=entry.body,
		content_type="html",
		author={'name': entry.author, 'uri': entry.author_link},
		url="https://theos.kyriasis.com/~kyrias/journal/"+entry.page,
		updated=entry.date
	)
	return entry

def build_journal(jinja_env):
	template   = jinja_env.get_template('journal.html')
	files      = glob.glob('src/journal/*.rst')
	files.sort(key=filename_to_id, reverse=True)

	atom_feed = AtomFeed(
		title     = "kyrias’ journal",
		subtitle  = "The lost journal of Kyrias",
		feed_url  = "https://theos.kyriasis.com/~kyrias/journal.atom",
		url       = "https://theos.kyriasis.com/~kyrias/journal.html",
		author    = { 'name': 'Johannes Löthberg',
		               'uri': 'https://theos.kyriasis.com/~kyrias/about.html' },
		updated   = datetime.datetime.utcnow(),
		generator = ('bupa', '', '0.0.1')
	)

	entries = []
	for file in files:
		entry = build_entry(file, atom_feed)
		write_page(entry, 'The lost journal', 'journal/'+entry.page, 'entry.html', jinja_env)
		entries += [entry]

	with open('build/journal.html', 'wb') as file:
		journal = template.render(entries=entries, title='~/journal', header='The lost journal')
		file.write(journal.encode('utf-8'))

	with open('build/journal.atom', 'wb') as file:
		file.write(atom_feed.to_string().encode('utf-8'))

	make_sitemap(jinja_env, entries)

def build_page(jinja_env, template, pagename):
	page = parse('src/'+pagename+'.rst')
	page.body = reST_to_html(page.body)
	page.article_id = pagename
	write_page(page, page.header, pagename+'.html', template, jinja_env)

def write_page(page, header, filename, template, jinja_env):
	template = jinja_env.get_template(template)
	with open('build/'+filename, 'wb') as file:
		rendered_page = template.render(page=page, title=page.title, header=header)
		file.write(rendered_page.encode('utf-8'))

def make_sitemap(jinja_env, entries):
	template = jinja_env.get_template('sitemap.xml')
	with open('build/sitemap.xml', 'wb') as f:
		rendered = template.render(entries=entries)
		f.write(rendered.encode('utf-8'))

def main():
	arguments = docopt.docopt(__doc__, version='bupa 0.0.1')
	jinja_env = Environment(loader=FileSystemLoader('src/templates')i
	                        trim_blocks=True, lstrip_blocks=True)

	if arguments['<page>'] == 'journal.html':
		build_journal(jinja_env)
	elif arguments['<page>'] == 'index.html':
		build_page(jinja_env, 'page.html', 'index')
	elif arguments['<page>'] == 'about.html':
		build_page(jinja_env, 'page.html', 'about')

if __name__ == '__main__':
	main()