add --output-urls-to flag

livecycle · Nov 23, 2023 · 00e10e9 · 00e10e9
1 parent 2f1549f
commit 00e10e9
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 4 deletions.
diff --git a/packages/cli-common/src/lib/common-flags.ts b/packages/cli-common/src/lib/common-flags.ts
@@ -70,4 +70,8 @@ export const urlFlags = {
     default: 'browser',
     required: true,
   })(),
+  'output-urls-to': Flags.file({
+    description: 'Output URLs to file',
+    required: false,
+  }),
 } as const
diff --git a/packages/cli/src/commands/proxy/connect.ts b/packages/cli/src/commands/proxy/connect.ts
@@ -6,7 +6,7 @@ import { formatPublicKey } from '@preevy/common'
 import { spawn } from 'child_process'
 import { connectToTunnelServerSsh } from '../../tunnel-server-client'
 import ProfileCommand from '../../profile-command'
-import { filterUrls, printUrls } from '../urls'
+import { filterUrls, printUrls, writeUrlsToFile } from '../urls'
 
 // eslint-disable-next-line no-use-before-define
 export default class Connect extends ProfileCommand<typeof Connect> {
@@ -141,6 +141,8 @@ export default class Connect extends ProfileCommand<typeof Connect> {
       filters: this.config.preevyHooks.filterUrls,
     })
 
+    await writeUrlsToFile({ log: this.logger }, flags, urls)
+
     if (flags.json) {
       return urls
     }

diff --git a/packages/cli/src/commands/up.ts b/packages/cli/src/commands/up.ts
@@ -12,7 +12,7 @@ import { inspect } from 'util'
 import { editUrl, tunnelNameResolver } from '@preevy/common'
 import MachineCreationDriverCommand from '../machine-creation-driver-command'
 import { envIdFlags, urlFlags } from '../common-flags'
-import { filterUrls, printUrls } from './urls'
+import { filterUrls, printUrls, writeUrlsToFile } from './urls'
 import { connectToTunnelServerSsh } from '../tunnel-server-client'
 
 // eslint-disable-next-line no-use-before-define
@@ -175,6 +175,8 @@ export default class Up extends MachineCreationDriverCommand<typeof Up> {
       )),
     )
 
+    await writeUrlsToFile({ log: this.logger }, flags, urls)
+
     if (flags.json) {
       return urls
     }

diff --git a/packages/cli/src/commands/urls.ts b/packages/cli/src/commands/urls.ts
@@ -1,12 +1,25 @@
+import fs from 'fs'
+import yaml from 'yaml'
 import { Args, ux, Interfaces } from '@oclif/core'
-import { FlatTunnel, ProfileStore, TunnelOpts, addBaseComposeTunnelAgentService, commands, findComposeTunnelAgentUrl, findEnvId, getTunnelNamesToServicePorts, profileStore } from '@preevy/core'
-import { HooksListeners, PluginContext, tunnelServerFlags } from '@preevy/cli-common'
+import { FlatTunnel, Logger, ProfileStore, TunnelOpts, addBaseComposeTunnelAgentService, commands, findComposeTunnelAgentUrl, findEnvId, getTunnelNamesToServicePorts, profileStore } from '@preevy/core'
+import { HooksListeners, PluginContext, text, tunnelServerFlags } from '@preevy/cli-common'
 import { asyncReduce } from 'iter-tools-es'
 import { tunnelNameResolver } from '@preevy/common'
 import { connectToTunnelServerSsh } from '../tunnel-server-client'
 import ProfileCommand from '../profile-command'
 import { envIdFlags, urlFlags } from '../common-flags'
 
+export const writeUrlsToFile = async (
+  { log }: { log: Logger },
+  { 'output-urls-to': outputUrlsTo }: Interfaces.InferredFlags<Pick<typeof urlFlags, 'output-urls-to'>>,
+  urls: FlatTunnel[],
+) => {
+  if (!outputUrlsTo) return
+  const contents = /\.ya?ml$/.test(outputUrlsTo) ? yaml.stringify(urls) : JSON.stringify(urls)
+  await fs.promises.writeFile(outputUrlsTo, contents, { encoding: 'utf8' })
+  log.info(`URLs written to file ${text.code(outputUrlsTo)}`)
+}
+
 export const printUrls = (
   flatTunnels: FlatTunnel[],
   flags: Interfaces.InferredFlags<typeof ux.table.Flags> & { 'include-access-credentials': boolean },
@@ -133,6 +146,8 @@ export default class Urls extends ProfileCommand<typeof Urls> {
       filters: this.config.preevyHooks.filterUrls,
     })
 
+    await writeUrlsToFile({ log: this.logger }, flags, urls)
+
     if (flags.json) {
       return urls
     }