149 lines
4.7 KiB
TypeScript
149 lines
4.7 KiB
TypeScript
// 1. 从 `astro:content` 导入工具函数
|
||
import { defineCollection, z } from 'astro:content';
|
||
import { glob } from 'astro/loaders';
|
||
import fs from 'node:fs';
|
||
import path from 'node:path';
|
||
|
||
// 2. 定义内容结构接口
|
||
export interface ContentStructure {
|
||
articles: string[];
|
||
sections: SectionStructure[];
|
||
}
|
||
|
||
export interface SectionStructure {
|
||
name: string;
|
||
path: string;
|
||
articles: string[];
|
||
sections: SectionStructure[];
|
||
}
|
||
|
||
// 辅助函数：获取相对于content目录的路径
|
||
export function getRelativePath(fullPath: string, basePath = './src/content'): string {
|
||
// 统一路径分隔符
|
||
const normalizedPath = fullPath.replace(/\\/g, '/');
|
||
const normalizedBasePath = basePath.replace(/\\/g, '/');
|
||
|
||
// 移除基础路径
|
||
let relativePath = normalizedPath;
|
||
|
||
// 如果路径包含基础路径，则移除它
|
||
if (normalizedPath.includes(normalizedBasePath)) {
|
||
relativePath = normalizedPath.replace(normalizedBasePath, '');
|
||
}
|
||
|
||
// 移除开头的斜杠
|
||
relativePath = relativePath.startsWith('/') ? relativePath.substring(1) : relativePath;
|
||
|
||
// 如果路径以articles/开头，移除它（适配Astro内容集合）
|
||
if (relativePath.startsWith('articles/')) {
|
||
relativePath = relativePath.substring('articles/'.length);
|
||
}
|
||
|
||
return relativePath;
|
||
}
|
||
|
||
// 辅助函数：从文件路径中提取文件名（不带扩展名）
|
||
export function getBasename(filePath: string): string {
|
||
// 统一路径分隔符
|
||
const normalizedPath = filePath.replace(/\\/g, '/');
|
||
|
||
// 分割路径并获取最后一部分（文件名）
|
||
const parts = normalizedPath.split('/');
|
||
const fileName = parts[parts.length - 1];
|
||
|
||
// 移除扩展名
|
||
const basename = fileName.replace(/\.(md|mdx)$/, '');
|
||
|
||
return basename;
|
||
}
|
||
|
||
// 辅助函数：从文件路径中提取目录路径
|
||
export function getDirPath(filePath: string, basePath = './src/content'): string {
|
||
const basename = getBasename(filePath);
|
||
const relativePath = getRelativePath(filePath, basePath);
|
||
|
||
// 移除文件名部分，获取目录路径
|
||
const dirPath = relativePath.replace(`${basename}.md`, '').replace(/\/$/, '');
|
||
|
||
return dirPath;
|
||
}
|
||
|
||
// 3. 定义目录结构处理函数
|
||
function getContentStructure(contentDir = './src/content', basePath = './src/content'): ContentStructure {
|
||
// 检查目录是否存在
|
||
if (!fs.existsSync(contentDir)) {
|
||
return { articles: [], sections: [] };
|
||
}
|
||
// 获取目录下的所有文件和文件夹
|
||
const items = fs.readdirSync(contentDir, { withFileTypes: true });
|
||
|
||
// 分离文章和目录
|
||
const articles = items
|
||
.filter(item => item.isFile() && item.name.endsWith('.md'))
|
||
.map(item => {
|
||
// 生成相对于content目录的路径，用于在页面中查找文章
|
||
const fullPath = path.join(contentDir, item.name);
|
||
// 将路径转换为相对于content目录的格式
|
||
return fullPath.replace(/\\/g, '/');
|
||
});
|
||
|
||
// 获取子目录（作为章节）
|
||
const sections: SectionStructure[] = items
|
||
.filter(item => item.isDirectory())
|
||
.map(item => {
|
||
const sectionPath = path.join(contentDir, item.name);
|
||
// 递归获取子目录的结构
|
||
const sectionContent: ContentStructure = getContentStructure(sectionPath, basePath);
|
||
|
||
// 确保路径格式正确
|
||
const normalizedPath = sectionPath.replace(/\\/g, '/');
|
||
|
||
return {
|
||
name: item.name,
|
||
path: normalizedPath,
|
||
articles: sectionContent.articles,
|
||
sections: sectionContent.sections
|
||
};
|
||
});
|
||
|
||
return { articles, sections };
|
||
}
|
||
|
||
// 4. 定义你的集合
|
||
const articles = defineCollection({
|
||
// 使用glob加载器从content目录加载所有markdown文件
|
||
loader: glob({ pattern: "**/*.md", base: "./src/content" }),
|
||
schema: z.object({
|
||
title: z.string(),
|
||
date: z.date(),
|
||
tags: z.array(z.string()).optional(),
|
||
summary: z.string().optional(),
|
||
image: z.string().optional(),
|
||
author: z.string().optional(),
|
||
draft: z.boolean().optional().default(false),
|
||
// 添加section字段，用于标识文章所属的目录
|
||
section: z.string().optional(),
|
||
// 添加weight字段，用于排序
|
||
weight: z.number().optional(),
|
||
}),
|
||
});
|
||
|
||
// 5. 定义目录集合
|
||
const sections = defineCollection({
|
||
type: 'data',
|
||
schema: z.object({
|
||
name: z.string(),
|
||
title: z.string().optional(),
|
||
description: z.string().optional(),
|
||
weight: z.number().optional(),
|
||
articles: z.array(z.string()).optional(),
|
||
subsections: z.array(z.string()).optional(),
|
||
}),
|
||
});
|
||
|
||
// 6. 导出一个 `collections` 对象来注册你的集合
|
||
export const collections = { articles, sections };
|
||
|
||
// 7. 导出内容结构，可以在构建时使用
|
||
export const contentStructure = getContentStructure();
|