autosnippet 3.0.2 → 3.0.6

package/lib/core/enhancement/django-enhancement.js

@@ -1,6 +1,15 @@
 /**
  * Django Enhancement Pack
  * 条件: { languages: ['python'], frameworks: ['django'] }
+ *
+ * 覆盖 Django 全栈生态:
+ *   - Model 字段/关系/Manager/QuerySet
+ *   - DRF (Django REST Framework) Serializer/ViewSet
+ *   - URL 路由约定
+ *   - Signal (pre_save/post_save)
+ *   - Middleware
+ *   - Management Command
+ *   - Celery Task 集成
  */
 
 import { EnhancementPack } from './EnhancementPack.js';
@@ -10,7 +19,7 @@ class DjangoEnhancement extends EnhancementPack {
     return 'django';
   }
   get displayName() {
-    return 'Django Enhancement';
+    return 'Django / DRF Enhancement';
   }
   get conditions() {
     return { languages: ['python'], frameworks: ['django'] };
@@ -22,7 +31,7 @@ class DjangoEnhancement extends EnhancementPack {
         id: 'django-model-scan',
         label: 'Django Model 扫描',
         guide:
-          'Model 字段与关系扫描（ForeignKey/ManyToMany/OneToOne）、Manager 自定义、QuerySet 链路',
+          'Model 字段与关系扫描（ForeignKey/ManyToMany/OneToOne）、Manager 自定义、QuerySet 链路、Meta 选项 (ordering/indexes/constraints)',
         knowledgeTypes: ['architecture', 'code-pattern'],
         skillWorthy: true,
         dualOutput: true,
@@ -32,11 +41,82 @@ class DjangoEnhancement extends EnhancementPack {
             'Django Model definitions, field relationships and custom managers (auto-generated by enhancement)',
         },
       },
+      {
+        id: 'django-drf-scan',
+        label: 'DRF API 分析',
+        guide:
+          'Django REST Framework 分析 — Serializer/ModelSerializer 字段映射、ViewSet/GenericAPIView 层级、Router 注册、Permission/Throttle 策略、Pagination 配置',
+        knowledgeTypes: ['architecture', 'code-pattern'],
+        skillWorthy: true,
+        dualOutput: true,
+        skillMeta: {
+          name: 'project-django-drf',
+          description:
+            'DRF serializers, viewsets, permissions and router configuration (auto-generated by enhancement)',
+        },
+      },
+      {
+        id: 'django-signal-middleware-scan',
+        label: 'Signal / Middleware 分析',
+        guide:
+          'Signal 分析 (pre_save/post_save/pre_delete/post_delete/m2m_changed 等) 及 Middleware 链路 (MIDDLEWARE 顺序)、自定义 Middleware 实现',
+        knowledgeTypes: ['architecture'],
+        skillWorthy: true,
+        dualOutput: true,
+        skillMeta: {
+          name: 'project-django-signals',
+          description:
+            'Django signal handlers, middleware chain ordering and custom middleware (auto-generated by enhancement)',
+        },
+      },
+    ];
+  }
+
+  getGuardRules() {
+    return [
+      {
+        ruleId: 'django-n-plus-one',
+        category: 'performance',
+        dimension: 'file',
+        severity: 'warning',
+        languages: ['python'],
+        pattern: /\.objects\.(?:all|filter)\([^)]*\)[\s\S]*?for\s+\w+\s+in/,
+        message: 'QuerySet 在循环中可能触发 N+1 查询 — 使用 select_related() / prefetch_related() 优化',
+      },
+      {
+        ruleId: 'django-raw-sql',
+        category: 'safety',
+        dimension: 'file',
+        severity: 'warning',
+        languages: ['python'],
+        pattern: /\.raw\s*\(|cursor\.execute\s*\(/,
+        message: '使用 raw SQL 需确保参数化查询，防止 SQL 注入 — 优先使用 ORM QuerySet',
+      },
+      {
+        ruleId: 'django-no-null-charfield',
+        category: 'correctness',
+        dimension: 'file',
+        severity: 'info',
+        languages: ['python'],
+        pattern: /(?:CharField|TextField|SlugField|URLField|EmailField)\s*\([^)]*null\s*=\s*True/,
+        message: 'Django 约定: CharField/TextField 不应使用 null=True — 使用 blank=True + default="" 代替',
+      },
+      {
+        ruleId: 'django-model-str',
+        category: 'style',
+        dimension: 'file',
+        severity: 'info',
+        languages: ['python'],
+        pattern: /class\s+\w+\((?:models\.)?Model\):\s*\n(?:(?!\s*def\s+__str__)[\s\S])*?(?=\nclass\s|\Z)/,
+        message: 'Django Model 建议实现 __str__ 方法 — 便于 admin 和 shell 调试',
+      },
     ];
   }
 
   detectPatterns(astSummary) {
     const patterns = [];
+
+    // ── Django Models ──
     for (const cls of astSummary.classes || []) {
       if (cls.superclass && /Model$/.test(cls.superclass)) {
         patterns.push({
@@ -46,7 +126,11 @@ class DjangoEnhancement extends EnhancementPack {
           confidence: 0.9,
         });
       }
-      if (cls.superclass && /View$|ViewSet$/.test(cls.superclass)) {
+    }
+
+    // ── Django Views / ViewSets ──
+    for (const cls of astSummary.classes || []) {
+      if (cls.superclass && /View$|ViewSet$|APIView$|GenericAPIView/.test(cls.superclass)) {
         patterns.push({
           type: 'django-view',
           className: cls.name,
@@ -54,6 +138,22 @@ class DjangoEnhancement extends EnhancementPack {
           confidence: 0.85,
         });
       }
+    }
+
+    // ── DRF Serializers ──
+    for (const cls of astSummary.classes || []) {
+      if (cls.superclass && /Serializer$|ModelSerializer|HyperlinkedModelSerializer/.test(cls.superclass)) {
+        patterns.push({
+          type: 'drf-serializer',
+          className: cls.name,
+          line: cls.line,
+          confidence: 0.9,
+        });
+      }
+    }
+
+    // ── Django Management Commands ──
+    for (const cls of astSummary.classes || []) {
       if (cls.superclass && /BaseCommand/.test(cls.superclass)) {
         patterns.push({
           type: 'django-management-command',
@@ -63,6 +163,72 @@ class DjangoEnhancement extends EnhancementPack {
         });
       }
     }
+
+    // ── Django Middleware ──
+    for (const cls of astSummary.classes || []) {
+      const nameLower = cls.name?.toLowerCase() || '';
+      if (nameLower.includes('middleware')) {
+        patterns.push({
+          type: 'django-middleware',
+          className: cls.name,
+          line: cls.line,
+          confidence: 0.8,
+        });
+      }
+    }
+
+    // ── Django Admin ──
+    for (const cls of astSummary.classes || []) {
+      if (cls.superclass && /ModelAdmin|TabularInline|StackedInline/.test(cls.superclass)) {
+        patterns.push({
+          type: 'django-admin',
+          className: cls.name,
+          line: cls.line,
+          confidence: 0.9,
+        });
+      }
+    }
+
+    // ── Signal handler functions ──
+    for (const m of astSummary.methods || []) {
+      if (m.decorators?.some((d) => /receiver/.test(d))) {
+        patterns.push({
+          type: 'django-signal-handler',
+          methodName: m.name,
+          line: m.line,
+          confidence: 0.9,
+        });
+      }
+    }
+
+    // ── Celery tasks ──
+    for (const m of astSummary.methods || []) {
+      if (m.decorators?.some((d) => /@(?:shared_task|app\.task|task)/.test(d))) {
+        patterns.push({
+          type: 'django-celery-task',
+          methodName: m.name,
+          line: m.line,
+          confidence: 0.9,
+        });
+      }
+    }
+
+    // ── Django ecosystem imports ──
+    const djangoImports = (astSummary.imports || []).filter(
+      (imp) =>
+        imp.includes('django') ||
+        imp.includes('rest_framework') ||
+        imp.includes('celery') ||
+        imp.includes('django_filters')
+    );
+    if (djangoImports.length > 0) {
+      patterns.push({
+        type: 'django-ecosystem-usage',
+        importCount: djangoImports.length,
+        confidence: 0.85,
+      });
+    }
+
     return patterns;
   }
 }

package/lib/core/enhancement/fastapi-enhancement.js

@@ -1,6 +1,13 @@
 /**
  * FastAPI Enhancement Pack
  * 条件: { languages: ['python'], frameworks: ['fastapi'] }
+ *
+ * 覆盖 FastAPI 全栈生态:
+ *   - 路由装饰器 + Pydantic 模型
+ *   - Depends() 依赖注入
+ *   - Middleware / CORS / BackgroundTask
+ *   - WebSocket
+ *   - SQLAlchemy / Tortoise ORM 集成模式
  */
 
 import { EnhancementPack } from './EnhancementPack.js';
@@ -21,7 +28,7 @@ class FastAPIEnhancement extends EnhancementPack {
       {
         id: 'fastapi-route-scan',
         label: 'FastAPI 路由分析',
-        guide: 'Route 装饰器(@app.get/post/put/delete)、Pydantic Model/Schema、DI (Depends())',
+        guide: 'Route 装饰器(@app.get/post/put/delete)、APIRouter 分组、Pydantic Model/Schema、DI (Depends())、路由参数声明',
         knowledgeTypes: ['architecture', 'code-pattern'],
         skillWorthy: true,
         dualOutput: true,
@@ -31,13 +38,84 @@ class FastAPIEnhancement extends EnhancementPack {
             'FastAPI route definitions, Pydantic schemas and dependency injection (auto-generated by enhancement)',
         },
       },
+      {
+        id: 'fastapi-middleware-scan',
+        label: 'FastAPI 中间件分析',
+        guide:
+          'FastAPI Middleware 分析 — @app.middleware("http") 自定义中间件、CORS 配置、Trusted Host、GZip、BackgroundTasks 使用模式、Lifespan 事件 (startup/shutdown)',
+        knowledgeTypes: ['architecture', 'code-pattern'],
+        skillWorthy: true,
+        dualOutput: true,
+        skillMeta: {
+          name: 'project-fastapi-middleware',
+          description:
+            'FastAPI middleware stack, CORS config, lifespan events and background tasks (auto-generated by enhancement)',
+        },
+      },
+      {
+        id: 'fastapi-orm-scan',
+        label: 'ORM / DB 模式分析',
+        guide:
+          'ORM 集成分析 — SQLAlchemy 2.0 async session 模式 / Tortoise ORM model 定义、数据库迁移 (Alembic)、Repository Pattern、Session 生命周期管理 (get_db Depends)',
+        knowledgeTypes: ['architecture', 'code-pattern'],
+        skillWorthy: true,
+        dualOutput: true,
+        skillMeta: {
+          name: 'project-fastapi-orm',
+          description:
+            'FastAPI ORM integration — SQLAlchemy async sessions, Alembic migrations and repository patterns (auto-generated by enhancement)',
+        },
+      },
+    ];
+  }
+
+  getGuardRules() {
+    return [
+      {
+        ruleId: 'fastapi-sync-in-async',
+        category: 'performance',
+        dimension: 'file',
+        severity: 'warning',
+        languages: ['python'],
+        pattern: /async\s+def\s+\w+[\s\S]*?(?:time\.sleep|open\(|os\.read|subprocess\.run)/,
+        message: 'async 路由中不应使用阻塞操作 (time.sleep/open/subprocess.run) — 使用 asyncio 版本或 run_in_executor',
+      },
+      {
+        ruleId: 'fastapi-no-response-model',
+        category: 'style',
+        dimension: 'file',
+        severity: 'info',
+        languages: ['python'],
+        pattern: /@(?:app|router)\.(?:get|post|put|delete|patch)\s*\([^)]*\)\s*\n\s*(?:async\s+)?def\s+\w+\([^)]*\)(?!\s*->)/,
+        message: 'FastAPI 路由建议声明 response_model 或返回类型注解 — 便于自动生成 OpenAPI 文档',
+      },
+      {
+        ruleId: 'fastapi-bare-except',
+        category: 'correctness',
+        dimension: 'file',
+        severity: 'warning',
+        languages: ['python'],
+        pattern: /except\s*:/,
+        message: '避免 bare except — 使用 except Exception 或具体异常类，否则会吞掉 KeyboardInterrupt/SystemExit',
+      },
+      {
+        ruleId: 'fastapi-no-star-import',
+        category: 'style',
+        dimension: 'file',
+        severity: 'warning',
+        languages: ['python'],
+        pattern: /from\s+\w+\s+import\s+\*/,
+        message: '避免 star import (from x import *) — 使用显式 import 以保证类型检查和 IDE 支持',
+      },
     ];
   }
 
   detectPatterns(astSummary) {
     const patterns = [];
+
+    // ── Pydantic models ──
     for (const cls of astSummary.classes || []) {
-      if (cls.superclass && /BaseModel/.test(cls.superclass)) {
+      if (cls.superclass && /BaseModel|BaseSettings/.test(cls.superclass)) {
         patterns.push({
           type: 'pydantic-model',
           className: cls.name,
@@ -46,8 +124,10 @@ class FastAPIEnhancement extends EnhancementPack {
         });
       }
     }
+
+    // ── FastAPI route handlers ──
     for (const m of astSummary.methods || []) {
-      if (m.decorators?.some((d) => /@app\.(get|post|put|delete|patch)/.test(d))) {
+      if (m.decorators?.some((d) => /@(?:app|router)\.(?:get|post|put|delete|patch|options|head)/.test(d))) {
         patterns.push({
           type: 'fastapi-route',
           methodName: m.name,
@@ -56,6 +136,72 @@ class FastAPIEnhancement extends EnhancementPack {
         });
       }
     }
+
+    // ── Dependency injection functions ──
+    for (const m of astSummary.methods || []) {
+      if (!m.className) {
+        const nameLower = m.name?.toLowerCase() || '';
+        if (
+          nameLower.startsWith('get_') && (
+            nameLower.includes('db') ||
+            nameLower.includes('session') ||
+            nameLower.includes('current_user') ||
+            nameLower.includes('settings') ||
+            nameLower.includes('service')
+          )
+        ) {
+          patterns.push({
+            type: 'fastapi-dependency',
+            methodName: m.name,
+            line: m.line,
+            confidence: 0.8,
+          });
+        }
+      }
+    }
+
+    // ── SQLAlchemy models ──
+    for (const cls of astSummary.classes || []) {
+      if (cls.superclass && /Base$|DeclarativeBase/.test(cls.superclass)) {
+        patterns.push({
+          type: 'sqlalchemy-model',
+          className: cls.name,
+          line: cls.line,
+          confidence: 0.85,
+        });
+      }
+    }
+
+    // ── Exception handlers ──
+    for (const cls of astSummary.classes || []) {
+      if (cls.superclass && /HTTPException|Exception/.test(cls.superclass)) {
+        patterns.push({
+          type: 'fastapi-exception',
+          className: cls.name,
+          line: cls.line,
+          confidence: 0.75,
+        });
+      }
+    }
+
+    // ── FastAPI ecosystem imports ──
+    const fastapiImports = (astSummary.imports || []).filter(
+      (imp) =>
+        imp.includes('fastapi') ||
+        imp.includes('pydantic') ||
+        imp.includes('sqlalchemy') ||
+        imp.includes('tortoise') ||
+        imp.includes('alembic') ||
+        imp.includes('uvicorn')
+    );
+    if (fastapiImports.length > 0) {
+      patterns.push({
+        type: 'fastapi-ecosystem-usage',
+        importCount: fastapiImports.length,
+        confidence: 0.85,
+      });
+    }
+
     return patterns;
   }
 }

package/lib/core/enhancement/go-grpc-enhancement.js

@@ -59,6 +59,8 @@ class GoGrpcEnhancement extends EnhancementPack {
     return [
       {
         ruleId: 'go-grpc-unimplemented-embed',
+        category: 'correctness',
+        dimension: 'file',
         severity: 'warning',
         languages: ['go'],
         pattern: /Unimplemented\w+Server\b/,
@@ -67,6 +69,8 @@ class GoGrpcEnhancement extends EnhancementPack {
       },
       {
         ruleId: 'go-grpc-status-error',
+        category: 'correctness',
+        dimension: 'file',
         severity: 'info',
         languages: ['go'],
         pattern: /fmt\.Errorf|errors\.New/,

package/lib/core/enhancement/go-web-enhancement.js

@@ -59,6 +59,8 @@ class GoWebEnhancement extends EnhancementPack {
     return [
       {
         ruleId: 'go-handler-abort-return',
+        category: 'correctness',
+        dimension: 'file',
         severity: 'warning',
         languages: ['go'],
         pattern: /\.Abort\w*\([^)]*\)[^}]*\n\s*[^}\s/]/,
@@ -66,6 +68,8 @@ class GoWebEnhancement extends EnhancementPack {
       },
       {
         ruleId: 'go-no-naked-goroutine',
+        category: 'correctness',
+        dimension: 'file',
         severity: 'warning',
         languages: ['go'],
         pattern: /go\s+func\s*\(/,
@@ -74,6 +78,8 @@ class GoWebEnhancement extends EnhancementPack {
       },
       {
         ruleId: 'go-context-first-param',
+        category: 'style',
+        dimension: 'file',
         severity: 'info',
         languages: ['go'],
         pattern:

package/lib/core/enhancement/index.js

@@ -40,14 +40,19 @@ export async function initEnhancementRegistry() {
 
   const packImports = [
     import('./react-enhancement.js'),
+    import('./nextjs-enhancement.js'),
     import('./vue-enhancement.js'),
     import('./node-server-enhancement.js'),
     import('./django-enhancement.js'),
     import('./fastapi-enhancement.js'),
+    import('./ml-enhancement.js'),
+    import('./langchain-enhancement.js'),
     import('./spring-enhancement.js'),
     import('./android-enhancement.js'),
     import('./go-web-enhancement.js'),
     import('./go-grpc-enhancement.js'),
+    import('./rust-web-enhancement.js'),
+    import('./rust-tokio-enhancement.js'),
   ];
 
   const results = await Promise.allSettled(packImports);