chore: expose api

libp2p · Jun 6, 2024 · 38be2ab · 38be2ab
1 parent bb7b36c
commit 38be2ab
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 28 deletions.
diff --git a/packages/upnp-nat/src/index.ts b/packages/upnp-nat/src/index.ts
@@ -35,11 +35,11 @@
  * ```
  */
 
-import { UPnPNAT, type NatAPI, type MapPortOptions } from './upnp-nat.js'
+import { UPnPNAT as UPnPNATClass, type NatAPI, type MapPortOptions } from './upnp-nat.js'
 import type { ComponentLogger, NodeInfo, PeerId } from '@libp2p/interface'
 import type { AddressManager, TransportManager } from '@libp2p/interface-internal'
 
-export type { UPnPNAT, NatAPI, MapPortOptions }
+export type { NatAPI, MapPortOptions }
 
 export interface PMPOptions {
   /**
@@ -88,8 +88,12 @@ export interface UPnPNATComponents {
   addressManager: AddressManager
 }
 
+export interface UPnPNAT {
+  client: NatAPI
+}
+
 export function uPnPNAT (init: UPnPNATInit = {}): (components: UPnPNATComponents) => UPnPNAT {
   return (components: UPnPNATComponents) => {
-    return new UPnPNAT(components, init)
+    return new UPnPNATClass(components, init)
   }
 }
diff --git a/packages/upnp-nat/src/upnp-nat.ts b/packages/upnp-nat/src/upnp-nat.ts
@@ -4,7 +4,7 @@ import { isLoopback } from '@libp2p/utils/multiaddr/is-loopback'
 import { isPrivateIp } from '@libp2p/utils/private-ip'
 import { fromNodeAddress } from '@multiformats/multiaddr'
 import { isBrowser } from 'wherearewe'
-import type { UPnPNATComponents, UPnPNATInit } from './index.js'
+import type { UPnPNATComponents, UPnPNATInit, UPnPNAT as UPnPNATInterface } from './index.js'
 import type { Logger, Startable } from '@libp2p/interface'
 
 const DEFAULT_TTL = 7200
@@ -15,7 +15,8 @@ function highPort (min = 1024, max = 65535): number {
   return Math.floor(Math.random() * (max - min + 1) + min)
 }
 
-export class UPnPNAT implements Startable {
+export class UPnPNAT implements Startable, UPnPNATInterface {
+  public client: NatAPI
   private readonly components: UPnPNATComponents
   private readonly externalAddress?: string
   private readonly localAddress?: string
@@ -24,7 +25,6 @@ export class UPnPNAT implements Startable {
   private readonly keepAlive: boolean
   private readonly gateway?: string
   private started: boolean
-  private client?: NatAPI
   private readonly log: Logger
 
   constructor (components: UPnPNATComponents, init: UPnPNATInit) {
@@ -42,6 +42,13 @@ export class UPnPNAT implements Startable {
     if (this.ttl < DEFAULT_TTL) {
       throw new CodeError(`NatManager ttl should be at least ${DEFAULT_TTL} seconds`, ERR_INVALID_PARAMETERS)
     }
+
+    this.client = upnpNat({
+      description: this.description,
+      ttl: this.ttl,
+      keepAlive: this.keepAlive,
+      gateway: this.gateway
+    })
   }
 
   isStarted (): boolean {
@@ -95,8 +102,7 @@ export class UPnPNAT implements Startable {
         continue
       }
 
-      const client = this._getClient()
-      const publicIp = this.externalAddress ?? await client.externalIp()
+      const publicIp = this.externalAddress ?? await this.client.externalIp()
       const isPrivate = isPrivateIp(publicIp)
 
       if (isPrivate === true) {
@@ -111,7 +117,7 @@ export class UPnPNAT implements Startable {
 
       this.log(`opening uPnP connection from ${publicIp}:${publicPort} to ${host}:${port}`)
 
-      await client.map({
+      await this.client.map({
         publicPort,
         localPort: port,
         localAddress: this.localAddress,
@@ -126,21 +132,6 @@ export class UPnPNAT implements Startable {
     }
   }
 
-  _getClient (): NatAPI {
-    if (this.client != null) {
-      return this.client
-    }
-
-    this.client = upnpNat({
-      description: this.description,
-      ttl: this.ttl,
-      keepAlive: this.keepAlive,
-      gateway: this.gateway
-    })
-
-    return this.client
-  }
-
   /**
    * Stops the NAT manager
    */
@@ -151,7 +142,6 @@ export class UPnPNAT implements Startable {
 
     try {
       await this.client.close()
-      this.client = undefined
     } catch (err: any) {
       this.log.error(err)
     }

diff --git a/packages/upnp-nat/test/index.spec.ts b/packages/upnp-nat/test/index.spec.ts
@@ -39,9 +39,7 @@ describe('UPnP NAT (TCP)', () => {
 
     client = stubInterface<NatAPI>()
 
-    natManager._getClient = () => {
-      return client
-    }
+    natManager.client = client
 
     teardown.push(async () => {
       await stop(natManager)