simplemdg-dev-cli 2.4.4 → 2.5.0

package/src/core/db/db-studio-server.ts

@@ -10,9 +10,10 @@ import {
   listPublicConnections,
   removeConnection,
   renameConnection,
+  updateConnectionFields,
   upsertConnectionFromDraft,
 } from "./db-cache";
-import type { TConnectionDraft } from "./db-cache";
+import type { TConnectionDraft, TConnectionFieldPatch } from "./db-cache";
 import { testConnectionProfile } from "./db-connection";
 import {
   analyzeSqlSafety,
@@ -22,14 +23,14 @@ import {
   generateSelectSql,
   looksLikeProduction,
 } from "./db-metadata";
-import type { TResolvedDatabaseConnection } from "./db-types";
+import type { TResolvedDatabaseConnection, TTableChangeSet } from "./db-types";
 import {
   deleteSavedQuery,
   listSavedQueries,
   renameSavedQuery,
   saveQuery,
 } from "./db-query-files";
-import { deleteRow, insertRow, updateRow } from "./db-row";
+import { deleteRow, insertRow, saveTableChanges, updateRow } from "./db-row";
 import { appendQueryHistory, listQueryHistory } from "./db-query-history";
 import {
   detectAppDatabaseServices,
@@ -125,10 +126,20 @@ function getNumber(body: TJsonBody, key: string, fallback: number): number {
   return Number.isFinite(value) ? value : fallback;
 }
 
+const VALID_ENVIRONMENTS = new Set(["DEV", "QAS", "PROD", "SANDBOX", "CUSTOM"]);
+
+function getEnvironment(body: TJsonBody): TConnectionDraft["environment"] {
+  const value = getString(body, "environment").toUpperCase();
+  return VALID_ENVIRONMENTS.has(value) ? (value as TConnectionDraft["environment"]) : undefined;
+}
+
 function draftFromBody(body: TJsonBody): TConnectionDraft {
   const type = getString(body, "type") === "hana" ? "hana" : "postgresql";
   return {
     name: getString(body, "name") || `${type} connection`,
+    color: getString(body, "color") || undefined,
+    environment: getEnvironment(body),
+    isFavorite: body.isFavorite === undefined ? undefined : Boolean(body.isFavorite),
     type,
     host: getString(body, "host"),
     port: getNumber(body, "port", type === "hana" ? 443 : 5432),
@@ -226,6 +237,21 @@ export async function startStudioServer(options: TStudioServerOptions = {}): Pro
       return;
     }
 
+    if (pathname === "/api/connections/update" && method === "POST") {
+      const body = await readJsonBody(req);
+      const patch: TConnectionFieldPatch = {};
+      if (typeof body.name === "string") patch.name = body.name;
+      if (typeof body.color === "string") patch.color = body.color;
+      if (body.environment !== undefined) patch.environment = getEnvironment(body);
+      if (body.isFavorite !== undefined) patch.isFavorite = Boolean(body.isFavorite);
+      if (Array.isArray(body.tags)) patch.tags = body.tags as string[];
+      const profile = await updateConnectionFields(getString(body, "id"), patch);
+      const { encryptedPassword: _omit, ...publicProfile } = profile;
+      void _omit;
+      sendJson(res, { connection: publicProfile });
+      return;
+    }
+
     if (pathname === "/api/connections/duplicate" && method === "POST") {
       const body = await readJsonBody(req);
       const profile = await duplicateConnection(getString(body, "id"));
@@ -337,6 +363,25 @@ export async function startStudioServer(options: TStudioServerOptions = {}): Pro
       return;
     }
 
+    if (pathname === "/api/catalog/primary-key" && method === "GET") {
+      const adapter = await pool.getAdapter(url.searchParams.get("connectionId") ?? "");
+      const primaryKey = await adapter.getPrimaryKey(url.searchParams.get("schema") ?? "", url.searchParams.get("table") ?? "");
+      sendJson(res, { primaryKey });
+      return;
+    }
+
+    if (pathname === "/api/catalog/constraints" && method === "GET") {
+      const adapter = await pool.getAdapter(url.searchParams.get("connectionId") ?? "");
+      const schema = url.searchParams.get("schema") ?? "";
+      const table = url.searchParams.get("table") ?? "";
+      const [primaryKey, indexes] = await Promise.all([
+        adapter.getPrimaryKey(schema, table),
+        adapter.listIndexes(schema, table).catch(() => []),
+      ]);
+      sendJson(res, { primaryKey, indexes });
+      return;
+    }
+
     // --- Table data ----------------------------------------------------------
     if (pathname === "/api/table/data" && method === "POST") {
       const body = await readJsonBody(req);
@@ -391,6 +436,28 @@ export async function startStudioServer(options: TStudioServerOptions = {}): Pro
       return;
     }
 
+    if (pathname === "/api/table/save-changes" && method === "POST") {
+      const body = await readJsonBody(req);
+      const readOnly = body.readOnly === undefined ? serverReadOnlyDefault : Boolean(body.readOnly);
+
+      if (readOnly) {
+        sendJson(res, { ok: false, blocked: true, error: "Read-only mode is on. Turn it off to save changes." });
+        return;
+      }
+
+      const adapter = await pool.getAdapter(getString(body, "connectionId"));
+      const result = await saveTableChanges(adapter, {
+        schema: getString(body, "schema"),
+        table: getString(body, "table"),
+        primaryKeyColumns: Array.isArray(body.primaryKeyColumns) ? (body.primaryKeyColumns as string[]) : [],
+        updates: Array.isArray(body.updates) ? (body.updates as TTableChangeSet["updates"]) : [],
+        inserts: Array.isArray(body.inserts) ? (body.inserts as TTableChangeSet["inserts"]) : [],
+        deletes: Array.isArray(body.deletes) ? (body.deletes as TTableChangeSet["deletes"]) : [],
+      });
+      sendJson(res, { ok: true, result });
+      return;
+    }
+
     if (pathname === "/api/table/sql" && method === "POST") {
       const body = await readJsonBody(req);
       const adapter = await pool.getAdapter(getString(body, "connectionId"));

package/src/core/db/db-studio-styles.ts

@@ -0,0 +1,221 @@
+export const STUDIO_STYLES = `
+:root{
+  --bg:#0b0f17;--bg-2:#0e1420;--bg-3:#121a28;--panel:#0d131e;--elev:#16203250;
+  --border:#1f2c44;--border-2:#27374f;--text:#dce6f5;--muted:#8295b5;--faint:#5b6c8a;
+  --accent:#3b82f6;--accent-2:#1d4ed8;--green:#22c55e;--red:#ef4444;--yellow:#f59e0b;
+  --amber:#fbbf24;--chip:#1a2840;--sel:#16243d;--editbg:#3a2f08;--delbg:#3a1414;--insbg:#0f2e18;
+}
+*{box-sizing:border-box}
+html,body{height:100%;margin:0}
+body{background:var(--bg);color:var(--text);font:13px/1.45 "Segoe UI",Roboto,Arial,sans-serif;overflow:hidden}
+button,input,select,textarea{font:inherit;color:inherit;outline:none}
+::-webkit-scrollbar{width:10px;height:10px}
+::-webkit-scrollbar-thumb{background:#22314c;border-radius:8px}
+::-webkit-scrollbar-track{background:transparent}
+.app{display:flex;flex-direction:column;height:100vh}
+.hidden{display:none!important}
+.spin{width:14px;height:14px;border:2px solid var(--border-2);border-top-color:var(--accent);border-radius:50%;display:inline-block;animation:sp .7s linear infinite;vertical-align:middle}
+@keyframes sp{to{transform:rotate(360deg)}}
+svg.ic{width:15px;height:15px;vertical-align:-2px;fill:none;stroke:currentColor;stroke-width:1.7;stroke-linecap:round;stroke-linejoin:round}
+
+/* top bar */
+.topbar{display:flex;align-items:center;gap:9px;padding:8px 12px;background:linear-gradient(180deg,#101a2b,#0d1521);border-bottom:1px solid var(--border);flex:0 0 auto}
+.brand{font-weight:700;color:#cfe0ff;letter-spacing:.2px;white-space:nowrap}
+.brand .b2{color:var(--faint);font-weight:600}
+.badge{padding:3px 9px;border-radius:999px;background:var(--chip);border:1px solid var(--border);color:var(--muted);font-size:12px;white-space:nowrap;display:inline-flex;align-items:center;gap:5px}
+.badge.on{color:#cfe0ff;border-color:var(--accent)}
+.badge.hana{color:#7dd3fc;border-color:#0e7490}
+.badge.pg{color:#a5b4fc;border-color:#4338ca}
+.badge.ro{cursor:pointer}
+.badge.ro.active{color:#fff;background:#7c2d12;border-color:#b45309}
+.badge.prod{color:#fff;background:#7f1d1d;border-color:#b91c1c}
+.grow{flex:1}
+.top-search{flex:0 1 320px}
+.iconbtn{background:transparent;border:1px solid var(--border);border-radius:8px;padding:6px 8px;color:var(--muted);cursor:pointer;display:inline-flex;align-items:center;gap:6px}
+.iconbtn:hover{color:#cfe0ff;border-color:var(--border-2);background:var(--bg-3)}
+.iconbtn.primary{background:var(--accent);border-color:var(--accent-2);color:#fff}
+.iconbtn.primary:hover{filter:brightness(1.08);color:#fff}
+
+/* body */
+.body{display:flex;flex:1;min-height:0}
+.sidebar{width:320px;min-width:220px;max-width:560px;background:var(--panel);border-right:1px solid var(--border);display:flex;flex-direction:column;overflow:hidden}
+.resizer{width:5px;cursor:col-resize}
+.resizer:hover{background:var(--accent)}
+.workspace{flex:1;display:flex;flex-direction:column;min-width:0}
+
+/* sidebar sections */
+.side-sec{display:flex;flex-direction:column;min-height:0;border-bottom:1px solid var(--border)}
+.side-sec.flex{flex:1}
+.side-head{display:flex;align-items:center;gap:7px;padding:9px 11px;cursor:pointer;color:var(--muted);user-select:none}
+.side-head:hover{color:#cfe0ff}
+.side-head .h-title{font-size:11px;text-transform:uppercase;letter-spacing:.7px;font-weight:700;flex:1}
+.side-head .chev{transition:transform .15s}
+.side-sec.collapsed .chev{transform:rotate(-90deg)}
+.side-sec.collapsed .side-body{display:none}
+.side-body{padding:0 9px 10px;overflow:auto;min-height:0}
+.side-sec.flex .side-body{flex:1}
+.side-actions{display:flex;gap:6px;margin-bottom:7px}
+
+/* search box */
+.searchbox{display:flex;align-items:center;gap:6px;background:var(--bg-3);border:1px solid var(--border);border-radius:8px;padding:5px 8px;margin-bottom:7px}
+.searchbox svg{color:var(--faint)}
+.searchbox input{flex:1;background:transparent;border:0;color:var(--text)}
+.searchbox .sbtn{background:transparent;border:0;color:var(--muted);cursor:pointer}
+
+/* buttons + inputs */
+.btn{background:var(--accent);border:1px solid var(--accent-2);color:#fff;border-radius:8px;padding:6px 11px;cursor:pointer}
+.btn:hover{filter:brightness(1.08)}
+.btn.sec{background:#22304a;border-color:var(--border-2);color:#cfe0ff}
+.btn.ghost{background:transparent;border-color:var(--border);color:var(--muted)}
+.btn.danger{background:#7f1d1d;border-color:#b91c1c;color:#fff}
+.btn.sm{padding:3px 8px;font-size:12px}
+.btn:disabled{opacity:.45;cursor:not-allowed;filter:none}
+.input,.select{width:100%;background:var(--bg-3);border:1px solid var(--border);border-radius:8px;padding:7px 9px;color:var(--text)}
+.field{display:flex;flex-direction:column;gap:4px;margin-bottom:9px}
+.field label{color:var(--muted);font-size:12px}
+.note{color:var(--muted);font-size:12px}
+.faint{color:var(--faint)}
+.row{display:flex;gap:7px;align-items:center;flex-wrap:wrap}
+.right{justify-content:flex-end}
+
+/* connection cards */
+.conn-card{position:relative;background:var(--bg-3);border:1px solid var(--border);border-left:3px solid var(--faint);border-radius:9px;padding:8px 10px;margin-bottom:7px;cursor:pointer}
+.conn-card:hover{border-color:var(--border-2);background:#152033}
+.conn-card.active{border-color:var(--accent);background:var(--sel)}
+.conn-top{display:flex;align-items:center;gap:7px}
+.conn-name{font-weight:600;flex:1;white-space:nowrap;overflow:hidden;text-overflow:ellipsis}
+.conn-sub{color:var(--muted);font-size:12px;white-space:nowrap;overflow:hidden;text-overflow:ellipsis;margin-top:2px}
+.conn-tags{display:flex;gap:5px;margin-top:5px;flex-wrap:wrap}
+.tag{font-size:10.5px;padding:1px 7px;border-radius:999px;background:var(--chip);border:1px solid var(--border);color:var(--muted)}
+.tag.env-DEV{color:#86efac;border-color:#166534}
+.tag.env-QAS{color:#fde68a;border-color:#a16207}
+.tag.env-PROD{color:#fecaca;border-color:#b91c1c}
+.tag.env-SANDBOX{color:#a5b4fc;border-color:#4338ca}
+.tag.type{color:#7dd3fc}
+.star{cursor:pointer;color:var(--faint)}
+.star.on{color:var(--amber)}
+.skel{height:56px;border-radius:9px;margin-bottom:7px;background:linear-gradient(90deg,#121a28,#1a2740,#121a28);background-size:200% 100%;animation:shimmer 1.3s infinite}
+@keyframes shimmer{0%{background-position:200% 0}100%{background-position:-200% 0}}
+
+/* tree */
+.tree{font-size:12.5px}
+.trow{display:flex;align-items:center;gap:5px;padding:3px 5px;border-radius:6px;cursor:pointer;white-space:nowrap}
+.trow:hover{background:var(--bg-3)}
+.trow.sel{background:var(--sel)}
+.tchev{width:13px;color:var(--faint);flex:0 0 auto;text-align:center;transition:transform .12s}
+.tchev.open{transform:rotate(90deg)}
+.tchev.leaf{visibility:hidden}
+.ticon{flex:0 0 auto;color:var(--muted);display:inline-flex}
+.ticon.tbl{color:#7dd3fc}.ticon.viw{color:#c4b5fd}.ticon.prc{color:#fca5a5}.ticon.fun{color:#fcd34d}.ticon.syn{color:#67e8f9}.ticon.sch{color:#93c5fd}.ticon.db{color:#86efac}.ticon.fld{color:var(--muted)}
+.tlabel{flex:1;overflow:hidden;text-overflow:ellipsis}
+.tbadge{color:var(--faint);font-size:11px}
+.tchildren{margin-left:13px;border-left:1px solid var(--border);padding-left:3px}
+.tnote{color:var(--faint);font-size:11.5px;padding:3px 6px}
+.tsearch{margin:3px 0 4px}
+
+/* workspace tabs */
+.tabbar{display:flex;align-items:stretch;background:var(--bg-2);border-bottom:1px solid var(--border);overflow-x:auto;flex:0 0 auto;min-height:38px}
+.wtab{display:flex;align-items:center;gap:7px;padding:8px 13px;border-right:1px solid var(--border);color:var(--muted);cursor:pointer;white-space:nowrap;max-width:260px}
+.wtab:hover{color:var(--text);background:var(--bg-3)}
+.wtab.active{color:#bcd4ff;background:var(--bg-3);box-shadow:inset 0 -2px 0 var(--accent)}
+.wtab .t-ico{display:inline-flex;color:var(--faint)}
+.wtab .t-title{overflow:hidden;text-overflow:ellipsis}
+.wtab .dot{width:7px;height:7px;border-radius:50%;background:var(--amber)}
+.wtab .x{color:var(--faint);border-radius:4px;padding:0 3px}
+.wtab .x:hover{color:#fff;background:#37425c}
+.tabcontent{flex:1;min-height:0;position:relative;background:var(--bg)}
+.tabpane{position:absolute;inset:0;display:flex;flex-direction:column;overflow:auto}
+
+/* welcome */
+.welcome{padding:26px;max-width:1000px;margin:0 auto;width:100%}
+.welcome h1{margin:0 0 4px;font-size:22px}
+.welcome .lede{color:var(--muted);margin-bottom:22px}
+.wcards{display:grid;grid-template-columns:repeat(auto-fit,minmax(210px,1fr));gap:12px;margin-bottom:24px}
+.wcard{background:var(--bg-2);border:1px solid var(--border);border-radius:12px;padding:16px;cursor:pointer}
+.wcard:hover{border-color:var(--accent);background:var(--bg-3)}
+.wcard .wc-ic{color:var(--accent);margin-bottom:8px}
+.wcard h3{margin:0 0 4px;font-size:14px}
+.wcard p{margin:0;color:var(--muted);font-size:12px}
+.wcols{display:grid;grid-template-columns:1fr 1fr;gap:18px}
+.wcol h4{color:var(--muted);font-size:11px;text-transform:uppercase;letter-spacing:.6px;margin:0 0 8px}
+.wlist .wli{padding:8px 10px;border:1px solid var(--border);border-radius:8px;margin-bottom:6px;cursor:pointer;background:var(--bg-2)}
+.wlist .wli:hover{border-color:var(--border-2);background:var(--bg-3)}
+
+/* toolbar + editor */
+.toolbar{display:flex;gap:6px;align-items:center;flex-wrap:wrap;padding:8px 10px;border-bottom:1px solid var(--border);background:var(--bg-2)}
+.pane-body{flex:1;min-height:0;display:flex;flex-direction:column;padding:10px;gap:8px;overflow:auto}
+.editor{width:100%;min-height:160px;resize:vertical;background:#0a1018;border:1px solid var(--border);border-radius:8px;padding:11px;color:#e7eefc;font-family:Consolas,"Cascadia Code",monospace;font-size:13px}
+.errbox{background:#2a0f10;border:1px solid #7f1d1d;color:#fca5a5;border-radius:8px;padding:9px 11px;white-space:pre-wrap;font-family:Consolas,monospace;font-size:12px}
+
+/* grid */
+.gridwrap{flex:1;min-height:120px;overflow:auto;border:1px solid var(--border);border-radius:8px;background:var(--bg-3)}
+table.grid{border-collapse:separate;border-spacing:0;width:100%;font-size:12.5px}
+table.grid th,table.grid td{border-bottom:1px solid var(--border);border-right:1px solid var(--border);padding:6px 9px;text-align:left;white-space:nowrap;max-width:440px;overflow:hidden;text-overflow:ellipsis}
+table.grid th{position:sticky;top:0;z-index:1;background:#172339;color:#cfe0ff;cursor:pointer;user-select:none}
+table.grid th .sort{color:var(--accent);margin-left:4px}
+table.grid tr:hover td{background:#13203450}
+table.grid td.num{text-align:right;color:#a7f3d0}
+table.grid th.rowhdr,table.grid td.rowhdr{width:54px;text-align:right;color:var(--faint);background:#15203550;cursor:pointer;position:sticky;left:0}
+table.grid tr.selrow td{background:var(--sel)}
+table.grid td.edited{background:var(--editbg);box-shadow:inset 0 0 0 1px var(--amber)}
+table.grid tr.row-del td{background:var(--delbg);text-decoration:line-through;color:#fca5a5}
+table.grid tr.row-ins td{background:var(--insbg)}
+table.grid tr.row-err td{box-shadow:inset 0 0 0 1px var(--red)}
+.rowflag{display:inline-block;width:8px;height:8px;border-radius:50%;margin-right:4px}
+.rowflag.d{background:var(--amber)}.rowflag.del{background:var(--red)}.rowflag.ins{background:var(--green)}
+.cellinput{width:100%;background:#0a1018;border:1px solid var(--accent);border-radius:4px;color:#fff;padding:3px 5px;font:inherit}
+.rowerr-msg{color:#fca5a5;font-size:11px}
+
+/* sql tabs inside editor */
+.sqltabs{display:flex;gap:4px;flex-wrap:wrap;padding:6px 10px 0}
+.sqltab{padding:4px 10px;background:var(--bg-3);border:1px solid var(--border);border-bottom:0;border-radius:7px 7px 0 0;cursor:pointer;color:var(--muted);font-size:12px}
+.sqltab.active{color:#cfe0ff;border-color:var(--accent);background:var(--sel)}
+.sqltab .x{margin-left:7px}
+
+/* status bar */
+.statusbar{display:flex;align-items:center;gap:16px;padding:5px 12px;background:var(--bg-2);border-top:1px solid var(--border);color:var(--muted);font-size:12px;flex:0 0 auto}
+.st-item{display:inline-flex;align-items:center;gap:6px}
+.st-dot{width:8px;height:8px;border-radius:50%;background:var(--faint)}
+.st-dot.ok{background:var(--green)}.st-dot.err{background:var(--red)}.st-dot.run{background:var(--amber);animation:sp 1s linear infinite}
+.st-pending{color:var(--amber)}
+
+/* metadata */
+.meta-tabs{display:flex;gap:5px;padding:8px 10px 0}
+.meta-tab{padding:5px 11px;border:1px solid var(--border);border-bottom:0;border-radius:7px 7px 0 0;cursor:pointer;color:var(--muted);font-size:12px}
+.meta-tab.active{color:#cfe0ff;border-color:var(--accent);background:var(--sel)}
+.kvs{display:grid;grid-template-columns:160px 1fr;gap:5px 14px;padding:6px}
+.kvs .k{color:var(--muted)}
+.pill{display:inline-block;padding:1px 7px;border-radius:999px;background:var(--chip);border:1px solid var(--border);font-size:11px;color:var(--muted)}
+.pill.pk{color:#fde68a;border-color:#a16207}
+
+/* modal + wizard */
+.modal{position:fixed;inset:0;background:rgba(2,6,15,.62);display:flex;align-items:center;justify-content:center;z-index:60}
+.dialog{background:var(--bg-2);border:1px solid var(--border-2);border-radius:14px;padding:18px;width:560px;max-width:94vw;max-height:90vh;overflow:auto;box-shadow:0 24px 80px rgba(0,0,0,.5)}
+.dialog h3{margin:0 0 14px}
+.steps{display:flex;gap:6px;margin-bottom:14px;flex-wrap:wrap}
+.step{flex:1;min-width:80px;text-align:center;font-size:11px;color:var(--faint);padding:6px 4px;border-radius:8px;background:var(--bg-3);border:1px solid var(--border)}
+.step.active{color:#cfe0ff;border-color:var(--accent)}
+.step.done{color:var(--green)}
+.wlistbox{max-height:240px;overflow:auto;border:1px solid var(--border);border-radius:9px}
+.wrow{padding:8px 10px;border-bottom:1px solid var(--border);cursor:pointer;display:flex;align-items:center;gap:8px}
+.wrow:hover{background:var(--bg-3)}
+.wrow.sel{background:var(--sel)}
+.swatches{display:flex;gap:7px;flex-wrap:wrap}
+.swatch{width:24px;height:24px;border-radius:7px;cursor:pointer;border:2px solid transparent}
+.swatch.sel{border-color:#fff}
+
+/* context menu */
+.ctxmenu{position:fixed;z-index:80;background:var(--bg-2);border:1px solid var(--border-2);border-radius:9px;padding:5px;min-width:190px;box-shadow:0 16px 50px rgba(0,0,0,.5)}
+.ctxitem{display:flex;align-items:center;gap:9px;padding:7px 10px;border-radius:7px;cursor:pointer;color:var(--text);font-size:12.5px}
+.ctxitem:hover{background:var(--accent);color:#fff}
+.ctxitem.danger{color:#fca5a5}
+.ctxitem.danger:hover{background:#7f1d1d;color:#fff}
+.ctxsep{height:1px;background:var(--border);margin:4px 6px}
+
+/* toasts */
+.toasts{position:fixed;right:16px;bottom:16px;display:flex;flex-direction:column;gap:8px;z-index:90}
+.toast{background:var(--bg-2);border:1px solid var(--border-2);border-left:3px solid var(--accent);border-radius:9px;padding:9px 13px;min-width:240px;max-width:380px;box-shadow:0 10px 34px rgba(0,0,0,.4);animation:slideup .18s ease}
+.toast.ok{border-left-color:var(--green)}.toast.err{border-left-color:var(--red)}.toast.warn{border-left-color:var(--yellow)}
+@keyframes slideup{from{transform:translateY(8px);opacity:0}to{transform:translateY(0);opacity:1}}
+.empty{color:var(--faint);padding:16px;text-align:center}
+`;

package/src/core/db/db-types.ts

@@ -1,8 +1,13 @@
 export type TDatabaseType = "hana" | "postgresql";
 
+export type TConnectionEnvironment = "DEV" | "QAS" | "PROD" | "SANDBOX" | "CUSTOM";
+
 export type TDatabaseConnectionProfile = {
   id: string;
   name: string;
+  color?: string;
+  environment?: TConnectionEnvironment;
+  isFavorite?: boolean;
   type: TDatabaseType;
   region?: string;
   org?: string;
@@ -174,11 +179,54 @@ export type TTableDataOptions = {
  * specifics (quoting, system catalog queries) so the rest of the studio is
  * dialect-agnostic.
  */
+export type TDatabasePrimaryKey = {
+  columns: string[];
+  constraintName?: string;
+};
+
+export type TRowUpdate = {
+  key: Record<string, unknown>;
+  changes: Record<string, unknown>;
+};
+
+export type TRowInsert = {
+  values: Record<string, unknown>;
+};
+
+export type TRowDelete = {
+  key: Record<string, unknown>;
+};
+
+export type TTableChangeSet = {
+  schema: string;
+  table: string;
+  primaryKeyColumns: string[];
+  updates: TRowUpdate[];
+  inserts: TRowInsert[];
+  deletes: TRowDelete[];
+};
+
+export type TSaveRowResult = {
+  type: "update" | "insert" | "delete";
+  success: boolean;
+  key?: Record<string, unknown>;
+  error?: string;
+};
+
+export type TSaveTableChangesResult = {
+  success: boolean;
+  updated: number;
+  inserted: number;
+  deleted: number;
+  rowResults: TSaveRowResult[];
+};
+
 export interface IDatabaseAdapter {
   readonly type: TDatabaseType;
   connect(): Promise<void>;
   disconnect(): Promise<void>;
   testConnection(): Promise<TConnectionTestResult>;
+  getPrimaryKey(schema: string, table: string): Promise<TDatabasePrimaryKey>;
   runQuery(sql: string, options?: { maxRows?: number }): Promise<TDatabaseQueryResult>;
   /** Run a parameterized statement. Placeholders come from `placeholder()`. */
   runParameterized(sql: string, params: unknown[], options?: { maxRows?: number }): Promise<TDatabaseQueryResult>;