From f278892f76aafe20b1e8c538aa2dbbca6c1d9870 Mon Sep 17 00:00:00 2001
From: Derrick Hammer <derrick@derrickhammer.com>
Date: Sun, 15 Jan 2023 02:58:08 -0500
Subject: [PATCH] *Add optional options object to peer discovery

---
 src/index.ts | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index 98c6a4f..3331ddc 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -5,7 +5,10 @@ export interface Peer {
   port: number;
 }
 
-export type PeerSource = (pubkey: Buffer) => Promise<boolean | Peer>;
+export type PeerSource = (
+  pubkey: Buffer,
+  options?: any
+) => Promise<boolean | Peer>;
 
 export class PeerDiscovery {
   private _sources: Map<string, PeerSource> = new Map<string, PeerSource>();
@@ -29,13 +32,16 @@ export class PeerDiscovery {
     return true;
   }
 
-  public async discover(pubkey: string | Buffer): Promise<Peer | boolean> {
+  public async discover(
+    pubkey: string | Buffer,
+    options = {}
+  ): Promise<Peer | boolean> {
     if (!b4a.isBuffer(pubkey)) {
       pubkey = b4a.from(pubkey, "hex") as Buffer;
     }
 
     for (const source of this._sources.values()) {
-      const result = await source(pubkey);
+      const result = await source(pubkey, options);
 
       if (result) {
         return result;