Ai Coding

src/modules/base/service/coding.ts

@@ -0,0 +1,109 @@
+import { App, IMidwayApplication, Init, Inject, Provide } from '@midwayjs/core';
+import { BaseService } from '@cool-midway/core';
+import { InjectEntityModel } from '@midwayjs/typeorm';
+import { Repository } from 'typeorm';
+import * as fs from 'fs';
+import * as path from 'path';
+
+/**
+ * Ai编码
+ */
+@Provide()
+export class BaseCodingService extends BaseService {
+  @App()
+  app: IMidwayApplication;
+
+  /**
+   * 获得模块目录结构
+   */
+  async getModuleTree() {
+    if (this.app.getEnv() !== 'local') {
+      return [];
+    }
+
+    const moduleDir = await this.app.getBaseDir();
+    const modulesPath = path.join(moduleDir, 'modules');
+    const result = [];
+
+    // 递归扫描目录函数
+    const scanDir = (dir: string) => {
+      const files = fs.readdirSync(dir);
+      
+      for (const file of files) {
+        // 跳过 .DS_Store 文件
+        if (file === '.DS_Store') continue;
+
+        const fullPath = path.join(dir, file);
+        const stat = fs.statSync(fullPath);
+        
+        if (stat.isDirectory()) {
+          scanDir(fullPath);
+        } else {
+          // 获取相对于项目根目录的路径
+          let relativePath = path.relative(moduleDir, fullPath);
+          // 只返回模块下的文件
+          if (relativePath.startsWith('modules/')) {
+            // 将 .js 后缀改为 .ts
+            if (relativePath.endsWith('.js')) {
+              relativePath = relativePath.replace(/\.js$/, '.ts');
+            }
+            result.push(`src/${relativePath}`);
+          }
+        }
+      }
+    };
+
+    // 开始扫描
+    scanDir(modulesPath);
+    return result;
+  }
+
+  /**
+   * 创建代码
+   * @param codes 代码
+   */
+  async createCode(codes: {
+    path: string;
+    content: string;
+  }[]) {
+    if (this.app.getEnv() !== 'local') {
+      throw new Error('只能在开发环境下创建代码');
+    }
+
+    const moduleDir = await this.app.getBaseDir();
+
+    for (const code of codes) {
+      // 格式化代码内容
+      const formattedContent = await this.formatContent(code.content);
+      
+      // 获取完整的文件路径
+      const filePath = path.join(moduleDir, code.path);
+      
+      // 确保目录存在
+      const dirPath = path.dirname(filePath);
+      if (!fs.existsSync(dirPath)) {
+        fs.mkdirSync(dirPath, { recursive: true });
+      }
+
+      // 写入文件
+      fs.writeFileSync(filePath, formattedContent, 'utf8');
+    }
+  }
+
+  /**
+   * 格式化内容
+   * @param content
+   */
+  async formatContent(content: string) {
+    // 使用prettier格式化内容
+    const prettier = require('prettier');
+    return prettier.format(content, {
+      parser: 'typescript',
+      singleQuote: true,
+      trailingComma: 'all',
+      bracketSpacing: true,
+      arrowParens: 'avoid',
+      printWidth: 80,
+    });
+  }
+}