feat: support multiple environments with --mode

src/commands/clear.ts

@@ -22,9 +22,9 @@ export const clear = async (args?: string[]) => {
       return;
     }
 
-    await clearAsset(file);
+    await clearAsset({fullPath: file, args});
     return;
   }
 
-  await clearServices();
+  await clearServices(args);
 };

src/commands/config.ts

@@ -1,15 +1,16 @@
 import {setConfig} from '@junobuild/admin';
 import ora from 'ora';
 import {junoConfigExist, readSatelliteConfig} from '../configs/juno.config';
+import {configEnv} from '../utils/config.utils';
 import {satelliteParameters} from '../utils/satellite.utils';
 import {init} from './init';
 
-export const config = async () => {
+export const config = async (args?: string[]) => {
   if (!(await junoConfigExist())) {
     await init();
   }
 
-  const {satelliteId, storage} = await readSatelliteConfig();
+  const {satelliteId, storage} = await readSatelliteConfig(configEnv(args));
 
   const spinner = ora(`Configuring...`).start();
 

src/commands/deploy.ts

@@ -31,6 +31,7 @@ import {assertSatelliteMemorySize} from '../services/deploy.services';
 import {links} from '../services/links.services';
 import {hasArgs} from '../utils/args.utils';
 import {gzipFiles} from '../utils/compress.utils';
+import {configEnv} from '../utils/config.utils';
 import {listSourceFiles} from '../utils/deploy.utils';
 import {satelliteParameters} from '../utils/satellite.utils';
 import {init} from './init';
@@ -49,22 +50,22 @@ export const deploy = async (args?: string[]) => {
   }
 
   if (hasArgs({args, options: ['-c', '--clear']})) {
-    await clear();
+    await clear(args);
   }
 
-  await executeDeploy();
+  await executeDeploy(args);
 
-  await links();
+  await links(args);
 };
 
-const executeDeploy = async () => {
+const executeDeploy = async (args?: string[]) => {
   const {
     satelliteId,
     source = DEPLOY_DEFAULT_SOURCE,
     ignore = DEPLOY_DEFAULT_IGNORE,
     encoding = DEPLOY_DEFAULT_ENCODING,
     gzip = DEPLOY_DEFAULT_GZIP
-  } = await readSatelliteConfig();
+  } = await readSatelliteConfig(configEnv(args));
 
   const sourceAbsolutePath = join(process.cwd(), source);
 
@@ -75,7 +76,7 @@ const executeDeploy = async () => {
     return;
   }
 
-  await assertSatelliteMemorySize();
+  await assertSatelliteMemorySize(args);
 
   const satellite = satelliteParameters(satelliteId);
 

src/commands/help.ts

@@ -33,6 +33,8 @@ Commands:
   ${cyan('whoami')}              Display the current controller.
 `;
 
+export const helpMode = `${yellow('-m, --mode')}        Set env mode. For example production, development or a custom string. Default is production.`;
+
 export const helpUpgrade = `
 ${TITLE}
 
@@ -47,6 +49,7 @@ Options:
   ${yellow('-s, --src')}             A local gzipped wasm file for the upgrade.
   ${yellow('-r, --reset')}           Reset to the initial state.
   ${yellow('-n, --nocheck')}         Skip assertions and execute upgrade without prompts.
+  ${helpMode}
   ${yellow('-h, --help')}            Output usage information.
 
 Notes:
@@ -93,6 +96,7 @@ Usage: ${green('juno')} ${cyan('open')} ${yellow('[options]')}
 Options:
   ${yellow('-b, --browser')}         A particular browser to open. supported: chrome|firefox|edge.
   ${yellow('-c, --console')}         Open satellite in the console.
+  ${helpMode}
   ${yellow('-h, --help')}            Output usage information.
 `;
 
@@ -103,6 +107,7 @@ Usage: ${green('juno')} ${cyan('deploy')} ${yellow('[options]')}
 
 Options:
   ${yellow('-c, --clear')}           Clear existing dapp files before proceeding with deployment.
+  ${helpMode}
   ${yellow('-h, --help')}            Output usage information.
 `;
 
@@ -113,6 +118,17 @@ Usage: ${green('juno')} ${cyan('clear')} ${yellow('[options]')}
 
 Options:
   ${yellow('-f, --fullPath')}        Clear a particular file of your dapp.
+  ${helpMode}
+  ${yellow('-h, --help')}            Output usage information.
+`;
+
+export const helpConfig = `
+${TITLE}
+
+Usage: ${green('juno')} ${cyan('config')} ${yellow('[options]')}
+
+Options:
+  ${helpMode}
   ${yellow('-h, --help')}            Output usage information.
 `;
 

src/commands/open.ts

@@ -2,6 +2,7 @@ import {listCustomDomains, type CustomDomain} from '@junobuild/admin';
 import prompts from 'prompts';
 import {junoConfigExist, readSatelliteConfig} from '../configs/juno.config';
 import {hasArgs, nextArg} from '../utils/args.utils';
+import {configEnv} from '../utils/config.utils';
 import {consoleUrl, defaultSatelliteDomain} from '../utils/domain.utils';
 import {consoleNoConfigFound} from '../utils/msg.utils';
 import {openUrl} from '../utils/open.utils';
@@ -16,7 +17,7 @@ export const open = async (args?: string[]) => {
 
   const browser = nextArg({args, option: '-b'}) ?? nextArg({args, option: '--browser'});
 
-  const {satelliteId} = await readSatelliteConfig();
+  const {satelliteId} = await readSatelliteConfig(configEnv(args));
   if (hasArgs({args, options: ['-c', '--console']})) {
     await openUrl({url: consoleUrl(satelliteId), browser});
     return;

src/commands/version.ts

@@ -18,13 +18,14 @@ import {githubCliLastRelease} from '../rest/github.rest';
 import type {AssetKey} from '../types/asset-key';
 import {actorParameters} from '../utils/actor.utils';
 import {toAssetKeys} from '../utils/asset-key.utils';
+import {configEnv} from '../utils/config.utils';
 import {orbiterKey, satelliteKey, satelliteParameters} from '../utils/satellite.utils';
 import {lastRelease} from '../utils/upgrade.utils';
 
-export const version = async () => {
+export const version = async (args?: string[]) => {
   await cliVersion();
   await missionControlVersion();
-  await satelliteVersion();
+  await satelliteVersion(args);
   await orbitersVersion();
 };
 
@@ -83,13 +84,13 @@ const missionControlVersion = async () => {
   });
 };
 
-const satelliteVersion = async () => {
+const satelliteVersion = async (args?: string[]) => {
   if (!(await junoConfigExist())) {
     console.log(`No ${yellow('config')} file found.`);
     return;
   }
 
-  const {satelliteId} = await readSatelliteConfig();
+  const {satelliteId} = await readSatelliteConfig(configEnv(args));
 
   const satellite = satelliteParameters(satelliteId);
 

src/commands/whoami.ts

@@ -4,14 +4,14 @@ import {green} from 'kleur';
 import {getToken, getUse, isDefaultProfile} from '../configs/cli.config';
 import {links} from '../services/links.services';
 
-export const whoami = async () => {
+export const whoami = async (args?: string[]) => {
   const {success} = info();
 
   if (!success) {
     return;
   }
 
-  await links();
+  await links(args);
 };
 
 const info = (): {success: boolean} => {

src/configs/juno.config.ts

@@ -1,4 +1,9 @@
-import type {JunoConfig, JunoConfigFnOrObject, SatelliteConfig} from '@junobuild/config';
+import type {
+  JunoConfig,
+  JunoConfigEnv,
+  JunoConfigFnOrObject,
+  SatelliteConfig
+} from '@junobuild/config';
 import {nonNullish} from '@junobuild/utils';
 import {existsSync} from 'node:fs';
 import {access, readFile, writeFile} from 'node:fs/promises';
@@ -26,8 +31,8 @@ export const saveConfig = async ({
   });
 };
 
-export const readSatelliteConfig = async (): Promise<SatelliteConfig> => {
-  const {satellite} = await readJunoConfig();
+export const readSatelliteConfig = async (env: JunoConfigEnv): Promise<SatelliteConfig> => {
+  const {satellite} = await readJunoConfig(env);
   return satellite;
 };
 
@@ -134,11 +139,11 @@ const writeJunoConfig = async ({
   }
 };
 
-const readJunoConfig = async (): Promise<JunoConfig> => {
+const readJunoConfig = async (env: JunoConfigEnv): Promise<JunoConfig> => {
   const {configPath, configType} = junoConfigFile();
 
   const config = (userConfig: JunoConfigFnOrObject): JunoConfig =>
-    typeof userConfig === 'function' ? userConfig({mode: 'production'}) : userConfig;
+    typeof userConfig === 'function' ? userConfig(env) : userConfig;
 
   switch (configType) {
     case 'ts': {

src/index.ts

@@ -8,6 +8,7 @@ import {
   help,
   helpClear,
   helpCommand,
+  helpConfig,
   helpDeploy,
   helpDev,
   helpLogin,
@@ -61,6 +62,9 @@ export const run = async () => {
       case 'clear':
         console.log(helpClear);
         break;
+      case 'config':
+        console.log(helpConfig);
+        break;
       case 'deploy':
         console.log(helpDeploy);
         break;
@@ -93,7 +97,7 @@ export const run = async () => {
       await clear(args);
       break;
     case 'version':
-      await versionCommand();
+      await versionCommand(args);
       break;
     case 'open':
       await open(args);
@@ -102,7 +106,7 @@ export const run = async () => {
       await upgrade(args);
       break;
     case 'whoami':
-      await whoami();
+      await whoami(args);
       break;
     case 'use':
       use(args);

src/services/clear.services.ts

@@ -3,10 +3,11 @@ import {deleteAsset} from '@junobuild/core-peer';
 import ora from 'ora';
 import {readSatelliteConfig} from '../configs/juno.config';
 import {DAPP_COLLECTION} from '../constants/constants';
+import {configEnv} from '../utils/config.utils';
 import {satelliteParameters} from '../utils/satellite.utils';
 
-export const clear = async () => {
-  const {satelliteId} = await readSatelliteConfig();
+export const clear = async (args?: string[]) => {
+  const {satelliteId} = await readSatelliteConfig(configEnv(args));
 
   const spinner = ora('Clearing dapp assets...').start();
 
@@ -25,8 +26,8 @@ const cleanFullPath = (fullPath: string): string => {
   return `${path.startsWith('/') ? '' : '/'}${path}`;
 };
 
-export const clearAsset = async (fullPath: string) => {
-  const {satelliteId} = await readSatelliteConfig();
+export const clearAsset = async ({fullPath, args}: {fullPath: string; args?: string[]}) => {
+  const {satelliteId} = await readSatelliteConfig(configEnv(args));
 
   const spinner = ora(`Clearing ${fullPath}...`).start();
 

src/services/deploy.services.ts

@@ -3,11 +3,12 @@ import {yellow} from 'kleur';
 import {compare} from 'semver';
 import {readSatelliteConfig} from '../configs/juno.config';
 import {MEMORY_HEAP_WARNING, MEMORY_SIZE_ENDPOINT_VERSION} from '../constants/deploy.constants';
+import {configEnv} from '../utils/config.utils';
 import {NEW_CMD_LINE, confirmAndExit} from '../utils/prompt.utils';
 import {satelliteParameters} from '../utils/satellite.utils';
 
-export const assertSatelliteMemorySize = async () => {
-  const {satelliteId, assertions} = await readSatelliteConfig();
+export const assertSatelliteMemorySize = async (args?: string[]) => {
+  const {satelliteId, assertions} = await readSatelliteConfig(configEnv(args));
 
   if (assertions?.heapMemory === false) {
     return;

src/services/links.services.ts

@@ -1,12 +1,13 @@
 import {listCustomDomains} from '@junobuild/admin';
 import {red} from 'kleur';
 import {junoConfigExist, readSatelliteConfig} from '../configs/juno.config';
+import {configEnv} from '../utils/config.utils';
 import {consoleUrl, defaultSatelliteDomain} from '../utils/domain.utils';
 import {terminalLink} from '../utils/links.utils';
 import {isProcessToken} from '../utils/process.utils';
 import {satelliteParameters} from '../utils/satellite.utils';
 
-export const links = async () => {
+export const links = async (args?: string[]) => {
   // If a developer is using a JUNO_TOKEN to execute command(s), the links will not be printed.
   // This is particularly useful for CI environment where such output is not needed and also because only ADMIN controllers can list custom domains.
   if (isProcessToken()) {
@@ -17,7 +18,7 @@ export const links = async () => {
     return;
   }
 
-  const {satelliteId} = await readSatelliteConfig();
+  const {satelliteId} = await readSatelliteConfig(configEnv(args));
 
   const defaultUrl = defaultSatelliteDomain(satelliteId);
   const adminUrl = consoleUrl(satelliteId);

src/services/upgrade/upgrade.satellite.services.ts

@@ -11,6 +11,7 @@ import {junoConfigExist, readSatelliteConfig} from '../../configs/juno.config';
 import {SATELLITE_WASM_NAME} from '../../constants/constants';
 import type {UpgradeWasm, UpgradeWasmModule} from '../../types/upgrade';
 import {hasArgs, nextArg} from '../../utils/args.utils';
+import {configEnv} from '../../utils/config.utils';
 import {consoleNoConfigFound} from '../../utils/msg.utils';
 import {NEW_CMD_LINE} from '../../utils/prompt.utils';
 import {satelliteKey, satelliteParameters} from '../../utils/satellite.utils';
@@ -29,7 +30,7 @@ export const upgradeSatellite = async (args?: string[]) => {
     return;
   }
 
-  const {satelliteId} = await readSatelliteConfig();
+  const {satelliteId} = await readSatelliteConfig(configEnv(args));
 
   console.log(
     `${NEW_CMD_LINE}Initiating upgrade for satellite ${cyan(satelliteId)}.${NEW_CMD_LINE}`

src/utils/config.utils.ts

@@ -0,0 +1,9 @@
+import type {JunoConfigEnv} from '@junobuild/config';
+import {nextArg} from './args.utils';
+
+export const configEnv = (args?: string[]): JunoConfigEnv => {
+  const mode = nextArg({args, option: '-m'}) ?? nextArg({args, option: '--mode'});
+  return {
+    mode: mode ?? 'production'
+  };
+};