Pass page.url to relativize_links() for consistent build and serve

Author: martinjoiner

mkdocs_snippets/plugin.py

@@ -82,7 +82,7 @@ def resolve_snippet(self, markdown: str, page: Page) -> str:
 
                 snippet_content = self.snippets[snippet_address][snippet_id]
 
-                snippet_content = relativize_links(snippet_content, page.file.src_uri, snippet_id)
+                snippet_content = relativize_links(snippet_content, page.url, snippet_id)
 
                 snippet_content = self.preserve_indentation(snippet_content, markdown)
 
@@ -119,6 +119,10 @@ def read_snippets_from_file(file_content) -> dict[str, str]:
 
 
 def relativize_links(snippet_content, current_path, snippet_id) -> str:
+    current_path = '/' + current_path.strip('/') + '/'
+    current_path_without_filename = current_path.rsplit("/", 2)[0]
+    current_path_without_filename = remove_mike_version_from_path(current_path_without_filename)
+
     matches = re.findall(markdown_link_pattern, snippet_content)
     for match in matches:
         link_text = match[0]
@@ -143,10 +147,6 @@ def relativize_links(snippet_content, current_path, snippet_id) -> str:
         link_without_filetype = link_and_filetype[0]
         filetype = link_and_filetype[1]
 
-        current_path_without_filename = current_path.rsplit("/", 2)[0]
-
-        current_path_without_filename = remove_mike_version_from_path(current_path_without_filename)
-
         relative_link = os.path.relpath(f"docs/{link_without_filetype}", f"docs/{current_path_without_filename}")
 
         relative_link += "." + filetype