From 0e5735cba2f7539a05fbee78ab03b026d94bf969 Mon Sep 17 00:00:00 2001
From: Greenscreener <git@grsc.cz>
Date: Thu, 7 Mar 2024 17:58:18 +0100
Subject: [PATCH] Added FileLink element for publishing and linking local
 files.

---
 src/formatitko/elements.py         |  5 ++++-
 src/formatitko/html_generator.py   | 18 ++++++++++++++++++
 src/formatitko/images.py           |  3 +++
 src/formatitko/nop_processor.py    |  7 ++++++-
 src/formatitko/output_generator.py |  7 +++++--
 5 files changed, 36 insertions(+), 4 deletions(-)

diff --git a/src/formatitko/elements.py b/src/formatitko/elements.py
index 17840f4..0ff016e 100644
--- a/src/formatitko/elements.py
+++ b/src/formatitko/elements.py
@@ -1,4 +1,4 @@
-from panflute import Quoted
+from panflute import Quoted, Link
 
 
 from .command import Command, InlineCommand, BlockCommand, CodeCommand
@@ -14,3 +14,6 @@ class FQuoted(Quoted):
 		del kwargs["style"]
 		super().__init__(*args, **kwargs)
 
+
+class FileLink(Link):
+	pass
diff --git a/src/formatitko/html_generator.py b/src/formatitko/html_generator.py
index e2bc787..e2332c2 100644
--- a/src/formatitko/html_generator.py
+++ b/src/formatitko/html_generator.py
@@ -19,6 +19,7 @@ from .output_generator import OutputGenerator
 from .katex import KatexClient
 from .images import ImageProcessor, ImageProcessorNamespaceSearcher
 from .util import inlinify
+from .elements import FileLink
 
 
 class HTMLGenerator(OutputGenerator):
@@ -219,6 +220,23 @@ class HTMLGenerator(OutputGenerator):
 
 		self.generate(link)
 
+	def generate_FileLink(self, e: FileLink):
+		url = e.url
+
+		# The directory of the current file relative to the current working directory
+		source_dir = self.context.dir
+		# The directory of the current file relative to the md file we were called on
+		rel_dir = self.context.rel_dir
+
+		searcher = self.imageProcessor.get_searcher_by_path(url, rel_dir, source_dir)
+		url = self.imageProcessor.get_path_without_namespace(url)
+
+		url = self.imageProcessor.process_image(url, "", searcher, self.context)
+		searcher.publish_image(url)
+		url = searcher.get_web_path() + "/" + url
+
+		self.generate_Link(Link(*e.content, url=url))
+
 	def generate_InlineGroup(self, e: InlineGroup):
 		self.generate_Group(e)
 
diff --git a/src/formatitko/images.py b/src/formatitko/images.py
index c447c30..553ab18 100644
--- a/src/formatitko/images.py
+++ b/src/formatitko/images.py
@@ -173,6 +173,9 @@ class ImageProcessor:
 		if format == "jpg":
 			format = "jpeg"
 
+		if format == "":
+			format = ext
+
 		# Locate all dependencies
 		deps_full = [full_path]
 		for dep in deps:
diff --git a/src/formatitko/nop_processor.py b/src/formatitko/nop_processor.py
index 8218b2e..b200f5c 100644
--- a/src/formatitko/nop_processor.py
+++ b/src/formatitko/nop_processor.py
@@ -6,7 +6,7 @@ from panflute import MetaValue
 from typing import Union, Callable
 
 from .whitespace import NBSP
-from .elements import FQuoted
+from .elements import FQuoted, FileLink
 from .context import Group, InlineGroup, BlockGroup, Context
 from .whitespace import Whitespace
 from .command import BlockCommand, InlineCommand, CodeCommand, Command
@@ -88,6 +88,7 @@ class NOPProcessor:
 			Underline: self.transform_Underline,
 			NBSP: self.transform_NBSP,
 			FQuoted: self.transform_FQuoted,
+			FileLink: self.transform_FileLink,
 
 			InlineCommand: self.transform_InlineCommand,
 			BlockCommand: self.transform_BlockCommand,
@@ -299,6 +300,10 @@ class NOPProcessor:
 		e.content = self.transform(e.content)
 		return e
 
+	def transform_FileLink(self, e: FileLink) -> FileLink:
+		e.content = self.transform(e.content)
+		return e
+
 	def transform_Figure(self, e: Figure) -> Figure:
 		e.content = self.transform(e.content)
 		e.caption = self.transform(e.caption)
diff --git a/src/formatitko/output_generator.py b/src/formatitko/output_generator.py
index 716e52a..dd051d3 100644
--- a/src/formatitko/output_generator.py
+++ b/src/formatitko/output_generator.py
@@ -7,7 +7,7 @@ from panflute import stringify
 from typing import Union, Callable
 
 from .whitespace import NBSP
-from .elements import FQuoted
+from .elements import FQuoted, FileLink
 from .context import Group, InlineGroup, BlockGroup, Context
 
 
@@ -127,6 +127,7 @@ class OutputGenerator:
 			Underline: self.generate_Underline,
 			NBSP: self.generate_NBSP,
 			FQuoted: self.generate_FQuoted,
+			FileLink: self.generate_FileLink,
 			InlineGroup: self.generate_InlineGroup
 		}
 
@@ -367,7 +368,9 @@ class OutputGenerator:
 				self.write("\"")
 				self.generate(e.content)
 				self.write("\"")
-
+	
+	def generate_FileLink(self, e: FileLink):
+		self.generate_simple_tag(e)
 
 	# Inline Elements
 	def generate_Cite(self, e: Cite):