|
1 | 1 | var fs = require('fs') |
| 2 | +var path = require('path') |
2 | 3 | var express = require('express') |
3 | 4 | var logger = require('morgan') |
4 | 5 | var cors = require('cors') |
5 | 6 | var errorhandler = require('errorhandler') |
6 | 7 |
|
7 | | -var arr = [] |
| 8 | +module.exports = function (opts) { |
| 9 | + var userDir = path.join(process.cwd(), 'public') |
| 10 | + var defaultDir = path.join(__dirname, 'public') |
| 11 | + var staticDir = fs.existsSync(userDir) ? |
| 12 | + userDir : |
| 13 | + defaultDir |
8 | 14 |
|
9 | | -// Logger |
10 | | -arr.push(logger('dev', { |
11 | | - skip: function (req, res) { |
12 | | - return process.env.NODE_ENV === 'test' || |
13 | | - req.path === '/favicon.ico' |
14 | | - } |
15 | | -})) |
| 15 | + opts = opts || { static: staticDir } |
16 | 16 |
|
17 | | -// Enable CORS for all the requests, including static files |
18 | | -arr.push(cors({ origin: true, credentials: true })) |
| 17 | + var arr = [] |
19 | 18 |
|
20 | | -if (process.env.NODE_ENV === 'development') { |
21 | | - // only use in development |
22 | | - arr.push(errorhandler()) |
23 | | -} |
| 19 | + // Logger |
| 20 | + arr.push(logger('dev', { |
| 21 | + skip: function (req, res) { |
| 22 | + return process.env.NODE_ENV === 'test' || |
| 23 | + req.path === '/favicon.ico' |
| 24 | + } |
| 25 | + })) |
24 | 26 |
|
25 | | -// Serve static files |
26 | | -if (fs.existsSync(process.cwd() + '/public')) { |
27 | | - arr.push(express.static(process.cwd() + '/public')) |
28 | | -} else { |
29 | | - arr.push(express.static(__dirname + '/public')) |
30 | | -} |
| 27 | + // Enable CORS for all the requests, including static files |
| 28 | + arr.push(cors({ origin: true, credentials: true })) |
31 | 29 |
|
32 | | -// No cache for IE |
33 | | -// https://support.microsoft.com/en-us/kb/234067 |
34 | | -arr.push(function (req, res, next) { |
35 | | - res.header('Cache-Control', 'no-cache') |
36 | | - res.header('Pragma', 'no-cache') |
37 | | - res.header('Expires', '-1') |
38 | | - next() |
39 | | -}) |
| 30 | + if (process.env.NODE_ENV === 'development') { |
| 31 | + // only use in development |
| 32 | + arr.push(errorhandler()) |
| 33 | + } |
40 | 34 |
|
41 | | -module.exports = arr |
| 35 | + // Serve static files |
| 36 | + arr.push(express.static(opts.static)) |
| 37 | + |
| 38 | + // No cache for IE |
| 39 | + // https://support.microsoft.com/en-us/kb/234067 |
| 40 | + arr.push(function (req, res, next) { |
| 41 | + res.header('Cache-Control', 'no-cache') |
| 42 | + res.header('Pragma', 'no-cache') |
| 43 | + res.header('Expires', '-1') |
| 44 | + next() |
| 45 | + }) |
| 46 | + |
| 47 | + return arr |
| 48 | +} |
0 commit comments