1 |
|
2 |
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 |
|
10 | 'use strict'
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 | var debug = require('debug')('method-override')
|
17 | var methods = require('methods')
|
18 | var parseurl = require('parseurl')
|
19 | var querystring = require('querystring')
|
20 | var vary = require('vary')
|
21 |
|
22 |
|
23 |
|
24 |
|
25 |
|
26 |
|
27 |
|
28 |
|
29 |
|
30 |
|
31 |
|
32 |
|
33 |
|
34 |
|
35 |
|
36 |
|
37 |
|
38 |
|
39 |
|
40 |
|
41 |
|
42 |
|
43 |
|
44 | module.exports = function methodOverride (getter, options) {
|
45 | var opts = options || {}
|
46 |
|
47 |
|
48 | var get = typeof getter === 'function'
|
49 | ? getter
|
50 | : createGetter(getter || 'X-HTTP-Method-Override')
|
51 |
|
52 | // get allowed request methods to examine
|
53 | var methods = opts.methods === undefined
|
54 | ? ['POST']
|
55 | : opts.methods
|
56 |
|
57 | return function methodOverride (req, res, next) {
|
58 | var method
|
59 | var val
|
60 |
|
61 | req.originalMethod = req.originalMethod || req.method
|
62 |
|
63 |
|
64 | if (methods && methods.indexOf(req.originalMethod) === -1) {
|
65 | return next()
|
66 | }
|
67 |
|
68 | val = get(req, res)
|
69 | method = Array.isArray(val)
|
70 | ? val[0]
|
71 | : val
|
72 |
|
73 | // replace
|
74 | if (method !== undefined && supports(method)) {
|
75 | req.method = method.toUpperCase()
|
76 | debug('override %s as %s', req.originalMethod, req.method)
|
77 | }
|
78 |
|
79 | next()
|
80 | }
|
81 | }
|
82 |
|
83 |
|
84 |
|
85 |
|
86 |
|
87 | function createGetter (str) {
|
88 | if (str.substr(0, 2).toUpperCase() === 'X-') {
|
89 |
|
90 | return createHeaderGetter(str)
|
91 | }
|
92 |
|
93 | return createQueryGetter(str)
|
94 | }
|
95 |
|
96 |
|
97 |
|
98 |
|
99 |
|
100 | function createQueryGetter (key) {
|
101 | return function (req, res) {
|
102 | var url = parseurl(req)
|
103 | var query = querystring.parse(url.query || '')
|
104 | return query[key]
|
105 | }
|
106 | }
|
107 |
|
108 |
|
109 |
|
110 |
|
111 |
|
112 | function createHeaderGetter (str) {
|
113 | var name = str.toLowerCase()
|
114 |
|
115 | return function (req, res) {
|
116 |
|
117 | vary(res, str)
|
118 |
|
119 |
|
120 | var header = req.headers[name]
|
121 |
|
122 | if (!header) {
|
123 | return undefined
|
124 | }
|
125 |
|
126 |
|
127 | var index = header.indexOf(',')
|
128 |
|
129 |
|
130 | return index !== -1
|
131 | ? header.substr(0, index).trim()
|
132 | : header.trim()
|
133 | }
|
134 | }
|
135 |
|
136 |
|
137 |
|
138 |
|
139 |
|
140 | function supports (method) {
|
141 | return method &&
|
142 | typeof method === 'string' &&
|
143 | methods.indexOf(method.toLowerCase()) !== -1
|
144 | }
|