staticoso/Build.py

#!/usr/bin/env python3
import os
import shutil
from pathlib import Path
from Libs.markdown import Markdown

def ReadFile(p):
	try:
		with open(p, 'r') as f:
			return f.read()
	except Exception:
		print("Error reading file {}".format(p))
		return None

def WriteFile(p, c):
	try:
		with open(p, 'w') as f:
			f.write(c)
		return True
	except Exception:
		print("Error writing file {}".format(p))
		return False

def ResetPublic():
	try:
		shutil.rmtree('public')
	except FileNotFoundError:
		pass
#	os.mkdir('public')

def GetTitles(c):
	Titles = []
	
	return Titles

def FormatTitles(Titles):
	return ''

"""
def GetAssociations():
	f = ReadFile('Associations.txt')
	Templates = f.splitlines()
	Uses = {}
	for t in Templates:
		t = t.split(' ')
		Uses.update({t[0][:-1]: t[1:]})
	print(Uses)
	return Uses

def LoadTemplates():
	Templates = {}
	for p in Path('Templates').rglob('*.html'):
		p = str(p)[len('Templates')+1:]
		Templates.update({p: ReadFile('Parts/{}'.format(p))})
	return Templates

	Templates = {}
	for t in Associations:
		Templates.update({t: ReadFile('Templates/{}'.format(t))})
	return Templates

def LoadParts():
	Parts = {}
	for p in Path('Parts').rglob('*.html'):
		p = str(p)[len('Parts')+1:]
		Parts.update({p: ReadFile('Parts/{}'.format(p))})
	return Parts
"""

def LoadFromDir(Dir):
	Contents = {}
	for File in Path(Dir).rglob('*.html'):
		File = str(File)[len(Dir)+1:]
		Contents.update({File: ReadFile('{}/{}'.format(Dir, File))})
	return Contents

def PreProcessor(File):
	File = ReadFile(File)
	Content, Titles, Meta = '', [], {
		'Template': 'Standard.html',
		'Background': '',
		'Style': ''}
	for l in File.splitlines():
		ls = l.lstrip()
		if ls.startswith('\%'):
			Content += ls[1:] + '\n'
		elif ls.startswith('% - '):
			Content += '<!-- {} -->'.format(ls[4:]) + '\n'
		elif ls.startswith('% Template: '):
			Meta['Template'] = ls[len('% Template: '):]
		elif ls.startswith('% Background: '):
			Meta['Background'] = ls[len('% Background: '):]
		elif ls.startswith('% Style: '):
			Meta['Style'] += ls[len('% Style: '):] + ' '
		else:
			Content += l + '\n'
			Heading = ls.split(' ')[0].count('#')
			if Heading > 0:
				Titles += [ls]
	return Content, Titles, Meta

"""
def MetaProcessor(Meta):
	Meta[]
	return Meta
"""

def PatchHTML(Template, Parts, Content, Titles, Meta):
	HTMLTitles = FormatTitles(Titles)

	Template = Template.replace('[HTML:Page:Title]', 'Untitled' if not Titles else Titles[0])
	Template = Template.replace('[HTML:Page:Style]', Meta['Style'])
	Template = Template.replace('[HTML:Page:RightBox]', str(Titles))
	Template = Template.replace('[HTML:Page:MainBox]', Content)

	for p in Parts:
		Template = Template.replace('[HTML:Part:{}]'.format(p), Parts[p])
	print(Template)
	return Template

def MakeSite(Templates, Parts):
	Pages = []
	for File in Path('Pages').rglob('*.md'):
		File = str(File)[len('Pages/'):]
		Content, Titles, Meta = PreProcessor('Pages/{}'.format(File))

		Template = Templates[Meta['Template']]
		Template = Template.replace(
			'Style.css',
			'{}Style.css'.format('../'*File.count('/')))

		print(Content, Titles, Meta)
		WriteFile(
			'public/{}.html'.format(File.rstrip('.md')), 
			PatchHTML(
				Template,
				Parts,
				Markdown().convert(Content),
				Titles,
				Meta))

def IgnoreFiles(dir, files):
    return [f for f in files if os.path.isfile(os.path.join(dir, f))]
def CopyAssets():
	shutil.copytree('Pages', 'public', ignore=IgnoreFiles)
#	for File in Path('Assets').rglob('*'):
#		File = str(File)
#		shutil.copy(File, 'public/{}'.format(File[len('Assets/'):]))
	os.system("cp -R Assets/* public/")

def Main():
	os.chdir(os.path.dirname(os.path.abspath(__file__)))
	ResetPublic()
#	Associations = GetAssociations()
	Templates = LoadFromDir('Templates') #LoadTemplates(Associations)
	Parts = LoadFromDir('Parts') #LoadParts()
	print(Templates, Parts)
	CopyAssets()
	MakeSite(Templates, Parts)

if __name__ == '__main__':
	Main()