From ab665d213eb6ded201922b53df140cf29ab88f38 Mon Sep 17 00:00:00 2001
From: Joachim Basmaison <jbasmaison@dashlane.com>
Date: Wed, 23 Nov 2022 17:55:00 +0100
Subject: [PATCH] Support got agent option

---
 pkg/dohdec/lib/doh.js     | 5 +++++
 pkg/dohdec/types/doh.d.ts | 4 ++++
 2 files changed, 9 insertions(+)

diff --git a/pkg/dohdec/lib/doh.js b/pkg/dohdec/lib/doh.js
index 5c1459c..2998ccc 100644
--- a/pkg/dohdec/lib/doh.js
+++ b/pkg/dohdec/lib/doh.js
@@ -55,6 +55,8 @@ export class DNSoverHTTPS extends DNSutils {
    * @param {Writable} [opts.verboseStream=process.stderr] Where to write
    *   verbose output.
    * @param {boolean} [opts.http2=false] Use http/2 if it is available.
+   *   GOT Agent objects
+   * @param {object} [opts.agent=Agents] Pass down got agent option
    */
   constructor(opts = {}) {
     const {
@@ -69,6 +71,7 @@ export class DNSoverHTTPS extends DNSutils {
       preferPost: true,
       contentType: WF_DNS,
       http2: false,
+      agent: undefined,
       ...rest,
     }
 
@@ -132,6 +135,7 @@ export class DNSoverHTTPS extends DNSutils {
       retry: {
         limit: 0,
       },
+      agent: this.opts.agent,
     }).buffer()
     this.hexDump(2, response)
     this.verbose(1, 'RESPONSE:', () => packet.decode(response))
@@ -181,6 +185,7 @@ export class DNSoverHTTPS extends DNSutils {
         retry: {
           limit: 0,
         },
+        agent: this.opts.agent,
       }
     )
 
diff --git a/pkg/dohdec/types/doh.d.ts b/pkg/dohdec/types/doh.d.ts
index 0ea728e..b8dad2a 100644
--- a/pkg/dohdec/types/doh.d.ts
+++ b/pkg/dohdec/types/doh.d.ts
@@ -1,3 +1,5 @@
+import type {Agents} from 'got';
+
 /**
  * Options for doing DOH lookups.
  *
@@ -46,6 +48,7 @@ export class DNSoverHTTPS extends DNSutils {
         verbose?: number;
         verboseStream?: Writable;
         http2?: boolean;
+        agent?: Agents;
     });
     opts: {
         userAgent: string;
@@ -53,6 +56,7 @@ export class DNSoverHTTPS extends DNSutils {
         preferPost: boolean;
         contentType: string;
         http2: boolean;
+        agent: Agents;
     };
     hooks: {
         beforeRequest: ((options: any) => void)[];