feat: add Dameng database client support

Author: Aaron3S

ui/chen/utils/sqlEditor.test.ts

@@ -1,11 +1,12 @@
 import type { CompletionSource } from "@codemirror/autocomplete";
 
 import { CompletionContext } from "@codemirror/autocomplete";
-import { PostgreSQL, sql } from "@codemirror/lang-sql";
+import { PLSQL, PostgreSQL, sql } from "@codemirror/lang-sql";
 import { EditorState } from "@codemirror/state";
 import { describe, expect, it } from "vitest";
 import {
   chenSqlExtensions,
+  chenSqlDialect,
   chenSqlStatementAtCursor,
   chenSqlStatementRanges,
   executableChenSql
@@ -50,6 +51,11 @@ describe("sql editor execution targets", () => {
 });
 
 describe("sql editor keyword completion", () => {
+  it("uses the PL/SQL grammar for Dameng protocol names", () => {
+    expect(chenSqlDialect("dameng")).toBe(PLSQL);
+    expect(chenSqlDialect("dm")).toBe(PLSQL);
+  });
+
   async function completionLabels(keywordCase: "lower" | "upper") {
     const state = EditorState.create({
       doc: "sel",

ui/chen/utils/sqlEditor.ts

@@ -20,6 +20,8 @@ export function chenSqlDialect(dbType: string) {
     case "postgresql":
       return PostgreSQL;
     case "oracle":
+    case "dameng":
+    case "dm":
       return PLSQL;
     case "sqlserver":
     case "sql server":

ui/chen/utils/sqlFormat.ts

@@ -11,7 +11,8 @@ const formatterLanguageMap = {
   db2: "db2",
   // ponytail: Dameng reuses PL/SQL only for display formatting. Add a dedicated
   // formatter before supporting stored procedures or Dameng-specific DDL.
-  dameng: "plsql"
+  dameng: "plsql",
+  dm: "plsql"
 } as const;
 
 export function formatChenSql(statement: string, dbType: string) {

ui/components/ScriptEditor.client.vue

@@ -27,7 +27,7 @@ function languageExtension(module: string): Extension {
   if (module === "mariadb") return sql({ dialect: MariaSQL });
   if (module === "postgresql") return sql({ dialect: PostgreSQL });
   if (module === "sqlserver") return sql({ dialect: MSSQL });
-  if (module === "oracle") return sql({ dialect: PLSQL });
+  if (["oracle", "dameng"].includes(module)) return sql({ dialect: PLSQL });
   return [];
 }
 

ui/components/SideBar/assetTreeNode.vue

@@ -68,7 +68,7 @@ const typeGroupIcon = computed(() => {
   const hasExact = (...keywords: string[]) => iconCandidates.value.some((value) => keywords.includes(value));
 
   if (has("k8s", "kubernetes", "container")) return "i-lucide-container";
-  if (has("database", "mysql", "mariadb", "oracle", "postgres", "sqlserver", "redis", "mongodb", "db2")) {
+  if (has("database", "mysql", "mariadb", "oracle", "dameng", "postgres", "sqlserver", "redis", "mongodb", "db2")) {
     return "i-lucide-database";
   }
   if (has("directory service", "directory_service", "directory-service", "windows_ad")) return "i-lucide-network";

ui/components/SideBar/bottomPanels.vue

@@ -29,7 +29,7 @@ type PanelKind = "favorites" | "snippets";
 type SnippetGroupKey = "shell" | "win_shell" | "python" | "raw" | "database" | "other";
 
 const FAVORITE_ROOT_ID = "__favorite_root__";
-const DATABASE_SNIPPET_MODULES = new Set(["mysql", "mariadb", "postgresql", "sqlserver", "oracle"]);
+const DATABASE_SNIPPET_MODULES = new Set(["mysql", "mariadb", "postgresql", "sqlserver", "oracle", "dameng"]);
 
 const { t } = useI18n();
 const toast = useToast();
@@ -91,7 +91,8 @@ const snippetCreateItems = computed<DropdownMenuItem[]>(() =>
     ["MariaDB", "mariadb", "i-lucide-database"],
     ["PostgreSQL", "postgresql", "i-lucide-database"],
     ["SQLServer", "sqlserver", "i-lucide-database"],
-    ["Oracle", "oracle", "i-lucide-database"]
+    ["Oracle", "oracle", "i-lucide-database"],
+    ["Dameng", "dameng", "i-lucide-database"]
   ].map(([label, module, icon]) => ({
     label,
     icon,
@@ -342,6 +343,7 @@ function getSnippetIcon(snippet: Snippet) {
     case "postgresql":
     case "sqlserver":
     case "oracle":
+    case "dameng":
       return "i-lucide-database";
     case "raw":
       return "i-lucide-file-text";

ui/koko/composables/terminal/terminalCommandSuggestions.ts

@@ -176,6 +176,23 @@ const CATALOGS: Record<KokoTerminalCommandProfile, string[]> = {
     "@",
     "EXIT"
   ],
+  dameng: [
+    "SELECT",
+    "INSERT",
+    "UPDATE",
+    "DELETE",
+    "CREATE",
+    "ALTER",
+    "DROP",
+    "DESC",
+    "EXPLAIN",
+    "SET",
+    "COMMIT",
+    "ROLLBACK",
+    "CONNECT",
+    "START",
+    "EXIT"
+  ],
   sqlserver: [
     "SELECT",
     "INSERT",
@@ -205,14 +222,15 @@ const CASE_INSENSITIVE_PROFILES = new Set<KokoTerminalCommandProfile>([
   "redis",
   "mongodb",
   "oracle",
+  "dameng",
   "sqlserver"
 ]);
 
 export function resolveTerminalCommandProfile(profile: ConnectorTerminalProfile = {}): KokoTerminalCommandProfile {
   const protocol = String(profile.protocol || "").toLowerCase();
   if (protocol === "mariadb" || protocol === "mysql") return "mysql";
   if (protocol === "pg" || protocol === "postgres" || protocol === "postgresql") return "postgresql";
-  if (["redis", "mongodb", "oracle", "sqlserver"].includes(protocol)) {
+  if (["redis", "mongodb", "oracle", "dameng", "sqlserver"].includes(protocol)) {
     return protocol as KokoTerminalCommandProfile;
   }
   const platform =

ui/koko/host.ts

@@ -31,6 +31,7 @@ export type KokoTerminalCommandProfile =
   | "redis"
   | "mongodb"
   | "oracle"
+  | "dameng"
   | "sqlserver";
 
 const SENSITIVE_TERMINAL_COMMAND_PATTERNS = [

ui/koko/tests/terminal/terminalCommandSuggestions.test.ts

@@ -16,6 +16,7 @@ describe("terminal command suggestions", () => {
     expect(resolveTerminalCommandProfile({ protocol: "mariadb" })).toBe("mysql");
     expect(resolveTerminalCommandProfile({ protocol: "pg" })).toBe("postgresql");
     expect(resolveTerminalCommandProfile({ protocol: "redis" })).toBe("redis");
+    expect(resolveTerminalCommandProfile({ protocol: "dameng" })).toBe("dameng");
     expect(resolveTerminalCommandProfile({})).toBe("linux");
   });
 

ui/shared/connectors/GuideSessionSurface.vue

@@ -29,6 +29,7 @@ const defaultPorts: Record<string, number> = {
   postgresql: 5432,
   redis: 6379,
   oracle: 1521,
+  dameng: 5236,
   sqlserver: 1433,
   mongodb: 27017
 };