get custom request headers with options.getSigHeaders

Author: nleush

lib/utils.js

@@ -88,9 +88,11 @@ export function prepareRequestOptions(request_options, options) {
                 request_options.uri = request_options.url;
             }
             var uri = request_options.uri;
+            var original_uri = uri;
             delete request_options.url;
 
             var disable_language = false;
+            var request_sig = false;
 
             function setPrerender(prerender_option) {
                 if (options?.allowPrerender && CONFIG.PRERENDER_URL) {
@@ -145,6 +147,9 @@ export function prepareRequestOptions(request_options, options) {
                     if (proxy && proxy.timeout > 0) {
                         request_options.timeout = proxy.timeout > 100 ? proxy.timeout : proxy.timeout * 1000;
                     }
+                    if (proxy.sig && options.getSigHeaders) {
+                        request_sig = true;
+                    }
                 }
             }
 
@@ -174,8 +179,22 @@ export function prepareRequestOptions(request_options, options) {
             prepareEncodedUri(request_options, 'uri');
 
             setCookieFromJar(uri, request_options.headers, options?.jar)
+
+            if (request_sig) {
+                options.getSigHeaders(original_uri, function(error, headers) {
+                    if (error) {
+                        // Skip error.
+                        resolve(request_options);
+                    } else {
+                        request_options.headers = request_options.headers || {};
+                        Object.assign(request_options.headers, headers)
+                        resolve(request_options);
+                    }
+                });
+            } else {
+                resolve(request_options);
+            }
 
-            resolve(request_options);
         } catch(ex) {
             reject(ex);
         }