|
1 | 1 | const express = require('express') |
2 | | -const url = require('url') |
3 | | -const _ = require('lodash') |
4 | | -function updateQueryString(target, sourceUrl) { |
5 | | - return ~sourceUrl.indexOf('?') |
6 | | - ? _.assign(target, url.parse(sourceUrl, true).query) |
7 | | - : {} |
8 | | -} |
| 2 | +const rewrite = require('express-urlrewrite') |
| 3 | + |
9 | 4 | module.exports = routes => { |
10 | 5 | const router = express.Router() |
11 | 6 |
|
12 | 7 | router.get('/__rules', (req, res) => { |
13 | 8 | res.json(routes) |
14 | 9 | }) |
15 | 10 |
|
16 | | - Object.keys(routes).forEach(route => { |
17 | | - if (route.indexOf(':') !== -1) { |
18 | | - router.all(route, (req, res, next) => { |
19 | | - // Rewrite target url using params |
20 | | - let target = routes[route] |
21 | | - for (let param in req.params) { |
22 | | - target = target.replace(`:${param}`, req.params[param]) |
23 | | - } |
24 | | - req.url = target |
25 | | - req.query = updateQueryString(req.query, req.url) |
26 | | - next() |
27 | | - }) |
28 | | - } else { |
29 | | - router.all(`${route}*`, (req, res, next) => { |
30 | | - // Rewrite url by replacing prefix |
31 | | - req.url = req.url.replace(route, routes[route]) |
32 | | - req.query = updateQueryString(req.query, req.url) |
33 | | - next() |
34 | | - }) |
35 | | - } |
| 11 | + Object.keys(routes).forEach(key => { |
| 12 | + router.use(rewrite(key, routes[key])) |
36 | 13 | }) |
37 | 14 |
|
38 | 15 | return router |
|
0 commit comments