2022-05-16 20:12:06 +02:00
|
|
|
#!/usr/bin/env python3
|
2022-05-17 22:58:51 +02:00
|
|
|
""" ================================= |
|
2022-06-26 00:06:34 +02:00
|
|
|
| This file is part of |
|
|
|
|
| staticoso |
|
2022-05-17 22:58:51 +02:00
|
|
|
| Just a simple Static Site Generator |
|
2022-05-21 20:03:27 +02:00
|
|
|
| |
|
|
|
|
| Licensed under the AGPLv3 license |
|
2022-06-26 00:06:34 +02:00
|
|
|
| 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-06-07 13:42:32 +02:00
|
|
|
from ast import literal_eval
|
2022-06-24 20:00:38 +02:00
|
|
|
from datetime import datetime
|
2022-06-29 20:09:13 +02:00
|
|
|
from pathlib import Path
|
|
|
|
|
2022-07-02 22:09:09 +02:00
|
|
|
try:
|
|
|
|
from Modules.ActivityPub import *
|
|
|
|
ActivityPub = True
|
|
|
|
except:
|
2022-07-08 15:50:40 +02:00
|
|
|
print("[E] Can't load the ActivityPub module. Its use is disabled. Make sure the 'requests' library is installed.")
|
2022-07-02 22:09:09 +02:00
|
|
|
ActivityPub = False
|
|
|
|
|
2022-07-11 17:28:59 +02:00
|
|
|
from Modules.Config import *
|
2022-06-29 00:17:52 +02:00
|
|
|
from Modules.Gemini import *
|
2022-07-14 20:38:45 +02:00
|
|
|
from Modules.Markdown import *
|
|
|
|
from Modules.Site import *
|
2022-07-13 17:40:09 +02:00
|
|
|
from Modules.Sitemap import *
|
2022-06-26 00:06:34 +02:00
|
|
|
from Modules.Utils import *
|
2022-05-16 20:12:06 +02:00
|
|
|
|
|
|
|
def ResetPublic():
|
2022-06-29 00:17:52 +02:00
|
|
|
for i in ('public', 'public.gmi'):
|
|
|
|
try:
|
|
|
|
shutil.rmtree(i)
|
|
|
|
except FileNotFoundError:
|
|
|
|
pass
|
2022-05-16 20:12:06 +02:00
|
|
|
|
2022-05-17 22:58:51 +02:00
|
|
|
def DelTmp():
|
2022-07-14 20:38:45 +02:00
|
|
|
for Ext in FileExtensions['Pages']:
|
2022-06-20 16:16:41 +02:00
|
|
|
for File in Path('public').rglob('*.{}'.format(Ext)):
|
|
|
|
os.remove(File)
|
2022-06-30 18:50:18 +02:00
|
|
|
for Dir in ('public', 'public.gmi'):
|
|
|
|
for File in Path(Dir).rglob('*.tmp'):
|
|
|
|
os.remove(File)
|
2022-05-17 22:58:51 +02:00
|
|
|
|
2022-06-23 17:20:06 +02:00
|
|
|
def SetSorting(Sorting):
|
|
|
|
Default = {
|
|
|
|
'Pages':'Standard',
|
|
|
|
'Posts':'Inverse'}
|
|
|
|
for i in Default:
|
|
|
|
if i not in Sorting:
|
|
|
|
Sorting.update({i:Default[i]})
|
|
|
|
return Sorting
|
|
|
|
|
2022-07-15 00:12:58 +02:00
|
|
|
def GetConfMenu(Conf, MarkdownExts):
|
2022-07-11 17:28:59 +02:00
|
|
|
Entries = ReadConf(Conf, 'Menu')
|
|
|
|
if Entries:
|
|
|
|
Menu, Max = [], 0
|
|
|
|
for i in Entries:
|
|
|
|
if int(i) > Max:
|
|
|
|
Max = int(i)
|
|
|
|
for i in range(Max+1):
|
|
|
|
Menu += [[]]
|
|
|
|
for i in Entries:
|
|
|
|
e = Entries[i]
|
|
|
|
if (e.startswith('<') and e.endswith('>')) or (e.startswith('[') and e.endswith(')')):
|
|
|
|
Menu[int(i)] = markdown(e, extensions=MarkdownExts)
|
|
|
|
else:
|
2022-07-15 00:12:58 +02:00
|
|
|
if not e.lower().endswith('.html'):
|
|
|
|
e += '.html'
|
|
|
|
Menu[int(i)] = e
|
2022-07-11 17:28:59 +02:00
|
|
|
print(Menu)
|
|
|
|
return Menu
|
|
|
|
|
2022-07-13 16:53:09 +02:00
|
|
|
def Main(Args, FeedEntries):
|
2022-07-11 17:28:59 +02:00
|
|
|
HavePages, HavePosts = False, False
|
|
|
|
SiteConf = LoadConf('Site.ini')
|
|
|
|
|
|
|
|
SiteName = Args.SiteName if Args.SiteName else ReadConf(SiteConf, 'Site', 'Name') if ReadConf(SiteConf, 'Site', 'Name') else ''
|
2022-07-11 22:56:42 +02:00
|
|
|
BlogName = Args.BlogName if Args.BlogName else ReadConf(SiteConf, 'Site', 'BlogName') if ReadConf(SiteConf, 'Site', 'BlogName') else ''
|
2022-07-11 17:28:59 +02:00
|
|
|
SiteTagline = Args.SiteTagline if Args.SiteTagline else ReadConf(SiteConf, 'Site', 'Tagline') if ReadConf(SiteConf, 'Site', 'Tagline') else ''
|
|
|
|
SiteDomain = Args.SiteDomain.rstrip('/') if Args.SiteDomain else ReadConf(SiteConf, 'Site', 'Domain') if ReadConf(SiteConf, 'Site', 'Domain') else ''
|
|
|
|
SiteLang = Args.SiteLang if Args.SiteLang else ReadConf(SiteConf, 'Site', 'Lang') if ReadConf(SiteConf, 'Site', 'Lang') else 'en'
|
2022-06-30 23:29:44 +02:00
|
|
|
Locale = LoadLocale(SiteLang)
|
|
|
|
MastodonURL = Args.MastodonURL if Args.MastodonURL else ''
|
|
|
|
MastodonToken = Args.MastodonToken if Args.MastodonToken else ''
|
2022-07-11 17:28:59 +02:00
|
|
|
MarkdownExts = literal_eval(Args.MarkdownExts) if Args.MarkdownExts else EvalOpt(ReadConf(SiteConf, 'Site', 'MarkdownExts')) if ReadConf(SiteConf, 'Site', 'MarkdownExts') else ['attr_list', 'def_list', 'markdown_del_ins', 'mdx_subscript', 'mdx_superscript']
|
|
|
|
|
2022-07-13 17:40:09 +02:00
|
|
|
Minify = False
|
2022-07-13 16:53:09 +02:00
|
|
|
if Args.Minify != None:
|
|
|
|
if Args.Minify not in ('False', 'None'):
|
|
|
|
Minify = True
|
|
|
|
else:
|
|
|
|
if ReadConf(SiteConf, 'Site', 'Minify') != None:
|
|
|
|
if ReadConf(SiteConf, 'Site', 'Minify') not in ('False', 'None'):
|
|
|
|
Minify = True
|
|
|
|
|
2022-07-11 17:28:59 +02:00
|
|
|
AutoCategories = False
|
|
|
|
if Args.AutoCategories != None:
|
|
|
|
if literal_eval(Args.AutoCategories) == True:
|
|
|
|
AutoCategories = True
|
|
|
|
else:
|
|
|
|
if ReadConf(SiteConf, 'Site', 'AutoCategories') != None:
|
|
|
|
if EvalOpt(ReadConf(SiteConf, 'Site', 'AutoCategories')) == True:
|
|
|
|
AutoCategories = True
|
2022-06-24 17:19:22 +02:00
|
|
|
|
2022-05-16 20:12:06 +02:00
|
|
|
ResetPublic()
|
2022-07-11 17:28:59 +02:00
|
|
|
|
2022-06-23 16:05:02 +02:00
|
|
|
if os.path.isdir('Pages'):
|
2022-07-11 17:28:59 +02:00
|
|
|
HavePages = True
|
2022-06-23 16:05:02 +02:00
|
|
|
shutil.copytree('Pages', 'public')
|
2022-06-29 20:09:13 +02:00
|
|
|
if Args.GemtextOut:
|
|
|
|
shutil.copytree('Pages', 'public.gmi', ignore=IgnoreFiles)
|
2022-06-23 16:05:02 +02:00
|
|
|
if os.path.isdir('Posts'):
|
2022-07-11 17:28:59 +02:00
|
|
|
HavePosts = True
|
2022-06-23 16:05:02 +02:00
|
|
|
shutil.copytree('Posts', 'public/Posts')
|
2022-06-29 20:09:13 +02:00
|
|
|
if Args.GemtextOut:
|
|
|
|
shutil.copytree('Posts', 'public.gmi/Posts', ignore=IgnoreFiles)
|
2022-06-24 17:19:22 +02:00
|
|
|
|
2022-07-11 17:28:59 +02:00
|
|
|
if not HavePages and not HavePosts:
|
|
|
|
print("[E] No Pages or posts found. Nothing to do, exiting!")
|
|
|
|
exit()
|
|
|
|
|
2022-07-01 16:39:54 +02:00
|
|
|
print("[I] Generating HTML")
|
2022-06-24 20:00:38 +02:00
|
|
|
Pages = MakeSite(
|
2022-06-20 16:16:41 +02:00
|
|
|
TemplatesText=LoadFromDir('Templates', '*.html'),
|
|
|
|
PartsText=LoadFromDir('Parts', '*.html'),
|
|
|
|
ContextParts=literal_eval(Args.ContextParts) if Args.ContextParts else {},
|
|
|
|
ContextPartsText=LoadFromDir('ContextParts', '*.html'),
|
2022-07-15 00:12:58 +02:00
|
|
|
ConfMenu=[],#GetConfMenu(SiteConf, MarkdownExts),
|
2022-06-24 17:19:22 +02:00
|
|
|
SiteName=SiteName,
|
2022-07-11 22:56:42 +02:00
|
|
|
BlogName=BlogName,
|
2022-06-24 17:19:22 +02:00
|
|
|
SiteTagline=SiteTagline,
|
|
|
|
SiteDomain=SiteDomain,
|
2022-06-20 16:16:41 +02:00
|
|
|
SiteRoot=Args.SiteRoot if Args.SiteRoot else '/',
|
|
|
|
FolderRoots=literal_eval(Args.FolderRoots) if Args.FolderRoots else {},
|
2022-07-12 22:32:53 +02:00
|
|
|
SiteLang=SiteLang,
|
2022-06-30 23:29:44 +02:00
|
|
|
Locale=Locale,
|
2022-07-13 17:40:09 +02:00
|
|
|
Minify=Minify,
|
2022-06-29 23:56:20 +02:00
|
|
|
Sorting=SetSorting(literal_eval(Args.ContextParts) if Args.ContextParts else {}),
|
2022-07-11 17:28:59 +02:00
|
|
|
MarkdownExts=MarkdownExts,
|
2022-07-13 17:40:09 +02:00
|
|
|
AutoCategories=AutoCategories)
|
2022-06-24 17:19:22 +02:00
|
|
|
|
|
|
|
if FeedEntries != 0:
|
2022-07-01 16:39:54 +02:00
|
|
|
print("[I] Generating Feeds")
|
2022-07-13 16:53:09 +02:00
|
|
|
for FeedType in (True, False):
|
|
|
|
MakeFeed(
|
|
|
|
Pages=Pages,
|
|
|
|
SiteName=SiteName,
|
|
|
|
SiteTagline=SiteTagline,
|
|
|
|
SiteDomain=SiteDomain,
|
|
|
|
MaxEntries=FeedEntries,
|
|
|
|
Lang=SiteLang,
|
|
|
|
FullSite=FeedType,
|
2022-07-13 17:40:09 +02:00
|
|
|
Minify=Minify)
|
|
|
|
|
|
|
|
if Args.SitemapOut:
|
|
|
|
print("[I] Generating Sitemap")
|
|
|
|
MakeSitemap(Pages, SiteDomain)
|
2022-06-24 17:19:22 +02:00
|
|
|
|
2022-07-02 22:09:09 +02:00
|
|
|
if ActivityPub and MastodonURL and MastodonToken and SiteDomain:
|
2022-07-01 16:39:54 +02:00
|
|
|
print("[I] Mastodon Stuff")
|
2022-07-01 14:00:25 +02:00
|
|
|
MastodonPosts = MastodonShare(
|
2022-06-30 23:29:44 +02:00
|
|
|
MastodonURL,
|
|
|
|
MastodonToken,
|
|
|
|
Pages,
|
|
|
|
SiteDomain,
|
|
|
|
SiteLang,
|
|
|
|
Locale)
|
2022-07-01 14:00:25 +02:00
|
|
|
else:
|
|
|
|
MastodonPosts = []
|
|
|
|
|
2022-07-01 17:29:37 +02:00
|
|
|
for File, Content, Titles, Meta, ContentHTML, SlimHTML, Description, Image in Pages:
|
2022-07-01 14:00:25 +02:00
|
|
|
File = 'public/{}.html'.format(StripExt(File))
|
|
|
|
Content = ReadFile(File)
|
|
|
|
Post = ''
|
|
|
|
for p in MastodonPosts:
|
|
|
|
if p['Link'] == SiteDomain + '/' + File[len('public/'):]:
|
|
|
|
Post = '<br><h3>{StrComments}</h3><a href="{URL}" rel="noopener" target="_blank">{StrOpen} ↗️</a>'.format(
|
|
|
|
StrComments=Locale['Comments'],
|
|
|
|
StrOpen=Locale['OpenInNewTab'],
|
|
|
|
URL=p['Post'])
|
|
|
|
break
|
|
|
|
Content = Content.replace('[HTML:Comments]', Post)
|
|
|
|
WriteFile(File, Content)
|
|
|
|
|
|
|
|
if Args.GemtextOut:
|
2022-07-01 16:39:54 +02:00
|
|
|
print("[I] Generating Gemtext")
|
2022-07-04 20:26:50 +02:00
|
|
|
GemtextCompileList(
|
|
|
|
Pages,
|
2022-07-04 22:27:38 +02:00
|
|
|
Header=Args.GemtextHeader if Args.GemtextHeader else '# {}\n\n'.format(SiteName) if SiteName else '')
|
2022-06-29 00:17:52 +02:00
|
|
|
|
2022-07-11 17:28:59 +02:00
|
|
|
print("[I] Last Steps")
|
2022-06-23 17:20:06 +02:00
|
|
|
DelTmp()
|
2022-05-17 22:58:51 +02:00
|
|
|
os.system("cp -R Assets/* public/")
|
2022-07-11 17:28:59 +02:00
|
|
|
|
2022-07-01 16:39:54 +02:00
|
|
|
print("[I] Done!")
|
2022-05-16 20:12:06 +02:00
|
|
|
|
|
|
|
if __name__ == '__main__':
|
2022-05-24 00:22:32 +02:00
|
|
|
Parser = argparse.ArgumentParser()
|
2022-06-20 23:16:31 +02:00
|
|
|
Parser.add_argument('--Minify', type=str)
|
2022-06-23 17:20:06 +02:00
|
|
|
Parser.add_argument('--Sorting', type=str)
|
2022-06-20 16:16:41 +02:00
|
|
|
Parser.add_argument('--SiteLang', type=str)
|
2022-05-29 00:51:19 +02:00
|
|
|
Parser.add_argument('--SiteRoot', type=str)
|
2022-06-24 17:19:22 +02:00
|
|
|
Parser.add_argument('--SiteName', type=str)
|
2022-07-11 22:56:42 +02:00
|
|
|
Parser.add_argument('--BlogName', type=str)
|
2022-06-24 17:19:22 +02:00
|
|
|
Parser.add_argument('--SiteDomain', type=str)
|
2022-06-29 23:56:20 +02:00
|
|
|
Parser.add_argument('--GemtextOut', type=bool)
|
2022-07-04 20:26:50 +02:00
|
|
|
Parser.add_argument('--GemtextHeader', type=str)
|
2022-06-24 17:19:22 +02:00
|
|
|
Parser.add_argument('--SiteTagline', type=str)
|
2022-07-13 00:14:37 +02:00
|
|
|
Parser.add_argument('--SitemapOut', type=bool)
|
2022-06-24 17:19:22 +02:00
|
|
|
Parser.add_argument('--FeedEntries', type=int)
|
2022-06-19 23:07:23 +02:00
|
|
|
Parser.add_argument('--FolderRoots', type=str)
|
2022-06-07 13:42:32 +02:00
|
|
|
Parser.add_argument('--ContextParts', type=str)
|
2022-06-29 23:56:20 +02:00
|
|
|
Parser.add_argument('--MarkdownExts', type=str)
|
2022-06-30 18:50:18 +02:00
|
|
|
Parser.add_argument('--MastodonURL', type=str)
|
|
|
|
Parser.add_argument('--MastodonToken', type=str)
|
2022-07-11 17:28:59 +02:00
|
|
|
Parser.add_argument('--AutoCategories', type=str)
|
2022-06-29 23:56:20 +02:00
|
|
|
Args = Parser.parse_args()
|
|
|
|
|
|
|
|
try:
|
|
|
|
import lxml
|
|
|
|
from Modules.Feed import *
|
2022-07-04 18:02:11 +02:00
|
|
|
FeedEntries = Args.FeedEntries if Args.FeedEntries or Args.FeedEntries == 0 else 10
|
2022-06-29 23:56:20 +02:00
|
|
|
except:
|
2022-07-13 16:53:09 +02:00
|
|
|
print("[E] Can't load the XML libraries. XML Feeds Generation is Disabled. Make sure the 'lxml' library is installed.")
|
2022-06-29 23:56:20 +02:00
|
|
|
FeedEntries = 0
|
|
|
|
|
2022-06-20 16:16:41 +02:00
|
|
|
Main(
|
2022-06-29 23:56:20 +02:00
|
|
|
Args=Args,
|
2022-07-13 16:53:09 +02:00
|
|
|
FeedEntries=FeedEntries)
|