@sugarat/theme 0.2.12 → 0.2.14

package/node.js

@@ -442,12 +442,15 @@ function patchVPThemeConfig(cfg, vpThemeConfig = {}) {
 var import_node_path3 = __toESM(require("path"));
 var import_node_child_process2 = require("child_process");
 var import_node_process2 = __toESM(require("process"));
+var import_node_fs2 = require("fs");
+var import_node_buffer = require("buffer");
 var import_vitepress_plugin_pagefind = require("vitepress-plugin-pagefind");
 var import_vitepress_plugin_rss = require("vitepress-plugin-rss");
 function getVitePlugins(cfg) {
   const plugins = [];
   const buildEndFn = [];
   plugins.push(inlineBuildEndPlugin(buildEndFn));
+  plugins.push(coverImgTransform());
   if (cfg && cfg.search !== false) {
     const ops = cfg.search instanceof Object ? cfg.search : {};
     plugins.push(
@@ -490,6 +493,47 @@ function inlineBuildEndPlugin(buildEndFn) {
     }
   };
 }
+function coverImgTransform() {
+  let blogConfig;
+  let vitepressConfig;
+  let assetsDir;
+  return {
+    name: "@sugarat/theme-plugin-cover-transform",
+    apply: "build",
+    enforce: "pre",
+    configResolved(config) {
+      vitepressConfig = config.vitepress;
+      assetsDir = vitepressConfig.assetsDir;
+      blogConfig = config.vitepress.site.themeConfig.blog;
+    },
+    async generateBundle(_, bundle) {
+      const assetsMap = Object.entries(bundle).filter(([key]) => {
+        return key.startsWith(assetsDir);
+      }).map(([_2, value]) => {
+        return value;
+      });
+      for (const page of blogConfig.pagesData) {
+        const { cover } = page.meta;
+        if (!cover?.startsWith("/")) {
+          continue;
+        }
+        try {
+          const realPath = import_node_path3.default.join(vitepressConfig.root, cover);
+          if (!(0, import_node_fs2.existsSync)(realPath)) {
+            continue;
+          }
+          const fileBuffer = (0, import_node_fs2.readFileSync)(realPath);
+          const matchAsset = assetsMap.find((v) => import_node_buffer.Buffer.compare(fileBuffer, v.source) === 0);
+          if (matchAsset) {
+            page.meta.cover = joinPath("/", matchAsset.fileName);
+          }
+        } catch (e) {
+          vitepressConfig.logger.warn(e?.message);
+        }
+      }
+    }
+  };
+}
 
 // src/node.ts
 function getThemeConfig(cfg) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sugarat/theme",
-  "version": "0.2.12",
+  "version": "0.2.14",
   "description": "简约风的 Vitepress 博客主题，sugarat vitepress blog theme",
   "author": "sugar",
   "license": "MIT",
@@ -51,6 +51,7 @@
     "@documate/vue": "^0.3.1",
     "@element-plus/icons-vue": "^2.1.0",
     "element-plus": "^2.3.4",
+    "javascript-stringify": "^2.1.0",
     "pagefind": "1.0.3",
     "sass": "^1.56.1",
     "typescript": "^4.8.2",

package/src/utils/node/vitePlugins.ts

@@ -1,13 +1,17 @@
 import path from 'node:path'
 import { execSync } from 'node:child_process'
 import process from 'node:process'
+import { existsSync, readFileSync } from 'node:fs'
+import { Buffer } from 'node:buffer'
 import type { SiteConfig } from 'vitepress'
+
 import {
   chineseSearchOptimize,
   pagefindPlugin
 } from 'vitepress-plugin-pagefind'
 import { RssPlugin } from 'vitepress-plugin-rss'
 import type { Theme } from '../../composables/config/index'
+import { joinPath } from './index'
 
 export function getVitePlugins(cfg?: Partial<Theme.BlogConfig>) {
   const plugins: any[] = []
@@ -16,7 +20,8 @@ export function getVitePlugins(cfg?: Partial<Theme.BlogConfig>) {
   const buildEndFn: any[] = []
   // 执行自定义的 buildEnd 钩子
   plugins.push(inlineBuildEndPlugin(buildEndFn))
-
+  // 处理cover image的路径（暂只支持自动识别的文章首图）
+  plugins.push(coverImgTransform())
   // 内置简化版的pagefind
   if (cfg && cfg.search !== false) {
     const ops = cfg.search instanceof Object ? cfg.search : {}
@@ -118,3 +123,49 @@ export function inlineBuildEndPlugin(buildEndFn: any[]) {
     }
   }
 }
+
+// TODO: 支持frontmatter中的相对路径图片自动处理
+export function coverImgTransform() {
+  let blogConfig: Theme.BlogConfig
+  let vitepressConfig: SiteConfig
+  let assetsDir: string
+  return {
+    name: '@sugarat/theme-plugin-cover-transform',
+    apply: 'build',
+    enforce: 'pre',
+    configResolved(config: any) {
+      vitepressConfig = config.vitepress
+      assetsDir = vitepressConfig.assetsDir
+      blogConfig = config.vitepress.site.themeConfig.blog
+    },
+    async generateBundle(_: any, bundle: Record<string, any>) {
+      const assetsMap = Object.entries(bundle).filter(([key]) => {
+        return key.startsWith(assetsDir)
+      }).map(([_, value]) => {
+        return value
+      })
+      for (const page of blogConfig.pagesData) {
+        const { cover } = page.meta
+        // 是否相对路径引用
+        if (!cover?.startsWith('/')) {
+          continue
+        }
+        try {
+          // 寻找构建后的
+          const realPath = path.join(vitepressConfig.root, cover)
+          if (!existsSync(realPath)) {
+            continue
+          }
+          const fileBuffer = readFileSync(realPath)
+          const matchAsset = assetsMap.find(v => Buffer.compare(fileBuffer, v.source) === 0)
+          if (matchAsset) {
+            page.meta.cover = joinPath('/', matchAsset.fileName)
+          }
+        }
+        catch (e: any) {
+          vitepressConfig.logger.warn(e?.message)
+        }
+      }
+    }
+  }
+}