get db type from plugins

chat-query/components/import_modal.js

 import { Modal, Notification, Select, Tabs, Spin } from '@arco-design/web-react';
 import { Parser } from '@dbml/core';
-import { useRef, useState } from 'react';
+import { useCallback, useRef, useState } from 'react';
 import { nanoid } from 'nanoid';
 import Editor from '@monaco-editor/react';
 import graphState from '../hooks/use-graph-state';
@@ -434,54 +434,65 @@ export default function ImportModal({ showModal, onCloseModal, cb = _p => { }, t
     const [dbList, setDbList] = useState([]);
     const searchParams = useSearchParams();
     const applicationId = searchParams.get('applicationId');
+    const organizationId = searchParams.get('organizationId');
 
-    useEffect(() => {
-        if (type === 5 && applicationId) {
-            setLoading(true); // 开始加载时显示loading
-            fetch(`/ziwei/api/v1/datasources?applicationId=${applicationId}`, {
+    const fetchDatasources = useCallback(async (applicationId, organizationId) => {
+        try {
+            const [response, pluginsResponse] = await Promise.all([fetch(`/ziwei/api/v1/datasources?applicationId=${applicationId}`, {
                 method: "GET",
                 credentials: "include",
-            }).finally(() => {
-                setLoading(false); // 请求完成后隐藏loading
-            })
-                .then(res => res.json())
-                .then(async data => {
-                    if (data?.data && Array.isArray(data?.data)) {
-                        const formattedData = await Promise.all(data.data.map(async item => {
-                            // 从 datasourceConfiguration 中获取连接信息
-                            const endpoint = item.datasourceConfiguration?.endpoints?.[0] || {};
-                            // 获取认证信息，包含加密的密钥
-                            const auth = item.datasourceConfiguration?.authentication || {};
-                            // 如果存在加密密钥，需要进行解密
-                            if (auth.privateKey && auth.encryptedAesKey) {
-                                try {
-                                    // 解密密码
-                                    auth.password = await decryptMessage(auth.privateKey, auth.encryptedAesKey)
-                                } catch (error) {
-                                    console.error('Failed to decrypt keys:', error);
-                                    Message.error(t('Failed to decrypt database credentials'));
-                                }
-                            }
+            }), fetch(`/ziwei/api/v1/plugins?organizationId==${organizationId}`, {
+                method: "GET",
+                credentials: "include",
+            })]);
+            const data = await response.json();
+            const plugins = await pluginsResponse.json();
+            const pluginsMap = _.keyBy(plugins?.data, 'id');
 
-                            return {
-                                host: endpoint.host || '127.0.0.1',
-                                port: endpoint.port || 3306,
-                                user: auth.username || 'root',
-                                password: auth.password || '',
-                                database: auth.databaseName || '',
-                                name: item.name || '',
-                                client: item.dbType === 1 ? 'oracle' : 'mysql2' // 根据 dbType 判断数据库类型
-                            };
-                        }));
-                        setDbList(formattedData);
-                        if (formattedData.length > 0) {
-                            form.setFieldsValue(formattedData[0]);
+            if (data?.data && Array.isArray(data?.data)) {
+                return await Promise.all(data.data.map(async item => {
+                    const endpoint = item.datasourceConfiguration?.endpoints?.[0] || {};
+                    const auth = item.datasourceConfiguration?.authentication || {};
+                    if (auth.privateKey && auth.encryptedAesKey) {
+                        try {
+                            auth.password = await decryptMessage(auth.privateKey, auth.encryptedAesKey);
+                        } catch (error) {
+                            console.error('Failed to decrypt keys:', error);
+                            Message.error(t('Failed to decrypt database credentials'));
                         }
                     }
+
+                    return {
+                        host: endpoint.host || '127.0.0.1',
+                        port: endpoint.port || 3306,
+                        user: auth.username || 'root',
+                        password: auth.password || '',
+                        database: auth.databaseName || '',
+                        name: item.name || '',
+                        client: pluginsMap[item.pluginId]?.name === 'Oracle' ? 'oracle' : 'mysql2'
+                    };
+                }));
+            }
+            return [];
+        } catch (error) {
+            console.error('Error fetching datasources:', error);
+            Message.error(t('Failed to fetch datasources'));
+            return [];
+        }
+    }, [t])
+
+    useEffect(() => {
+        if (type === 5 && applicationId) {
+            setLoading(true);
+            fetchDatasources(applicationId, organizationId)
+                .then(formattedData => {
+                    setDbList(formattedData);
+                    if (formattedData.length > 0) {
+                        form.setFieldsValue(formattedData[0]);
+                    }
                 })
-                .catch(error => {
-                    console.error('Error fetching datasources:', error);
-                    Message.error(t('Failed to fetch datasources'));
+                .finally(() => {
+                    setLoading(false);
                 });
         }
     }, [type, applicationId]);