2022-05-16 20:12:06 +02:00
|
|
|
#!/usr/bin/env python3
|
2022-05-17 22:58:51 +02:00
|
|
|
""" ================================= |
|
|
|
|
| staticoso |
|
|
|
|
| Just a simple Static Site Generator |
|
2022-05-21 20:03:27 +02:00
|
|
|
| |
|
|
|
|
| Licensed under the AGPLv3 license |
|
|
|
|
| Copyright (C) 2022, OctoSpacc |
|
2022-05-17 22:58:51 +02:00
|
|
|
| ================================= """
|
2022-05-17 18:16:39 +02:00
|
|
|
|
2022-05-24 00:22:32 +02:00
|
|
|
import argparse
|
2022-05-16 20:12:06 +02:00
|
|
|
import os
|
|
|
|
import shutil
|
2022-05-17 18:16:39 +02:00
|
|
|
from markdown import Markdown
|
2022-05-17 22:58:51 +02:00
|
|
|
from pathlib import Path
|
2022-05-16 20:12:06 +02:00
|
|
|
|
|
|
|
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
|
|
|
|
|
2022-05-17 22:58:51 +02:00
|
|
|
def DashifyStr(s, Limit=32):
|
|
|
|
Str, lc = '', Limit
|
|
|
|
for c in s[:Limit].replace(' ','-').replace(' ','-'):
|
|
|
|
if c.lower() in '0123456789qwfpbjluyarstgmneiozxcdvkh-':
|
|
|
|
Str += c
|
2022-05-21 19:14:01 +02:00
|
|
|
return '-' + Str
|
|
|
|
|
2022-05-30 20:10:43 +02:00
|
|
|
def GetTitle(Meta, Titles, Prefer='MetaTitle'):
|
|
|
|
if Prefer == 'Title':
|
|
|
|
return Titles[0].lstrip('#') if Titles else Meta['Title'] if Meta['Title'] else 'Untitled'
|
|
|
|
elif Prefer == 'MetaTitle':
|
|
|
|
return Meta['Title'] if Meta['Title'] else Titles[0].lstrip('#') if Titles else 'Untitled'
|
|
|
|
elif Prefer == 'HTMLTitle':
|
|
|
|
return Meta['HTMLTitle'] if Meta['HTMLTitle'] else Meta['Title'] if Meta['Title'] else Titles[0].lstrip('#') if Titles else 'Untitled'
|
|
|
|
|
2022-05-21 19:14:01 +02:00
|
|
|
def GetTitleIdLine(Line, Title):
|
|
|
|
Title = DashifyStr(Title.lstrip('#'))
|
|
|
|
Index = Line.find('h')
|
|
|
|
NewLine = ''
|
|
|
|
NewLine += Line[:Index]
|
|
|
|
NewLine += "{}(id='{}')".format(Line[Index:Index+2], Title)
|
|
|
|
NewLine += Line[Index+2:]
|
|
|
|
return NewLine
|
2022-05-17 22:58:51 +02:00
|
|
|
|
2022-05-16 20:12:06 +02:00
|
|
|
def FormatTitles(Titles):
|
2022-05-21 19:14:01 +02:00
|
|
|
MDTitles = ''
|
2022-05-16 21:16:36 +02:00
|
|
|
for t in Titles:
|
2022-05-17 22:58:51 +02:00
|
|
|
n = t.split(' ')[0].count('#')
|
2022-05-22 23:19:02 +02:00
|
|
|
Heading = '- ' * n
|
2022-05-21 13:04:05 +02:00
|
|
|
Title = t.lstrip('#')
|
|
|
|
Title = '[{}](#{})'.format(Title, DashifyStr(Title))
|
2022-05-22 23:19:02 +02:00
|
|
|
MDTitles += Heading + Title + '\n'
|
2022-05-21 13:04:05 +02:00
|
|
|
return Markdown().convert(MDTitles)
|
2022-05-16 20:12:06 +02:00
|
|
|
|
2022-05-27 11:11:33 +02:00
|
|
|
def LoadFromDir(Dir, Rglob):
|
2022-05-16 20:12:06 +02:00
|
|
|
Contents = {}
|
2022-05-27 11:11:33 +02:00
|
|
|
for File in Path(Dir).rglob(Rglob):
|
2022-05-16 20:12:06 +02:00
|
|
|
File = str(File)[len(Dir)+1:]
|
|
|
|
Contents.update({File: ReadFile('{}/{}'.format(Dir, File))})
|
|
|
|
return Contents
|
|
|
|
|
2022-05-29 00:51:19 +02:00
|
|
|
def PreProcessor(p, SiteRoot):
|
2022-05-17 22:58:51 +02:00
|
|
|
File = ReadFile(p)
|
2022-05-16 20:12:06 +02:00
|
|
|
Content, Titles, Meta = '', [], {
|
|
|
|
'Template': 'Standard.html',
|
2022-05-21 13:04:05 +02:00
|
|
|
'Style': '',
|
2022-05-23 22:36:50 +02:00
|
|
|
'Index': 'True',
|
2022-05-24 20:24:12 +02:00
|
|
|
'Title': '',
|
2022-05-30 20:10:43 +02:00
|
|
|
'HTMLTitle': '',
|
2022-05-24 20:24:12 +02:00
|
|
|
'Order': None}
|
2022-05-16 20:12:06 +02:00
|
|
|
for l in File.splitlines():
|
|
|
|
ls = l.lstrip()
|
2022-05-30 20:10:43 +02:00
|
|
|
if ls.startswith('// '):
|
2022-06-01 00:08:49 +02:00
|
|
|
lss = ls[3:]
|
|
|
|
for Item in ['Template', 'Index', 'Title', 'HTMLTitle']:
|
|
|
|
ItemText = '{}: '.format(Item)
|
|
|
|
if lss.startswith(ItemText):
|
|
|
|
Meta[Item] = lss[len(ItemText):]
|
|
|
|
if lss.startswith('Background: '):
|
2022-05-24 20:24:12 +02:00
|
|
|
Meta['Style'] += "#MainBox{Background:" + ls[len('// Background: '):] + ";} "
|
2022-06-01 00:08:49 +02:00
|
|
|
elif lss.startswith('Style: '):
|
2022-05-24 20:24:12 +02:00
|
|
|
Meta['Style'] += ls[len('// Style: '):] + ' '
|
2022-06-01 00:08:49 +02:00
|
|
|
elif lss.startswith('Order: '):
|
2022-05-24 20:24:12 +02:00
|
|
|
Meta['Order'] = int(ls[len('// Order: '):])
|
|
|
|
elif ls.startswith(('h1', 'h2', 'h3', 'h4', 'h5', 'h6')):
|
|
|
|
if ls[2:].startswith(("(class='NoTitle", '(class="NoTitle')):
|
2022-05-17 22:58:51 +02:00
|
|
|
Content += l + '\n'
|
|
|
|
else:
|
2022-05-24 20:24:12 +02:00
|
|
|
Title = '#'*int(ls[1]) + str(ls[3:])
|
|
|
|
Titles += [Title]
|
|
|
|
# We should handle headers that for any reason already have parenthesis
|
|
|
|
if ls[2:] == '(':
|
|
|
|
Content += l + '\n'
|
|
|
|
else:
|
|
|
|
Content += GetTitleIdLine(l, Title) + '\n'
|
|
|
|
else:
|
|
|
|
Content += l + '\n'
|
2022-05-16 20:12:06 +02:00
|
|
|
return Content, Titles, Meta
|
|
|
|
|
2022-05-17 22:58:51 +02:00
|
|
|
def PugCompileList(Pages):
|
2022-05-29 13:10:48 +02:00
|
|
|
# Pug-cli seems to shit itself with folder paths as input, so we pass ALL the files as arguments
|
2022-05-17 22:58:51 +02:00
|
|
|
Paths = ''
|
|
|
|
for File, Content, Titles, Meta in Pages:
|
2022-05-21 19:14:01 +02:00
|
|
|
FilePath = 'public/{}'.format(File)
|
|
|
|
WriteFile(FilePath, Content)
|
|
|
|
Paths += '"{}" '.format(FilePath)
|
2022-05-17 22:58:51 +02:00
|
|
|
os.system('pug {} > /dev/null'.format(Paths))
|
2022-05-17 18:16:39 +02:00
|
|
|
|
2022-05-29 00:51:19 +02:00
|
|
|
def PatchHTML(Template, Parts, HTMLPagesList, Content, Titles, Meta, SiteRoot):
|
2022-05-16 20:12:06 +02:00
|
|
|
HTMLTitles = FormatTitles(Titles)
|
2022-05-29 01:13:34 +02:00
|
|
|
for p in Parts:
|
|
|
|
Template = Template.replace('[HTML:Part:{}]'.format(p), Parts[p])
|
2022-05-29 13:10:48 +02:00
|
|
|
Template = Template.replace('[HTML:Site:AbsoluteRoot]', SiteRoot)
|
2022-05-23 22:36:50 +02:00
|
|
|
Template = Template.replace('[HTML:Page:LeftBox]', HTMLPagesList)
|
2022-05-16 21:16:36 +02:00
|
|
|
Template = Template.replace('[HTML:Page:RightBox]', HTMLTitles)
|
2022-05-30 20:10:43 +02:00
|
|
|
Template = Template.replace('[HTML:Page:Title]', GetTitle(Meta, Titles, 'MetaTitle'))
|
2022-05-29 13:10:48 +02:00
|
|
|
Template = Template.replace('[HTML:Page:Style]', Meta['Style'])
|
2022-05-16 20:12:06 +02:00
|
|
|
Template = Template.replace('[HTML:Page:MainBox]', Content)
|
|
|
|
return Template
|
|
|
|
|
2022-05-23 22:36:50 +02:00
|
|
|
def FileToStr(File, Truncate=''):
|
|
|
|
return str(File)[len(Truncate):]
|
|
|
|
|
2022-05-24 20:24:12 +02:00
|
|
|
def OrderPages(Old):
|
|
|
|
New = []
|
|
|
|
Max = 0
|
2022-05-30 20:10:43 +02:00
|
|
|
#Off = 0
|
2022-05-24 20:24:12 +02:00
|
|
|
for i,e in enumerate(Old):
|
2022-05-30 20:10:43 +02:00
|
|
|
Curr = e[3]['Order'] #if e[3]['Order'] else 0
|
2022-05-24 20:24:12 +02:00
|
|
|
if Curr > Max:
|
|
|
|
Max = Curr
|
|
|
|
for i in range(Max+1):
|
|
|
|
New += [[]]
|
|
|
|
for i,e in enumerate(Old):
|
2022-05-30 20:10:43 +02:00
|
|
|
#if e[3]['Order']:
|
2022-05-24 20:24:12 +02:00
|
|
|
New[e[3]['Order']] = e
|
2022-05-30 20:10:43 +02:00
|
|
|
#else:
|
|
|
|
#Off += 1
|
|
|
|
#New += [[e]]
|
2022-05-24 20:24:12 +02:00
|
|
|
while [] in New:
|
|
|
|
New.remove([])
|
2022-05-30 20:10:43 +02:00
|
|
|
#for i in New:
|
|
|
|
#print(i)
|
2022-05-24 20:24:12 +02:00
|
|
|
return New
|
|
|
|
|
2022-05-29 00:51:19 +02:00
|
|
|
def GetHTMLPagesList(Pages, SiteRoot):
|
2022-05-23 22:36:50 +02:00
|
|
|
List = ''
|
|
|
|
LastParent = []
|
2022-06-01 00:08:49 +02:00
|
|
|
IndexPages = Pages.copy()
|
|
|
|
for e in IndexPages:
|
|
|
|
if e[3]['Index'] == 'False':
|
|
|
|
IndexPages.remove(e)
|
|
|
|
IndexPages = OrderPages(IndexPages)
|
|
|
|
for File, Content, Titles, Meta in IndexPages:
|
2022-05-30 20:10:43 +02:00
|
|
|
if Meta['Index'] == 'True' and GetTitle(Meta, Titles, Prefer='HTMLTitle') != 'Untitled':
|
2022-05-23 22:36:50 +02:00
|
|
|
n = File.count('/') + 1
|
|
|
|
if n > 1:
|
|
|
|
CurParent = File.split('/')[:-1]
|
|
|
|
for i,s in enumerate(CurParent):
|
|
|
|
if LastParent != CurParent:
|
|
|
|
LastParent = CurParent
|
|
|
|
Levels = '- ' * (n-1+i)
|
2022-05-30 20:10:43 +02:00
|
|
|
if File[:-3].endswith('index.'):
|
|
|
|
Title = GetTitle(Meta, Titles, 'HTMLTitle')
|
|
|
|
Title = '[{}]({})'.format(
|
|
|
|
Title,
|
|
|
|
'{}{}html'.format(SiteRoot, File[:-3]))
|
|
|
|
else:
|
|
|
|
Title = CurParent[n-2+i]
|
2022-05-23 22:36:50 +02:00
|
|
|
List += Levels + Title + '\n'
|
2022-05-30 20:10:43 +02:00
|
|
|
if not (n > 1 and File[:-3].endswith('index.')):
|
|
|
|
Levels = '- ' * n
|
|
|
|
Title = GetTitle(Meta, Titles, 'HTMLTitle')
|
|
|
|
Title = '[{}]({})'.format(
|
|
|
|
Title,
|
|
|
|
'{}{}html'.format(SiteRoot, File[:-3]))
|
|
|
|
List += Levels + Title + '\n'
|
2022-05-23 22:36:50 +02:00
|
|
|
return Markdown().convert(List)
|
|
|
|
|
2022-05-17 22:58:51 +02:00
|
|
|
def DelTmp():
|
|
|
|
for File in Path('public').rglob('*.pug'):
|
|
|
|
os.remove(File)
|
|
|
|
for File in Path('public').rglob('*.md'):
|
|
|
|
os.remove(File)
|
|
|
|
|
2022-05-29 00:51:19 +02:00
|
|
|
def MakeSite(Templates, Parts, SiteRoot):
|
2022-05-16 20:12:06 +02:00
|
|
|
Pages = []
|
2022-05-17 18:16:39 +02:00
|
|
|
for File in Path('Pages').rglob('*.pug'):
|
2022-05-23 22:36:50 +02:00
|
|
|
File = FileToStr(File, 'Pages/')
|
2022-05-29 00:51:19 +02:00
|
|
|
Content, Titles, Meta = PreProcessor('Pages/{}'.format(File), SiteRoot)
|
2022-05-17 22:58:51 +02:00
|
|
|
Pages += [[File, Content, Titles, Meta]]
|
|
|
|
PugCompileList(Pages)
|
2022-05-29 00:51:19 +02:00
|
|
|
HTMLPagesList = GetHTMLPagesList(Pages, SiteRoot)
|
2022-05-17 22:58:51 +02:00
|
|
|
for File, Content, Titles, Meta in Pages:
|
2022-05-16 20:12:06 +02:00
|
|
|
Template = Templates[Meta['Template']]
|
|
|
|
Template = Template.replace(
|
2022-05-29 00:51:19 +02:00
|
|
|
'[HTML:Site:AbsoluteRoot]',
|
|
|
|
SiteRoot)
|
|
|
|
Template = Template.replace(
|
|
|
|
'[HTML:Site:RelativeRoot]',
|
|
|
|
'../'*File.count('/'))
|
2022-05-16 20:12:06 +02:00
|
|
|
WriteFile(
|
2022-05-27 11:11:33 +02:00
|
|
|
'public/{}html'.format(File[:-3]),
|
2022-05-17 22:58:51 +02:00
|
|
|
PatchHTML(
|
2022-05-23 22:36:50 +02:00
|
|
|
Template, Parts, HTMLPagesList,
|
2022-05-27 11:11:33 +02:00
|
|
|
ReadFile('public/{}html'.format(File[:-3])),
|
2022-05-29 00:51:19 +02:00
|
|
|
Titles, Meta, SiteRoot))
|
2022-05-17 22:58:51 +02:00
|
|
|
for File in Path('Pages').rglob('*.md'):
|
2022-05-23 22:36:50 +02:00
|
|
|
File = FileToStr(File, 'Pages/')
|
2022-05-29 00:51:19 +02:00
|
|
|
Content, Titles, Meta = PreProcessor('Pages/{}'.format(File), SiteRoot)
|
2022-05-17 22:58:51 +02:00
|
|
|
Template = Templates[Meta['Template']]
|
|
|
|
Template = Template.replace(
|
2022-05-29 00:51:19 +02:00
|
|
|
'[HTML:Site:AbsoluteRoot]',
|
|
|
|
SiteRoot)
|
|
|
|
Template = Template.replace(
|
|
|
|
'[HTML:Site:RelativeRoot]',
|
|
|
|
'../'*File.count('/'))
|
2022-05-17 22:58:51 +02:00
|
|
|
WriteFile(
|
2022-05-27 11:11:33 +02:00
|
|
|
'public/{}html'.format(File[:-2]),
|
2022-05-17 22:58:51 +02:00
|
|
|
PatchHTML(
|
2022-05-23 22:36:50 +02:00
|
|
|
Template, Parts, HTMLPagesList,
|
2022-05-17 22:58:51 +02:00
|
|
|
Markdown().convert(Content),
|
2022-05-29 00:51:19 +02:00
|
|
|
Titles, Meta, SiteRoot))
|
2022-05-17 22:58:51 +02:00
|
|
|
DelTmp()
|
2022-05-16 20:12:06 +02:00
|
|
|
|
2022-05-24 00:22:32 +02:00
|
|
|
def Main(Args):
|
2022-05-16 20:12:06 +02:00
|
|
|
ResetPublic()
|
2022-05-29 00:51:19 +02:00
|
|
|
SiteRoot = Args.SiteRoot if Args.SiteRoot else '/'
|
2022-05-24 00:22:32 +02:00
|
|
|
|
2022-05-27 11:11:33 +02:00
|
|
|
Templates = LoadFromDir('Templates', '*.html')
|
|
|
|
Parts = LoadFromDir('Parts', '*.html')
|
2022-05-17 22:58:51 +02:00
|
|
|
shutil.copytree('Pages', 'public')
|
2022-05-29 00:51:19 +02:00
|
|
|
MakeSite(Templates, Parts, SiteRoot)
|
2022-05-17 22:58:51 +02:00
|
|
|
os.system("cp -R Assets/* public/")
|
2022-05-16 20:12:06 +02:00
|
|
|
|
|
|
|
if __name__ == '__main__':
|
2022-05-24 00:22:32 +02:00
|
|
|
Parser = argparse.ArgumentParser()
|
2022-05-29 00:51:19 +02:00
|
|
|
Parser.add_argument('--SiteRoot', type=str)
|
2022-05-24 00:22:32 +02:00
|
|
|
Args = Parser.parse_args()
|
|
|
|
|
|
|
|
Main(Args)
|
2022-05-30 20:10:43 +02:00
|
|
|
|