fdb2 1.0.6 → 1.0.7

package/packages/vscode/src/service/DatabaseServiceBridge.ts

@@ -0,0 +1,414 @@
+/**
+ * 数据库服务桥接层
+ * 用于 VSCode 扩展调用 server 目录下的现有数据库服务
+ * 为了简化部署，这里直接实现必要的功能，复用 server 目录的逻辑
+ */
+
+import * as fs from 'fs';
+import * as path from 'path';
+import * as os from 'os';
+import { DataSource, type DataSourceOptions } from 'typeorm';
+
+// 引入数据库服务接口（从本地复制的服务）
+import { MySQLService } from '../database-services/mysql.service';
+import { PostgreSQLService } from '../database-services/postgres.service';
+import { SQLiteService } from '../database-services/sqlite.service';
+import { OracleService } from '../database-services/oracle.service';
+import { SQLServerService } from '../database-services/mssql.service';
+import { CockroachDBService } from '../database-services/cockroachdb.service';
+import { MongoDBService } from '../database-services/mongodb.service';
+import { SAPHANADatabaseService } from '../database-services/sap.service';
+import { BaseDatabaseService } from '../database-services/base.service';
+import { ConnectionEntity } from '../database-services/model/connection.entity';
+
+/**
+ * 数据库服务桥接类
+ * 直接复用 server 目录下的服务
+ */
+export class DatabaseServiceBridge {
+  /**
+   * 连接配置文件路径
+   */
+  private readonly configPath: string;
+
+  /**
+   * 活跃的数据库连接实例
+   */
+  private activeConnections: Map<string, DataSource> = new Map();
+
+  /**
+   * 数据库服务实例
+   */
+  private mysqlService: MySQLService;
+  private postgreSQLService: PostgreSQLService;
+  private sqliteService: SQLiteService;
+  private oracleService: OracleService;
+  private sqlServerService: SQLServerService;
+  private cockroachDBService: CockroachDBService;
+  private mongoDBService: MongoDBService;
+  private sapHANADatabaseService: SAPHANADatabaseService;
+
+  constructor() {
+    const dataDir = process.env.DB_TOOL_DATA_DIR || path.join(os.homedir(), '.fdb2');
+    this.configPath = path.join(dataDir, 'connections.json');
+
+    this.mysqlService = new MySQLService();
+    this.postgreSQLService = new PostgreSQLService();
+    this.sqliteService = new SQLiteService();
+    this.oracleService = new OracleService();
+    this.sqlServerService = new SQLServerService();
+    this.cockroachDBService = new CockroachDBService();
+    this.mongoDBService = new MongoDBService();
+    this.sapHANADatabaseService = new SAPHANADatabaseService();
+  }
+
+  /**
+   * 初始化服务
+   */
+  async init() {
+    const configDir = path.dirname(this.configPath);
+    if (!fs.existsSync(configDir)) {
+      fs.mkdirSync(configDir, { recursive: true });
+    }
+  }
+
+  /**
+   * 连接管理接口
+   */
+  get connection() {
+    return {
+      getAllConnections: this.getAllConnections.bind(this),
+      getConnectionById: this.getConnectionById.bind(this),
+      addConnection: this.addConnection.bind(this),
+      updateConnection: this.updateConnection.bind(this),
+      deleteConnection: this.deleteConnection.bind(this),
+      testConnection: this.testConnection.bind(this)
+    };
+  }
+
+  /**
+   * 数据库管理接口
+   */
+  get database() {
+    return {
+      getDatabases: this.getDatabases.bind(this),
+      getTables: this.getTables.bind(this),
+      executeQuery: this.executeQuery.bind(this),
+      getViews: this.getViews.bind(this),
+      getProcedures: this.getProcedures.bind(this)
+    };
+  }
+
+  /**
+   * 连接管理实现
+   */
+  private async getAllConnections(): Promise<ConnectionEntity[]> {
+    try {
+      if (!fs.existsSync(this.configPath)) {
+        return [];
+      }
+      const data = fs.readFileSync(this.configPath, 'utf8');
+      return JSON.parse(data) as ConnectionEntity[];
+    } catch (error) {
+      console.error('读取连接配置失败:', error);
+      return [];
+    }
+  }
+
+  private async getConnectionById(id: string): Promise<ConnectionEntity | null> {
+    const connections = await this.getAllConnections();
+    return connections.find(conn => conn.id === id) || null;
+  }
+
+  private async addConnection(connection: ConnectionEntity): Promise<ConnectionEntity> {
+    const connections = await this.getAllConnections();
+
+    if (connections.find(conn => conn.name === connection.name)) {
+      throw new Error('连接名称已存在');
+    }
+
+    connection.id = this.generateId();
+    connection.createdAt = new Date();
+    connection.updatedAt = new Date();
+    connection.enabled = connection.enabled !== undefined ? connection.enabled : true;
+
+    connections.push(connection);
+    await this.saveConnections(connections);
+
+    return connection;
+  }
+
+  private async updateConnection(id: string, updates: Partial<ConnectionEntity>): Promise<ConnectionEntity> {
+    const connections = await this.getAllConnections();
+    const index = connections.findIndex(conn => conn.id === id);
+
+    if (index === -1) {
+      throw new Error('连接配置不存在');
+    }
+
+    if (updates.name && connections.find((conn, idx) => conn.name === updates.name && idx !== index)) {
+      throw new Error('连接名称已存在');
+    }
+
+    connections[index] = { ...connections[index], ...updates, updatedAt: new Date() } as ConnectionEntity;
+    await this.saveConnections(connections);
+
+    return connections[index];
+  }
+
+  private async deleteConnection(id: string): Promise<void> {
+    const connections = await this.getAllConnections();
+    const filteredConnections = connections.filter(conn => conn.id !== id);
+
+    if (filteredConnections.length === connections.length) {
+      throw new Error('连接配置不存在');
+    }
+
+    if (this.activeConnections.has(id)) {
+      await this.activeConnections.get(id)?.destroy();
+      this.activeConnections.delete(id);
+    }
+
+    await this.saveConnections(filteredConnections);
+  }
+
+  private async testConnection(connection: ConnectionEntity): Promise<boolean> {
+    try {
+      const tempDataSource = await this.createTypeORMDataSource(connection);
+      await tempDataSource.query('SELECT 1');
+      await tempDataSource.destroy();
+      return true;
+    } catch (error) {
+      console.error(error);
+      return false;
+    }
+  }
+
+  /**
+   * 数据库管理实现
+   */
+  private async getDatabases(connectionId: string): Promise<string[]> {
+    const dataSource = await this.getActiveConnection(connectionId);
+    const databaseService = this.getDatabaseService(dataSource.options.type as string);
+    return databaseService.getDatabases(dataSource);
+  }
+
+  private async getTables(connectionId: string, databaseName: string): Promise<any[]> {
+    const dataSource = await this.getActiveConnection(connectionId, databaseName);
+    const databaseService = this.getDatabaseService(dataSource.options.type as string);
+    return databaseService.getTables(dataSource, databaseName);
+  }
+
+  private async executeQuery(connectionId: string, sql: string, databaseName?: string): Promise<any> {
+    const dataSource = await this.getActiveConnection(connectionId, databaseName);
+    const databaseService = this.getDatabaseService(dataSource.options.type as string);
+    return databaseService.executeQuery(dataSource, sql);
+  }
+
+  private async getViews(connectionId: string, databaseName: string): Promise<any[]> {
+    const dataSource = await this.getActiveConnection(connectionId, databaseName);
+    const databaseService = this.getDatabaseService(dataSource.options.type as string);
+    return databaseService.getViews(dataSource, databaseName);
+  }
+
+  private async getProcedures(connectionId: string, databaseName: string): Promise<any[]> {
+    const dataSource = await this.getActiveConnection(connectionId, databaseName);
+    const databaseService = this.getDatabaseService(dataSource.options.type as string);
+    return databaseService.getProcedures(dataSource, databaseName);
+  }
+
+  /**
+   * 获取活跃的数据库连接
+   */
+  private async getActiveConnection(id: string, database?: string): Promise<DataSource> {
+    const key = database ? `${id}_${database}` : id;
+    if (this.activeConnections.has(key)) {
+      const db = this.activeConnections.get(key);
+      if (db?.isInitialized) {
+        return db;
+      } else {
+        this.activeConnections.delete(key);
+      }
+    }
+
+    if (this.activeConnections.size >= 10) {
+      const oldestKey = this.activeConnections.keys().next().value || '';
+      const oldestConnection = this.activeConnections.get(oldestKey);
+      try {
+        await oldestConnection?.destroy();
+      } catch (error) {
+        console.error(`关闭旧连接 ${oldestKey} 失败:`, error);
+      }
+      this.activeConnections.delete(oldestKey);
+    }
+
+    const connectionConfig = await this.getConnectionById(id);
+    if (!connectionConfig) {
+      throw new Error('连接配置不存在');
+    }
+
+    const updatedConnectionConfig: ConnectionEntity = {
+      ...connectionConfig,
+      database: database || connectionConfig.database
+    };
+
+    const dataSource = await this.createTypeORMDataSource(updatedConnectionConfig);
+    this.activeConnections.set(key, dataSource);
+
+    return dataSource;
+  }
+
+  /**
+   * 创建TypeORM数据源
+   */
+  private async createTypeORMDataSource(connectionConfig: ConnectionEntity): Promise<DataSource> {
+    const connectionOptions = this.getTypeORMOptions(connectionConfig);
+    return new DataSource(connectionOptions).initialize();
+  }
+
+  /**
+   * 获取TypeORM连接配置
+   */
+  private getTypeORMOptions(connectionConfig: ConnectionEntity): DataSourceOptions {
+    const baseOptions: DataSourceOptions = {
+      type: connectionConfig.type as any,
+      host: connectionConfig.host,
+      port: connectionConfig.port,
+      username: connectionConfig.username,
+      password: connectionConfig.password,
+      database: connectionConfig.database,
+      synchronize: false,
+      logging: false,
+      extra: {
+        multipleStatements: true
+      },
+      ...connectionConfig.options
+    };
+
+    switch (connectionConfig.type.toLowerCase()) {
+      case 'sqlite':
+        return {
+          ...baseOptions,
+          type: 'sqlite' as any,
+          database: connectionConfig.database,
+          host: undefined,
+          port: undefined,
+          username: undefined,
+          password: undefined
+        };
+      case 'postgres':
+      case 'postgresql':
+        return {
+          ...baseOptions,
+          type: 'postgres' as any,
+          ssl: connectionConfig.options?.ssl || false
+        };
+      case 'oracle':
+        return {
+          ...baseOptions,
+          type: 'oracle' as any,
+          connectString: `${connectionConfig.host}:${connectionConfig.port}/${connectionConfig.database}`,
+          host: undefined,
+          port: undefined,
+          database: undefined,
+          extra: {
+            connectionTimeout: 60000,
+            poolMax: 10,
+            poolMin: 1,
+            poolIncrement: 1
+          }
+        };
+      case 'mssql':
+      case 'sqlserver':
+        return {
+          ...baseOptions,
+          type: 'mssql' as any,
+          options: {
+            encrypt: connectionConfig.options?.encrypt || false,
+            trustServerCertificate: true
+          },
+          extra: {
+            connectionTimeout: 60000,
+            requestTimeout: 15000
+          }
+        };
+      default:
+        return baseOptions;
+    }
+  }
+
+  /**
+   * 获取数据库服务实例
+   */
+  private getDatabaseService(type: string): BaseDatabaseService {
+    switch (type.toLowerCase()) {
+      case 'mysql':
+      case 'aurora-mysql':
+      case 'auroramysql':
+        return this.mysqlService;
+      case 'postgres':
+      case 'postgresql':
+      case 'aurora-postgres':
+      case 'aurorapostgres':
+      case 'aurora-postgresql':
+        return this.postgreSQLService;
+      case 'sqlite':
+      case 'better-sqlite3':
+      case 'bettersqlite3':
+        return this.sqliteService;
+      case 'oracle':
+        return this.oracleService;
+      case 'mssql':
+      case 'sqlserver':
+        return this.sqlServerService;
+      case 'cockroachdb':
+      case 'cockroach':
+        return this.cockroachDBService;
+      case 'mongodb':
+      case 'mongo':
+        return this.mongoDBService;
+      case 'sap':
+      case 'sap-hana':
+      case 'saphana':
+        return this.sapHANADatabaseService;
+      default:
+        throw new Error(`不支持的数据库类型: ${type}`);
+    }
+  }
+
+  /**
+   * 保存连接配置到文件
+   */
+  private async saveConnections(connections: ConnectionEntity[]): Promise<void> {
+    try {
+      const dir = path.dirname(this.configPath);
+      if (!fs.existsSync(dir)) {
+        fs.mkdirSync(dir, { recursive: true });
+      }
+      fs.writeFileSync(this.configPath, JSON.stringify(connections, null, 2), 'utf8');
+    } catch (error) {
+      console.error('保存连接配置失败:', error);
+      throw error;
+    }
+  }
+
+  /**
+   * 生成唯一ID
+   */
+  private generateId(): string {
+    return Date.now().toString(36) + Math.random().toString(36).substr(2);
+  }
+}
+
+// 创建全局单例
+let bridgeInstance: DatabaseServiceBridge | null = null;
+
+/**
+ * 获取数据库服务桥接实例
+ */
+export function getDatabaseServiceBridge(): DatabaseServiceBridge {
+  if (!bridgeInstance) {
+    bridgeInstance = new DatabaseServiceBridge();
+  }
+  return bridgeInstance;
+}

package/packages/vscode/src/typings/connection.ts

@@ -0,0 +1,90 @@
+/**
+ * 数据库连接配置
+ */
+export interface DatabaseConnection {
+    id: string;
+    name: string;
+    type: DatabaseType;
+    host: string;
+    port: number;
+    database?: string;
+    username?: string;
+    password?: string;
+    ssl?: boolean;
+    charset?: string;
+    timezone?: string;
+    connectionTimeout?: number;
+    createdAt?: number;
+    updatedAt?: number;
+}
+
+/**
+ * 支持的数据库类型
+ */
+export type DatabaseType =
+    | 'mysql'
+    | 'postgresql'
+    | 'sqlite'
+    | 'sqlserver'
+    | 'oracle'
+    | 'cockroachdb'
+    | 'mongodb'
+    | 'sap_hana';
+
+/**
+ * 数据库类型配置
+ */
+export interface DatabaseTypeConfig {
+    type: DatabaseType;
+    name: string;
+    defaultPort: number;
+    icon: string;
+    driver: string;
+    supports: string[];
+}
+
+/**
+ * 查询结果
+ */
+export interface QueryResult {
+    columns: string[];
+    rows: any[];
+    rowCount: number;
+    executionTime: number;
+}
+
+/**
+ * 表信息
+ */
+export interface TableInfo {
+    name: string;
+    type: 'table' | 'view' | 'system';
+    engine?: string;
+    rows?: number;
+    size?: string;
+    comment?: string;
+}
+
+/**
+ * 列信息
+ */
+export interface ColumnInfo {
+    name: string;
+    type: string;
+    nullable: boolean;
+    key: string;
+    default?: any;
+    extra?: string;
+    comment?: string;
+}
+
+/**
+ * 数据库信息
+ */
+export interface DatabaseInfo {
+    name: string;
+    size?: string;
+    tables?: number;
+    charset?: string;
+    collation?: string;
+}

package/packages/vscode/tsconfig.json

@@ -0,0 +1,21 @@
+{
+  "compilerOptions": {
+    "module": "commonjs",
+    "target": "ES2020",
+    "outDir": "out",
+    "lib": ["ES2020"],
+    "sourceMap": true,
+    "rootDir": "src",
+    "strict": false,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "resolveJsonModule": true,
+    "moduleResolution": "node",
+    "types": ["node", "vscode"],
+    "baseUrl": ".",
+    "noErrorTruncation": true
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", ".vscode-test"]
+}

package/server/index.ts

@@ -259,16 +259,11 @@ export async function handleDatabaseRoutes(pathname: string, body: any) {
 
     // /api/database/alterTable
     if (pathname === '/api/database/alterTable') {
-      const { id, database, tableName, columns, oldColumns } = body;
-      if (!id || !database || !tableName || !columns) throw Error('Missing parameters');
-
-      const sqlStatements = generateAlterTableSQL(tableName, columns, oldColumns);
-      const results = [] as Array<any>;
-      for (const sql of sqlStatements) {
-        const result = await databaseService.executeQuery(id, sql, database);
-        results.push({ sql, result });
-      }
-      return { success: true, statements: sqlStatements, results };
+      const { id, database, tableDiff } = body;
+      if (!id || !database || !tableDiff) throw Error('Missing parameters');
+
+      const result = await databaseService.alterTable(id, database, tableDiff);
+      return result;
     }
 
     // /api/database/insertData

package/server/model/connection.entity.js

@@ -0,0 +1,11 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ConnectionEntity = void 0;
+/**
+ * 数据库连接配置实体
+ * 注意：此实体仅用于代码结构，实际数据保存在本地JSON文件中
+ */
+class ConnectionEntity {
+}
+exports.ConnectionEntity = ConnectionEntity;
+//# sourceMappingURL=connection.entity.js.map

package/server/model/connection.entity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connection.entity.js","sourceRoot":"","sources":["connection.entity.ts"],"names":[],"mappings":";;;AACA;;;GAGG;AACH,MAAa,gBAAgB;CA4D5B;AA5DD,4CA4DC"}

package/server/model/database.entity.js

@@ -0,0 +1,35 @@
+"use strict";
+/**
+ * 数据库信息实体
+ * 用于存储数据库的基本信息和结构
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ForeignKeyEntity = exports.IndexEntity = exports.ColumnEntity = exports.TableEntity = exports.DatabaseEntity = void 0;
+class DatabaseEntity {
+}
+exports.DatabaseEntity = DatabaseEntity;
+/**
+ * 数据库表实体
+ */
+class TableEntity {
+}
+exports.TableEntity = TableEntity;
+/**
+ * 数据库列实体
+ */
+class ColumnEntity {
+}
+exports.ColumnEntity = ColumnEntity;
+/**
+ * 数据库索引实体
+ */
+class IndexEntity {
+}
+exports.IndexEntity = IndexEntity;
+/**
+ * 外键实体
+ */
+class ForeignKeyEntity {
+}
+exports.ForeignKeyEntity = ForeignKeyEntity;
+//# sourceMappingURL=database.entity.js.map

package/server/model/database.entity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"database.entity.js","sourceRoot":"","sources":["database.entity.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,MAAa,cAAc;CAmC1B;AAnCD,wCAmCC;AAED;;GAEG;AACH,MAAa,WAAW;CAsEvB;AAtED,kCAsEC;AAED;;GAEG;AACH,MAAa,YAAY;CA4DxB;AA5DD,oCA4DC;AAED;;GAEG;AACH,MAAa,WAAW;CAyBvB;AAzBD,kCAyBC;AAED;;GAEG;AACH,MAAa,gBAAgB;CA8B5B;AA9BD,4CA8BC"}

package/server/service/connection.service.ts

@@ -41,6 +41,7 @@ export class ConnectionService {
    */
   async getAllConnections(): Promise<ConnectionEntity[]> {
     try {
+      console.log('configPath', this.configPath);
       if (!fs.existsSync(this.configPath)) {
         return [];
       }

package/server/service/database/base.service.ts

@@ -174,7 +174,7 @@ export abstract class BaseDatabaseService {
                 success++;
               } catch (error) {
                 failed++;
-                errors.push({ statement, error: error.message });
+                errors.push({ statement, error: error instanceof Error ? error.message : String(error) });
                 if (!continueOnError) {
                   throw error;
                 }
@@ -189,7 +189,7 @@ export abstract class BaseDatabaseService {
               success++;
             } catch (error) {
               failed++;
-              errors.push({ statement, error: error.message });
+              errors.push({ statement, error: error instanceof Error ? error.message : String(error) });
               if (!continueOnError) {
                 throw error;
               }
@@ -199,7 +199,7 @@ export abstract class BaseDatabaseService {
       } catch (batchError) {
         console.error(`批次执行失败 (${i}-${i + batchSize}):`, batchError);
         if (!continueOnError) {
-          throw new Error(`批次执行失败: ${batchError.message}`);
+          throw new Error(`批次执行失败: ${batchError instanceof Error ? batchError.message : String(batchError)}`);
         }
       }
     }
@@ -360,4 +360,9 @@ export abstract class BaseDatabaseService {
    * 导出表数据到 Excel 文件 - 子类实现
    */
   abstract exportTableDataToExcel(dataSource: DataSource, databaseName: string, tableName: string, options?: any): Promise<string>;
+
+  /**
+   * 修改表结构 - 子类实现
+   */
+  abstract alterTable(dataSource: DataSource, databaseName: string, tableDiff: any): Promise<any>;
 }