Fix terminal connections

src/backend/ssh/tunnel.ts

@@ -17,6 +17,7 @@ import type {
 import { CONNECTION_STATES } from "../../types/index.js";
 import { tunnelLogger } from "../utils/logger.js";
 import { SystemCrypto } from "../utils/system-crypto.js";
+import { SimpleDBOps } from "../utils/simple-db-ops.js";
 
 const app = express();
 app.use(
@@ -490,7 +491,8 @@ async function connectSSHTunnel(
 
   if (tunnelConfig.sourceCredentialId && tunnelConfig.sourceUserId) {
     try {
-      const credentials = await getDb()
+      const credentials = await SimpleDBOps.select(
+        getDb()
           .select()
           .from(sshCredentials)
           .where(
@@ -498,6 +500,9 @@ async function connectSSHTunnel(
               eq(sshCredentials.id, tunnelConfig.sourceCredentialId),
               eq(sshCredentials.userId, tunnelConfig.sourceUserId),
             ),
+          ),
+        "ssh_credentials",
+        tunnelConfig.sourceUserId,
       );
 
       if (credentials.length > 0) {
@@ -564,7 +569,8 @@ async function connectSSHTunnel(
 
   if (tunnelConfig.endpointCredentialId && tunnelConfig.endpointUserId) {
     try {
-      const credentials = await getDb()
+      const credentials = await SimpleDBOps.select(
+        getDb()
           .select()
           .from(sshCredentials)
           .where(
@@ -572,6 +578,9 @@ async function connectSSHTunnel(
               eq(sshCredentials.id, tunnelConfig.endpointCredentialId),
               eq(sshCredentials.userId, tunnelConfig.endpointUserId),
             ),
+          ),
+        "ssh_credentials",
+        tunnelConfig.endpointUserId,
       );
 
       if (credentials.length > 0) {
@@ -990,7 +999,8 @@ async function killRemoteTunnelByMarker(
 
   if (tunnelConfig.sourceCredentialId && tunnelConfig.sourceUserId) {
     try {
-      const credentials = await getDb()
+      const credentials = await SimpleDBOps.select(
+        getDb()
           .select()
           .from(sshCredentials)
           .where(
@@ -998,6 +1008,9 @@ async function killRemoteTunnelByMarker(
               eq(sshCredentials.id, tunnelConfig.sourceCredentialId),
               eq(sshCredentials.userId, tunnelConfig.sourceUserId),
             ),
+          ),
+        "ssh_credentials",
+        tunnelConfig.sourceUserId,
       );
 
       if (credentials.length > 0) {