From 8bc7a50dfa91dd7ba6b657aca6dbc437fa6df6b7 Mon Sep 17 00:00:00 2001
From: Jacky Zhao <j.zhao2k19@gmail.com>
Date: Fri, 18 Aug 2023 04:54:42 +0000
Subject: [PATCH] format

---
 quartz/plugins/transformers/links.ts |  135 ++++++++++++++++++++++++++------------------
 1 files changed, 79 insertions(+), 56 deletions(-)

diff --git a/quartz/plugins/transformers/links.ts b/quartz/plugins/transformers/links.ts
index 4bf0e08..d867039 100644
--- a/quartz/plugins/transformers/links.ts
+++ b/quartz/plugins/transformers/links.ts
@@ -1,77 +1,100 @@
 import { QuartzTransformerPlugin } from "../types"
-import { relative, relativeToRoot, slugify } from "../../path"
+import {
+  CanonicalSlug,
+  RelativeURL,
+  TransformOptions,
+  _stripSlashes,
+  canonicalizeServer,
+  joinSegments,
+  splitAnchor,
+  transformLink,
+} from "../../util/path"
 import path from "path"
-import { visit } from 'unist-util-visit'
+import { visit } from "unist-util-visit"
 import isAbsoluteUrl from "is-absolute-url"
 
 interface Options {
   /** How to resolve Markdown paths */
-  markdownLinkResolution: 'absolute' | 'relative'
+  markdownLinkResolution: TransformOptions["strategy"]
   /** Strips folders from a link so that it looks nice */
   prettyLinks: boolean
 }
 
 const defaultOptions: Options = {
-  markdownLinkResolution: 'absolute',
-  prettyLinks: true
+  markdownLinkResolution: "absolute",
+  prettyLinks: true,
 }
 
-export const ResolveLinks: QuartzTransformerPlugin<Partial<Options> | undefined> = (userOpts) => {
+export const CrawlLinks: QuartzTransformerPlugin<Partial<Options> | undefined> = (userOpts) => {
   const opts = { ...defaultOptions, ...userOpts }
   return {
     name: "LinkProcessing",
-    markdownPlugins() {
-      return []
-    },
-    htmlPlugins() {
-      return [() => {
-        return (tree, file) => {
-          const curSlug = file.data.slug!
-          const transformLink = (target: string) => {
-            const targetSlug = slugify(decodeURI(target).trim())
-            if (opts.markdownLinkResolution === 'relative' && !path.isAbsolute(targetSlug)) {
-              return './' + relative(curSlug, targetSlug)
-            } else {
-              return './' + relativeToRoot(curSlug, targetSlug)
+    htmlPlugins(ctx) {
+      return [
+        () => {
+          return (tree, file) => {
+            const curSlug = canonicalizeServer(file.data.slug!)
+            const outgoing: Set<CanonicalSlug> = new Set()
+
+            const transformOptions: TransformOptions = {
+              strategy: opts.markdownLinkResolution,
+              allSlugs: ctx.allSlugs,
             }
+
+            visit(tree, "element", (node, _index, _parent) => {
+              // rewrite all links
+              if (
+                node.tagName === "a" &&
+                node.properties &&
+                typeof node.properties.href === "string"
+              ) {
+                let dest = node.properties.href as RelativeURL
+                node.properties.className ??= []
+                node.properties.className.push(isAbsoluteUrl(dest) ? "external" : "internal")
+
+                // don't process external links or intra-document anchors
+                if (!(isAbsoluteUrl(dest) || dest.startsWith("#"))) {
+                  dest = node.properties.href = transformLink(curSlug, dest, transformOptions)
+                  const canonicalDest = path.posix.normalize(joinSegments(curSlug, dest))
+                  const [destCanonical, _destAnchor] = splitAnchor(canonicalDest)
+                  outgoing.add(destCanonical as CanonicalSlug)
+                }
+
+                // rewrite link internals if prettylinks is on
+                if (
+                  opts.prettyLinks &&
+                  node.children.length === 1 &&
+                  node.children[0].type === "text" &&
+                  !node.children[0].value.startsWith("#")
+                ) {
+                  node.children[0].value = path.basename(node.children[0].value)
+                }
+              }
+
+              // transform all other resources that may use links
+              if (
+                ["img", "video", "audio", "iframe"].includes(node.tagName) &&
+                node.properties &&
+                typeof node.properties.src === "string"
+              ) {
+                if (!isAbsoluteUrl(node.properties.src)) {
+                  let dest = node.properties.src as RelativeURL
+                  dest = node.properties.src = transformLink(curSlug, dest, transformOptions)
+                  node.properties.src = dest
+                }
+              }
+            })
+
+            file.data.links = [...outgoing]
           }
+        },
+      ]
+    },
+  }
+}
 
-          visit(tree, 'element', (node, _index, _parent) => {
-            // rewrite all links
-            if (
-              node.tagName === 'a' &&
-              node.properties &&
-              typeof node.properties.href === 'string'
-            ) {
-              node.properties.className = isAbsoluteUrl(node.properties.href) ? "external" : "internal"
-
-              // don't process external links or intra-document anchors
-              if (!(isAbsoluteUrl(node.properties.href) || node.properties.href.startsWith("#"))) {
-                node.properties.href = transformLink(node.properties.href)
-              } else {
-
-              }
-
-              // rewrite link internals if prettylinks is on
-              if (opts.prettyLinks && node.children.length === 1 && node.children[0].type === 'text') {
-                node.children[0].value = path.basename(node.children[0].value)
-              }
-            }
-
-            // transform all images
-            if (
-              node.tagName === 'img' &&
-              node.properties &&
-              typeof node.properties.src === 'string'
-            ) {
-              if (!isAbsoluteUrl(node.properties.src)) {
-                const ext = path.extname(node.properties.src)
-                node.properties.src = transformLink(path.join("assets", node.properties.src)) + ext
-              }
-            }
-          })
-        }
-      }]
-    }
+declare module "vfile" {
+  interface DataMap {
+    links: CanonicalSlug[]
   }
 }

--
Gitblit v1.10.0