movehat 0.1.9 → 0.2.1

package/src/commands/update.ts

@@ -1,4 +1,3 @@
-import { spawn } from "child_process";
 import { readFileSync, existsSync } from "fs";
 import { join, dirname, resolve } from "path";
 import { fileURLToPath } from "url";
@@ -7,6 +6,7 @@ import prompts from "prompts";
 import { isNewerVersion } from "../helpers/semver-utils.js";
 import { fetchLatestVersion } from "../helpers/npm-registry.js";
 import { logger, withSpinner, box, colors } from "../ui/index.js";
+import { runCli } from "../utils/runCli.js";
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = dirname(__filename);
@@ -159,35 +159,38 @@ export default async function updateCommand() {
         break;
     }
 
-    // Execute update
-    // Use home directory as cwd to avoid packageManager conflicts from local package.json
-    const child = spawn(packageManager, updateArgs, {
-      stdio: "inherit",
-      cwd: homedir() || process.cwd(),
-    });
-
-    child.on("exit", (code) => {
-      if (code === 0) {
+    // Execute update.
+    // Use home directory as cwd to avoid packageManager conflicts from local package.json.
+    try {
+      const result = await runCli(
+        {
+          command: packageManager,
+          args: updateArgs,
+          cwd: homedir() || process.cwd(),
+          inheritStdio: true,
+        },
+        { throwOnNonZeroExit: false }
+      );
+
+      if (result.exitCode === 0) {
         logger.newline();
         logger.success(`Successfully updated to version ${latestVersion}!`);
         logger.newline();
         process.exit(0);
       } else {
         logger.newline();
-        logger.error('Update failed');
+        logger.error("Update failed");
         logger.plain(`   Try manually: ${packageManager} ${updateArgs.join(" ")}`);
         logger.newline();
         process.exit(1);
       }
-    });
-
-    child.on("error", (error) => {
+    } catch (error) {
       logger.newline();
-      logger.error(`Failed to update: ${error.message}`);
+      logger.error(`Failed to update: ${(error as Error).message}`);
       logger.plain(`   Try manually: ${packageManager} ${updateArgs.join(" ")}`);
       logger.newline();
       process.exit(1);
-    });
+    }
   } catch (error) {
     const message = error instanceof Error ? error.message : String(error);
     logger.newline();

package/src/core/AccountManager.ts

@@ -2,23 +2,17 @@ import {
   Account,
   Ed25519PrivateKey,
 } from "@aptos-labs/ts-sdk";
-import { existsSync, mkdirSync, readFileSync, writeFileSync } from "fs";
+import { chmodSync, existsSync, mkdirSync, readFileSync, writeFileSync } from "fs";
 import { join } from "path";
 import { MovehatConfig } from "../types/config.js";
 
-/**
- * Represents a stored account in the pool
- */
 export interface StoredAccount {
-  label?: string;
+  label?: string | undefined;
   privateKey: string;
   address: string;
   createdAt: number;
 }
 
-/**
- * Account pool storage structure
- */
 interface AccountPool {
   accounts: StoredAccount[];
   labelMap: Record<string, string>; // label → address
@@ -49,7 +43,6 @@ export class AccountManager {
    * const randomAccount = AccountManager.getTestAccount();
    */
   static getTestAccount(label?: string): Account {
-    // If label provided, try to get existing labeled account
     if (label) {
       const existingAccount = this.getAccountByLabel(label);
       if (existingAccount) {
@@ -57,7 +50,6 @@ export class AccountManager {
       }
     }
 
-    // Create new account with optional label
     return this.createAccount(label, false);
   }
 
@@ -76,13 +68,9 @@ export class AccountManager {
     const account = Account.generate();
     const address = account.accountAddress.toString();
 
-    // Add to pool
     this.pool.set(address, account);
-
-    // Store private key
     this.privateKeys.set(address, account.privateKey.toString());
 
-    // If label provided, add to label map
     if (label) {
       this.labelMap.set(label, address);
     }
@@ -173,9 +161,10 @@ export class AccountManager {
       try {
         const account = this.loadAccountFromPrivateKey(privateKeyHex);
         accounts.push(account);
-      } catch (error: any) {
+      } catch (error) {
+        const msg = error instanceof Error ? error.message : String(error);
         console.warn(
-          `Warning: Failed to load account from config: ${error.message}`
+          `Warning: Failed to load account from config: ${msg}`
         );
       }
     }
@@ -216,10 +205,14 @@ export class AccountManager {
   static saveAccountPool(poolPath?: string): void {
     const basePath = poolPath || this.defaultPoolPath;
 
-    // Ensure directory exists
+    // Ensure directory exists with restrictive perms (the pool file holds
+    // plaintext private keys, so the directory must not be world-readable).
+    // Note: mkdirSync's mode is masked by the process umask, so we chmod
+    // explicitly afterwards to guarantee 0o700 regardless of umask.
     if (!existsSync(basePath)) {
-      mkdirSync(basePath, { recursive: true });
+      mkdirSync(basePath, { recursive: true, mode: 0o700 });
     }
+    chmodSync(basePath, 0o700);
 
     // Build stored accounts array
     const storedAccounts: StoredAccount[] = [];
@@ -255,9 +248,16 @@ export class AccountManager {
       labelMap: labelMapObject,
     };
 
-    // Write to file
+    // Write to file with owner-only permissions — file contains plaintext
+    // private keys for test accounts. writeFileSync's mode is masked by
+    // the process umask, so we chmod explicitly afterwards to guarantee
+    // 0o600 regardless of umask.
     const poolFilePath = join(basePath, "test-pool.json");
-    writeFileSync(poolFilePath, JSON.stringify(poolData, null, 2), "utf-8");
+    writeFileSync(poolFilePath, JSON.stringify(poolData, null, 2), {
+      encoding: "utf-8",
+      mode: 0o600,
+    });
+    chmodSync(poolFilePath, 0o600);
   }
 
   /**
@@ -311,9 +311,10 @@ export class AccountManager {
 
       this.poolLoaded = true;
       return true;
-    } catch (error: any) {
+    } catch (error) {
+      const msg = error instanceof Error ? error.message : String(error);
       console.warn(
-        `Warning: Failed to load account pool: ${error.message}`
+        `Warning: Failed to load account pool: ${msg}`
       );
       return false;
     }

package/src/core/Publisher.ts

@@ -0,0 +1,314 @@
+import { homedir } from "os";
+import { join } from "path";
+import { randomUUID } from "crypto";
+import { Account } from "@aptos-labs/ts-sdk";
+import { MovehatConfig } from "../types/config.js";
+import { extractNamedAddresses } from "../commands/compile.js";
+import {
+  saveDeployment,
+  loadDeployment,
+  DeploymentInfo,
+  validateSafeName,
+} from "./deployments.js";
+import { validatePathSafety, validateProfileSafety } from "./shell.js";
+import { CliExecutionError, ModuleAlreadyDeployedError, PostPublishError } from "../errors.js";
+import { runCli } from "../utils/runCli.js";
+import { logger } from "../ui/index.js";
+import type { ChildProcessAdapter } from "../utils/childProcessAdapter.js";
+import {
+  withYamlLock,
+  addProfile,
+  removeProfile,
+  removeProfileSync,
+  ensureSignalHandler,
+  cleanupCallbacks,
+} from "./movementProfile.js";
+import { parseTxHash } from "../utils/parseCliOutput.js";
+
+/** @internal */
+export interface PublisherDeps {
+  adapter?: ChildProcessAdapter | undefined;
+}
+
+/** @internal */
+export interface PublishInput {
+  moduleName: string;
+  config: MovehatConfig;
+  account: Account;
+  packageDir?: string | undefined;
+}
+
+/**
+ * Publishes a Move module via the Movement CLI.
+ *
+ * @internal
+ */
+export class Publisher {
+  constructor(private readonly deps: PublisherDeps = {}) {}
+
+  async deploy(input: PublishInput): Promise<DeploymentInfo> {
+    const { moduleName, config, account } = input;
+
+    validateSafeName(moduleName, "module");
+
+    const forceRedeploy = process.env.MH_CLI_REDEPLOY === "true";
+
+    const existingDeployment = loadDeployment(config.network, moduleName);
+    if (existingDeployment && !forceRedeploy) {
+      // Build detailed error message with all deployment info
+      const errorDetails = [
+        `Module "${moduleName}" is already deployed on ${config.network}`,
+        `Address: ${existingDeployment.address}`,
+        `Deployed at: ${new Date(existingDeployment.timestamp).toLocaleString()}`,
+        existingDeployment.txHash ? `Transaction: ${existingDeployment.txHash}` : null,
+        `\nTo redeploy, run with the --redeploy flag:`,
+        `movehat run <script> --network ${config.network} --redeploy`,
+      ]
+        .filter(Boolean)
+        .join("\n");
+
+      // Log formatted error message for user
+      logger.error(`Module "${moduleName}" is already deployed on ${config.network}`);
+      logger.plain(`   Address: ${existingDeployment.address}`);
+      logger.plain(
+        `   Deployed at: ${new Date(existingDeployment.timestamp).toLocaleString()}`
+      );
+      if (existingDeployment.txHash) {
+        logger.plain(`   Transaction: ${existingDeployment.txHash}`);
+      }
+      logger.newline();
+      logger.info("To redeploy, run with the --redeploy flag:");
+      logger.plain(`   movehat run <script> --network ${config.network} --redeploy`);
+      logger.newline();
+
+      // Throw custom error with complete context for programmatic handling
+      throw new ModuleAlreadyDeployedError(
+        errorDetails,
+        moduleName,
+        config.network,
+        existingDeployment.address,
+        existingDeployment.timestamp,
+        existingDeployment.txHash
+      );
+    }
+
+    if (forceRedeploy && existingDeployment) {
+      logger.info(`Redeploying module "${moduleName}" on ${config.network}...`);
+    }
+
+    const dir = input.packageDir || config.moveDir;
+
+    // Bug #37: use a UUID-suffixed profile name per deploy so concurrent
+    // Publisher.deploy() calls in the same process don't fight over the
+    // same key in ~/.aptos/config.yaml. The previous code reused
+    // config.profile (default "default"), which meant two parallel
+    // deploys would clobber each other's profile data mid-publish.
+    const profile = `movehat-deploy-${randomUUID().slice(0, 8)}`;
+
+    // Validate (no shell escape — runCli uses spawn, which takes args
+    // verbatim and would treat the single-quote wrapping as part of the
+    // literal path/profile, breaking Movement CLI argument parsing).
+    const safeDir = validatePathSafety(dir, "package directory");
+    const safeProfile = validateProfileSafety(profile);
+
+    logger.step(`Publishing module "${moduleName}" from ${dir}...`);
+
+    try {
+      // Get the deployer address to use for named addresses
+      const deployerAddress = account.accountAddress.toString();
+
+      // Detect named addresses from Move files
+      const detectedAddresses = extractNamedAddresses(dir);
+
+      // Build named addresses argument - use deployer address for all detected addresses.
+      // Stored as a pre-split args fragment so the spawn path never has to parse
+      // shell tokens; an empty fragment becomes a no-op via spread.
+      const namedAddrArgs: string[] =
+        detectedAddresses.size > 0
+          ? [
+              "--named-addresses",
+              Array.from(detectedAddresses)
+                .map((name) => `${name}=${deployerAddress}`)
+                .join(","),
+            ]
+          : [];
+
+      // Build first with named addresses
+      logger.step("Building package...");
+      const buildResult = await runCli(
+        {
+          command: "movement",
+          args: ["move", "build", "--package-dir", safeDir, ...namedAddrArgs],
+          timeoutMs: 120000, // 2 minutes for git dependency downloads
+        },
+        { adapter: this.deps.adapter }
+      );
+      if (buildResult.stdout) console.log(buildResult.stdout.trim());
+
+      // Publish using direct parameters (avoid config file issues)
+      logger.step("Publishing to blockchain...");
+
+      // Use parameters directly instead of relying on config file
+      // Strip any ed25519-priv- prefix if present
+      let cleanPrivateKey = config.privateKey;
+      if (cleanPrivateKey.startsWith("ed25519-priv-")) {
+        cleanPrivateKey = cleanPrivateKey.replace("ed25519-priv-", "");
+      }
+
+      // Bug #38: Move.toml is NOT mutated. All address overrides flow
+      // through the `--named-addresses` flag above, which Movement CLI
+      // applies during build + publish. The previous regex rewrite +
+      // restore-in-finally was destructive: if the process died between
+      // write and restore, the user's Move.toml stayed mutated.
+
+      let publishOut = "";
+      let publishErr = "";
+
+      // Setup Movement CLI config with private key securely.
+      // Movement CLI uses .aptos config directory (not .movement).
+      const movementConfigPath = join(homedir(), ".aptos", "config.yaml");
+
+      // Register a sync cleanup hook BEFORE writing the private key.
+      // If the user Ctrl+C's (or the process is SIGTERM'd) between the
+      // yaml write and our async finally, the SIGINT handler iterates
+      // every registered callback and removes this deploy's profile
+      // synchronously — closes bug #36 (private key persisting on disk
+      // after abnormal exit).
+      ensureSignalHandler();
+      const syncCleanup = () => removeProfileSync(movementConfigPath, profile);
+      cleanupCallbacks.add(syncCleanup);
+
+      // Add our deploy profile under the unique key. The mutex serializes
+      // read-modify-write cycles so concurrent deploys in the same process
+      // can't drop each other's profiles. Other user profiles in the same
+      // file are preserved untouched.
+      await withYamlLock(() =>
+        addProfile(movementConfigPath, profile, {
+          private_key: cleanPrivateKey,
+          public_key: account.publicKey.toString(),
+          account: deployerAddress,
+          rest_url: config.rpc,
+        })
+      );
+
+      try {
+        // Execute publish command without exposing private key in CLI.
+        // Routed through runCli so stdout/stderr are redacted of any
+        // `ed25519-priv-…` shape before reaching console.log/console.error
+        // or the thrown CliExecutionError — that's bug #43.
+        const publishResult = await runCli(
+          {
+            command: "movement",
+            args: [
+              "move",
+              "publish",
+              "--package-dir",
+              safeDir,
+              "--url",
+              config.rpc,
+              "--profile",
+              safeProfile,
+              "--assume-yes",
+              ...namedAddrArgs,
+            ],
+            timeoutMs: 120000, // 2 minutes for blockchain transactions
+          },
+          { adapter: this.deps.adapter }
+        );
+        publishOut = publishResult.stdout;
+        publishErr = publishResult.stderr;
+        if (publishOut) console.log(publishOut.trim());
+        if (publishErr) console.error(publishErr.trim());
+      } finally {
+        // Always remove our profile from the shared yaml — never restore
+        // a "snapshot" of the whole file (that's what the old code did,
+        // and that's the bug #37 race). Removing only our key leaves
+        // other concurrent deploys' profiles intact.
+        //
+        // CRITICAL: catch + log instead of throwing. `await` in a finally
+        // block that throws will clobber both the try block's successful
+        // return value AND any error already propagating. Without this
+        // catch, a yaml-write failure here would mask a successful
+        // publish (making the deploy look failed and inviting a redeploy)
+        // or mask the real publish error.
+        await withYamlLock(() => removeProfile(movementConfigPath, profile)).catch((err) => {
+          const cleanupMsg = err instanceof Error ? err.message : String(err);
+          logger.warning(
+            `Failed to remove deploy profile "${profile}" from ${movementConfigPath}: ${cleanupMsg}. ` +
+            `Run 'movement config delete-profile --profile ${profile}' to clean up manually.`
+          );
+        });
+        // Unregister the sync cleanup hook — normal path. (The signal
+        // handler stays installed for the process lifetime; cheap.)
+        cleanupCallbacks.delete(syncCleanup);
+      }
+
+      // Extract transaction hash from output via the shared helper
+      // (`utils/parseCliOutput.ts`). Same regex pair as before; lifted
+      // for reuse by harness/codeObject.ts and harness/script.ts.
+      const txHash = parseTxHash(publishOut);
+
+      logger.success("Module published successfully!");
+
+      // ←← "Publish succeeded" boundary. Anything thrown below this
+      // point did NOT cause the publish to fail — the module is on
+      // chain. We surface those failures as PostPublishError so callers
+      // can distinguish a genuine publish failure from a local
+      // bookkeeping failure (and avoid a wasteful redeploy).
+
+      const deployment: DeploymentInfo = {
+        address: account.accountAddress.toString(),
+        moduleName,
+        network: config.network,
+        deployer: account.accountAddress.toString(),
+        timestamp: Date.now(),
+        txHash,
+      };
+
+      try {
+        saveDeployment(deployment);
+      } catch (error) {
+        const err = error instanceof Error ? error : new Error(String(error));
+        throw new PostPublishError(
+          `Module "${moduleName}" published to ${deployment.address} ` +
+          `but local deployment record could not be written: ${err.message}`,
+          deployment,
+          err
+        );
+      }
+
+      return deployment;
+    } catch (error) {
+      if (error instanceof PostPublishError) {
+        // Publish DID succeed; only local persistence failed. Log as
+        // warning (not error) so the user knows the deploy is real on
+        // chain. Re-throw so programmatic callers can react.
+        logger.warning(
+          `Module published successfully to ${error.deployment.address} ` +
+          `(tx=${error.deployment.txHash ?? "unknown"}) ` +
+          `but local deployment record could not be written.`
+        );
+        logger.warning(`   Cause: ${error.cause.message}`);
+        logger.warning(
+          `   To recover, manually write the deployment to ` +
+          `deployments/${error.deployment.network}/${error.deployment.moduleName}.json.`
+        );
+        throw error;
+      }
+      if (error instanceof CliExecutionError) {
+        // stdout/stderr are already redacted by runCli before reaching here,
+        // so this branch is safe to log verbatim.
+        if (error.stdoutPreview) console.log(error.stdoutPreview);
+        logger.error(`Failed to publish module: ${error.message}\n${error.stderr}`);
+      } else {
+        // Preserve existing behaviour for non-CLI errors (filesystem write
+        // failures from Move.toml / ~/.aptos/config.yaml, yaml parse errors,
+        // etc.). These paths can't carry private-key material so logging raw
+        // is safe.
+        const err = error instanceof Error ? error : new Error(String(error));
+        logger.error(`Failed to publish module: ${err.message}`);
+      }
+      throw error;
+    }
+  }
+}