@pygmalionjs/pygmalion 0.2.14 → 0.2.15

package/README.md

@@ -86,6 +86,18 @@ When the mirror preview reloads the host Vite configuration,
 `PYGMALION_PREVIEW_MODE=1` prevents the integration plugins from starting
 recursively.
 
+Mirror and session previews run as their own Vite processes, so the mode you
+pass to the editor does not reach them. Set `preview.mode` when a screen only
+exists under a specific env file — a feature flag that has to be off, a mock
+profile a recipe depends on:
+
+```ts
+export default definePygmalionProject({
+  // ...
+  preview: { mode: "e2e" },
+});
+```
+
 ## Register screens and scenarios
 
 The host owns application-specific routes, fixtures, authentication, mock data,

package/node/design-session.mjs

@@ -7,6 +7,7 @@ import os from 'node:os';
 import path from 'node:path';
 import { fileURLToPath } from 'node:url';
 import { promisify } from 'node:util';
+import { normalizeViteMode } from './dev-mirror.mjs';
 import { writeStyleEdit, writeTextEdit } from './inspect-plugin.mjs';
 import { createUnifiedDiff } from './source-diff.mjs';
 import { writeSourceOperations } from './source-operations.mjs';
@@ -289,6 +290,9 @@ export function pygmalionDesignSessionPlugin(options) {
     options.previewConfig ?? fileURLToPath(new URL('./dev-view.vite.mjs', import.meta.url)),
   );
   const viteConfig = path.resolve(options.viteConfig ?? path.join(mirrorAppRoot, 'vite.config.ts'));
+  // Session previews are separate Vite processes too, so they need the same
+  // host-declared mode as the mirror or a session would render a different app.
+  const previewMode = normalizeViteMode(options.previewMode);
   const sourceDirectory = (options.sourceDirectory ?? 'src').replace(/^\.?\//, '').replace(/\/$/, '') || '.';
   const dependencies = options.dependencies ?? {};
   const inspect = options.inspect ?? {};
@@ -503,7 +507,17 @@ export function pygmalionDesignSessionPlugin(options) {
 
     const child = spawn(
       process.execPath,
-      [viteBin, '--config', wrapperConfig, '--host', '127.0.0.1', '--port', String(port), '--strictPort'],
+      [
+        viteBin,
+        '--config',
+        wrapperConfig,
+        ...(previewMode ? ['--mode', previewMode] : []),
+        '--host',
+        '127.0.0.1',
+        '--port',
+        String(port),
+        '--strictPort',
+      ],
       {
         cwd: sessionAppRoot,
         env: {

package/node/dev-mirror.mjs

@@ -122,6 +122,24 @@ function normalizeLocalSourceRef(value) {
   return value.trim();
 }
 
+/**
+ * The mode becomes an argv entry for the spawned preview Vite, so it is
+ * restricted to the identifier shape Vite mode names actually use. Anything
+ * else could smuggle a second flag into that command line.
+ */
+export function normalizeViteMode(value) {
+  if (value == null) return null;
+  if (typeof value !== 'string' || !/^[A-Za-z0-9._-]+$/.test(value)) {
+    throw new Error(
+      'Pygmalion preview.mode must be a Vite mode name ([A-Za-z0-9._-]+)',
+    );
+  }
+  if (value.startsWith('-')) {
+    throw new Error('Pygmalion preview.mode must not start with "-"');
+  }
+  return value;
+}
+
 function normalizeManagedOutputPaths(value) {
   if (value == null) return [];
   if (!Array.isArray(value)) {
@@ -278,6 +296,10 @@ export function pygmalionDevMirrorPlugin(options) {
     options.previewConfig ?? fileURLToPath(new URL('./dev-view.vite.mjs', import.meta.url)),
   );
   const viteConfig = path.resolve(options.viteConfig ?? path.join(mirrorAppRoot, 'vite.config.ts'));
+  // The mirror is a separate Vite process, so the editor's own `--mode` never
+  // reaches it. Hosts whose captured screens need a specific env file (a feature
+  // flag a screen depends on, a mock profile) declare that mode here.
+  const previewMode = normalizeViteMode(options.previewMode);
   const inventory = options.inventory;
   const inventoryOutputs = normalizeManagedOutputPaths(inventory?.outputs);
   const managedPaths = normalizeManagedOutputPaths(
@@ -450,6 +472,7 @@ export function pygmalionDevMirrorPlugin(options) {
         viteBin,
         '--config',
         wrapperConfig,
+        ...(previewMode ? ['--mode', previewMode] : []),
         '--host',
         '127.0.0.1',
         '--port',

package/node/vite.mjs

@@ -267,6 +267,7 @@ export function createPygmalionVitePlugins(config) {
         previewConfig: project.preview.configFile,
         viteConfig: project.preview.viteConfig,
         viteBin: project.preview.viteBin,
+        previewMode: project.preview.mode,
         previewPort: project.mirror.previewPort ?? project.preview.mirrorPort,
         prefix: normalizeEndpoint(
           project.mirror.prefix,
@@ -298,6 +299,7 @@ export function createPygmalionVitePlugins(config) {
         previewConfig: project.preview.configFile,
         viteConfig: project.preview.viteConfig,
         viteBin: project.preview.viteBin,
+        previewMode: project.preview.mode,
         previewPort:
           project.sessions.previewPort ?? project.preview.sessionPort,
         worktreesRoot: project.sessions.worktreesRoot,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pygmalionjs/pygmalion",
-  "version": "0.2.14",
+  "version": "0.2.15",
   "description": "Code-backed DOM design sandbox and visual QA editor",
   "license": "UNLICENSED",
   "publishConfig": {

package/vite.d.ts

@@ -60,6 +60,15 @@ export interface PygmalionPreviewConfig {
   configFile?: string;
   viteConfig?: string;
   viteBin?: string;
+  /**
+   * Vite mode for the mirror and session preview servers.
+   *
+   * Those run as separate Vite processes, so the editor's own `--mode` never
+   * reaches the previewed application. Set this when a captured screen depends
+   * on a specific env file — for example a feature flag that has to be off for
+   * the screen to exist at all. Defaults to Vite's own default mode.
+   */
+  mode?: string;
   mirrorPort?: number;
   sessionPort?: number;
   artifactFile?: string;