优化 new命令,生成新页面时自动生成 4 位 slug 属性

Author: CatCodeMe

package.json

@@ -11,7 +11,7 @@
     "check": "astro check",
     "astro": "astro",
     "pure": "astro-pure",
-    "new": "astro-pure new",
+    "new": "bun packages/pure/scripts/new.mjs",
     "format": "prettier --write  '**/*.{js,jsx,ts,tsx,md,mdx,astro}'",
     "lint": "eslint --fix 'src/**/*.{js,ts,jsx,tsx,astro}'",
     "yijiansilian": "bun lint && bun sync && bun check && bun format",

packages/pure/scripts/new.mjs

@@ -20,6 +20,17 @@ import path from 'node:path'
 import minimist from './libs/minimist.cjs'
 import slugify from './libs/slugify.cjs'
 
+const ID_LENGTH = 4;
+const ID_CHARS = 'abcdefghijklmnopqrstuvwxyz0123456789';
+
+function generateShortId(length) {
+  let result = '';
+  for (let i = 0; i < length; i++) {
+    result += ID_CHARS.charAt(Math.floor(Math.random() * ID_CHARS.length));
+  }
+  return result;
+}
+
 function getDate() {
   const date = new Date()
   const year = date.getFullYear()
@@ -54,6 +65,28 @@ Example:
 `
 const TARGET_DIR = 'src/content/blog/'
 
+function getExistingSlugs() {
+  const existingSlugs = new Set();
+  const files = fs.readdirSync(TARGET_DIR);
+
+  for (const file of files) {
+    if (file.endsWith('.md') || file.endsWith('.mdx')) {
+      const filePath = path.join(TARGET_DIR, file);
+      const content = fs.readFileSync(filePath, 'utf8');
+      const frontmatterMatch = content.match(/^---\n([\s\S]*?)\n---/);
+
+      if (frontmatterMatch && frontmatterMatch[1]) {
+        const frontmatter = frontmatterMatch[1];
+        const slugMatch = frontmatter.match(/^slug:\s*(.*)$/m);
+        if (slugMatch && slugMatch[1]) {
+          existingSlugs.add(slugMatch[1].trim());
+        }
+      }
+    }
+  }
+  return existingSlugs;
+}
+
 export default function main(args) {
   const parsedArgs = minimist(args, {
     string: ['lang'],
@@ -93,19 +126,21 @@ export default function main(args) {
     process.exit(1)
   }
 
-  let content = `---
-title: ${postTitle}
-description: 'Write your description here.'
-publishDate: ${getDate()}
-`
+  let slug = generateShortId(ID_LENGTH);
+  const existingSlugs = getExistingSlugs();
+
+  while (existingSlugs.has(slug)) {
+    slug = generateShortId(ID_LENGTH);
+  }
+
+  let content = `---\ntitle: ${postTitle}\ndescription: 'Write your description here.'\npublishDate: ${getDate()}\nslug: ${slug}\n`
   content += parsedArgs.draft ? 'draft: true\n' : ''
   content += parsedArgs.lang ? `lang: ${parsedArgs.lang}\n` : ''
-  content += `tags: ['tag1', 'tag2']
----
-
-Write your content here.
+  content += `tags: ['tag1', 'tag2']\n---\n\nWrite your content here.
 `
 
   fs.writeFileSync(fullPath, content)
   console.log(`Post "${postTitle}" created at ${fullPath}`)
 }
+
+main(process.argv.slice(2));

src/data/.gitkeep public/fonts/.gitkeepsrc/data/.gitkeep renamed to public/fonts/.gitkeep

@@ -13,7 +13,7 @@ heroImage:
   src: ./attachments/bit_op.png
   color: '#188a3e'
 toc: 1
-slug: c1
+slug: test
 ---
 
 import Aside from '@/custom/components/user/Aside.astro'