""" ================================= | | This file is part of | | staticoso | | Just a simple Static Site Generator | | | | Licensed under the AGPLv3 license | | Copyright (C) 2022, OctoSpacc | | ================================= """ import shutil from datetime import datetime from multiprocessing import Pool, cpu_count from Libs.bs4 import BeautifulSoup from Modules.Config import * from Modules.Elements import * from Modules.Globals import * from Modules.HTML import * from Modules.Logging import * from Modules.Markdown import * from Modules.Pug import * from Modules.Utils import * # Menu styles: # - Simple: Default, Flat, Line # - Others: Excerpt, Image, Preview (Excerpt + Image), Full def GetHTMLPagesList(Pages, BlogName, SiteRoot, PathPrefix, CallbackFile=None, Unite=[], Type=None, Limit=None, PathFilter='', Category=None, For='Menu', MarkdownExts=(), MenuStyle='Default', ShowPaths=True): Flatten, SingleLine, DoneCount, PrevDepth = False, False, 0, 0 if MenuStyle == 'Flat': Flatten = True elif MenuStyle == 'Line': ShowPaths, SingleLine = False, True List, ToPop, LastParent = '', [], [] IndexPages = Pages.copy() for e in IndexPages: if e[3]['Index'].lower() in PageIndexStrNeg: IndexPages.remove(e) for i,e in enumerate(IndexPages): if Type and e[3]['Type'] != Type: ToPop += [i] ToPop = RevSort(ToPop) for i in ToPop: IndexPages.pop(i) if Type == 'Page': IndexPages = OrderPages(IndexPages) for i,e in enumerate(Unite): if e: IndexPages.insert(i, [e, None, None, {'Type':Type, 'Index':'True', 'Order':'Unite'}]) for File, Content, Titles, Meta in IndexPages: # Allow for the virtual "Pages/" prefix to be used in path filtering TmpPathFilter = PathFilter if TmpPathFilter.startswith('Pages/'): TmpPathFilter = TmpPathFilter[len('Pages/'):] if File.startswith('Posts/'): continue if (not Type or (Meta['Type'] == Type and CanIndex(Meta['Index'], For))) and (not Category or Category in Meta['Categories']) and File.startswith(TmpPathFilter) and File != CallbackFile and (not Limit or Limit > DoneCount): Depth = (File.count('/') + 1) if Meta['Order'] != 'Unite' else 1 # Folder names are handled here if Depth > 1 and Meta['Order'] != 'Unite': CurParent = File.split('/')[:-1] for i,s in enumerate(CurParent): if LastParent != CurParent and ShowPaths: LastParent = CurParent Levels = '.' * ((Depth-2+i) if not Flatten else 0) + ':' # If search node endswith index, it's a page; else, it's a folder if StripExt(File).endswith('index'): Title = MakeListTitle(File, Meta, Titles, 'HTMLTitle', BlogName, PathPrefix) DoneCount += 1 else: Title = CurParent[Depth-2+i] if SingleLine: List += f' {Title} ' else: List += f'{Levels}{Title}\n' # Pages with any other path if not (Depth > 1 and StripExt(File).split('/')[-1] == 'index'): Levels = '.' * ((Depth-1) if not Flatten else 0) + ':' DoneCount += 1 if Meta['Order'] == 'Unite': Title = markdown(MarkdownHTMLEscape(File, MarkdownExts), extensions=MarkdownExts).removeprefix('

').removesuffix('

') else: Title = MakeListTitle(File, Meta, Titles, 'HTMLTitle', BlogName, PathPrefix) if SingleLine: List += ' ' + Title + ' ' else: List += Levels + Title + '\n' if MenuStyle in ('Default', 'Flat'): return GenHTMLTreeList(List, Class="staticoso-PagesList") elif MenuStyle in ('Line', 'Excerpt', 'Image', 'Preview', 'Full'): return List def CheckHTMLCommentLine(Line): if Line.startswith(''): return Line return None def TemplatePreprocessor(Text): Meta, MetaDefault = '', { 'MenuStyle': 'Default'} for l in Text.splitlines(): ll = l.lstrip().rstrip() lll = CheckHTMLCommentLine(ll) if lll: if lll.startswith('%'): Meta += lll[1:-3].lstrip().rstrip() + '\n' Meta = dict(ReadConf(LoadConfStr('[Meta]\n' + Meta), 'Meta')) for i in MetaDefault: if not i in Meta: Meta.update({i:MetaDefault[i]}) return Meta def FindPreprocLine(Line, Meta, Macros): Changed = False Line = Line.lstrip().rstrip() lll = CheckHTMLCommentLine(Line) if Line.startswith('//') or lll: # Find preprocessor lines lll = Line[2:].lstrip() if lll.startswith('%'): Meta += lll[1:].lstrip() + '\n' Changed = True elif lll.startswith('$'): Macros += lll[1:].lstrip() + '\n' Changed = True #if ll.startswith(''): # Find comment and code blocks # IgnoreBlocksStart += [l] return (Meta, Macros, Changed) def PagePreprocessor(Path:str, TempPath:str, Type, SiteTemplate, SiteRoot, GlobalMacros, CategoryUncategorized, LightRun=False): File = ReadFile(Path) Path = Path.lower() Content, Titles, DashyTitles, HTMLTitlesFound, Macros, Meta, MetaDefault = '', [], [], False, '', '', { 'Template': SiteTemplate, 'Head': '', 'Style': '', 'Type': Type, 'Index': 'Unspecified', 'Feed': 'True', 'Title': '', 'HTMLTitle': '', 'Description': '', 'Image': '', 'Macros': {}, 'Categories': [], 'URLs': [], 'CreatedOn': '', 'UpdatedOn': '', 'EditedOn': '', 'Order': None, 'Language': None, 'Downsync': None} # Find all positions of '', add them in a list=[[pos0,pos1,line0,line1],...] for l in File.splitlines(): ll = l.lstrip().rstrip() Meta, Macros, Changed = FindPreprocLine(ll, Meta, Macros) if not Changed: # Find headings #if line in ignore block: # continue Headings = ('h1', 'h2', 'h3', 'h4', 'h5', 'h6') #if Path.endswith(FileExtensions['HTML']): # if ll[1:].startswith(Headings): # if ll[3:].startswith((" class='NoTitle", ' class="NoTitle')): # Content += l + '\n' # elif ll.replace(' ', ' ').startswith('// %'): # pass # else: # Title = '#'*int(ll[2]) + ' ' + ll[4:] # DashTitle = DashifyTitle(Title.lstrip('#'), DashyTitles) # DashyTitles += [DashTitle] # Titles += [Title] # Content += MakeLinkableTitle(l, Title, DashTitle, 'pug') + '\n' # else: # Content += l + '\n' if Path.endswith(FileExtensions['HTML']) and not HTMLTitlesFound: Soup = BeautifulSoup(File, 'html.parser') Tags = Soup.find_all() for t in Tags: if t.name in Headings: Title = '#'*int(t.name[1]) + ' ' + str(t.text) DashTitle = DashifyTitle(Title.lstrip('#'), DashyTitles) DashyTitles += [DashTitle] Titles += [Title] t.replace_with(MakeLinkableTitle(None, Title, DashTitle, 'md')) HTMLTitlesFound = True Content = '' TmpContent = str(Soup.prettify(formatter=None)) for cl in TmpContent.splitlines(): _, _, IsMetaLine = FindPreprocLine(cl, Meta, Macros) if not IsMetaLine: #print(cl) Content += cl + '\n' break elif Path.endswith(FileExtensions['Markdown']): lsuffix = '' if ll.startswith(('-', '+', '*')): lsuffix += ll[0] ll = ll[1:].lstrip() if ll.startswith('#') or (ll.startswith('<') and ll[1:].startswith(Headings)): if ll.startswith('#'): Title = ll elif ll.startswith('<'): if ll[3:].startswith((" class='NoTitle", ' class="NoTitle')): Content += l + '\n' continue else: Title = '#'*int(ll[2]) + ' ' + ll[4:] DashTitle = DashifyTitle(MkSoup(Title.lstrip('#')).get_text(), DashyTitles) DashyTitles += [DashTitle] Titles += [Title] Title = MakeLinkableTitle(None, Title, DashTitle, 'md') # I can't remember why I put this but it was needed Title = Title.replace('> Max: Max = int(Curr) else: NoOrder += [e] New = [None] * (Max+1) for i,e in enumerate(Old): Curr = e[3]['Order'] if Curr: New[int(Curr)] = e while None in New: New.remove(None) return New + NoOrder def CanIndex(Index:str, For:str): if Index.lower() in PageIndexStrNeg: return False elif Index.lower() in PageIndexStrPos: return True else: return True if Index == For else False def PatchHTML(File, HTML, StaticPartsText, DynamicParts, DynamicPartsText, HTMLPagesList, PagePath, Content, Titles, Meta, SiteDomain, SiteRoot, SiteName, BlogName, FolderRoots, Categories, SiteLang, Locale, LightRun): HTMLTitles = FormatTitles(Titles) BodyDescription, BodyImage = '', '' if not File.lower().endswith('.txt'): Soup = MkSoup(Content) if not BodyDescription:# and Soup.p: #BodyDescription = Soup.p.get_text()[:150].replace('\n', ' ').replace('"', "'") + '...' for t in Soup.find_all('p'): if t.get_text(): BodyDescription = t.get_text()[:150].replace('\n', ' ').replace('"', "'") + '...' break if not BodyImage and Soup.img and Soup.img['src']: BodyImage = Soup.img['src'] #Content = SquareFnrefs(Content) if '', '[', ']') if any(_ in Content for _ in ('', '', '', '')): Content = DictReplWithEsc( Content, { '': '', '--->': '', '': '', '': '', '': '', '': ''}) Title = GetTitle(File.split('/')[-1], Meta, Titles, 'MetaTitle', BlogName) Description = GetDescription(Meta, BodyDescription, 'MetaDescription') Image = GetImage(Meta, BodyImage, 'MetaImage') ContentHeader = MakeContentHeader(Meta, Locale, MakeCategoryLine(File, Meta)) TimeNow = datetime.now().strftime('%Y-%m-%d %H:%M') RelativeRoot = GetPathLevels(PagePath) if 'staticoso:DynamicPart:' in HTML: # Reduce risk of unnecessary cycles for Line in HTML.splitlines(): Line = Line.lstrip().rstrip() if (Line.startswith('[staticoso:DynamicPart:') and Line.endswith(']')) or (Line.startswith('')): Path = Line[len('", Text) for i in range(2): for e in StaticPartsText: HTML = ReplWithEsc(HTML, f"[staticoso:StaticPart:{e}]", StaticPartsText[e]) HTML = ReplWithEsc(HTML, f"", StaticPartsText[e]) if LightRun: HTML = None else: HTML = WrapDictReplWithEsc(HTML, { #'[staticoso:PageHead]': Meta['Head'], #'': Meta['Head'], # #DEPRECATION # 'staticoso:Site:Menu': HTMLPagesList, 'staticoso:Page:Lang': Meta['Language'] if Meta['Language'] else SiteLang, 'staticoso:Page:Chapters': HTMLTitles, 'staticoso:Page:Title': Title, 'staticoso:Page:Description': Description, 'staticoso:Page:Image': Image, 'staticoso:Page:Path': PagePath, 'staticoso:Page:Style': Meta['Style'], ################ 'staticoso:SiteMenu': HTMLPagesList, 'staticoso:PageLang': Meta['Language'] if Meta['Language'] else SiteLang, 'staticoso:PageLanguage': Meta['Language'] if Meta['Language'] else SiteLang, 'staticoso:PageSections': HTMLTitles, 'staticoso:PageTitle': Title, 'staticoso:PageDescription': Description, 'staticoso:PageImage': Image, 'staticoso:PagePath': PagePath, 'staticoso:PageHead': Meta['Head'], 'staticoso:PageStyle': Meta['Style'], # NOTE: Content is injected in page only at this point! Keep in mind for other substitutions # #DEPRECATION # 'staticoso:Page:Content': Content, 'staticoso:Page:ContentInfo': ContentHeader, 'staticoso:Site:Name': SiteName, 'staticoso:Site:AbsoluteRoot': SiteRoot, 'staticoso:Site:RelativeRoot': RelativeRoot, ################ 'staticoso:PageContent': Content, 'staticoso:PageContentInfo': ContentHeader, 'staticoso:BuildTime': TimeNow, 'staticoso:SiteDomain': SiteDomain, 'staticoso:SiteName': SiteName, 'staticoso:SiteAbsoluteRoot': SiteRoot, 'staticoso:SiteRelativeRoot': RelativeRoot, }, InternalMacrosWraps) for e in Meta['Macros']: HTML = ReplWithEsc(HTML, f"[:{e}:]", Meta['Macros'][e]) for e in FolderRoots: HTML = WrapDictReplWithEsc(HTML, { f'staticoso:CustomPath:{e}': FolderRoots[e], f'staticoso:Folder:{e}:AbsoluteRoot': FolderRoots[e], #DEPRECATED }, InternalMacrosWraps) for e in Categories: HTML = WrapDictReplWithEsc(HTML, { f'staticoso:Category:{e}': Categories[e], f'staticoso:CategoryList:{e}': Categories[e], }, InternalMacrosWraps) HTML = ReplWithEsc(HTML, f'[staticoso:Category:{e}]', Categories[e]) #DEPRECATED # TODO: Clean this doubling? ContentHTML = Content ContentHTML = WrapDictReplWithEsc(ContentHTML, { # #DEPRECATION # '[staticoso:Page:Title]': Title, '[staticoso:Page:Description]': Description, '[staticoso:Site:Name]': SiteName, '[staticoso:Site:AbsoluteRoot]': SiteRoot, '[staticoso:Site:RelativeRoot]': RelativeRoot, ################ '': Title, '': Description, '': SiteDomain, '': SiteName, '': SiteRoot, '': RelativeRoot, }, InternalMacrosWraps) for e in Meta['Macros']: ContentHTML = ReplWithEsc(ContentHTML, f"[:{e}:]", Meta['Macros'][e]) for e in FolderRoots: ContentHTML = WrapDictReplWithEsc(ContentHTML, { f'staticoso:CustomPath:{e}': FolderRoots[e], f'staticoso:Folder:{e}:AbsoluteRoot': FolderRoots[e], #DEPRECATED }, InternalMacrosWraps) for e in Categories: ContentHTML = WrapDictReplWithEsc(ContentHTML, { f'staticoso:Category:{e}': Categories[e], f'staticoso:CategoryList:{e}': Categories[e], }, InternalMacrosWraps) ContentHTML = ReplWithEsc(ContentHTML, f'[staticoso:Category:{e}]', Categories[e]) #DEPRECATED return HTML, ContentHTML, Description, Image def HandlePage(Flags, Page, Pages, Categories, LimitFiles, Snippets, ConfMenu, Locale): File, Content, Titles, Meta = Page OutDir, MarkdownExts, Sorting, MinifyKeepComments = Flags['OutDir'], Flags['MarkdownExts'], Flags['Sorting'], Flags['MinifyKeepComments'] SiteName, BlogName, SiteTagline = Flags['SiteName'], Flags['BlogName'], Flags['SiteTagline'] SiteTemplate, SiteLang = Flags['SiteTemplate'], Flags['SiteLang'] SiteDomain, SiteRoot, FolderRoots = Flags['SiteDomain'], Flags['SiteRoot'], Flags['FolderRoots'] AutoCategories, CategoryUncategorized = Flags['CategoriesAutomatic'], Flags['CategoriesUncategorized'] ImgAltToTitle, ImgTitleToAlt = Flags['ImgAltToTitle'], Flags['ImgTitleToAlt'] DynamicParts, DynamicPartsText, StaticPartsText, TemplatesText = Flags['DynamicParts'], Snippets['DynamicParts'], Snippets['StaticParts'], Snippets['Templates'] FileLower = File.lower() PagePath = f"{OutDir}/{StripExt(File)}.html" LightRun = False if LimitFiles == False or File in LimitFiles else True if FileLower.endswith(FileExtensions['Markdown']): Content = markdown(PagePostprocessor('md', Content, Meta), extensions=MarkdownExts) elif FileLower.endswith(('.pug')): Content = PagePostprocessor('pug', ReadFile(PagePath), Meta) elif FileLower.endswith(('.txt')): Content = '

' + html.escape(Content) + '
' elif FileLower.endswith(FileExtensions['HTML']): Content = ReadFile(PagePath) if LightRun: HTMLPagesList = None else: TemplateMeta = TemplatePreprocessor(TemplatesText[Meta['Template']]) HTMLPagesList = GetHTMLPagesList( Pages=Pages, BlogName=BlogName, SiteRoot=SiteRoot, PathPrefix=GetPathLevels(File), Unite=ConfMenu, Type='Page', For='Menu', MarkdownExts=MarkdownExts, MenuStyle=TemplateMeta['MenuStyle']) HTML, ContentHTML, Description, Image = PatchHTML( File=File, HTML=TemplatesText[Meta['Template']], StaticPartsText=StaticPartsText, DynamicParts=DynamicParts, DynamicPartsText=DynamicPartsText, HTMLPagesList=HTMLPagesList, PagePath=PagePath[len(f"{OutDir}/"):], Content=Content, Titles=Titles, Meta=Meta, SiteDomain=SiteDomain, SiteRoot=SiteRoot, SiteName=SiteName, BlogName=BlogName, FolderRoots=FolderRoots, Categories=Categories, SiteLang=SiteLang, Locale=Locale, LightRun=LightRun) HTML = ReplWithEsc(HTML, f"", GetHTMLPagesList( Limit=Flags['FeedEntries'], Type='Post', Category=None if Flags['FeedCategoryFilter'] == '*' else Flags['FeedCategoryFilter'], Pages=Pages, BlogName=BlogName, SiteRoot=SiteRoot, PathPrefix=GetPathLevels(File), For='Categories', MarkdownExts=MarkdownExts, MenuStyle='Flat', ShowPaths=False)) if 'staticoso:DirectoryList:' in HTML: # Reduce risk of unnecessary cycles for Line in HTML.splitlines(): Line = Line.lstrip().rstrip() if Line.startswith(''): Path = Line[len('", DirectoryList) if Flags['MinifyOutput']: if not LightRun: HTML = DoMinifyHTML(HTML, MinifyKeepComments) ContentHTML = DoMinifyHTML(ContentHTML, MinifyKeepComments) if Flags['NoScripts'] and ('', f"""""") WriteFile(StripExt(PagePath)+'.Journal.html', HTML) return [File, Content, Titles, Meta, ContentHTML, SlimHTML, Description, Image] def MultiprocPagePreprocessor(d): PrintProcPercentDots(d['Process'], 2) return PagePreprocessor(d['Path'], d['TempPath'], d['Type'], d['Template'], d['SiteRoot'], d['GlobalMacros'], d['CategoryUncategorized'], d['LightRun']) def MultiprocHandlePage(d): PrintProcPercentDots(d['Process']) return HandlePage(d['Flags'], d['Page'], d['Pages'], d['Categories'], d['LimitFiles'], d['Snippets'], d['ConfMenu'], d['Locale']) def MakeSite(Flags, LimitFiles, Snippets, ConfMenu, GlobalMacros, Locale, Threads): PagesPaths, PostsPaths, Pages, MadePages, Categories = [], [], [], [], {} PoolSize = cpu_count() if Threads <= 0 else Threads OutDir, MarkdownExts, Sorting = Flags['OutDir'], Flags['MarkdownExts'], Flags['Sorting'] SiteName, BlogName, SiteTagline = Flags['SiteName'], Flags['BlogName'], Flags['SiteTagline'] SiteTemplate, SiteLang = Flags['SiteTemplate'], Flags['SiteLang'] SiteDomain, SiteRoot, FolderRoots = Flags['SiteDomain'], Flags['SiteRoot'], Flags['FolderRoots'] AutoCategories, CategoryUncategorized = Flags['CategoriesAutomatic'], Flags['CategoriesUncategorized'] ImgAltToTitle, ImgTitleToAlt = Flags['ImgAltToTitle'], Flags['ImgTitleToAlt'] DynamicParts, DynamicPartsText, StaticPartsText, TemplatesText = Flags['DynamicParts'], Snippets['DynamicParts'], Snippets['StaticParts'], Snippets['Templates'] for Ext in FileExtensions['Pages']: for File in Path('Pages').rglob(f"*.{Ext}"): PagesPaths += [FileToStr(File, 'Pages/')] for File in Path('Posts').rglob(f"*.{Ext}"): PostsPaths += [FileToStr(File, 'Posts/')] logging.info(f"Pages Found: {len(PagesPaths+PostsPaths)}") PagesPaths = FileNameDateSort(PagesPaths) if Sorting['Pages'] == 'Inverse': PagesPaths.reverse() PostsPaths = FileNameDateSort(PostsPaths) if Sorting['Posts'] == 'Inverse': PostsPaths.reverse() logging.info("Preprocessing Source Pages") MultiprocPages = [] for Type in ['Page', 'Post']: if Type == 'Page': Files = PagesPaths PathPrefix = '' elif Type == 'Post': Files = PostsPaths PathPrefix = 'Posts/' for i,File in enumerate(Files): TempPath = f"{PathPrefix}{File}" LightRun = False if LimitFiles == False or TempPath in LimitFiles else True MultiprocPages += [{'Process':{'Num':i, 'Count':len(Files)}, 'Path':f"{Type}s/{File}", 'TempPath':TempPath, 'Type':Type, 'Template':SiteTemplate, 'SiteRoot':SiteRoot, 'GlobalMacros':GlobalMacros, 'CategoryUncategorized':CategoryUncategorized, 'LightRun':LightRun}] os.system('printf "["') with Pool(PoolSize) as MultiprocPool: Pages = MultiprocPool.map(MultiprocPagePreprocessor, MultiprocPages) os.system('printf "]\n"') # Make newline after percentage dots for File, Content, Titles, Meta in Pages: for Cat in Meta['Categories']: Categories.update({Cat:''}) PugCompileList(OutDir, Pages, LimitFiles) if Categories: logging.info("Generating Category Lists") for Cat in Categories: for Type in ('Page', 'Post'): Categories[Cat] += GetHTMLPagesList( Pages=Pages, BlogName=BlogName, SiteRoot=SiteRoot, PathPrefix=GetPathLevels('Categories/'), Type=Type, Category=Cat, For='Categories', MarkdownExts=MarkdownExts, MenuStyle='Flat') if AutoCategories: Dir = f"{OutDir}/Categories" for Cat in Categories: Exists = False for File in Path(Dir).rglob(str(Cat)+'.*'): Exists = True break if not Exists: File = f"Categories/{Cat}.md" FilePath = f"{OutDir}/{File}" WriteFile(FilePath, CategoryPageTemplate.format(Name=Cat)) _, Content, Titles, Meta = PagePreprocessor(FilePath, FilePath, Type, SiteTemplate, SiteRoot, GlobalMacros, CategoryUncategorized, LightRun=LightRun) Pages += [[File, Content, Titles, Meta]] for i,e in enumerate(ConfMenu): for File, Content, Titles, Meta in Pages: File = StripExt(File)+'.html' if e == File: ConfMenu[i] = None logging.info("Writing Pages") MultiprocPages = [] for i,Page in enumerate(Pages): MultiprocPages += [{'Process':{'Num':i, 'Count':len(Pages)}, 'Flags':Flags, 'Page':Page, 'Pages':Pages, 'Categories':Categories, 'LimitFiles':LimitFiles, 'Snippets':Snippets, 'ConfMenu':ConfMenu, 'Locale':Locale}] os.system('printf "["') with Pool(PoolSize) as MultiprocPool: MadePages = MultiprocPool.map(MultiprocHandlePage, MultiprocPages) os.system('printf "]\n"') # Make newline after percentage dots # Do page transclusions here (?) #while True: # Operated = False # for di,Dest in enumerate(MadePages): # #print(Dest[0]) # #TempPath = f'{PathPrefix}{Dest["File"]}' # #LightRun = False if LimitFiles == False or TempPath in LimitFiles else True # #if not LightRun: # if '[staticoso:Transclude:' in Dest[4] and (LimitFiles == False or f'{PathPrefix}{Dest[0]}' in LimitFiles): # for Item in MadePages: # SrcPrefix = '' if Item[0].startswith('Posts/') else 'Pages/' # print(SrcPrefix, Item[0]) # if Item[0] != Dest[0] and f'[staticoso:Transclude:{SrcPrefix}{Item[0]}]' in Dest[4]: # MadePages[di][4] = ReplWithEsc(Dest[4], f'', Item[4]) # print(f'[staticoso:Transclude:{SrcPrefix}{Item[0]}]', Item[4]) # Operated = True # if not Operated: # break return MadePages