fixing small bug with ui

Author: alizamili00

src/cmd/run.ts

@@ -8,7 +8,7 @@ import type { Command } from "commander";
 export async function runCommand(
     projectconfig: ProjectConfig,
     cmd: string[],
-    options?: { env?: string },
+    options?: { env?: string; unmountSpinner?: () => void },
 ): Promise<void> {
     const provider = providerRegistry.get(projectconfig.provider);
     if (!provider) {
@@ -22,6 +22,11 @@ export async function runCommand(
     const secrets = await provider.run(projectconfig, { env: options?.env });
     const env = buildEnvWithSecrets(secrets);
 
+    // Unmount spinner right after secrets are fetched, before command runs
+    if (options?.unmountSpinner) {
+        options.unmountSpinner();
+    }
+
     if (cmd.length === 0) {
         throw new Error("Command is required. Please provide a command to run.");
     }
@@ -60,8 +65,11 @@ export function registerRunCommand(program: Command) {
 
                 await RunFlow({
                     envName: opts.env,
-                    run: async () => {
-                        await runCommand(projectConfig, cmd, opts);
+                    run: async (unmountSpinner) => {
+                        await runCommand(projectConfig, cmd, {
+                            ...opts,
+                            unmountSpinner,
+                        });
                     },
                 });
             } catch (error) {

src/ui/RunFlow.tsx

@@ -3,7 +3,7 @@ import Spinner from "ink-spinner";
 
 export interface RunFlowProps {
     envName?: string;
-    run: () => Promise<void>;
+    run: (unmountSpinner: () => void) => Promise<void>;
 }
 
 function SpinnerComponent({ message }: { message: string }) {
@@ -24,17 +24,18 @@ export async function RunFlow({ envName, run }: RunFlowProps): Promise<void> {
         stdout: process.stderr,
     });
 
+    const unmountSpinner = () => spinner.unmount();
+
     try {
-        await run();
+        await run(unmountSpinner);
 
-        spinner.unmount();
         const successMessage = envName
             ? `Secrets injected successfully for environment "${envName}".\n`
             : "Secrets injected successfully.\n";
+
         process.stderr.write(successMessage);
     } catch (error) {
         spinner.unmount();
-        // Re-throw error - let command handler log it
         throw error;
     }
 }

test.ts

@@ -0,0 +1,5 @@
+for (let i = 0; i < 2; i++) {
+    //dealy of 1 sceond
+    await new Promise((resolve) => setTimeout(resolve, 1000));
+    console.log("Hello, world!");
+}