summaryrefslogtreecommitdiffstats
path: root/scripts/bupa
blob: 3802982517a89e402007f982010e7510c9f15d39 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
#!/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')) # something something jinja2 templates

	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()