refactor: print vite version at end of runMain

Move version printing from installVitePlus to the end of runMain,
after runViteInstall completes. Print the multi-line vite --version
output directly.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: fengmk2

dist/index.mjs

@@ -1,10 +1,11 @@
-import { saveState, getState, setFailed } from "@actions/core";
+import { saveState, getState, setFailed, info, setOutput, warning } from "@actions/core";
+import { getExecOutput } from "@actions/exec";
 import { getInputs } from "./inputs.js";
 import { installVitePlus } from "./install-viteplus.js";
 import { runViteInstall } from "./run-install.js";
 import { restoreCache } from "./cache-restore.js";
 import { saveCache } from "./cache-save.js";
-import { State } from "./types.js";
+import { State, Outputs } from "./types.js";
 import type { Inputs } from "./types.js";
 
 async function runMain(inputs: Inputs): Promise<void> {
@@ -23,6 +24,26 @@ async function runMain(inputs: Inputs): Promise<void> {
   if (inputs.runInstall.length > 0) {
     await runViteInstall(inputs);
   }
+
+  // Print version info at the end
+  await printViteVersion();
+}
+
+async function printViteVersion(): Promise<void> {
+  try {
+    const result = await getExecOutput("vite", ["--version"], { silent: true });
+    const versionOutput = result.stdout.trim();
+    info(versionOutput);
+
+    // Extract global version for output (e.g., "- Global: v0.0.0" -> "0.0.0")
+    const globalMatch = versionOutput.match(/Global:\s*v?([\d.]+[^\s]*)/i);
+    const version = globalMatch?.[1] || "unknown";
+    saveState(State.InstalledVersion, version);
+    setOutput(Outputs.Version, version);
+  } catch (error) {
+    warning(`Could not get vite version: ${error}`);
+    setOutput(Outputs.Version, "unknown");
+  }
 }
 
 async function runPost(inputs: Inputs): Promise<void> {

src/index.ts

@@ -1,7 +1,7 @@
-import { info, debug, warning, saveState, setOutput, addPath } from "@actions/core";
+import { info, debug, warning, addPath } from "@actions/core";
 import { exec, getExecOutput } from "@actions/exec";
 import type { Inputs } from "./types.js";
-import { PACKAGE_NAME, GITHUB_REGISTRY, State, Outputs } from "./types.js";
+import { PACKAGE_NAME, GITHUB_REGISTRY } from "./types.js";
 
 export async function installVitePlus(inputs: Inputs): Promise<void> {
   const { version, registry, githubToken } = inputs;
@@ -52,42 +52,10 @@ export async function installVitePlus(inputs: Inputs): Promise<void> {
     throw new Error(`Failed to install ${PACKAGE_NAME}. Exit code: ${exitCode}`);
   }
 
-  // Verify installation and get version
-  const installedVersion = await getInstalledVersion();
-  info(`Successfully installed ${PACKAGE_NAME}@${installedVersion}`);
-
-  // Save state for outputs
-  saveState(State.InstalledVersion, installedVersion);
-  setOutput(Outputs.Version, installedVersion);
-
   // Ensure global bin is in PATH
   await ensureGlobalBinInPath();
 }
 
-async function getInstalledVersion(): Promise<string> {
-  try {
-    const result = await getExecOutput("vite", ["--version"], {
-      silent: true,
-    });
-    return result.stdout.trim();
-  } catch {
-    // Fallback: check npm list
-    try {
-      const result = await getExecOutput(
-        "npm",
-        ["list", "-g", PACKAGE_NAME, "--depth=0", "--json"],
-        { silent: true },
-      );
-      const data = JSON.parse(result.stdout) as {
-        dependencies?: Record<string, { version?: string }>;
-      };
-      return data.dependencies?.[PACKAGE_NAME]?.version || "unknown";
-    } catch {
-      return "unknown";
-    }
-  }
-}
-
 async function ensureGlobalBinInPath(): Promise<void> {
   try {
     // Use 'npm config get prefix' instead of deprecated 'npm bin -g'