Make nodejs proxy in charge of redirections

Author: qboot

src/Client.js

@@ -79,9 +79,17 @@ export default class Client {
 
         response = JSON.parse(response)
 
-        return response.status_code === 410
-            ? new Response(410)
-            : new RedirectResponse(response.location, Number(response.status_code))
+        let ruleId = null
+
+        if (response.matched_rule && response.matched_rule.id) {
+            ruleId = response.matched_rule.id
+        }
+
+        if (410 === response.status_code) {
+            return new Response(410, ruleId)
+        }
+
+        return new RedirectResponse(response.location, Number(response.status_code), ruleId)
     }
 
     /**
@@ -99,6 +107,14 @@ export default class Client {
             'use_json': true,
         }
 
+        if (response instanceof RedirectResponse) {
+            context.target = response.location
+        }
+
+        if (response.ruleId) {
+            context['rule_id'] = response.ruleId
+        }
+
         try {
             return await this.request('LOG', context) == true
         } catch (error) {

src/HttpMessage/RedirectResponse.js

@@ -1,8 +1,8 @@
 import Response from './Response'
 
 export default class RedirectResponse extends Response {
-    constructor(location, statusCode = 301) {
-        super(statusCode)
+    constructor(location, statusCode = 301, ruleId = null) {
+        super(statusCode, ruleId)
         this.location = location
     }
 }

src/HttpMessage/Response.js

@@ -1,5 +1,6 @@
 export default class Response {
-    constructor(statusCode = 200) {
+    constructor(statusCode = 200, ruleId = null) {
         this.statusCode = statusCode
+        this.ruleId = ruleId
     }
 }

src/RedirectionIO.js

@@ -14,11 +14,12 @@ export default class RedirectionIO {
         }
 
         app.use(async (req, res, next) => {
-            this.attachLogEvent(req, res)
-
             const request = new Request(req.get('Host'), req.originalUrl, req.get('User-Agent'), req.get('Referer'), req.protocol)
             const response = await this.handleRequest(request)
 
+            req.rio = { request: request, response: response }
+            this.attachLogEvent(req, res)
+
             if (!response) {
                 return next()
             }
@@ -38,12 +39,13 @@ export default class RedirectionIO {
             this.connections = connections
         }
 
-        this.attachLogEvent(req, res)
-
         const scheme = req.headers['x-forwarded-proto'] ? req.headers['x-forwarded-proto'] : 'http'
         const request = new Request(req.headers['host'], req.url.split('?')[0], req.headers['user-agent'], scheme)
         const response = await this.handleRequest(request)
 
+        req.rio = { request: request, response: response }
+        this.attachLogEvent(req, res)
+
         if (!response) {
             return false
         }
@@ -87,10 +89,8 @@ export default class RedirectionIO {
      */
     static attachLogEvent(req, res) {
         req.on('end', async () => {
-            const path = req.originalUrl ? req.originalUrl : req.url.split('?')[0]
-            const scheme = req.headers['x-forwarded-proto'] ? req.headers['x-forwarded-proto'] : 'http'
-            const request = new Request(req.headers['host'], path, req.headers['user-agent'], req.headers['referer'], scheme)
-            const response = new Response(res.statusCode)
+            const request = req.rio.request
+            const response = req.rio.response || new Response(res.statusCode)
             const client = new Client(this.connections, 10, false)
 
             try {