创建 HTTP 服务器实现了一个最简单的静态资源服务器,可以对代码进行写改造,增加文件夹预览功能,暴露出一些配置,变成一个可定制的静态资源服务器模块

模块化

可定制的静态资源服务器理想的使用方式应该是这样的

  1. const StaticServer = require('YOUR_STATIC_SERVER_FILE_PATH');
  2. const staticServer = new StaticServer({
  3. port: 9527,
  4. root: '/public',
  5. });
  6. staticServer.start();
  7. staticServer.close();

这样的使用方式就要求代码实现模块化,Node.js 实现一个模块非常简单

  1. const http = require('http');
  2. const fs = require('fs');
  3. const path = require('path');
  4. const mime = require('mime-types');
  5. const defaultConf = require('./config');
  6. class StaticServer {
  7. constructor(options = {}) {
  8. this.config = Object.assign(defaultConf, options);
  9. }
  10. start() {
  11. const { port, root } = this.config;
  12. this.server = http.createServer((req, res) => {
  13. const { url, method } = req;
  14. if (method !== 'GET') {
  15. res.writeHead(404, {
  16. 'content-type': 'text/html',
  17. });
  18. res.end('请使用 GET 方法访问文件!');
  19. return false;
  20. }
  21. const filePath = path.join(root, url);
  22. fs.access(filePath, fs.constants.R_OK, err => {
  23. if (err) {
  24. res.writeHead(404, {
  25. 'content-type': 'text/html',
  26. });
  27. res.end('文件不存在!');
  28. } else {
  29. res.writeHead(200, {
  30. 'content-type': mime.contentType(path.extname(url)),
  31. });
  32. fs.createReadStream(filePath).pipe(res);
  33. }
  34. });
  35. }).listen(port, () => {
  36. console.log(`Static server started at port ${port}`);
  37. });
  38. }
  39. stop() {
  40. this.server.close(() => {
  41. console.log(`Static server closed.`);
  42. });
  43. }
  44. }
  45. module.exports = StaticServer;

完整代码:https://github.com/Samaritan89/static-server/tree/v1
执行 npm run test 可以测试
image.png
image.png

支持文件夹预览

当访问的路径是文件夹的时候程序会报错

  1. Error: EISDIR: illegal operation on a directory, read
  2. Emitted 'error' event on ReadStream instance at:
  3. at internal/fs/streams.js:217:14
  4. at FSReqCallback.wrapper [as oncomplete] (fs.js:524:5) {
  5. errno: -21,
  6. code: 'EISDIR',
  7. syscall: 'read'
  8. }

因为 fs.createReadStream 尝试读取文件夹,需要兼容下访问路径是文件夹的时候,返回一个目录页,也就是在 fs.access 之后判断文件类型

  1. fs.access(filePath, fs.constants.R_OK, err => {
  2. if (err) {
  3. res.writeHead(404, {
  4. 'content-type': 'text/html',
  5. });
  6. res.end('文件不存在!');
  7. } else {
  8. const stats = fs.statSync(filePath);
  9. const list = [];
  10. if (stats.isDirectory()) {
  11. // 如果是文件夹则遍历文件夹,生成改文件夹内的文件树
  12. // 遍历文件内容,生成 html
  13. } else {
  14. res.writeHead(200, {
  15. 'content-type': mime.contentType(path.extname(url)),
  16. });
  17. fs.createReadStream(filePath).pipe(res);
  18. }
  19. }
  20. });

遍历生成 html 部分需要用到 文件夹操作 章节介绍的知识,为了方便生成 HTML,demo 使用了 Handlebar 模板引擎,主要逻辑

  1. if (stats.isDirectory()) {
  2. // 如果是文件夹则遍历文件夹,生成改文件夹内的文件树
  3. const dir = fs.opendirSync(filePath);
  4. let dirent = dir.readSync();
  5. while (dirent) {
  6. list.push({
  7. name: dirent.name,
  8. path: path.join(url, dirent.name),
  9. type: dirent.isDirectory() ? 'folder' : 'file',
  10. });
  11. dirent = dir.readSync();
  12. }
  13. dir.close();
  14. res.writeHead(200, {
  15. 'content-type': 'text/html',
  16. });
  17. // 对文件顺序重排,文件夹在文件前面,相同类型按字母排序,不区分大小写
  18. list.sort((x, y) => {
  19. if (x.type > y.type) {
  20. // 'folder' > 'file', 返回 -1,folder 在 file 之前
  21. return -1;
  22. } else if (x.type == y.type) {
  23. return compare(x.name.toLowerCase(), y.name.toLowerCase());
  24. } else {
  25. return 1;
  26. }
  27. });
  28. // 使用 handlebars 模板引擎,生成目录页面 html
  29. const html = template({ list });
  30. res.end(html);
  31. }

通过 git 代码修改记录可以清晰看到本次的变更:https://github.com/Samaritan89/static-server/commit/5565788dc317f29372f6e67e6fd55ec92323d0ea

同样在项目根目录执行 npm run test ,使用浏览器访问 127.0.0.1:9527 可以看到目录文件的展示
image.png
完整代码:https://github.com/Samaritan89/static-server/tree/v2