+ {!readOnly && (
+
+ } onClick={openCreateIndexModal}>新增
+ } disabled={!selectedIndex} onClick={openEditIndexModal}>修改
+ } danger disabled={!selectedIndex} onClick={handleDeleteIndex}>删除
+ {!supportsMysqlSchemaOps() && (
+
+ 当前数据库暂不支持索引编辑，仅支持查看
+
+ )}
+ {supportsMysqlSchemaOps() && selectedIndex && (
+
+ 已选择：{selectedIndex.name}
+
+ )}
+
+ )}
+
+ 索引数：{groupedIndexes.length}，索引字段：{groupedIndexFieldCount}
+
+
(
+
+
+ {text}
+
+
+ ),
+ },
+ {
+ title: '字段',
+ dataIndex: 'columnNames',
+ key: 'columnNames',
+ render: (columnNames: string[]) => {
+ if (!columnNames || columnNames.length === 0) {
+ return '-';
+ }
+ return (
+
+ {columnNames.map((columnName, idx) => (
+
+ {columnName}
+
+ ))}
+
+ );
+ }
+ },
+ {
+ title: '索引类型',
+ dataIndex: 'indexType',
+ key: 'indexType',
+ width: 140,
+ render: (text: string) => text || '-',
+ },
+ {
+ title: '唯一性',
+ dataIndex: 'nonUnique',
+ key: 'nonUnique',
+ width: 110,
+ render: (v: number) => (
+
+ {v === 0 ? '唯一' : '普通'}
+
+ ),
+ },
+ ]}
+ rowKey="key"
+ size="small"
+ pagination={false}
+ loading={loading}
+ scroll={{ x: 960, y: tableHeight }}
+ rowSelection={{
+ type: 'radio',
+ selectedRowKeys: selectedIndex ? [selectedIndex.key] : [],
+ onChange: (_, selectedRows) => setSelectedIndex((selectedRows[0] as IndexDisplayRow) || null),
+ }}
+ onRow={(record) => ({
+ onClick: () => {
+ if (selectedIndex?.key === record.key) {
+ setSelectedIndex(null);
+ } else {
+ setSelectedIndex(record);
+ }
+ },
+ style: { cursor: 'pointer' }
+ })}
+ />
+
)
},
{
key: 'foreignKeys',
label: '外键',
children: (
-
+
+ {!readOnly && (
+
+ } onClick={openCreateForeignKeyModal}>新增
+ } disabled={!selectedForeignKey} onClick={openEditForeignKeyModal}>修改
+ } danger disabled={!selectedForeignKey} onClick={handleDeleteForeignKey}>删除
+ {!supportsMysqlSchemaOps() && (
+
+ 当前数据库暂不支持外键编辑，仅支持查看
+
+ )}
+ {supportsMysqlSchemaOps() && selectedForeignKey && (
+
+ 已选择：{selectedForeignKey.constraintName}
+
+ )}
+
+ )}
+
vals?.length ? vals.join(', ') : '-',
+ },
+ { title: '参考表', dataIndex: 'refTableName', key: 'refTableName', width: 220 },
+ {
+ title: '参考字段',
+ dataIndex: 'refColumnNames',
+ key: 'refColumnNames',
+ render: (vals: string[]) => vals?.length ? vals.join(', ') : '-',
+ },
+ ]}
+ rowKey="key"
+ size="small"
+ pagination={false}
+ loading={loading}
+ scroll={{ x: 980, y: tableHeight }}
+ rowSelection={{
+ type: 'radio',
+ selectedRowKeys: selectedForeignKey ? [selectedForeignKey.key] : [],
+ onChange: (_, selectedRows) => setSelectedForeignKey((selectedRows[0] as ForeignKeyDisplayRow) || null),
+ }}
+ onRow={(record) => ({
+ onClick: () => {
+ if (selectedForeignKey?.key === record.key) {
+ setSelectedForeignKey(null);
+ } else {
+ setSelectedForeignKey(record);
+ }
+ },
+ style: { cursor: 'pointer' }
+ })}
+ />
+
)
},
{
@@ -1071,7 +1932,7 @@ ${selectedTrigger.statement}`;
)
}
] : []),
- ...(readOnly ? [{
+ ...(!isNewTable ? [{
key: 'ddl',
label: 'DDL',
icon: ,
@@ -1087,9 +1948,10 @@ ${selectedTrigger.statement}`;
minimap: { enabled: false },
fontSize: 14,
lineNumbers: 'on',
- scrollBeyondLastLine: false,
+ scrollBeyondLastLine: true,
wordWrap: 'on',
automaticLayout: true,
+ padding: { top: 8, bottom: 24 },
}}
/>
@@ -1098,6 +1960,200 @@ ${selectedTrigger.statement}`;
]}
/>
+ {
+ if (commentEditorColumnKey) {
+ handleColumnChange(commentEditorColumnKey, 'comment', commentEditorValue);
+ }
+ closeCommentEditor();
+ }}
+ okText="应用"
+ cancelText="取消"
+ width={640}
+ destroyOnClose
+ >
+ setCommentEditorValue(e.target.value)}
+ autoSize={{ minRows: 8, maxRows: 18 }}
+ placeholder="请输入字段注释"
+ maxLength={2000}
+ />
+
+
+ setIsCopyColumnsModalOpen(false)}
+ onOk={handleExecuteCopySelectedColumns}
+ okText="创建新表"
+ cancelText="取消"
+ confirmLoading={copyExecuting}
+ width={560}
+ >
+
+
+ 已选择字段：{selectedColumns.length}
+
+ setCopyTableName(e.target.value)}
+ maxLength={128}
+ />
+
+
+
+
+
+ setIsTableCommentModalOpen(false)}
+ onOk={handleSaveTableComment}
+ okText="保存"
+ cancelText="取消"
+ confirmLoading={tableCommentSaving}
+ width={640}
+ >
+ setTableCommentDraft(e.target.value)}
+ autoSize={{ minRows: 5, maxRows: 12 }}
+ placeholder="请输入表备注"
+ maxLength={2048}
+ />
+
+ 当前备注：{tableComment || '(空)'}
+
+
+
+ setIsIndexModalOpen(false)}
+ onOk={handleSubmitIndex}
+ okText={indexModalMode === 'create' ? '创建' : '保存'}
+ cancelText="取消"
+ confirmLoading={indexSaving}
+ width={620}
+ >
+
+ setIndexForm(prev => ({ ...prev, name: e.target.value }))}
+ maxLength={128}
+ disabled={indexForm.kind === 'PRIMARY'}
+ />
+
+
+
+ setIsForeignKeyModalOpen(false)}
+ onOk={handleSubmitForeignKey}
+ okText={foreignKeyModalMode === 'create' ? '创建' : '保存'}
+ cancelText="取消"
+ confirmLoading={foreignKeySaving}
+ width={700}
+ >
+
+ setForeignKeyForm(prev => ({ ...prev, constraintName: e.target.value }))}
+ maxLength={128}
+ />
+
+
+
;
tableSortPreference: Record;
@@ -287,7 +293,7 @@ interface AppState {
setTheme: (theme: 'light' | 'dark') => void;
setAppearance: (appearance: Partial<{ opacity: number; blur: number }>) => void;
setSqlFormatOptions: (options: { keywordCase: 'upper' | 'lower' }) => void;
- setQueryOptions: (options: Partial<{ maxRows: number }>) => void;
+ setQueryOptions: (options: Partial) => void;
addSqlLog: (log: SqlLog) => void;
clearSqlLogs: () => void;
@@ -326,13 +332,15 @@ const sanitizeSqlFormatOptions = (value: unknown): { keywordCase: 'upper' | 'low
return { keywordCase: raw.keywordCase === 'lower' ? 'lower' : 'upper' };
};
-const sanitizeQueryOptions = (value: unknown): { maxRows: number } => {
+const sanitizeQueryOptions = (value: unknown): QueryOptions => {
const raw = (value && typeof value === 'object') ? value as Record : {};
const maxRows = Number(raw.maxRows);
+ const showColumnComment = typeof raw.showColumnComment === 'boolean' ? raw.showColumnComment : true;
+ const showColumnType = typeof raw.showColumnType === 'boolean' ? raw.showColumnType : true;
if (!Number.isFinite(maxRows) || maxRows <= 0) {
- return { maxRows: 5000 };
+ return { maxRows: 5000, showColumnComment, showColumnType };
}
- return { maxRows: Math.min(50000, Math.trunc(maxRows)) };
+ return { maxRows: Math.min(50000, Math.trunc(maxRows)), showColumnComment, showColumnType };
};
const sanitizeTableAccessCount = (value: unknown): Record => {
@@ -383,7 +391,7 @@ export const useStore = create()(
theme: 'light',
appearance: { ...DEFAULT_APPEARANCE },
sqlFormatOptions: { keywordCase: 'upper' },
- queryOptions: { maxRows: 5000 },
+ queryOptions: { maxRows: 5000, showColumnComment: true, showColumnType: true },
sqlLogs: [],
tableAccessCount: {},
tableSortPreference: {},
diff --git a/internal/app/app.go b/internal/app/app.go
index 80c5541..248093c 100644
--- a/internal/app/app.go
+++ b/internal/app/app.go
@@ -133,8 +133,17 @@ func formatConnSummary(config connection.ConnectionConfig) string {
}
var b strings.Builder
- b.WriteString(fmt.Sprintf("类型=%s 地址=%s:%d 数据库=%s 用户=%s 超时=%ds",
- config.Type, config.Host, config.Port, dbName, config.User, timeoutSeconds))
+ normalizedType := strings.ToLower(strings.TrimSpace(config.Type))
+ if normalizedType == "sqlite" || normalizedType == "duckdb" {
+ path := strings.TrimSpace(config.Host)
+ if path == "" {
+ path = "(未配置)"
+ }
+ b.WriteString(fmt.Sprintf("类型=%s 路径=%s 超时=%ds", config.Type, path, timeoutSeconds))
+ } else {
+ b.WriteString(fmt.Sprintf("类型=%s 地址=%s:%d 数据库=%s 用户=%s 超时=%ds",
+ config.Type, config.Host, config.Port, dbName, config.User, timeoutSeconds))
+ }
if len(config.Hosts) > 0 {
b.WriteString(fmt.Sprintf(" 节点数=%d", len(config.Hosts)))
diff --git a/internal/app/methods_driver.go b/internal/app/methods_driver.go
index 7c2cec9..61ae632 100644
--- a/internal/app/methods_driver.go
+++ b/internal/app/methods_driver.go
@@ -1,6 +1,7 @@
package app
import (
+ "archive/zip"
"crypto/sha256"
"encoding/hex"
"encoding/json"
@@ -111,13 +112,20 @@ type driverReleaseAssetSizeCacheEntry struct {
Err string
}
+type driverBundleAssetIndex struct {
+ Assets map[string]int64 `json:"assets"`
+}
+
const (
// 默认使用内置 manifest，避免依赖网络与外部仓库 404。
defaultDriverManifestURLValue = "builtin://manifest"
+ optionalDriverBundleAssetName = "GoNavi-DriverAgents.zip"
+ optionalDriverBundleIndexAssetName = "GoNavi-DriverAgents-Index.json"
driverManifestCacheTTL = 5 * time.Minute
driverReleaseAssetSizeCacheTTL = 30 * time.Minute
driverReleaseAssetSizeErrorCacheTTL = 30 * time.Second
driverReleaseAssetSizeProbeTimeout = 4 * time.Second
+ driverBundleIndexMaxSize = 1 << 20
driverManifestMaxSize = 2 << 20
driverChecksumPolicyStrict = "strict"
driverChecksumPolicyWarn = "warn"
@@ -1123,6 +1131,19 @@ func ensureOptionalDriverAgentBinary(a *App, definition driverDefinition, execut
downloadErrs = append(downloadErrs, fmt.Sprintf("%s: %s", candidateURL, strings.TrimSpace(dlErr.Error())))
}
}
+ bundleURLs := resolveOptionalDriverBundleDownloadURLs()
+ if len(bundleURLs) > 0 {
+ for _, bundleURL := range bundleURLs {
+ if a != nil {
+ a.emitDriverDownloadProgress(driverType, "downloading", 20, 100, fmt.Sprintf("从驱动总包提取 %s 代理", displayName))
+ }
+ source, hash, bundleErr := downloadOptionalDriverAgentFromBundle(a, definition, bundleURL, executablePath)
+ if bundleErr == nil {
+ return source, hash, nil
+ }
+ downloadErrs = append(downloadErrs, fmt.Sprintf("%s: %s", bundleURL, strings.TrimSpace(bundleErr.Error())))
+ }
+ }
if a != nil {
a.emitDriverDownloadProgress(driverType, "downloading", 92, 100, "未命中预编译包，尝试开发态本地构建")
}
@@ -1176,6 +1197,112 @@ func downloadOptionalDriverAgentBinary(a *App, definition driverDefinition, urlT
return hash, nil
}
+func downloadOptionalDriverAgentFromBundle(a *App, definition driverDefinition, bundleURL, executablePath string) (string, string, error) {
+ driverType := normalizeDriverType(definition.Type)
+ displayName := resolveDriverDisplayName(definition)
+ trimmedURL := strings.TrimSpace(bundleURL)
+ if trimmedURL == "" {
+ return "", "", fmt.Errorf("驱动总包下载地址为空")
+ }
+
+ bundleTempPath := executablePath + ".bundle.zip.tmp"
+ _ = os.Remove(bundleTempPath)
+ _, err := downloadFileWithHash(trimmedURL, bundleTempPath, func(downloaded, total int64) {
+ if a == nil {
+ return
+ }
+ scaledDownloaded, scaledTotal := scaleProgress(downloaded, total, 20, 78)
+ a.emitDriverDownloadProgress(driverType, "downloading", scaledDownloaded, scaledTotal, fmt.Sprintf("下载 %s 驱动总包", displayName))
+ })
+ if err != nil {
+ _ = os.Remove(bundleTempPath)
+ return "", "", fmt.Errorf("下载驱动总包失败：%w", err)
+ }
+ defer func() { _ = os.Remove(bundleTempPath) }()
+
+ reader, err := zip.OpenReader(bundleTempPath)
+ if err != nil {
+ return "", "", fmt.Errorf("打开驱动总包失败：%w", err)
+ }
+ defer reader.Close()
+
+ entryPath := optionalDriverBundleEntryPath(driverType)
+ expectedBaseName := optionalDriverReleaseAssetName(driverType)
+ findEntry := func() *zip.File {
+ for _, file := range reader.File {
+ name := filepath.ToSlash(strings.TrimPrefix(strings.TrimSpace(file.Name), "./"))
+ if name == entryPath {
+ return file
+ }
+ }
+ for _, file := range reader.File {
+ name := filepath.ToSlash(strings.TrimPrefix(strings.TrimSpace(file.Name), "./"))
+ if strings.EqualFold(name, entryPath) {
+ return file
+ }
+ }
+ for _, file := range reader.File {
+ name := filepath.ToSlash(strings.TrimPrefix(strings.TrimSpace(file.Name), "./"))
+ if strings.EqualFold(filepath.Base(name), expectedBaseName) {
+ return file
+ }
+ }
+ return nil
+ }
+
+ entry := findEntry()
+ if entry == nil {
+ return "", "", fmt.Errorf("驱动总包内未找到 %s（期望路径 %s）", displayName, entryPath)
+ }
+ if a != nil {
+ a.emitDriverDownloadProgress(driverType, "downloading", 84, 100, fmt.Sprintf("解压 %s 驱动代理", displayName))
+ }
+
+ src, err := entry.Open()
+ if err != nil {
+ return "", "", fmt.Errorf("读取驱动总包条目失败：%w", err)
+ }
+ defer src.Close()
+
+ tempPath := executablePath + ".tmp"
+ _ = os.Remove(tempPath)
+ dst, err := os.Create(tempPath)
+ if err != nil {
+ return "", "", fmt.Errorf("创建驱动代理临时文件失败：%w", err)
+ }
+ if _, err := io.Copy(dst, src); err != nil {
+ dst.Close()
+ _ = os.Remove(tempPath)
+ return "", "", fmt.Errorf("写入驱动代理失败：%w", err)
+ }
+ if err := dst.Sync(); err != nil {
+ dst.Close()
+ _ = os.Remove(tempPath)
+ return "", "", fmt.Errorf("落盘驱动代理失败：%w", err)
+ }
+ if err := dst.Close(); err != nil {
+ _ = os.Remove(tempPath)
+ return "", "", fmt.Errorf("关闭驱动代理文件失败：%w", err)
+ }
+ if chmodErr := os.Chmod(tempPath, 0o755); chmodErr != nil && stdRuntime.GOOS != "windows" {
+ _ = os.Remove(tempPath)
+ return "", "", fmt.Errorf("设置驱动代理权限失败：%w", chmodErr)
+ }
+ if err := os.Rename(tempPath, executablePath); err != nil {
+ _ = os.Remove(tempPath)
+ return "", "", fmt.Errorf("替换驱动代理失败：%w", err)
+ }
+ if chmodErr := os.Chmod(executablePath, 0o755); chmodErr != nil && stdRuntime.GOOS != "windows" {
+ return "", "", fmt.Errorf("设置驱动代理权限失败：%w", chmodErr)
+ }
+ hash, err := hashFileSHA256(executablePath)
+ if err != nil {
+ return "", "", fmt.Errorf("计算驱动代理摘要失败：%w", err)
+ }
+ source := fmt.Sprintf("%s#%s", trimmedURL, filepath.ToSlash(strings.TrimPrefix(strings.TrimSpace(entry.Name), "./")))
+ return source, hash, nil
+}
+
func buildOptionalDriverAgentFromSource(definition driverDefinition, executablePath string) (string, error) {
driverType := normalizeDriverType(definition.Type)
displayName := resolveDriverDisplayName(definition)
@@ -1282,6 +1409,46 @@ func optionalDriverReleaseAssetName(driverType string) string {
return name
}
+func optionalDriverBundlePlatformDir(goos string) string {
+ switch strings.ToLower(strings.TrimSpace(goos)) {
+ case "windows":
+ return "Windows"
+ case "darwin":
+ return "MacOS"
+ case "linux":
+ return "Linux"
+ default:
+ return "Unknown"
+ }
+}
+
+func optionalDriverBundleEntryPath(driverType string) string {
+ return filepath.ToSlash(filepath.Join(optionalDriverBundlePlatformDir(stdRuntime.GOOS), optionalDriverReleaseAssetName(driverType)))
+}
+
+func resolveOptionalDriverBundleDownloadURLs() []string {
+ candidates := make([]string, 0, 2)
+ seen := make(map[string]struct{}, 2)
+ appendURL := func(value string) {
+ trimmed := strings.TrimSpace(value)
+ if trimmed == "" {
+ return
+ }
+ if _, ok := seen[trimmed]; ok {
+ return
+ }
+ seen[trimmed] = struct{}{}
+ candidates = append(candidates, trimmed)
+ }
+
+ currentVersion := normalizeVersion(getCurrentVersion())
+ if currentVersion != "" && currentVersion != "0.0.0" {
+ appendURL(fmt.Sprintf("https://github.com/Syngnat/GoNavi/releases/download/v%s/%s", currentVersion, optionalDriverBundleAssetName))
+ }
+ appendURL(fmt.Sprintf("https://github.com/Syngnat/GoNavi/releases/latest/download/%s", optionalDriverBundleAssetName))
+ return candidates
+}
+
func resolveOptionalDriverAgentDownloadURLs(definition driverDefinition, rawURL string) []string {
driverType := normalizeDriverType(definition.Type)
candidates := make([]string, 0, 3)
@@ -1541,6 +1708,15 @@ func loadReleaseAssetSizesCached(cacheKey string, fetch func() (*githubRelease,
entry.Err = err.Error()
} else {
entry.SizeByKey = buildReleaseAssetSizeMap(release)
+ if indexSizes, indexErr := fetchDriverBundleAssetSizeIndex(release); indexErr == nil {
+ for name, size := range indexSizes {
+ trimmedName := strings.TrimSpace(name)
+ if trimmedName == "" || size <= 0 {
+ continue
+ }
+ entry.SizeByKey[trimmedName] = size
+ }
+ }
}
driverReleaseSizeMu.Lock()
@@ -1568,6 +1744,50 @@ func buildReleaseAssetSizeMap(release *githubRelease) map[string]int64 {
return sizes
}
+func fetchDriverBundleAssetSizeIndex(release *githubRelease) (map[string]int64, error) {
+ if release == nil {
+ return nil, fmt.Errorf("release 为空")
+ }
+ indexURL := ""
+ for _, asset := range release.Assets {
+ if strings.EqualFold(strings.TrimSpace(asset.Name), optionalDriverBundleIndexAssetName) {
+ indexURL = strings.TrimSpace(asset.BrowserDownloadURL)
+ break
+ }
+ }
+ if indexURL == "" {
+ return nil, fmt.Errorf("未找到驱动总包索引资产")
+ }
+
+ client := &http.Client{Timeout: driverReleaseAssetSizeProbeTimeout}
+ req, err := http.NewRequest(http.MethodGet, indexURL, nil)
+ if err != nil {
+ return nil, err
+ }
+ req.Header.Set("User-Agent", "GoNavi-DriverManager")
+ req.Header.Set("Accept", "application/json")
+
+ resp, err := client.Do(req)
+ if err != nil {
+ return nil, err
+ }
+ defer resp.Body.Close()
+ if resp.StatusCode != http.StatusOK {
+ return nil, fmt.Errorf("拉取驱动总包索引失败：HTTP %d", resp.StatusCode)
+ }
+
+ limited := io.LimitReader(resp.Body, driverBundleIndexMaxSize)
+ decoder := json.NewDecoder(limited)
+ var index driverBundleAssetIndex
+ if err := decoder.Decode(&index); err != nil {
+ return nil, fmt.Errorf("解析驱动总包索引失败：%w", err)
+ }
+ if len(index.Assets) == 0 {
+ return nil, fmt.Errorf("驱动总包索引为空")
+ }
+ return index.Assets, nil
+}
+
func fetchLatestReleaseForDriverAssets() (*githubRelease, error) {
return fetchDriverReleaseByURL(updateAPIURL)
}
diff --git a/internal/db/sqlite_impl.go b/internal/db/sqlite_impl.go
index 582937a..e8e8c06 100644
--- a/internal/db/sqlite_impl.go
+++ b/internal/db/sqlite_impl.go
@@ -6,6 +6,9 @@ import (
"context"
"database/sql"
"fmt"
+ "os"
+ "path/filepath"
+ "strconv"
"strings"
"time"
@@ -21,7 +24,14 @@ type SQLiteDB struct {
}
func (s *SQLiteDB) Connect(config connection.ConnectionConfig) error {
- dsn := config.Host
+ dsn, err := resolveSQLiteDSN(config)
+ if err != nil {
+ return err
+ }
+ if err := ensureSQLiteParentDir(dsn); err != nil {
+ return err
+ }
+
db, err := sql.Open("sqlite", dsn)
if err != nil {
return fmt.Errorf("打开数据库连接失败：%w", err)
@@ -31,11 +41,140 @@ func (s *SQLiteDB) Connect(config connection.ConnectionConfig) error {
// Force verification
if err := s.Ping(); err != nil {
+ _ = db.Close()
+ s.conn = nil
return fmt.Errorf("连接建立后验证失败：%w", err)
}
return nil
}
+func resolveSQLiteDSN(config connection.ConnectionConfig) (string, error) {
+ dsn := strings.TrimSpace(config.Host)
+ if dsn == "" {
+ dsn = strings.TrimSpace(config.Database)
+ }
+ dsn = normalizeSQLitePath(dsn)
+ if dsn == "" {
+ return "", fmt.Errorf("SQLite 需要本地数据库文件路径（例如 /path/to/demo.sqlite）")
+ }
+ if strings.EqualFold(dsn, ":memory:") {
+ return dsn, nil
+ }
+ if looksLikeHostPort(dsn) {
+ return "", fmt.Errorf("SQLite 需要本地数据库文件路径，当前输入看起来是主机地址：%s", dsn)
+ }
+ return dsn, nil
+}
+
+func normalizeSQLitePath(raw string) string {
+ text := strings.TrimSpace(raw)
+ if strings.HasPrefix(text, "/") && len(text) > 3 && isWindowsDrivePath(text[1:]) {
+ text = text[1:]
+ }
+ if isWindowsDrivePath(text) {
+ text = trimLegacyPortSuffix(text)
+ }
+ return text
+}
+
+func isWindowsDrivePath(path string) bool {
+ if len(path) < 3 {
+ return false
+ }
+ drive := path[0]
+ if !((drive >= 'a' && drive <= 'z') || (drive >= 'A' && drive <= 'Z')) {
+ return false
+ }
+ if path[1] != ':' {
+ return false
+ }
+ sep := path[2]
+ return sep == '\\' || sep == '/'
+}
+
+func trimLegacyPortSuffix(path string) string {
+ normalized := path
+ for {
+ idx := strings.LastIndex(normalized, ":")
+ if idx <= 1 || idx+1 >= len(normalized) {
+ return normalized
+ }
+ suffix := normalized[idx+1:]
+ validDigits := true
+ for _, ch := range suffix {
+ if ch < '0' || ch > '9' {
+ validDigits = false
+ break
+ }
+ }
+ if !validDigits {
+ return normalized
+ }
+ normalized = normalized[:idx]
+ }
+}
+
+func looksLikeHostPort(raw string) bool {
+ text := strings.TrimSpace(raw)
+ if text == "" {
+ return false
+ }
+ if strings.ContainsAny(text, `/\`) {
+ return false
+ }
+ if strings.HasPrefix(strings.ToLower(text), "file:") {
+ return false
+ }
+ if strings.HasPrefix(text, "[") {
+ closing := strings.LastIndex(text, "]")
+ if closing <= 0 || closing+1 >= len(text) {
+ return false
+ }
+ portText := strings.TrimSpace(strings.TrimPrefix(text[closing+1:], ":"))
+ return isValidPortText(portText)
+ }
+ if strings.Count(text, ":") != 1 {
+ return false
+ }
+ split := strings.LastIndex(text, ":")
+ if split <= 0 || split+1 >= len(text) {
+ return false
+ }
+ return isValidPortText(strings.TrimSpace(text[split+1:]))
+}
+
+func isValidPortText(text string) bool {
+ port, err := strconv.Atoi(text)
+ return err == nil && port > 0 && port <= 65535
+}
+
+func ensureSQLiteParentDir(dsn string) error {
+ text := strings.TrimSpace(dsn)
+ if text == "" || strings.EqualFold(text, ":memory:") {
+ return nil
+ }
+ // file: URI 由驱动处理，避免在这里误判路径格式。
+ if strings.HasPrefix(strings.ToLower(text), "file:") {
+ return nil
+ }
+ path := text
+ if idx := strings.Index(path, "?"); idx >= 0 {
+ path = path[:idx]
+ }
+ path = strings.TrimSpace(path)
+ if path == "" {
+ return nil
+ }
+ dir := filepath.Dir(path)
+ if dir == "." || dir == "" {
+ return nil
+ }
+ if err := os.MkdirAll(dir, 0o755); err != nil {
+ return fmt.Errorf("创建 SQLite 数据文件目录失败：%w", err)
+ }
+ return nil
+}
+
func (s *SQLiteDB) Close() error {
if s.conn != nil {
return s.conn.Close()
diff --git a/internal/db/sqlite_impl_test.go b/internal/db/sqlite_impl_test.go
new file mode 100644
index 0000000..15745b8
--- /dev/null
+++ b/internal/db/sqlite_impl_test.go
@@ -0,0 +1,79 @@
+//go:build gonavi_full_drivers || gonavi_sqlite_driver
+
+package db
+
+import (
+ "os"
+ "path/filepath"
+ "strings"
+ "testing"
+
+ "GoNavi-Wails/internal/connection"
+)
+
+func TestResolveSQLiteDSNRejectsHostPort(t *testing.T) {
+ _, err := resolveSQLiteDSN(connection.ConnectionConfig{Type: "sqlite", Host: "localhost:3306"})
+ if err == nil {
+ t.Fatalf("期望拦截 host:port 输入")
+ }
+ if !strings.Contains(err.Error(), "本地数据库文件路径") {
+ t.Fatalf("错误提示不符合预期: %v", err)
+ }
+}
+
+func TestResolveSQLiteDSNFallbackDatabase(t *testing.T) {
+ dsn, err := resolveSQLiteDSN(connection.ConnectionConfig{Type: "sqlite", Database: "/tmp/demo.sqlite"})
+ if err != nil {
+ t.Fatalf("解析 DSN 失败: %v", err)
+ }
+ if dsn != "/tmp/demo.sqlite" {
+ t.Fatalf("期望使用 database 作为 DSN，实际=%s", dsn)
+ }
+}
+
+func TestResolveSQLiteDSNNormalizesWindowsLegacyPath(t *testing.T) {
+ dsn, err := resolveSQLiteDSN(connection.ConnectionConfig{Type: "sqlite", Host: `F:\py\py\history.db:3306:3306`})
+ if err != nil {
+ t.Fatalf("解析 DSN 失败: %v", err)
+ }
+ if dsn != `F:\py\py\history.db` {
+ t.Fatalf("期望清理历史端口污染，实际=%s", dsn)
+ }
+}
+
+func TestResolveSQLiteDSNNormalizesWindowsPathWithLeadingSlash(t *testing.T) {
+ dsn, err := resolveSQLiteDSN(connection.ConnectionConfig{Type: "sqlite", Host: `/F:\py\py\history.db:3306`})
+ if err != nil {
+ t.Fatalf("解析 DSN 失败: %v", err)
+ }
+ if dsn != `F:\py\py\history.db` {
+ t.Fatalf("期望清理前导斜杠与端口污染，实际=%s", dsn)
+ }
+}
+
+func TestEnsureSQLiteParentDirCreatesNestedDir(t *testing.T) {
+ base := t.TempDir()
+ target := filepath.Join(base, "nested", "child", "demo.sqlite")
+ if err := ensureSQLiteParentDir(target); err != nil {
+ t.Fatalf("创建目录失败: %v", err)
+ }
+ info, err := os.Stat(filepath.Dir(target))
+ if err != nil {
+ t.Fatalf("目录不存在: %v", err)
+ }
+ if !info.IsDir() {
+ t.Fatalf("目标不是目录: %s", filepath.Dir(target))
+ }
+}
+
+func TestLooksLikeHostPort(t *testing.T) {
+ if !looksLikeHostPort("localhost:3306") {
+ t.Fatalf("localhost:3306 应识别为 host:port")
+ }
+ if looksLikeHostPort("/tmp/demo.sqlite") {
+ t.Fatalf("/tmp/demo.sqlite 不应识别为 host:port")
+ }
+ if looksLikeHostPort(`C:\sqlite\demo.db`) {
+ t.Fatalf("Windows 路径不应识别为 host:port")
+ }
+}