vite-plugin-ai-mock-generator 1.0.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Your Name
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,81 @@
+# vite-plugin-ai-mock-generator
+
+AI-powered Mock data generator for Vite. Generate realistic test data with AI or basic algorithms.
+
+## ✨ Features
+
+- 🤖 **AI Generation** - Use OpenAI to generate realistic business data
+- ⚡ **Fast Mode** - Basic algorithm generation without AI
+- 💾 **Data Persistence** - Save generated data to files
+- 🔄 **Auto Generation** - Generate data on server start
+- 🎯 **Type Safe** - Full TypeScript support
+- 🔍 **Query Support** - Filter, sort, and pagination
+- 🎭 **Mock Server** - Built-in mock server with request interception
+
+## 📦 Installation
+
+::: code-group
+
+```bash [npm]
+npm install -D vite-plugin-ai-mock-generator
+```
+
+```bash [yarn]
+yarn add -D vite-plugin-ai-mock-generator
+```
+
+```bash [pnpm]
+pnpm add -D vite-plugin-ai-mock-generator
+```
+
+:::
+
+## 🚀 Quick Start
+
+```typescript
+// vite.config.ts
+import { defineConfig } from "vite";
+import { vitePluginAIMockGenerator } from "vite-plugin-ai-mock-generator";
+
+export default defineConfig({
+  plugins: [
+    vitePluginAIMockGenerator({
+      apiKey: process.env.OPENAI_API_KEY,
+      enabled: true,
+      autoGenerate: true,
+      endpoints: [
+        {
+          path: "/api/users",
+          method: "GET",
+          response: {
+            name: "User",
+            properties: [
+              { name: "id", type: "number" },
+              { name: "name", type: "string", comment: "用户名" },
+              { name: "email", type: "string", comment: "邮箱" },
+            ],
+            isArray: true,
+          },
+          count: 20,
+        },
+      ],
+      generation: {
+        locale: "zh-CN",
+        quality: "fast", // 'fast' | 'balanced' | 'high'
+      },
+      storage: {
+        dir: "mock-data",
+        persist: true,
+      },
+    }),
+  ],
+});
+```
+
+## 📚 Documentation
+
+Full documentation: https://mo520.github.io/vite-plugin-ai/plugins/ai-mock-generator
+
+## 📄 License
+
+MIT

package/dist/index.d.mts

@@ -0,0 +1,272 @@
+import { ViteDevServer, Plugin } from 'vite';
+
+/**
+ * AI Mock Generator - 类型定义
+ */
+type HttpMethod = 'GET' | 'POST' | 'PUT' | 'DELETE' | 'PATCH';
+type Locale = 'zh-CN' | 'en-US';
+type Quality = 'fast' | 'balanced' | 'high';
+/**
+ * 插件配置选项
+ */
+interface MockGeneratorOptions {
+    apiKey?: string;
+    apiUrl?: string;
+    model?: string;
+    enabled?: boolean;
+    autoGenerate?: boolean;
+    sources?: {
+        types?: string[];
+        openapi?: string;
+        jsdoc?: boolean;
+    };
+    endpoints?: EndpointConfig[];
+    generation?: {
+        locale?: Locale;
+        count?: number;
+        seed?: number;
+        quality?: Quality;
+    };
+    storage?: {
+        dir?: string;
+        format?: 'json' | 'js' | 'ts';
+        persist?: boolean;
+        cache?: boolean;
+    };
+    server?: {
+        port?: number;
+        prefix?: string;
+        delay?: number | [number, number];
+        cors?: boolean;
+    };
+    output?: {
+        console?: boolean;
+        ui?: boolean;
+        logs?: boolean;
+    };
+}
+/**
+ * 端点配置
+ */
+interface EndpointConfig {
+    path: string;
+    method: HttpMethod;
+    response: string | TypeDefinition;
+    request?: string | TypeDefinition;
+    count?: number;
+    template?: string;
+    enabled?: boolean;
+    custom?: CustomHandler;
+    pagination?: PaginationConfig;
+}
+/**
+ * 类型定义
+ */
+interface TypeDefinition {
+    name: string;
+    properties: PropertyDefinition[];
+    isArray?: boolean;
+}
+/**
+ * 属性定义
+ */
+interface PropertyDefinition {
+    name: string;
+    type: string;
+    comment?: string;
+    required?: boolean;
+    constraints?: PropertyConstraints;
+}
+/**
+ * 属性约束
+ */
+interface PropertyConstraints {
+    min?: number;
+    max?: number;
+    pattern?: string;
+    unique?: boolean;
+    enum?: string[];
+}
+/**
+ * 分页配置
+ */
+interface PaginationConfig {
+    pageSize: number;
+    total: number;
+}
+/**
+ * 自定义处理函数
+ */
+type CustomHandler = (data: any, params?: any) => any | Promise<any>;
+/**
+ * 生成上下文
+ */
+interface GenerationContext {
+    locale: Locale;
+    count: number;
+    seed?: number;
+    quality: Quality;
+    type: TypeDefinition;
+}
+/**
+ * Mock 数据存储
+ */
+interface MockDataStore {
+    endpoint: string;
+    method: HttpMethod;
+    data: any;
+    metadata: {
+        count: number;
+        generatedAt: string;
+        version: string;
+        type: string;
+    };
+}
+
+/**
+ * AI Mock Generator - 数据存储
+ */
+
+declare class MockStorage {
+    private storageDir;
+    private cache;
+    private persist;
+    constructor(options?: {
+        dir?: string;
+        persist?: boolean;
+    });
+    /**
+     * 生成存储 key
+     */
+    private getKey;
+    /**
+     * 生成文件名
+     */
+    private getFileName;
+    /**
+     * 获取数据
+     */
+    get(endpoint: string, method?: HttpMethod): any;
+    /**
+     * 设置数据
+     */
+    set(endpoint: string, method: HttpMethod, data: any, metadata?: Partial<MockDataStore['metadata']>): void;
+    /**
+     * 删除数据
+     */
+    delete(endpoint: string, method?: HttpMethod): void;
+    /**
+     * 清空所有数据
+     */
+    clear(): void;
+    /**
+     * 获取所有端点
+     */
+    getAll(): MockDataStore[];
+    /**
+     * 保存到文件
+     */
+    private save;
+    /**
+     * 从文件加载
+     */
+    private load;
+    /**
+     * 导出所有数据
+     */
+    export(outputPath: string): void;
+    /**
+     * 导入数据
+     */
+    import(inputPath: string): void;
+}
+
+/**
+ * AI Mock Generator - 数据生成器
+ */
+
+declare class MockDataGenerator {
+    private apiKey;
+    private apiUrl;
+    private model;
+    constructor(options: {
+        apiKey: string;
+        apiUrl: string;
+        model: string;
+    });
+    /**
+     * 生成 Mock 数据
+     */
+    generate(context: GenerationContext): Promise<any>;
+    /**
+     * 构建 AI Prompt
+     */
+    private buildPrompt;
+    /**
+     * 格式化属性
+     */
+    private formatProperty;
+    /**
+     * 调用 AI API
+     */
+    private callAI;
+    /**
+     * 解析 AI 响应
+     */
+    private parseResponse;
+    /**
+     * 生成基础数据（不使用 AI）
+     */
+    generateBasic(type: TypeDefinition, count: number): any[];
+    /**
+     * 生成基础值
+     */
+    private generateBasicValue;
+}
+
+/**
+ * AI Mock Generator - Mock 服务器
+ */
+
+declare class MockServer {
+    private storage;
+    private endpoints;
+    private options;
+    constructor(storage: MockStorage, options: MockGeneratorOptions);
+    /**
+     * 配置服务器中间件
+     */
+    configureServer(server: ViteDevServer): void;
+    /**
+     * 匹配端点
+     */
+    private matchEndpoint;
+    /**
+     * 解析请求
+     */
+    private parseRequest;
+    /**
+     * 获取 Mock 数据
+     */
+    private getMockData;
+    /**
+     * 添加端点
+     */
+    addEndpoint(endpoint: EndpointConfig): void;
+    /**
+     * 移除端点
+     */
+    removeEndpoint(path: string, method: HttpMethod): void;
+    /**
+     * 获取所有端点
+     */
+    getEndpoints(): EndpointConfig[];
+}
+
+/**
+ * AI Mock Generator - Vite 插件入口
+ */
+
+declare function vitePluginAIMockGenerator(options?: MockGeneratorOptions): Plugin;
+
+export { type EndpointConfig, MockDataGenerator, type MockGeneratorOptions, MockServer, MockStorage, vitePluginAIMockGenerator };

package/dist/index.d.ts

@@ -0,0 +1,272 @@
+import { ViteDevServer, Plugin } from 'vite';
+
+/**
+ * AI Mock Generator - 类型定义
+ */
+type HttpMethod = 'GET' | 'POST' | 'PUT' | 'DELETE' | 'PATCH';
+type Locale = 'zh-CN' | 'en-US';
+type Quality = 'fast' | 'balanced' | 'high';
+/**
+ * 插件配置选项
+ */
+interface MockGeneratorOptions {
+    apiKey?: string;
+    apiUrl?: string;
+    model?: string;
+    enabled?: boolean;
+    autoGenerate?: boolean;
+    sources?: {
+        types?: string[];
+        openapi?: string;
+        jsdoc?: boolean;
+    };
+    endpoints?: EndpointConfig[];
+    generation?: {
+        locale?: Locale;
+        count?: number;
+        seed?: number;
+        quality?: Quality;
+    };
+    storage?: {
+        dir?: string;
+        format?: 'json' | 'js' | 'ts';
+        persist?: boolean;
+        cache?: boolean;
+    };
+    server?: {
+        port?: number;
+        prefix?: string;
+        delay?: number | [number, number];
+        cors?: boolean;
+    };
+    output?: {
+        console?: boolean;
+        ui?: boolean;
+        logs?: boolean;
+    };
+}
+/**
+ * 端点配置
+ */
+interface EndpointConfig {
+    path: string;
+    method: HttpMethod;
+    response: string | TypeDefinition;
+    request?: string | TypeDefinition;
+    count?: number;
+    template?: string;
+    enabled?: boolean;
+    custom?: CustomHandler;
+    pagination?: PaginationConfig;
+}
+/**
+ * 类型定义
+ */
+interface TypeDefinition {
+    name: string;
+    properties: PropertyDefinition[];
+    isArray?: boolean;
+}
+/**
+ * 属性定义
+ */
+interface PropertyDefinition {
+    name: string;
+    type: string;
+    comment?: string;
+    required?: boolean;
+    constraints?: PropertyConstraints;
+}
+/**
+ * 属性约束
+ */
+interface PropertyConstraints {
+    min?: number;
+    max?: number;
+    pattern?: string;
+    unique?: boolean;
+    enum?: string[];
+}
+/**
+ * 分页配置
+ */
+interface PaginationConfig {
+    pageSize: number;
+    total: number;
+}
+/**
+ * 自定义处理函数
+ */
+type CustomHandler = (data: any, params?: any) => any | Promise<any>;
+/**
+ * 生成上下文
+ */
+interface GenerationContext {
+    locale: Locale;
+    count: number;
+    seed?: number;
+    quality: Quality;
+    type: TypeDefinition;
+}
+/**
+ * Mock 数据存储
+ */
+interface MockDataStore {
+    endpoint: string;
+    method: HttpMethod;
+    data: any;
+    metadata: {
+        count: number;
+        generatedAt: string;
+        version: string;
+        type: string;
+    };
+}
+
+/**
+ * AI Mock Generator - 数据存储
+ */
+
+declare class MockStorage {
+    private storageDir;
+    private cache;
+    private persist;
+    constructor(options?: {
+        dir?: string;
+        persist?: boolean;
+    });
+    /**
+     * 生成存储 key
+     */
+    private getKey;
+    /**
+     * 生成文件名
+     */
+    private getFileName;
+    /**
+     * 获取数据
+     */
+    get(endpoint: string, method?: HttpMethod): any;
+    /**
+     * 设置数据
+     */
+    set(endpoint: string, method: HttpMethod, data: any, metadata?: Partial<MockDataStore['metadata']>): void;
+    /**
+     * 删除数据
+     */
+    delete(endpoint: string, method?: HttpMethod): void;
+    /**
+     * 清空所有数据
+     */
+    clear(): void;
+    /**
+     * 获取所有端点
+     */
+    getAll(): MockDataStore[];
+    /**
+     * 保存到文件
+     */
+    private save;
+    /**
+     * 从文件加载
+     */
+    private load;
+    /**
+     * 导出所有数据
+     */
+    export(outputPath: string): void;
+    /**
+     * 导入数据
+     */
+    import(inputPath: string): void;
+}
+
+/**
+ * AI Mock Generator - 数据生成器
+ */
+
+declare class MockDataGenerator {
+    private apiKey;
+    private apiUrl;
+    private model;
+    constructor(options: {
+        apiKey: string;
+        apiUrl: string;
+        model: string;
+    });
+    /**
+     * 生成 Mock 数据
+     */
+    generate(context: GenerationContext): Promise<any>;
+    /**
+     * 构建 AI Prompt
+     */
+    private buildPrompt;
+    /**
+     * 格式化属性
+     */
+    private formatProperty;
+    /**
+     * 调用 AI API
+     */
+    private callAI;
+    /**
+     * 解析 AI 响应
+     */
+    private parseResponse;
+    /**
+     * 生成基础数据（不使用 AI）
+     */
+    generateBasic(type: TypeDefinition, count: number): any[];
+    /**
+     * 生成基础值
+     */
+    private generateBasicValue;
+}
+
+/**
+ * AI Mock Generator - Mock 服务器
+ */
+
+declare class MockServer {
+    private storage;
+    private endpoints;
+    private options;
+    constructor(storage: MockStorage, options: MockGeneratorOptions);
+    /**
+     * 配置服务器中间件
+     */
+    configureServer(server: ViteDevServer): void;
+    /**
+     * 匹配端点
+     */
+    private matchEndpoint;
+    /**
+     * 解析请求
+     */
+    private parseRequest;
+    /**
+     * 获取 Mock 数据
+     */
+    private getMockData;
+    /**
+     * 添加端点
+     */
+    addEndpoint(endpoint: EndpointConfig): void;
+    /**
+     * 移除端点
+     */
+    removeEndpoint(path: string, method: HttpMethod): void;
+    /**
+     * 获取所有端点
+     */
+    getEndpoints(): EndpointConfig[];
+}
+
+/**
+ * AI Mock Generator - Vite 插件入口
+ */
+
+declare function vitePluginAIMockGenerator(options?: MockGeneratorOptions): Plugin;
+
+export { type EndpointConfig, MockDataGenerator, type MockGeneratorOptions, MockServer, MockStorage, vitePluginAIMockGenerator };