import path from 'path';
import {existsSync, readFileSync} from 'fs';
import {getMimeType} from '../utils/mime-type.js';
import '../utils/marked.min.js';
export default (fastify, options, done) => {
fastify.get('/docs/*', async (request, reply) => {
const fullPath = request.params['*']; // 捕获整个路径
console.log(`Request received for path: ${fullPath}`);
try {
const resolvedPath = path.resolve(options.docsDir, fullPath); // 将路径解析为绝对路径
// 确保 resolvedPath 在 docsDir 下
if (!resolvedPath.startsWith(options.docsDir)) {
reply.status(403).send(`403 Forbidden
Access to the requested file is forbidden.
`);
return;
}
fastify.log.info(`Resolved path: ${resolvedPath}`);
// 检查文件是否存在
if (!existsSync(resolvedPath)) {
reply.status(404).send(`404 Not Found
File "${fullPath}" not found in /docs.
`);
return;
}
// 获取文件扩展名
const ext = path.extname(resolvedPath).toLowerCase();
if (ext === '.md') {
// 处理 Markdown 文件
const markdownContent = readFileSync(resolvedPath, 'utf8');
const htmlContent = marked.parse(markdownContent);
reply.type('text/html').send(`
${fullPath}
${htmlContent}
`);
} else {
try {
const mimeType = getMimeType(ext);
if (mimeType.startsWith('text') || mimeType.includes('json') || mimeType.includes('javascript')) {
const fileContent = readFileSync(resolvedPath, 'utf8'); // 确保读取文本内容为 UTF-8
reply.type(mimeType).send(fileContent);
} else {
const fileContent = readFileSync(resolvedPath);
reply.type(mimeType).send(fileContent);
}
} catch (e) {
console.log(e);
}
}
} catch (error) {
reply.status(500).send(`500 Internal Server Error
Error reading or rendering file: ${error.message}
`);
}
});
done();
};