@mmapp/react-compiler 0.1.0-alpha.1

package/src/__tests__/project-compiler.test.ts

@@ -0,0 +1,546 @@
+/**
+ * ProjectCompiler Tests — validates multi-file project compilation and IR merging.
+ */
+
+import { describe, it, expect } from 'vitest';
+import { compileProject } from '../project-compiler';
+
+describe('ProjectCompiler', () => {
+  // =========================================================================
+  // Basic single-file projects
+  // =========================================================================
+
+  it('should compile a single workflow file', () => {
+    const result = compileProject({
+      'app.workflow.tsx': `
+        export function App() {
+          const [name, setName] = useState('');
+          const submit = useTransition('submit');
+          return <div><button onClick={submit.fire}>Submit</button></div>;
+        }
+      `,
+    });
+
+    expect(result.errors).toHaveLength(0);
+    expect(result.ir.fields).toHaveLength(1);
+    expect(result.ir.fields[0].name).toBe('name');
+    expect(result.ir.transitions).toHaveLength(1);
+    expect(result.ir.transitions[0].name).toBe('submit');
+    expect(Object.keys(result.fileIRs)).toHaveLength(1);
+  });
+
+  it('should return empty IR for empty project', () => {
+    const result = compileProject({});
+
+    expect(result.ir.slug).toBe('project');
+    expect(result.ir.fields).toHaveLength(0);
+    expect(result.ir.states).toHaveLength(1);
+    expect(result.ir.states[0].type).toBe('START');
+  });
+
+  // =========================================================================
+  // mm.config.ts resolution
+  // =========================================================================
+
+  it('should resolve slug/name/version from mm.config.ts', () => {
+    const result = compileProject({
+      'mm.config.ts': `
+        import { defineBlueprint } from '@mindmatrix/react';
+        export default defineBlueprint({
+          slug: 'my-app',
+          name: 'My Application',
+          version: '2.0.0',
+          category: 'blueprint',
+          description: 'A test app',
+        });
+      `,
+      'app.workflow.tsx': `
+        export function App() {
+          const [count, setCount] = useState(0);
+          return <div>{count}</div>;
+        }
+      `,
+    });
+
+    expect(result.errors).toHaveLength(0);
+    expect(result.ir.slug).toBe('my-app');
+    expect(result.ir.name).toBe('My Application');
+    expect(result.ir.version).toBe('2.0.0');
+    expect(result.ir.category).toBe('blueprint');
+    expect(result.ir.description).toBe('A test app');
+  });
+
+  it('should resolve mode from mm.config.ts', () => {
+    const result = compileProject({
+      'mm.config.ts': `
+        import { defineBlueprint } from '@mindmatrix/react';
+        export default defineBlueprint({
+          slug: 'strict-app',
+          name: 'Strict App',
+          version: '1.0.0',
+          mode: 'strict',
+        });
+      `,
+      'app.workflow.tsx': `
+        export function App() {
+          const [val, setVal] = useState('');
+          return <div>{val}</div>;
+        }
+      `,
+    });
+
+    // Should compile successfully in strict mode
+    expect(result.ir.slug).toBe('strict-app');
+  });
+
+  it('should use defaults when no mm.config.ts is present', () => {
+    const result = compileProject({
+      'main.workflow.tsx': `
+        export function Main() {
+          const [x, setX] = useState(0);
+          return <div>{x}</div>;
+        }
+      `,
+    });
+
+    // slug inferred from filename
+    expect(result.ir.slug).toBe('main');
+    expect(result.ir.version).toBe('0.1.0');
+  });
+
+  // =========================================================================
+  // Multi-file merging
+  // =========================================================================
+
+  it('should merge fields from multiple workflow files', () => {
+    const result = compileProject({
+      'order.workflow.tsx': `
+        export function Order() {
+          const [total, setTotal] = useState(0);
+          const [status, setStatus] = useState('draft');
+          return <div>{total}</div>;
+        }
+      `,
+      'customer.workflow.tsx': `
+        export function Customer() {
+          const [email, setEmail] = useState('');
+          const [name, setName] = useState('');
+          return <div>{email}</div>;
+        }
+      `,
+    });
+
+    expect(result.errors).toHaveLength(0);
+    expect(result.ir.fields).toHaveLength(4);
+
+    const fieldNames = result.ir.fields.map(f => f.name);
+    expect(fieldNames).toContain('total');
+    expect(fieldNames).toContain('status');
+    expect(fieldNames).toContain('email');
+    expect(fieldNames).toContain('name');
+  });
+
+  it('should deduplicate fields by name (first wins)', () => {
+    const result = compileProject({
+      'a.workflow.tsx': `
+        export function A() {
+          const [name, setName] = useState('Alice');
+          return <div>{name}</div>;
+        }
+      `,
+      'b.workflow.tsx': `
+        export function B() {
+          const [name, setName] = useState('Bob');
+          return <div>{name}</div>;
+        }
+      `,
+    });
+
+    const nameFields = result.ir.fields.filter(f => f.name === 'name');
+    expect(nameFields).toHaveLength(1);
+    expect(nameFields[0].default_value).toBe('Alice');
+  });
+
+  it('should merge transitions from multiple files', () => {
+    const result = compileProject({
+      'a.workflow.tsx': `
+        export function A() {
+          const approve = useTransition('approve');
+          return <div><button onClick={approve.fire}>Approve</button></div>;
+        }
+      `,
+      'b.workflow.tsx': `
+        export function B() {
+          const reject = useTransition('reject');
+          return <div><button onClick={reject.fire}>Reject</button></div>;
+        }
+      `,
+    });
+
+    expect(result.ir.transitions).toHaveLength(2);
+    const names = result.ir.transitions.map(t => t.name);
+    expect(names).toContain('approve');
+    expect(names).toContain('reject');
+  });
+
+  it('should merge states and combine actions', () => {
+    const result = compileProject({
+      'a.workflow.tsx': `
+        export function A() {
+          useOnEnter('submitted', () => {
+            console.log('a entered');
+          });
+          return <div>A</div>;
+        }
+      `,
+      'b.workflow.tsx': `
+        export function B() {
+          useOnEnter('submitted', () => {
+            console.log('b entered');
+          });
+          return <div>B</div>;
+        }
+      `,
+    });
+
+    const submitted = result.ir.states.find(s => s.name === 'submitted');
+    expect(submitted).toBeDefined();
+    // Both files contribute on_enter actions to the same state
+    expect(submitted!.on_enter.length).toBe(2);
+  });
+
+  it('should merge experience trees into a root Stack', () => {
+    const result = compileProject({
+      'header.workflow.tsx': `
+        export function Header() {
+          return <div><h1>Header</h1></div>;
+        }
+      `,
+      'content.workflow.tsx': `
+        export function Content() {
+          return <div><p>Content</p></div>;
+        }
+      `,
+    });
+
+    const views = (result.ir as any).views;
+    expect(views).toBeDefined();
+    expect(views.default).toBeDefined();
+    // Multiple views → wrapped in root Stack
+    expect(views.default.component).toBe('Stack');
+    expect(views.default.children).toHaveLength(2);
+  });
+
+  it('should not wrap single view in Stack', () => {
+    const result = compileProject({
+      'app.workflow.tsx': `
+        export function App() {
+          return <div><h1>Hello</h1></div>;
+        }
+      `,
+    });
+
+    const views = (result.ir as any).views;
+    expect(views).toBeDefined();
+    // Single view: no wrapping
+    expect(views.default.component).not.toBe('Stack');
+  });
+
+  // =========================================================================
+  // Cross-file imports
+  // =========================================================================
+
+  it('should compile model files and merge their fields', () => {
+    const result = compileProject({
+      'models/user.ts': `
+        export interface User {
+          email: string;
+          age: number;
+          status: 'active' | 'inactive';
+        }
+        export const transitions = {};
+      `,
+      'app.workflow.tsx': `
+        import { User } from './models/user';
+        export function App() {
+          const [name, setName] = useState('');
+          return <div>{name}</div>;
+        }
+      `,
+    });
+
+    expect(result.errors).toHaveLength(0);
+    // Both model fields and workflow fields are present
+    const fieldNames = result.ir.fields.map(f => f.name);
+    expect(fieldNames).toContain('email');
+    expect(fieldNames).toContain('age');
+    expect(fieldNames).toContain('name');
+  });
+
+  it('should compile model files before workflow files that import them', () => {
+    const result = compileProject({
+      'pages/index.workflow.tsx': `
+        import { Order } from '../models/order';
+        export function OrderPage() {
+          const [view, setView] = useState('list');
+          return <div>{view}</div>;
+        }
+      `,
+      'models/order.ts': `
+        export interface Order {
+          total: number;
+          customer: string;
+          status: 'pending' | 'shipped';
+        }
+        export const transitions = {};
+      `,
+    });
+
+    expect(result.errors).toHaveLength(0);
+    const fieldNames = result.ir.fields.map(f => f.name);
+    expect(fieldNames).toContain('total');
+    expect(fieldNames).toContain('customer');
+    expect(fieldNames).toContain('view');
+  });
+
+  // =========================================================================
+  // Error handling
+  // =========================================================================
+
+  it('should report errors from individual files', () => {
+    const result = compileProject({
+      'broken.workflow.tsx': `
+        export function Broken() {
+          this is not valid code{{{{
+        }
+      `,
+    });
+
+    expect(result.errors.length).toBeGreaterThan(0);
+    expect(result.errors[0].file).toBe('broken.workflow.tsx');
+    expect(result.errors[0].severity).toBe('error');
+  });
+
+  it('should continue compiling other files when one fails', () => {
+    const result = compileProject({
+      'broken.workflow.tsx': `
+        export function Broken() {
+          this is not valid{{{{
+        }
+      `,
+      'good.workflow.tsx': `
+        export function Good() {
+          const [ok, setOk] = useState(true);
+          return <div>{String(ok)}</div>;
+        }
+      `,
+    });
+
+    expect(result.errors.length).toBeGreaterThan(0);
+    expect(result.fileIRs['good.workflow.tsx']).toBeDefined();
+    expect(result.ir.fields.some(f => f.name === 'ok')).toBe(true);
+  });
+
+  it('should skip non-compilable files', () => {
+    const result = compileProject({
+      'README.md': '# Hello',
+      'package.json': '{}',
+      'styles.css': 'body { }',
+      'app.workflow.tsx': `
+        export function App() {
+          const [x, setX] = useState(0);
+          return <div>{x}</div>;
+        }
+      `,
+    });
+
+    expect(result.errors).toHaveLength(0);
+    expect(Object.keys(result.fileIRs)).toHaveLength(1);
+  });
+
+  // =========================================================================
+  // Config + multi-file integration
+  // =========================================================================
+
+  it('should apply config slug/name/version over file-inferred values', () => {
+    const result = compileProject({
+      'mm.config.ts': `
+        import { defineBlueprint } from '@mindmatrix/react';
+        export default defineBlueprint({
+          slug: 'invoice-app',
+          name: 'Invoice Manager',
+          version: '3.0.0',
+        });
+      `,
+      'pages/create.workflow.tsx': `
+        export function CreateInvoice() {
+          const [amount, setAmount] = useState(0);
+          const submit = useTransition('submit');
+          return <div><button onClick={submit.fire}>Create</button></div>;
+        }
+      `,
+      'pages/list.workflow.tsx': `
+        export function ListInvoices() {
+          const [filter, setFilter] = useState('all');
+          return <div>{filter}</div>;
+        }
+      `,
+      'models/invoice.ts': `
+        export interface Invoice {
+          total: number;
+          due_date: string;
+          status: 'draft' | 'sent' | 'paid';
+        }
+        export const transitions = {};
+      `,
+    });
+
+    expect(result.errors).toHaveLength(0);
+    expect(result.ir.slug).toBe('invoice-app');
+    expect(result.ir.name).toBe('Invoice Manager');
+    expect(result.ir.version).toBe('3.0.0');
+
+    // All fields merged
+    const fieldNames = result.ir.fields.map(f => f.name);
+    expect(fieldNames).toContain('amount');
+    expect(fieldNames).toContain('filter');
+    expect(fieldNames).toContain('total');
+    expect(fieldNames).toContain('due_date');
+
+    // Transition present
+    expect(result.ir.transitions.some(t => t.name === 'submit')).toBe(true);
+  });
+
+  // =========================================================================
+  // Event merging
+  // =========================================================================
+
+  it('should merge on_event subscriptions from multiple files', () => {
+    const result = compileProject({
+      'a.workflow.tsx': `
+        export function A() {
+          useOnEvent('order.created', () => {
+            console.log('order created in A');
+          });
+          return <div>A</div>;
+        }
+      `,
+      'b.workflow.tsx': `
+        export function B() {
+          useOnEvent('order.shipped', () => {
+            console.log('order shipped in B');
+          });
+          return <div>B</div>;
+        }
+      `,
+    });
+
+    expect(result.ir.on_event).toBeDefined();
+    expect(result.ir.on_event!.length).toBe(2);
+    const matches = result.ir.on_event!.map(e => e.match);
+    expect(matches).toContain('order.created');
+    expect(matches).toContain('order.shipped');
+  });
+
+  // =========================================================================
+  // Per-file IRs
+  // =========================================================================
+
+  it('should return per-file IRs for each compiled file', () => {
+    const result = compileProject({
+      'a.workflow.tsx': `
+        export function A() {
+          const [x, setX] = useState(0);
+          return <div>{x}</div>;
+        }
+      `,
+      'b.workflow.tsx': `
+        export function B() {
+          const [y, setY] = useState('');
+          return <div>{y}</div>;
+        }
+      `,
+    });
+
+    expect(result.fileIRs['a.workflow.tsx']).toBeDefined();
+    expect(result.fileIRs['b.workflow.tsx']).toBeDefined();
+    expect(result.fileIRs['a.workflow.tsx'].fields[0].name).toBe('x');
+    expect(result.fileIRs['b.workflow.tsx'].fields[0].name).toBe('y');
+  });
+
+  // =========================================================================
+  // Grammar island merging
+  // =========================================================================
+
+  it('should merge grammar islands (extensions) from multiple files', () => {
+    const result = compileProject({
+      'a.workflow.tsx': `
+        export function A() {
+          const policy = cedar\`permit(principal, action == Action::"read", resource);\`;
+          return <div>A</div>;
+        }
+      `,
+      'b.workflow.tsx': `
+        export function B() {
+          const query = sql\`SELECT * FROM users\`;
+          return <div>B</div>;
+        }
+      `,
+    });
+
+    if (result.ir.extensions) {
+      // Extensions should contain grammar islands from both files
+      const allKeys = Object.keys(result.ir.extensions);
+      expect(allKeys.length).toBeGreaterThanOrEqual(1);
+    }
+  });
+
+  // =========================================================================
+  // Server action files
+  // =========================================================================
+
+  it('should compile server action files', () => {
+    const result = compileProject({
+      'actions/approve.server.ts': `
+        export async function approve(ctx: any) {
+          await ctx.env.notify({ message: 'Approved' });
+        }
+      `,
+      'app.workflow.tsx': `
+        export function App() {
+          const [status, setStatus] = useState('pending');
+          return <div>{status}</div>;
+        }
+      `,
+    });
+
+    expect(result.errors).toHaveLength(0);
+    expect(result.fileIRs['actions/approve.server.ts']).toBeDefined();
+  });
+
+  // =========================================================================
+  // Metadata merging
+  // =========================================================================
+
+  it('should set provenance source to project', () => {
+    const result = compileProject({
+      'mm.config.ts': `
+        export default defineBlueprint({
+          slug: 'test-project',
+          name: 'Test',
+          version: '1.0.0',
+        });
+      `,
+      'app.workflow.tsx': `
+        export function App() {
+          return <div>Hello</div>;
+        }
+      `,
+    });
+
+    expect(result.ir.metadata?.provenance).toBeDefined();
+    const provenance = result.ir.metadata!.provenance as Record<string, string>;
+    expect(provenance.source).toBe('project');
+    expect(provenance.frontend).toBe('react-compiler');
+  });
+});