Method endInit

src/node/services/initStateManager.ts:285–337

* Finalize init hook execution. * Updates state, persists to disk, emits init-end event, and resolves completion promise. * * IMPORTANT: We persist BEFORE updating in-memory exitCode to prevent a race condition * where replay() sees exitCode !== null but the file doesn't exist yet. This

(workspaceId: string, exitCode: number)

   * the invariant: if init-end is visible (live or replay), the file MUST exist.
   */
  async endInit(workspaceId: string, exitCode: number): Promise<void> {
    const state = this.store.getState(workspaceId);

    if (!state) {
      log.error(`endInit called for workspace ${workspaceId} with no active init state`);
      return;
    }

    const endTime = Date.now();
    const finalStatus = exitCode === 0 ? "success" : "error";

    // Create complete state for persistence (don't mutate in-memory state yet)
    const stateToPerist: InitHookState = {
      ...state,
      status: finalStatus,
      exitCode,
      endTime,
    };

    // Persist FIRST - ensures file exists before in-memory state shows completion
    await this.store.persist(workspaceId, stateToPerist, {
      // If WorkspaceService.remove() cleared init state, do not recreate ~/.mux/sessions/<id>/
      shouldWrite: () => this.store.hasState(workspaceId),
    });

    // NOW update in-memory state (replay will now see file exists)
    state.status = finalStatus;
    state.exitCode = exitCode;
    state.endTime = endTime;

    log.info(
      `Init hook ${state.status} for workspace ${workspaceId} (exit code ${exitCode}, duration ${endTime - state.startTime}ms)`
    );

    // Emit init-end event
    this.emit("init-end", {
      type: "init-end",
      workspaceId,
      exitCode,
      timestamp: endTime,
      // Include truncation info so frontend can show indicator
      ...(state.truncatedLines ? { truncatedLines: state.truncatedLines } : {}),
    } satisfies WorkspaceInitEvent & { workspaceId: string });

    // Resolve completion promise for waiting tools
    const promiseEntry = this.initPromises.get(workspaceId);
    if (promiseEntry) {
      promiseEntry.resolve();
      this.initPromises.delete(workspaceId);
    }

    // Keep state in memory for replay (unlike streams which delete immediately)
  }

  /**
   * Get current in-memory init state for a workspace.