add --fetch-urls-timeout flag (#410)

livecycle · Jan 29, 2024 · 6dc4d60 · 6dc4d60
1 parent 7907599
commit 6dc4d60
Show file tree

Hide file tree

Showing 6 changed files with 13 additions and 1 deletion.
diff --git a/packages/cli-common/src/lib/common-flags/index.ts b/packages/cli-common/src/lib/common-flags/index.ts
@@ -101,4 +101,8 @@ export const urlFlags = {
     description: 'Output URLs to file',
     required: false,
   }),
+  'fetch-urls-timeout': Flags.integer({
+    summary: 'Timeout for fetching URLs request in milliseconds',
+    default: 2500,
+  }),
 } as const
diff --git a/packages/cli/src/commands/proxy/connect.ts b/packages/cli/src/commands/proxy/connect.ts
@@ -131,6 +131,7 @@ export default class Connect extends ProfileCommand<typeof Connect> {
         retries: 10,
         onFailedAttempt: e => { this.logger.debug(`Failed to query tunnels: ${inspect(e)}`) },
       },
+      fetchTimeout: flags['fetch-urls-timeout'],
     }), { text: 'Getting tunnel URLs...' })
 
     const urls = await filterUrls({

diff --git a/packages/cli/src/commands/up.ts b/packages/cli/src/commands/up.ts
@@ -219,6 +219,7 @@ export default class Up extends MachineCreationDriverCommand<typeof Up> {
         retries: 10,
         onFailedAttempt: e => { this.logger.debug(`Failed to query tunnels: ${inspect(e)}`) },
       },
+      fetchTimeout: flags['fetch-urls-timeout'],
     }), { text: 'Getting tunnel URLs...' })
 
     const urls = await filterUrls({

diff --git a/packages/cli/src/commands/urls.ts b/packages/cli/src/commands/urls.ts
@@ -133,6 +133,7 @@ export default class Urls extends ProfileCommand<typeof Urls> {
       includeAccessCredentials: flags['include-access-credentials'] && (flags['access-credentials-type'] as 'api' | 'browser'),
       showPreevyService: flags['show-preevy-service-urls'],
       retryOpts: { retries: 2 },
+      fetchTimeout: flags['fetch-urls-timeout'],
     })
 
     const urls = await filterUrls({

diff --git a/packages/core/src/commands/urls.ts b/packages/core/src/commands/urls.ts
@@ -25,13 +25,15 @@ export const urls = async ({
   retryOpts,
   showPreevyService,
   composeTunnelServiceUrl,
+  fetchTimeout,
 }: {
   serviceAndPort?: { service: string; port?: number }
   tunnelingKey: string | Buffer
   includeAccessCredentials: false | 'browser' | 'api'
   retryOpts: retry.Options
   showPreevyService: boolean
   composeTunnelServiceUrl: string
+  fetchTimeout: number
 }) => {
   const credentials = await generateBasicAuthCredentials(jwtGenerator(tunnelingKey))
 
@@ -40,6 +42,7 @@ export const urls = async ({
     retryOpts,
     credentials,
     includeAccessCredentials,
+    fetchTimeout,
   })
 
   return flattenTunnels(tunnels).filter(tunnelFilter({ serviceAndPort, showPreevyService }))

diff --git a/packages/core/src/compose-tunnel-agent-client.ts b/packages/core/src/compose-tunnel-agent-client.ts
@@ -164,16 +164,18 @@ export const queryTunnels = async ({
   composeTunnelServiceUrl,
   credentials,
   includeAccessCredentials,
+  fetchTimeout,
 }: {
   composeTunnelServiceUrl: string
   credentials: { user: string; password: string }
   retryOpts?: retry.Options
   includeAccessCredentials: false | 'browser' | 'api'
+  fetchTimeout: number
 }) => {
   const { tunnels } = await retry(async () => {
     const r = await fetch(
       `${composeTunnelServiceUrl}/tunnels`,
-      { signal: AbortSignal.timeout(2500), headers: { Authorization: `Bearer ${credentials.password}` } }
+      { signal: AbortSignal.timeout(fetchTimeout), headers: { Authorization: `Bearer ${credentials.password}` } }
     ).catch(e => { throw new Error(`Failed to connect to docker proxy at ${composeTunnelServiceUrl}: ${e}`, { cause: e }) })
     if (!r.ok) {
       throw new Error(`Failed to connect to docker proxy at ${composeTunnelServiceUrl}: ${r.status}: ${r.statusText}`)