import path from 'path';
import {readdirSync, readFileSync, writeFileSync, existsSync} from 'fs';
import '../utils/marked.min.js';
import {computeHash} from '../utils/utils.js';
import {validateBasicAuth} from "../utils/api_validate.js";
export default (fastify, options, done) => {
// 添加 / 接口
fastify.get('/', {preHandler: validateBasicAuth}, async (request, reply) => {
let readmePath = null;
const indexHtmlPath = path.join(options.rootDir, 'public/index.html');
// console.log(`indexHtmlPath:${indexHtmlPath}`);
const files = readdirSync(options.rootDir);
// console.log(files);
for (const file of files) {
if (/^readme\.md$/i.test(file)) {
readmePath = path.join(options.rootDir, file);
break;
}
}
// 如果未找到 README.md 文件
if (!readmePath && !process.env.VERCEL) {
let fileHtml = files.map(file => `${file}`).join('');
return reply.code(404).type('text/html;charset=utf-8').send(`README.md not found
`);
} else if (!readmePath && process.env.VERCEL) {
const tmpIndexHtml = readFileSync(indexHtmlPath, 'utf-8');
return reply.type('text/html;charset=utf-8').send(tmpIndexHtml);
}
// 读取 README.md 文件内容
const markdownContent = readFileSync(readmePath, 'utf-8');
// 将 Markdown 转换为 HTML
const htmlContent = marked.parse(markdownContent);
const indexHtml = `
drpyS-README
${htmlContent}
`;
const indexHtmlHash = computeHash(indexHtml);
if (!existsSync(indexHtmlPath)) {
console.log(`将readme.md 本地文件:${indexHtmlPath}`);
writeFileSync(indexHtmlPath, indexHtml, 'utf8');
} else {
const tmpIndexHtml = readFileSync(indexHtmlPath, 'utf-8');
const tmpIndexHtmlHash = computeHash(tmpIndexHtml);
if (indexHtmlHash !== tmpIndexHtmlHash) {
console.log(`readme.md发生了改变，更新本地文件:${indexHtmlPath}`);
writeFileSync(indexHtmlPath, indexHtml, 'utf8');
}
}
// 返回 HTML 内容
reply.type('text/html;charset=utf-8').send(indexHtml);
});
// 新增 /robots.txt 路由
fastify.get('/robots.txt', (request, reply) => {
reply.type('text/plain;charset=utf-8').sendFile('robots.txt', path.join(options.rootDir, 'public'));
});
// 新增 /favicon.ico 路由
fastify.get('/favicon.ico', async (request, reply) => {
try {
// 设置文件路径
const faviconPath = path.join(options.rootDir, 'public', 'favicon.ico');
// 如果文件存在，返回图片
if (existsSync(faviconPath)) {
return reply.sendFile('favicon.ico', path.join(options.rootDir, 'public')); // 直接返回图片
} else {
reply.status(404).send({error: 'Favicon not found'}); // 如果文件不存在，返回 404 错误
}
} catch (error) {
reply.status(500).send({error: 'Failed to fetch favicon', details: error.message});
}
});
done();
};