From b33f13ccaf4ec14a94ee0ee467dda04cf4981d00 Mon Sep 17 00:00:00 2001
From: Jacky Zhao <j.zhao2k19@gmail.com>
Date: Mon, 01 Jan 2024 22:20:34 +0000
Subject: [PATCH] fix: dont show last page if folder

---
 quartz/plugins/transformers/links.ts |  198 ++++++++++++++++++++++++++++++-------------------
 1 files changed, 120 insertions(+), 78 deletions(-)

diff --git a/quartz/plugins/transformers/links.ts b/quartz/plugins/transformers/links.ts
index 3083ce7..50d2d1a 100644
--- a/quartz/plugins/transformers/links.ts
+++ b/quartz/plugins/transformers/links.ts
@@ -1,104 +1,146 @@
 import { QuartzTransformerPlugin } from "../types"
-import { relative, relativeToRoot, slugify, trimPathSuffix } from "../../path"
+import {
+  FullSlug,
+  RelativeURL,
+  SimpleSlug,
+  TransformOptions,
+  _stripSlashes,
+  simplifySlug,
+  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"
+import { Root } from "hast"
 
 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
-  indexAnchorLinks: boolean
-  indexExternalLinks: boolean
+  openLinksInNewTab: boolean
+  lazyLoad: boolean
 }
 
 const defaultOptions: Options = {
-  markdownLinkResolution: 'absolute',
+  markdownLinkResolution: "absolute",
   prettyLinks: true,
-  indexAnchorLinks: false,
-  indexExternalLinks: false,
+  openLinksInNewTab: false,
+  lazyLoad: false,
 }
 
 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: Root, file) => {
+            const curSlug = simplifySlug(file.data.slug!)
+            const outgoing: Set<SimpleSlug> = 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
+                const classes = (node.properties.className ?? []) as string[]
+                classes.push(isAbsoluteUrl(dest) ? "external" : "internal")
+
+                // Check if the link has alias text
+                if (
+                  node.children.length === 1 &&
+                  node.children[0].type === "text" &&
+                  node.children[0].value !== dest
+                ) {
+                  // Add the 'alias' class if the text content is not the same as the href
+                  classes.push("alias")
+                }
+                node.properties.className = classes
+
+                if (opts.openLinksInNewTab) {
+                  node.properties.target = "_blank"
+                }
+
+                // don't process external links or intra-document anchors
+                const isInternal = !(isAbsoluteUrl(dest) || dest.startsWith("#"))
+                if (isInternal) {
+                  dest = node.properties.href = transformLink(
+                    file.data.slug!,
+                    dest,
+                    transformOptions,
+                  )
+
+                  // url.resolve is considered legacy
+                  // WHATWG equivalent https://nodejs.dev/en/api/v18/url/#urlresolvefrom-to
+                  const url = new URL(dest, `https://base.com/${curSlug}`)
+                  const canonicalDest = url.pathname
+                  let [destCanonical, _destAnchor] = splitAnchor(canonicalDest)
+                  if (destCanonical.endsWith("/")) {
+                    destCanonical += "index"
+                  }
+
+                  // need to decodeURIComponent here as WHATWG URL percent-encodes everything
+                  const full = decodeURIComponent(_stripSlashes(destCanonical, true)) as FullSlug
+                  const simple = simplifySlug(full)
+                  outgoing.add(simple)
+                  node.properties["data-slug"] = full
+                }
+
+                // rewrite link internals if prettylinks is on
+                if (
+                  opts.prettyLinks &&
+                  isInternal &&
+                  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 (opts.lazyLoad) {
+                  node.properties.loading = "lazy"
+                }
+
+                if (!isAbsoluteUrl(node.properties.src)) {
+                  let dest = node.properties.src as RelativeURL
+                  dest = node.properties.src = transformLink(
+                    file.data.slug!,
+                    dest,
+                    transformOptions,
+                  )
+                  node.properties.src = dest
+                }
+              }
+            })
+
+            file.data.links = [...outgoing]
           }
-
-          const outgoing: Set<string> = new Set()
-          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
-              node.properties.className = isAbsoluteUrl(dest) ? "external" : "internal"
-
-
-              // don't process external links or intra-document anchors
-              if (!(isAbsoluteUrl(dest) || dest.startsWith("#"))) {
-                node.properties.href = transformLink(dest)
-              }
-              
-              dest = node.properties.href
-              if (dest.startsWith(".")) {
-                const normalizedPath = path.normalize(path.join(curSlug, dest))
-                outgoing.add(trimPathSuffix(normalizedPath))
-              } else if (dest.startsWith("#")) {
-                if (opts.indexAnchorLinks) {
-                  outgoing.add(dest)
-                }
-              } else {
-                if (opts.indexExternalLinks) {
-                  outgoing.add(dest)
-                }
-              }
-
-              // 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
-              }
-            }
-          })
-
-          file.data.links = [...outgoing]
-        }
-      }]
-    }
+        },
+      ]
+    },
   }
 }
 
-declare module 'vfile' {
+declare module "vfile" {
   interface DataMap {
-    links: string[]
+    links: SimpleSlug[]
   }
 }

--
Gitblit v1.10.0