1 |
|
2 | 'use strict'
|
3 |
|
4 | var ref = require('path');
|
5 | var resolve = ref.resolve;
|
6 | var dirname = ref.dirname;
|
7 | var extname = ref.extname;
|
8 | var join = ref.join;
|
9 | var debug = require('debug')('koa-views')
|
10 | var consolidate = require('consolidate')
|
11 | var ref$1 = require('mz/fs');
|
12 | var stat = ref$1.stat;
|
13 | var send = require('koa-send')
|
14 |
|
15 | module.exports = viewsMiddleware
|
16 |
|
17 | function viewsMiddleware (path, ref) {
|
18 | if ( ref === void 0 ) ref = {};
|
19 | var extension = ref.extension; if ( extension === void 0 ) extension = 'html';
|
20 | var options = ref.options; if ( options === void 0 ) options = {};
|
21 | var map = ref.map;
|
22 |
|
23 | return function views (ctx, next) {
|
24 | if (ctx.render) return next()
|
25 |
|
26 | ctx.render = function (relPath, locals) {
|
27 | if ( locals === void 0 ) locals = {};
|
28 |
|
29 | extension = (extname(relPath) || '.' + extension).slice(1)
|
30 |
|
31 | return getPaths(path, relPath, extension)
|
32 | .then(function (paths) {
|
33 | var state = Object.assign(locals, options, ctx.state || {})
|
34 | debug('render `%s` with %j', paths.rel, state)
|
35 | ctx.type = 'text/html'
|
36 |
|
37 | if (isHtml(extension) && !map) {
|
38 | return send(ctx, paths.rel, {
|
39 | root: path
|
40 | })
|
41 | } else {
|
42 | var engineName = map && map[extension]
|
43 | ? map[extension]
|
44 | : extension
|
45 |
|
46 | var render = consolidate[engineName]
|
47 |
|
48 | if (!engineName || !render) return Promise.reject(new Error(
|
49 | ("Engine not found for the \"." + extension + "\" file extension")
|
50 | ))
|
51 |
|
52 | return render(resolve(paths.abs, paths.rel), state)
|
53 | .then(function (html) {
|
54 | ctx.body = html
|
55 | })
|
56 | }
|
57 | })
|
58 | }
|
59 |
|
60 | return next()
|
61 | }
|
62 | }
|
63 |
|
64 | function getPaths(abs, rel, ext) {
|
65 | return stat(join(abs, rel)).then(function (stats) {
|
66 | if (stats.isDirectory()) {
|
67 |
|
68 | return {
|
69 | rel: join(rel, toFile('index', ext)),
|
70 | abs: join(abs, dirname(rel), rel)
|
71 | }
|
72 | }
|
73 |
|
74 |
|
75 | return { rel: rel, abs: abs }
|
76 | })
|
77 | .catch(function (e) {
|
78 |
|
79 | if (!extname(rel)) {
|
80 |
|
81 |
|
82 | return getPaths(abs, (rel + "." + ext), ext)
|
83 | }
|
84 |
|
85 | throw e
|
86 | })
|
87 | }
|
88 |
|
89 | function isHtml (ext) {
|
90 | return ext === 'html'
|
91 | }
|
92 |
|
93 | function toFile (fileName, ext) {
|
94 | return (fileName + "." + ext)
|
95 | }
|