check internal links for validity

1 files changed, 23 insertions, 1 deletions

diff --git a/sitegen/site.py b/sitegen/site.py
index 2220eac..1767a2a 100644
--- a/sitegen/site.py
+++ b/sitegen/site.py
@@ -90,8 +90,30 @@ def make_helpers(root, in_file, locale):
                 + str(Path(abs_file).relative_to(root / "template")).rstrip(".j2")
             )
 
-    def url(x):
+    def url(x, virtual=False):
         abs_file = Path(in_file).resolve()
+
+        def check_exists():
+            if virtual:
+                return True
+            if Path(root / "static" / x).exists():
+                return True
+            if Path(root / "template" / (x + ".j2")).exists():
+                return True
+            if Path(root / "template" / (x + ".j2")).exists():
+                return True
+            slashpos = x.find("/")
+            if slashpos < 0:
+                return False
+            rest = x[slashpos+1:]
+            # Heuristic search for translated file
+            if Path(root / "template" / (rest + ".j2")).exists():
+                return True
+            return False
+
+        if not check_exists():
+            raise FileNotFoundError("can't find " + x)
+
         url = ""
         current_location = Path(abs_file).relative_to(root / "template")
         for p in current_location.parts: