Merge pull request #33 from OasAIStudio/fix/global-cli-entry

Fix global CLI entrypoint

Author: Octane0411

src/cli/main.ts

@@ -1,7 +1,8 @@
 #!/usr/bin/env node
 
+import { realpathSync } from "node:fs";
 import { resolve } from "node:path";
-import { pathToFileURL } from "node:url";
+import { fileURLToPath, pathToFileURL } from "node:url";
 
 import { resolveWorkflowConfig } from "../config/config-resolver.js";
 import { WORKFLOW_FILENAME } from "../config/defaults.js";
@@ -220,6 +221,23 @@ export async function main(): Promise<void> {
   process.exitCode = exitCode;
 }
 
+export function shouldRunAsCli(
+  importMetaUrl: string,
+  entryPath: string | undefined,
+): boolean {
+  if (!entryPath) {
+    return false;
+  }
+
+  try {
+    return (
+      realpathSync(fileURLToPath(importMetaUrl)) === realpathSync(entryPath)
+    );
+  } catch {
+    return importMetaUrl === pathToFileURL(entryPath).href;
+  }
+}
+
 function readValueFlag(
   argv: readonly string[],
   index: number,
@@ -268,9 +286,6 @@ function renderUsage(): string {
   ].join("\n");
 }
 
-if (
-  process.argv[1] &&
-  import.meta.url === pathToFileURL(process.argv[1]).href
-) {
+if (shouldRunAsCli(import.meta.url, process.argv[1])) {
   void main();
 }

tests/cli/main.test.ts

@@ -1,6 +1,7 @@
-import { mkdtemp, writeFile } from "node:fs/promises";
+import { mkdtemp, symlink, writeFile } from "node:fs/promises";
 import { tmpdir } from "node:os";
 import { join } from "node:path";
+import { pathToFileURL } from "node:url";
 
 import { describe, expect, it, vi } from "vitest";
 
@@ -9,6 +10,7 @@ import {
   applyCliOverrides,
   parseCliArgs,
   runCli,
+  shouldRunAsCli,
 } from "../../src/cli/main.js";
 import type { ResolvedWorkflowConfig } from "../../src/config/types.js";
 
@@ -61,6 +63,30 @@ describe("cli", () => {
     expect(runtime.logsRoot).toBe("/repo/runtime-logs");
   });
 
+  it("treats symlinked executables as the CLI entrypoint", async () => {
+    const workspace = await mkdtemp(join(tmpdir(), "symphony-task-cli-link-"));
+    const cliPath = join(workspace, "main.js");
+    const symlinkPath = join(workspace, "symphony");
+
+    await writeFile(cliPath, "#!/usr/bin/env node\n", "utf8");
+    await symlink(cliPath, symlinkPath);
+
+    expect(shouldRunAsCli(pathToFileURL(cliPath).href, symlinkPath)).toBe(true);
+  });
+
+  it("returns false when the resolved entrypoint differs from the module path", async () => {
+    const workspace = await mkdtemp(
+      join(tmpdir(), "symphony-task-cli-mismatch-"),
+    );
+    const cliPath = join(workspace, "main.js");
+    const otherPath = join(workspace, "other.js");
+
+    await writeFile(cliPath, "#!/usr/bin/env node\n", "utf8");
+    await writeFile(otherPath, "#!/usr/bin/env node\n", "utf8");
+
+    expect(shouldRunAsCli(pathToFileURL(cliPath).href, otherPath)).toBe(false);
+  });
+
   it("defaults to loading ./WORKFLOW.md from cwd when no workflow path is given", async () => {
     const workspace = await mkdtemp(join(tmpdir(), "symphony-task16-cli-"));
     const workflowPath = join(workspace, "WORKFLOW.md");