1

scripts/generate-sidebar.ts

@@ -0,0 +1,171 @@
+import fs from 'fs'
+import path from 'path'
+import { fileURLToPath } from 'url'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = path.dirname(__filename)
+const docsDir = path.resolve(__dirname, '../docs')
+
+interface SidebarItem {
+  text: string
+  link: string
+}
+
+interface SidebarGroup {
+  text: string
+  items: SidebarItem[]
+}
+
+interface SidebarConfig {
+  [path: string]: SidebarGroup[]
+}
+
+function getTitleFromFile(filePath: string): string {
+  const content = fs.readFileSync(filePath, 'utf-8')
+  
+  const frontmatterMatch = content.match(/^---\n([\s\S]*?)\n---/)
+  if (frontmatterMatch) {
+    const titleMatch = frontmatterMatch[1].match(/title:\s*["']?([^"'\n]+)["']?/)
+    if (titleMatch) {
+      return titleMatch[1].trim()
+    }
+  }
+  
+  const h1Match = content.match(/^#\s+(.+)$/m)
+  if (h1Match) {
+    return h1Match[1].trim()
+  }
+  
+  return path.basename(filePath, '.md')
+}
+
+function getSidebarOrder(filePath: string, fileName: string): number {
+  if (fileName === 'index.md') {
+    return 0
+  }
+  
+  const content = fs.readFileSync(filePath, 'utf-8')
+  
+  const frontmatterMatch = content.match(/^---\n([\s\S]*?)\n---/)
+  if (frontmatterMatch) {
+    const orderMatch = frontmatterMatch[1].match(/sidebarOrder:\s*(\d+)/)
+    if (orderMatch) {
+      return parseInt(orderMatch[1], 10)
+    }
+  }
+  
+  return 999
+}
+
+function generateSidebar(): SidebarConfig {
+  const sidebar: SidebarConfig = {}
+  
+  const entries = fs.readdirSync(docsDir, { withFileTypes: true })
+  
+  for (const entry of entries) {
+    if (!entry.isDirectory() || entry.name.startsWith('.') || entry.name === '.vitepress') {
+      continue
+    }
+    
+    const dirPath = path.join(docsDir, entry.name)
+    const files = fs.readdirSync(dirPath)
+      .filter(f => f.endsWith('.md'))
+      .map(f => ({
+        name: f,
+        path: path.join(dirPath, f),
+        order: getSidebarOrder(path.join(dirPath, f), f)
+      }))
+      .sort((a, b) => a.order - b.order)
+    
+    if (files.length === 0) continue
+    
+    const items: SidebarItem[] = files.map(file => {
+      let title = getTitleFromFile(file.path)
+      if (file.name === 'index.md') {
+        title = '概述'
+      }
+      const linkPath = file.name === 'index.md' 
+        ? `/${entry.name}/` 
+        : `/${entry.name}/${file.name.replace('.md', '')}`
+      
+      return {
+        text: title,
+        link: linkPath
+      }
+    })
+    
+    const dirName = entry.name.charAt(0).toUpperCase() + entry.name.slice(1)
+    
+    sidebar[`/${entry.name}/`] = [
+      {
+        text: `${dirName} 笔记`,
+        items
+      }
+    ]
+  }
+  
+  return sidebar
+}
+
+function generateNav(sidebar: SidebarConfig) {
+  return Object.keys(sidebar).map(p => {
+    const name = p.replace(/\//g, '').charAt(0).toUpperCase() + p.replace(/\//g, '').slice(1)
+    return {
+      text: name,
+      link: p
+    }
+  })
+}
+
+function formatObject(obj: unknown, indent: number = 0): string {
+  const spaces = ' '.repeat(indent)
+  const innerSpaces = ' '.repeat(indent + 2)
+  
+  if (Array.isArray(obj)) {
+    if (obj.length === 0) return '[]'
+    const items = obj.map(item => formatObject(item, indent + 2))
+    return `[\n${items.map(i => innerSpaces + i).join(',\n')}\n${spaces}]`
+  }
+  
+  if (typeof obj === 'object' && obj !== null) {
+    const entries = Object.entries(obj)
+    if (entries.length === 0) return '{}'
+    const items = entries.map(([key, value]) => {
+      const formattedValue = formatObject(value, indent + 2)
+      return `${innerSpaces}'${key}': ${formattedValue}`
+    })
+    return `{\n${items.join(',\n')}\n${spaces}}`
+  }
+  
+  if (typeof obj === 'string') {
+    return `'${obj}'`
+  }
+  
+  return String(obj)
+}
+
+function updateConfig() {
+  const sidebar = generateSidebar()
+  const nav = generateNav(sidebar)
+  
+  const configPath = path.resolve(__dirname, '../docs/.vitepress/config.mts')
+  let config = fs.readFileSync(configPath, 'utf-8')
+  
+  const sidebarStr = formatObject(sidebar, 4)
+  const navStr = formatObject(nav, 4)
+  
+  const sidebarRegex = /sidebar:\s*\{\s*\n\s*\}/
+  const navRegex = /nav:\s*\[\s*\n\s*\]/
+  
+  config = config.replace(sidebarRegex, `sidebar: ${sidebarStr}`)
+  config = config.replace(navRegex, `nav: ${navStr}`)
+  
+  fs.writeFileSync(configPath, config)
+  
+  console.log('Sidebar config updated:')
+  console.log(JSON.stringify(sidebar, null, 2))
+  console.log('\nNav config updated:')
+  console.log(JSON.stringify(nav, null, 2))
+}
+
+updateConfig()