add vite plugin

package.json

@@ -26,7 +26,8 @@
     "jsx.js",
     "loader.js",
     "macro.js",
-    "carmiRoot.js"
+    "carmiRoot.js",
+    "vite.js"
   ],
   "publishConfig": {
     "registry": "https://registry.npmjs.org/"
@@ -63,7 +64,8 @@
     "resolve": "^1.10.0",
     "tempy": "^0.2.1",
     "toposort": "^1.0.6",
-    "uuid": "^8.3.2"
+    "uuid": "^8.3.2",
+    "vite": "2.9.13"
   },
   "devDependencies": {
     "babel-plugin-tester": "^5.5.1",

src/vitePlugin/index.js

@@ -0,0 +1,40 @@
+const babel = require('@babel/core');
+const carmiBabelPlugin = require('../babelPlugin');
+const compileFile = require('../babelPlugin/compileFile');
+
+const isCarmiFile = (id) => /\.carmi\.(js|mjs)$/.test(id);
+
+function carmiVitePlugin(options = {}) {
+  return {
+    name: 'vite-plugin-carmi',
+    enforce: 'pre',
+
+    async transform(code, id) {
+      if (!isCarmiFile(id)) {
+        return null;
+      }
+
+      const babelResult = await babel.transformAsync(code, {
+        filename: id,
+        plugins: [carmiBabelPlugin],
+        ast: true
+      });
+
+      if (!babelResult || !babelResult.code) {
+        return null;
+      }
+
+      const compiledCode = compileFile(id, {
+        isDebug: options.debug,
+        disableCurrentLineFunctionName: options.disableCurrentLineFunctionName
+      });
+
+      return {
+        code: compiledCode,
+        map: babelResult.map
+      };
+    }
+  };
+}
+
+module.exports = carmiVitePlugin;

src/vitePlugin/test.carmi.js

@@ -0,0 +1,4 @@
+// @carmi
+
+const {root} = require('../../index');
+module.exports = {first: root.get(0), sum: root.call('sum')};

src/vitePlugin/vite-plugin.spec.js

@@ -0,0 +1,94 @@
+const vitePlugin = require('./index');
+
+describe('Carmi Vite Plugin', () => {
+  let mockPluginContext;
+
+
+  beforeEach(() => {
+    mockPluginContext = {
+      addWatchFile: jest.fn(),
+      emitFile: jest.fn()
+    };
+    jest.clearAllMocks();
+  });
+
+  const getCarmiTestFile = async (plugin, mockCode) => await plugin.transform.call(mockPluginContext, mockCode, './src/vitePlugin/test.carmi.js');
+
+  it('transforms .carmi.js files', async () => {
+    const plugin = vitePlugin();
+    const mockCode = `
+      const { root } = require('carmi');
+      module.exports = { first: root.get(0) };
+    `;
+
+    const result = await getCarmiTestFile(plugin, mockCode);
+
+    expect(result).toBeDefined();
+    expect(typeof result.code).toBe('string');
+    expect(result.code).toContain('function model');
+  });
+
+  it('does not transform regular .js files', async () => {
+    const plugin = vitePlugin();
+    const mockCode = `
+      const regularJsCode = 'Hello, World!';
+    `;
+
+    const result = await plugin.transform.call(mockPluginContext, mockCode, 'regular.js');
+
+    expect(result).toBeNull();
+  });
+
+  it('transforms files with @carmi comment', async () => {
+    const plugin = vitePlugin();
+    const mockCode = `
+      // @carmi
+      const { root } = require('carmi');
+      module.exports = { first: root.get(0) };
+    `;
+
+    const result = await getCarmiTestFile(plugin, mockCode);
+
+    expect(result).toBeDefined();
+    expect(result.code).toContain('function model');
+  });
+
+  it('does not transform files without .carmi.js extension or @carmi comment', async () => {
+    const plugin = vitePlugin();
+    const mockCode = `
+      const { root } = require('carmi');
+      module.exports = { first: root.get(0) };
+    `;
+
+    const result = await plugin.transform.call(mockPluginContext, mockCode, 'test.js');
+
+    expect(result).toBeNull();
+  });
+
+  it('handles carmi compiler options', async () => {
+    const customOptions = {optimize: true};
+    const plugin = vitePlugin(customOptions);
+    const mockCode = `
+      // @carmi
+      const { root } = require('carmi');
+      module.exports = { first: root.get(0) };
+    `;
+
+    const result = await getCarmiTestFile(plugin, mockCode);
+
+    expect(result).toBeDefined();
+    expect(typeof result.code).toBe('string');
+  });
+
+  it('handles syntax errors gracefully', async () => {
+    const plugin = vitePlugin();
+    const mockCode = `
+      // @carmi
+      const { root } = require('carmi');
+      module.exports = { first: root.get(0 };  // Missing closing parenthesis
+    `;
+
+    await expect(plugin.transform.call(mockPluginContext, mockCode, 'test.carmi.js'))
+      .rejects.toThrow();
+  });
+});

vite.config.js

@@ -0,0 +1,19 @@
+const {defineConfig} = require('vite');
+const carmiPlugin = require('./src/vitePlugin');
+const {resolve} = require('path');
+
+module.exports = api => defineConfig({
+  plugins: [carmiPlugin()],
+  build: {
+    outDir: 'dist',
+    rollupOptions: {
+      input: {
+        main: resolve(__dirname, 'test.carmi.js')
+      },
+      output: {
+        dir: 'dist',
+        format: 'es'
+      }
+    }
+  }
+});

vite.js

@@ -0,0 +1 @@
+module.exports = require('./src/vitePlugin');