From d77077f13dd42dfa57636c59cc4f8107f9a4eeb9 Mon Sep 17 00:00:00 2001
From: Alvie Rahman <alvierahman90@gmail.com>
Date: Sun, 15 Aug 2021 20:40:13 +0100
Subject: [PATCH] tags!

---
 notes2web.py                       | 37 +++++++++++++++++++++++++++++-
 requirements.txt                   |  2 ++
 templates/article.html             |  6 +++++
 templates/extra_index_content.html |  5 +++-
 templates/index.html               |  2 +-
 5 files changed, 49 insertions(+), 3 deletions(-)

diff --git a/notes2web.py b/notes2web.py
index c1899a5..fd5cf53 100755
--- a/notes2web.py
+++ b/notes2web.py
@@ -2,6 +2,7 @@
 
 
 from bs4 import BeautifulSoup as bs
+import frontmatter
 import magic
 import sys
 import pathlib
@@ -97,10 +98,11 @@ def main(args):
     dirs_with_index_article = []
 
     print(f"{markdown_files=}")
+    tag_dict = {}
     for filename in markdown_files:
         print(f"{filename=}")
         print(f"{os.path.basename(filename)=}")
-        html = pypandoc.convert_file(filename, 'html', extra_args=[f'--template={args.template}'])
+
         if os.path.basename(filename) in args.index_article_names:
             output_filename = os.path.join(
                     os.path.dirname(re.sub(f"^{args.notes.name}", args.output_dir.name, filename)),
@@ -109,6 +111,20 @@ def main(args):
             dirs_with_index_article.append(os.path.dirname(re.sub(f"^{args.notes.name}", args.output_dir.name, filename)))
         else:
             output_filename = os.path.splitext(re.sub(f"^{args.notes.name}", args.output_dir.name, filename))[0] + '.html'
+
+        fm = frontmatter.load(filename)
+        if isinstance(fm.get('tags'), list):
+            for tag in fm.get('tags'):
+                if tag in tag_dict.keys():
+                    tag_dict[tag].append({
+                        'path': str(pathlib.Path(*pathlib.Path(output_filename).parts[1:])),
+                        'title': fm.get('title')
+                        })
+                else:
+                    tag_dict[tag] = [ {
+                        'path': str(pathlib.Path(*pathlib.Path(output_filename).parts[1:])),
+                        'title': fm.get('title')
+                        } ]
         print(f"{output_filename=}")
 
         if update_required(filename, output_filename) or args.force:
@@ -124,6 +140,7 @@ def main(args):
         output_filename = re.sub(f"^{args.notes.name}", args.output_dir.name, filename) + '.html'
         os.makedirs(os.path.dirname(output_filename), exist_ok=True)
         html = re.sub(r'\$title\$', title, TEXT_ARTICLE_TEMPLATE_HEAD)
+        html = re.sub(r'\$h1title\$', title, html)
         html = re.sub(r'\$raw\$', os.path.basename(filename), html)
         with open(filename) as fp:
             html += fp.read()
@@ -138,6 +155,22 @@ def main(args):
         os.makedirs(os.path.dirname(output_filename), exist_ok=True)
         shutil.copyfile(filename, output_filename)
 
+    tagdir = os.path.join(args.output_dir, '.tags')
+    os.makedirs(tagdir, exist_ok=True)
+
+    for tag in tag_dict.keys():
+        html = re.sub(r'\$title\$', f'{tag}', INDEX_TEMPLATE_HEAD)
+        html = re.sub(r'\$h1title\$', f'tag: {tag}', html)
+        html = re.sub(r'\$extra_content\$', '', html)
+
+        for entry in tag_dict[tag]:
+            html += f"<div class=\"article\"><a href=\"/{entry['path']}\">{entry['title']}</a></div>"
+        html += INDEX_TEMPLATE_FOOT
+
+        with open(os.path.join(tagdir, f'{tag}.html'), 'w+') as fp:
+            fp.write(html)
+
+
 
     dirs_to_index = [args.output_dir.name] + get_dirs(args.output_dir)
     print(f"{dirs_to_index=}")
@@ -184,6 +217,7 @@ def main(args):
         indexentries.sort(key=lambda entry: entry['isdirectory'], reverse=True)
         
         html = re.sub(r'\$title\$', directory, INDEX_TEMPLATE_HEAD)
+        html = re.sub(r'\$h1title\$', directory, html)
         html = re.sub(r'\$extra_content\$',
                 EXTRA_INDEX_CONTENT if directory == os.path.commonpath(dirs_to_index) else '',
                 html
@@ -197,6 +231,7 @@ def main(args):
             fp.write(html)
 
     shutil.copyfile(args.stylesheet, os.path.join(args.output_dir.name, 'styles.css'))
+    print(tag_dict)
 
     return 0
 
diff --git a/requirements.txt b/requirements.txt
index 2b10189..3af4be9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,6 @@
 beautifulsoup4==4.9.3
 pypandoc==1.5
+python-frontmatter==1.0.0
 python-magic==0.4.24
+PyYAML==5.4.1
 soupsieve==2.2.1
diff --git a/templates/article.html b/templates/article.html
index 27f37e7..2948bf6 100644
--- a/templates/article.html
+++ b/templates/article.html
@@ -9,6 +9,12 @@
 </head>
 
 <body>
+	<p style="font-size: 0.7em"> tags:
+	$for(tags)$
+	<a href="/.tags/$it$.html">$it$</a>$sep$,
+	$endfor$
+	</p>
 	$body$
+
 	<p style="font-size: 0.7em;"> page generated by <a href="https://github.com/alvierahman90/notes2web">notes2web</a></p>
 </body>
diff --git a/templates/extra_index_content.html b/templates/extra_index_content.html
index b96810c..26e211a 100644
--- a/templates/extra_index_content.html
+++ b/templates/extra_index_content.html
@@ -1 +1,4 @@
-<p>These are my personal notes. Correctness is not guaranteed.</p>
+<p>
+These are my personal notes. Correctness is not guaranteed.
+Browse by tag <a href="/.tags">here</a>.
+</p>
diff --git a/templates/index.html b/templates/index.html
index fa29053..e6c1c29 100644
--- a/templates/index.html
+++ b/templates/index.html
@@ -6,7 +6,7 @@
 <title>$title$</title>
 </head>
 <body>
-	<h1>$title$</h1>
+	<h1>$h1title$</h1>
 $body$
 
 <p style="font-size: 0.7em;"> page generated by <a href="https://github.com/alvierahman90/notes2web">notes2web</a></p>