const path = require('path');
const fs = require('fs')
// 这里使用mime 这个库来判断文件的类型，从而设置响应头
const mime = require('mime');
/**
 * 获取文件名称
 * @param {*} filename 
 * @param {*} root 
 * @returns 
 */
async function getFileName(filename, root) {
  // 获取文件的全路径,node 中规定，如果是以 \ 开头，就是绝对路径是一个跟路径
  const fullName = path.resolve(root, filename.substring(1));
  // 判断当前路径是文件还是文件夹或者不存在
  try {
    const fsStat = await fs.promises.stat(fullName);
    if (fsStat.isDirectory()) {
      // 如果是目录的话，那就需要补上默认文件
      const adtFileName = path.join('/index.html');
      return await getFileName(adtFileName, root);
    } else {
      return fullName;
    }
  } catch (error) {
    return null;
  }
}
/**
 * 读取静态资源目录
 * @param {*} root 
 * @returns 
 */
module.exports = function (root) {
  return async (ctx, next) => {
    if (ctx.method !== 'GET') {
      await next();
      return;
    }
    const filename = await getFileName(ctx.path, root)
    if (!filename) {
      // 文件不存在
      await next();
      return;
    }
    // 根据文件的后缀名响应文件
    const mimeType = mime.getType(filename);
    // 返回一个文件流
    ctx.body = fs.createReadStream(filename);
    // 设置文件的类型
    ctx.type = mimeType;
    await next();
  }
}
复制代码

module.exports = async function (ctx, next) {
  // 判断当前请求是不是get请求，请求的路径中是否包含.，并且请求头的类型是html文件
  if (ctx.method === "GET" &&
    ctx.headers.accept.includes("text/html") &&
    !ctx.path.includes(".")) {
    ctx.path = '/index.html'
  }
  await next();
}
复制代码