记一次对Koa.js middleware的源码贡献

本文主要记录笔者在使用开源Node.js web框架Koa.js过程当中遇到的一个小bug，为修复此Bug查找Koa及其middleware源码的过程，以及最终发起Pull Request并被采纳的过程。

缘起

事情的原由是这样的，在我刚入职当前公司时，因为团队组件不久，开发人员还没有配备期权，尤为缺少服务端(Java)开发人员；而刚好有一个对内视频服务的需求比较紧急，因此本人虽然是一名（资深）前端工程师，依然主动承担起了Server端开发的责任。项目最终选择FE们最爱的Node.js进行开发，web框架则选择了Koa.js。

问题

Service中有一个功能是为生成的视频提供下载功能。由于仅是内部人员下载，加上天天都要生成，因此决定直接在存储在服务器，而后提供连接供用户下载。

因而在服务端选择中间件koa-static，煎蛋设置一下缓存便可。主要代码以下：

const Koa = require('koa');
const serve = require('koa-static');

const app = new Koa();
app.use(
    serve(path.join(__dirname + '/dist'), {
        extensions: ['mp4'],
        maxage: 1000 * 60 * 60 * 24 * 100
    })
);
复制代码

网页部分提供一个下载按钮，采用a标签加download，外面套button的形式，代码以下（vue）：

<button><a :href="video.outputPath" download>下载</a></button>
复制代码

因而，功能完成，顺利上线，运营小mm们效率提高，齐声夸赞，完满解决。

本集完。

---

若是生活是童话故事，那么上面即是结局。惋惜，生活不是童话。

大概在今年（2018）2月左右，突然你们反映，下载按钮不能用了，点击后，都是直接在新的Tab页打开连接。

归因

遇到bug后，第一反应是分析，能用 -> 不能用 的过程当中，发生了什么。通过大体判断，能够得出结论是chrome自动升级后，对download的支持发生了变化。

接下来，个人第一反应是，是否是download属性没有用好呢。因而去搜了搜标准，而后尝试给赋值，结果发现同样是不行。

这个时候我突然想到，能够去看看别的网站，是否有一样的问题，以及怎么作的。

找了很久以后，发现了一个网站，视频还能够下载，因而在chrome Develop Tools的Network面板下，苦苦寻找差别。终于发现，在Response Header的Content-Type中存在差别。个人请求状况以下：

而能够下载的视频请求，内容则是：Content-Type: video/mpeg4。因而我怀疑，是否是浏览器把本身可以识别的扩展名直接打开，不能识别的则进行保存操做。那么接下来要作的事情就简单了：修改咱们的响应头。

初次尝试

对于npm安装的package，我的建议直接去npm官网搜索，通常都会提供源码地址，文档地址。

因而直接进入npm官网，搜索koa-static，进入该package主页，发现以下内容：

• setHeaders Function to set custom headers on response.

既然官方直接提供了功能，那么事情好办了，直接加上吧。

修改Server端代码以下：

app.use(
    serve(path.join(__dirname + '/dist'), {
        extensions: ['mp4'],
        maxage: 1000 * 60 * 60 * 24 * 100,
        setHeaders: function (res) {
            res.setHeader('Content-Type', 'video/mpeg4');
        }
    })
);
复制代码

欢天洗地，打开浏览器刷新重试，结果呢，无效！

深刻源码探索

柴犬屁股一沉，发现事情并不简单

文档救不了咱们，只能去看源码了。好在这些中间件通常都短小精悍而且逻辑严谨，读一读仍是颇有价值的。

对于node/js的项目，用到的package，直接打开项目目录下的node_modules找到对应目录阅读就能够了，十分方便。PS：大多数package入口在目录下的 index.js 文件。

打开node_modules/koa-static/index.js后，发现koa-static直接把传入的options原封不动传递给了koa-send：

function serve (root, opts) {
    ......
done = await send(ctx, ctx.path, opts)
复制代码

因而继续，打开node_modules/koa-send/index.js，仔细阅读代码，发现对options中的setHeaders处理以下：

// 此处为一个Assertion，若setHeaders不是函数，直接抛出错误
const setHeaders = opts.setHeaders
if (setHeaders && typeof setHeaders !== 'function') {
    throw new TypeError('option setHeaders must be function')
}
......
// 若是是函数，则将其加入到reponse header
if (setHeaders) setHeaders(ctx.res, path, stats)
复制代码

这里关于Assertion能够多说一句，断言是编程中很使用的一种技巧，不论是开发、调试过程当中快速发现错误，仍是线上的防护性编程。在《代码大全》等经典书籍中都有介绍，推荐你们阅读相关章节。

这么看没问题啊，传入的config应该都使用了啊。因而继续往下读，发现玄机：

ctx.type = type(path, encodingExt)
...

/** * File type. */
function type (file, ext) {
  return ext !== '' ? extname(basename(file, ext)) : extname(file)
}
复制代码

原来，在setHeader以后，源代码又根据文件扩展名，修改了其content-type。为了验证本身的想法，我简单修改这里的代码，进行尝试：

if (!ctx.type) ctx.type = type(path, encodingExt)
复制代码

重启服务，刷新后，发现效果以下：

果真ok了。

Pull Request

既然折腾了这么一大圈，解决了问题，因而我决定一不作二不休，直接给koa-send开源项目Pull Request，若是被采纳，还算是给开源届作了Contribution。

过程很简单，到项目主页，fork项目。到本身主页，把fork的项目checkout到本地，修改代码，commit, push。

修改的代码很简单，可是注意，这些开源项目通常会有很重视测试，因此若是有UT，必定记得添加用例。个人代码具体以下（提交内容不包含注释）：

// 删除原来代码：ctx.type = type(path, encodingExt)
if (!ctx.type) ctx.type = type(path, encodingExt)

// 添加Test Case
it('should set the Content-Type', function (done) {
    const app = new Koa()

    app.use(async (ctx) => {
      await send(ctx, '/test/fixtures/user.json')
    })

    request(app.listen())
    .get('/')
    .expect('Content-Type', /application\/json/)
    .end(done)
})
复制代码

而后到仍是到本身fork的项目中，选择第二个Tab：Pull requests，而后点击New pull request按钮，选择本身想提交的分支便可。

结论

发起请求后，项目维护者愉快的采纳了，因而我也有了对Node.js生态开源圈的第一次贡献，内心仍是很高兴的。

Pull Request的地址在这里。

这件事情也给我带来了必定的思考，整理后，结论以下：

1. 写代码，解决问题，是充满快乐的，可以给咱们带来知足感。
2. 认真调研，阅读文档，甚至深刻源码，问题总归是能够解决的。
3. 我发现这些开源项目其实都有issue，而且有些维护者也公开说了pull request is welcomed，因此有时间能够多读一些源码，找机会多作一些贡献。

以上就是此次修复bug、贡献源码的全过程以及给我带来的思考。只作了一点小小的工做，谢谢你们。