npm - dbgate-api-premium - Versions diffs - 5.5.7-alpha.45 - Mend

dbgate-api-premium 5.5.7-alpha.45

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (125) hide show

package/.env +19 -0
package/.yarnrc +2 -0
package/README.md +1 -0
package/env/dblogin/.env +14 -0
package/env/portal/.env +70 -0
package/env/singledb/.env +17 -0
package/env/storage/.env +43 -0
package/package.json +89 -0
package/src/auth/authCommon.js +16 -0
package/src/auth/authProvider.js +343 -0
package/src/auth/storageAuthProvider.js +393 -0
package/src/controllers/apps.js +280 -0
package/src/controllers/archive.js +217 -0
package/src/controllers/auth.js +136 -0
package/src/controllers/config.js +271 -0
package/src/controllers/connections.js +486 -0
package/src/controllers/databaseConnections.js +561 -0
package/src/controllers/files.js +222 -0
package/src/controllers/jsldata.js +296 -0
package/src/controllers/metadata.js +47 -0
package/src/controllers/plugins.js +216 -0
package/src/controllers/queryHistory.js +54 -0
package/src/controllers/runners.js +234 -0
package/src/controllers/scheduler.js +46 -0
package/src/controllers/serverConnections.js +271 -0
package/src/controllers/sessions.js +243 -0
package/src/controllers/storage.js +380 -0
package/src/controllers/storageDb.js +215 -0
package/src/controllers/uploads.js +133 -0
package/src/currentVersion.js +5 -0
package/src/gistSecret.js +2 -0
package/src/index.js +139 -0
package/src/main.js +202 -0
package/src/packagedPluginsContent.js +1 -0
package/src/proc/connectProcess.js +38 -0
package/src/proc/databaseConnectionProcess.js +431 -0
package/src/proc/index.js +15 -0
package/src/proc/jslDatastoreProcess.js +60 -0
package/src/proc/serverConnectionProcess.js +188 -0
package/src/proc/sessionProcess.js +390 -0
package/src/proc/sshForwardProcess.js +75 -0
package/src/shell/archiveReader.js +11 -0
package/src/shell/archiveWriter.js +22 -0
package/src/shell/autoIndexForeignKeysTransform.js +19 -0
package/src/shell/collectorWriter.js +33 -0
package/src/shell/consoleObjectWriter.js +16 -0
package/src/shell/copyStream.js +48 -0
package/src/shell/dataDuplicator.js +63 -0
package/src/shell/dataTypeMapperTransform.js +21 -0
package/src/shell/dbModelToJson.js +16 -0
package/src/shell/deployDb.js +56 -0
package/src/shell/download.js +15 -0
package/src/shell/dropAllDbObjects.js +42 -0
package/src/shell/dumpDatabase.js +49 -0
package/src/shell/executeQuery.js +39 -0
package/src/shell/fakeObjectReader.js +35 -0
package/src/shell/finalizer.js +12 -0
package/src/shell/generateDeploySql.js +95 -0
package/src/shell/generateModelSql.js +30 -0
package/src/shell/importDatabase.js +85 -0
package/src/shell/index.js +80 -0
package/src/shell/initializeApiEnvironment.js +9 -0
package/src/shell/jslDataReader.js +9 -0
package/src/shell/jsonLinesReader.js +52 -0
package/src/shell/jsonLinesWriter.js +36 -0
package/src/shell/jsonReader.js +84 -0
package/src/shell/jsonToDbModel.js +9 -0
package/src/shell/jsonWriter.js +97 -0
package/src/shell/loadDatabase.js +27 -0
package/src/shell/loadFile.js +10 -0
package/src/shell/modifyJsonLinesReader.js +148 -0
package/src/shell/queryReader.js +30 -0
package/src/shell/registerPlugins.js +9 -0
package/src/shell/requirePlugin.js +43 -0
package/src/shell/runScript.js +19 -0
package/src/shell/sqlDataWriter.js +52 -0
package/src/shell/sqlTextReplacementTransform.js +32 -0
package/src/shell/tableReader.js +39 -0
package/src/shell/tableWriter.js +18 -0
package/src/storageModel.js +819 -0
package/src/utility/ColumnMapTransformStream.js +21 -0
package/src/utility/DatastoreProxy.js +106 -0
package/src/utility/EnsureStreamHeaderStream.js +31 -0
package/src/utility/JsonLinesDatabase.js +148 -0
package/src/utility/JsonLinesDatastore.js +232 -0
package/src/utility/LineReader.js +88 -0
package/src/utility/SSHConnection.js +251 -0
package/src/utility/authProxy.js +133 -0
package/src/utility/checkLicense.js +186 -0
package/src/utility/childProcessChecker.js +21 -0
package/src/utility/cleanDirectory.js +24 -0
package/src/utility/cloudUpgrade.js +61 -0
package/src/utility/connectUtility.js +111 -0
package/src/utility/crypting.js +105 -0
package/src/utility/diff2htmlPage.js +8 -0
package/src/utility/directories.js +179 -0
package/src/utility/downloadPackage.js +51 -0
package/src/utility/downloader.js +25 -0
package/src/utility/exceptions.js +9 -0
package/src/utility/exportDbModel.js +31 -0
package/src/utility/exportDbModelSql.js +80 -0
package/src/utility/getChartExport.js +55 -0
package/src/utility/getDiagramExport.js +25 -0
package/src/utility/getExpressPath.js +10 -0
package/src/utility/getJslFileName.js +16 -0
package/src/utility/getMapExport.js +77 -0
package/src/utility/hardwareFingerprint.js +89 -0
package/src/utility/hasPermission.js +101 -0
package/src/utility/importDbModel.js +9 -0
package/src/utility/loadFilesRecursive.js +20 -0
package/src/utility/loadModelFolder.js +29 -0
package/src/utility/loadModelTransform.js +36 -0
package/src/utility/pipeForkLogs.js +19 -0
package/src/utility/platformInfo.js +62 -0
package/src/utility/processArgs.js +39 -0
package/src/utility/processComm.js +18 -0
package/src/utility/requireEngineDriver.js +26 -0
package/src/utility/requirePluginFunction.js +16 -0
package/src/utility/socket.js +68 -0
package/src/utility/sshTunnel.js +106 -0
package/src/utility/sshTunnelProxy.js +36 -0
package/src/utility/timingSafeCheckToken.js +9 -0
package/src/utility/useController.js +99 -0
package/tsconfig.json +13 -0
package/webpack.config.js +55 -0

package/src/utility/getDiagramExport.js ADDED Viewed

@@ -0,0 +1,25 @@
+const getDiagramExport = (html, css, themeType, themeClassName) => {
+  return `<html>
+  <meta charset='utf-8'>
+  <head>
+      <style>
+        ${css}
+        body {
+          background: var(--theme-bg-1);
+          color: var(--theme-font-1);
+        }
+      </style>
+      <link rel="stylesheet" href='https://cdn.jsdelivr.net/npm/@mdi/font@6.5.95/css/materialdesignicons.css' />
+  </head>
+  <body class='${themeType == 'dark' ? 'theme-type-dark' : 'theme-type-light'} ${themeClassName}'>
+      ${html}
+  </body>
+  </html>`;
+};
+module.exports = getDiagramExport;

package/src/utility/getExpressPath.js ADDED Viewed

@@ -0,0 +1,10 @@
+function getExpressPath(path) {
+  path = path.replace(/\/*$/, '').replace(/^\/*/, '');
+  const root = (process.env.WEB_ROOT || '').replace(/^\/*/, '').replace(/\/*$/, '');
+  if (root) {
+    return `/${root}/${path}`;
+  }
+  return `/${path}`;
+}
+module.exports = getExpressPath;

package/src/utility/getJslFileName.js ADDED Viewed

@@ -0,0 +1,16 @@
+const path = require('path');
+const { jsldir, archivedir, resolveArchiveFolder } = require('./directories');
+function getJslFileName(jslid) {
+  const archiveMatch = jslid.match(/^archive:\/\/([^/]+)\/(.*)$/);
+  if (archiveMatch) {
+    return path.join(resolveArchiveFolder(archiveMatch[1]), `${archiveMatch[2]}.jsonl`);
+  }
+  const fileMatch = jslid.match(/^file:\/\/(.*)$/);
+  if (fileMatch) {
+    return fileMatch[1];
+  }
+  return path.join(jsldir(), `${jslid}.jsonl`);
+}
+module.exports = getJslFileName;

package/src/utility/getMapExport.js ADDED Viewed

@@ -0,0 +1,77 @@
+const getMapExport = (geoJson) => {
+  return `<html>
+  <meta charset='utf-8'>
+  <head>
+    <link rel="stylesheet" href="https://unpkg.com/leaflet@1.8.0/dist/leaflet.css"
+      integrity="sha512-hoalWLoI8r4UszCkZ5kL8vayOGVae1oxXe/2A4AO6J9+580uKHDO3JdHb7NzwwzK5xr/Fs0W40kiNHxM9vyTtQ=="
+      crossorigin=""/>
+    <script src="https://unpkg.com/leaflet@1.8.0/dist/leaflet.js"
+      integrity="sha512-BB3hKbKWOc9Ez/TAwyWxNXeoV9c1v6FIeYiBieIWkpLjauysF18NzgR1MBNBXf8/KABdlkX68nAhlwcDFLGPCQ=="
+      crossorigin=""></script>
+    <script>
+    function createMap() {
+    map = leaflet.map('map').setView([50, 15], 13);
+    leaflet
+      .tileLayer('https://{s}.tile.openstreetmap.org/{z}/{x}/{y}.png', {
+        maxZoom: 19,
+        attribution: '<a href="https://dbgate.org" title="Exported from DbGate">DbGate</a> | © OpenStreetMap',
+      })
+      .addTo(map);
+      const geoJsonObj = leaflet
+      .geoJSON(${JSON.stringify(geoJson)}, {
+        style: function () {
+          return {
+            weight: 2,
+            fillColor: '#ff7800',
+            color: '#ff7800',
+            opacity: 0.8,
+            fillOpacity: 0.4,
+          };
+        },
+        pointToLayer: (feature, latlng) => {
+          return leaflet.circleMarker(latlng, {
+            radius: 7,
+            weight: 2,
+            fillColor: '#ff0000',
+            color: '#ff0000',
+            opacity: 0.9,
+            fillOpacity: 0.9,
+          });
+        },
+        onEachFeature: (feature, layer) => {
+          // does this feature have a property named popupContent?
+          if (feature.properties && feature.properties.popupContent) {
+            layer.bindPopup(feature.properties.popupContent);
+            layer.bindTooltip(feature.properties.popupContent);
+          }
+        },
+      })
+      .addTo(map);
+    map.fitBounds(geoJsonObj.getBounds());
+    }
+    </script>
+      <style>
+        #map {
+          position: fixed;
+          left: 0;
+          top: 0;
+          right: 0;
+          bottom: 0;
+        }
+      </style>
+  </head>
+  <body onload='createMap()'>
+      <div id='map'></div>
+  </body>
+  </html>`;
+};
+module.exports = getMapExport;

package/src/utility/hardwareFingerprint.js ADDED Viewed

@@ -0,0 +1,89 @@
+const axios = require('axios');
+const os = require('os');
+const crypto = require('crypto');
+const platformInfo = require('./platformInfo');
+async function getPublicIpInfo() {
+  try {
+    const resp = await axios.default.get('https://ipinfo.io/json');
+    if (!resp.data?.ip) {
+      return { ip: 'unknown-ip' };
+    }
+    return resp.data;
+  } catch (err) {
+    return { ip: 'unknown-ip' };
+  }
+}
+function getMacAddress() {
+  try {
+    const interfaces = os.networkInterfaces();
+    for (let iface of Object.values(interfaces)) {
+      for (let config of iface) {
+        if (config.mac && config.mac !== '00:00:00:00:00:00') {
+          return config.mac;
+        }
+      }
+    }
+    return '00:00:00:00:00:00';
+  } catch (err) {
+    return '00:00:00:00:00:00';
+  }
+}
+async function getHardwareFingerprint() {
+  const publicIpInfo = await getPublicIpInfo();
+  const macAddress = getMacAddress();
+  const platform = os.platform();
+  const release = os.release();
+  const hostname = os.hostname();
+  const totalMemory = os.totalmem();
+  return {
+    publicIp: publicIpInfo.ip,
+    country: publicIpInfo.country,
+    region: publicIpInfo.region,
+    city: publicIpInfo.city,
+    macAddress,
+    platform,
+    release,
+    hostname,
+    totalMemory,
+  };
+}
+async function getHardwareFingerprintHash(data = undefined) {
+  if (!data) {
+    data = await getHardwareFingerprint();
+  }
+  const fingerprintData = JSON.stringify(data);
+  const hash = crypto.createHash('sha256').update(fingerprintData).digest('hex');
+  return hash;
+}
+async function getPublicHardwareFingerprint() {
+  const fingerprint = await getHardwareFingerprint();
+  const hash = await getHardwareFingerprintHash(fingerprint);
+  return {
+    hash,
+    payload: {
+      platform: fingerprint.platform,
+      city: fingerprint.city,
+      country: fingerprint.country,
+      region: fingerprint.region,
+      isDocker: platformInfo.isDocker,
+      isAwsUbuntuLayout: platformInfo.isAwsUbuntuLayout,
+      isElectron: platformInfo.isElectron,
+    },
+  };
+}
+// getHardwareFingerprint().then(console.log);
+// getHardwareFingerprintHash().then(console.log);
+// getPublicHardwareFingerprint().then(console.log);
+module.exports = {
+  getHardwareFingerprint,
+  getHardwareFingerprintHash,
+  getPublicHardwareFingerprint,
+};

package/src/utility/hasPermission.js ADDED Viewed

@@ -0,0 +1,101 @@
+const { compilePermissions, testPermission } = require('dbgate-tools');
+const _ = require('lodash');
+const { getAuthProviderFromReq } = require('../auth/authProvider');
+const cachedPermissions = {};
+function hasPermission(tested, req) {
+  if (!req) {
+    // request object not available, allow all
+    return true;
+  }
+  const permissions = getAuthProviderFromReq(req).getCurrentPermissions(req);
+  if (!cachedPermissions[permissions]) {
+    cachedPermissions[permissions] = compilePermissions(permissions);
+  }
+  return testPermission(tested, cachedPermissions[permissions]);
+  // const { user } = (req && req.auth) || {};
+  // const { login } = (process.env.OAUTH_PERMISSIONS && req && req.user) || {};
+  // const key = user || login || '';
+  // const logins = getLogins();
+  // if (!userPermissions[key]) {
+  //   if (logins) {
+  //     const login = logins.find(x => x.login == user);
+  //     userPermissions[key] = compilePermissions(login ? login.permissions : null);
+  //   } else {
+  //     userPermissions[key] = compilePermissions(process.env.PERMISSIONS);
+  //   }
+  // }
+  // return testPermission(tested, userPermissions[key]);
+}
+// let loginsCache = null;
+// let loginsLoaded = false;
+// function getLogins() {
+//   if (loginsLoaded) {
+//     return loginsCache;
+//   }
+//   const res = [];
+//   if (process.env.LOGIN && process.env.PASSWORD) {
+//     res.push({
+//       login: process.env.LOGIN,
+//       password: process.env.PASSWORD,
+//       permissions: process.env.PERMISSIONS,
+//     });
+//   }
+//   if (process.env.LOGINS) {
+//     const logins = _.compact(process.env.LOGINS.split(',').map(x => x.trim()));
+//     for (const login of logins) {
+//       const password = process.env[`LOGIN_PASSWORD_${login}`];
+//       const permissions = process.env[`LOGIN_PERMISSIONS_${login}`];
+//       if (password) {
+//         res.push({
+//           login,
+//           password,
+//           permissions,
+//         });
+//       }
+//     }
+//   } else if (process.env.OAUTH_PERMISSIONS) {
+//     const login_permission_keys = Object.keys(process.env).filter(key => _.startsWith(key, 'LOGIN_PERMISSIONS_'));
+//     for (const permissions_key of login_permission_keys) {
+//       const login = permissions_key.replace('LOGIN_PERMISSIONS_', '');
+//       const permissions = process.env[permissions_key];
+//       userPermissions[login] = compilePermissions(permissions);
+//     }
+//   }
+//   loginsCache = res.length > 0 ? res : null;
+//   loginsLoaded = true;
+//   return loginsCache;
+// }
+function connectionHasPermission(connection, req) {
+  if (!connection) {
+    return true;
+  }
+  if (_.isString(connection)) {
+    return hasPermission(`connections/${connection}`, req);
+  } else {
+    return hasPermission(`connections/${connection._id}`, req);
+  }
+}
+function testConnectionPermission(connection, req) {
+  if (!connectionHasPermission(connection, req)) {
+    throw new Error('Connection permission not granted');
+  }
+}
+module.exports = {
+  hasPermission,
+  connectionHasPermission,
+  testConnectionPermission,
+};

package/src/utility/importDbModel.js ADDED Viewed

@@ -0,0 +1,9 @@
+const { databaseInfoFromYamlModel, DatabaseAnalyser } = require('dbgate-tools');
+const loadModelFolder = require('./loadModelFolder');
+async function importDbModel(inputDir) {
+  const files = await loadModelFolder(inputDir);
+  return databaseInfoFromYamlModel(files);
+}
+module.exports = importDbModel;

package/src/utility/loadFilesRecursive.js ADDED Viewed

@@ -0,0 +1,20 @@
+const fs = require('fs-extra');
+const stream = require('stream');
+const readline = require('readline');
+const path = require('path');
+async function loadFilesRecursive(dir, root = null) {
+  if (!root) root = dir;
+  if (!root.endsWith(path.sep)) root += path.sep;
+  const dirents = await fs.readdir(dir, { withFileTypes: true });
+  const files = await Promise.all(
+    dirents.map(dirent => {
+      const res = path.join(dir, dirent.name);
+      return dirent.isDirectory() ? loadFilesRecursive(res, root) : res;
+    })
+  );
+  const flatten = Array.prototype.concat(...files);
+  return flatten.map(file => (file.startsWith(root) ? file.substr(root.length) : file));
+}
+module.exports = loadFilesRecursive;

package/src/utility/loadModelFolder.js ADDED Viewed

@@ -0,0 +1,29 @@
+const fs = require('fs-extra');
+const path = require('path');
+const yaml = require('js-yaml');
+const { resolveArchiveFolder } = require('./directories');
+const loadFilesRecursive = require('./loadFilesRecursive');
+async function loadModelFolder(inputDir) {
+  const files = [];
+  const dir = inputDir.startsWith('archive:')
+    ? resolveArchiveFolder(inputDir.substring('archive:'.length))
+    : path.resolve(inputDir);
+  for (const name of await loadFilesRecursive(dir)) {
+    if (name.endsWith('.table.yaml') || name.endsWith('.sql')) {
+      const text = await fs.readFile(path.join(dir, name), { encoding: 'utf-8' });
+      files.push({
+        name: path.parse(name).base,
+        text,
+        json: name.endsWith('.yaml') ? yaml.load(text) : null,
+      });
+    }
+  }
+  return files;
+}
+module.exports = loadModelFolder;

package/src/utility/loadModelTransform.js ADDED Viewed

@@ -0,0 +1,36 @@
+const { filesdir } = require('./directories');
+const path = require('path');
+const fs = require('fs-extra');
+const _ = require('lodash');
+const dbgateApi = require('../shell');
+const { getLogger, extractErrorLogData } = require('dbgate-tools');
+const logger = getLogger('loadModelTransform');
+function modelTransformFromJson(json) {
+  if (!dbgateApi[json.transform]) return null;
+  const creator = dbgateApi[json.transform];
+  return creator(...json.arguments);
+}
+async function loadModelTransform(file) {
+  if (!file) return null;
+  try {
+    const dir = filesdir();
+    const fullPath = path.join(dir, 'modtrans', file);
+    const text = await fs.readFile(fullPath, { encoding: 'utf-8' });
+    const json = JSON.parse(text);
+    if (_.isArray(json)) {
+      const array = _.compact(json.map(x => modelTransformFromJson(x)));
+      return array.length ? structure => array.reduce((acc, val) => val(acc), structure) : null;
+    }
+    if (_.isPlainObject(json)) {
+      return modelTransformFromJson(json);
+    }
+    return null;
+  } catch (err) {
+    logger.error(extractErrorLogData(err), `Error loading model transform ${file}`);
+    return null;
+  }
+}
+module.exports = loadModelTransform;

package/src/utility/pipeForkLogs.js ADDED Viewed

@@ -0,0 +1,19 @@
+const byline = require('byline');
+const { safeJsonParse, getLogger } = require('dbgate-tools');
+const logger = getLogger();
+const logDispatcher = method => data => {
+  const json = safeJsonParse(data.toString());
+  if (json && json.level) {
+    logger.log(json);
+  } else {
+    logger[method](json || data.toString());
+  }
+};
+function pipeForkLogs(subprocess) {
+  byline(subprocess.stdout).on('data', logDispatcher('info'));
+  byline(subprocess.stderr).on('data', logDispatcher('error'));
+}
+module.exports = pipeForkLogs;

package/src/utility/platformInfo.js ADDED Viewed

@@ -0,0 +1,62 @@
+const fs = require('fs');
+const os = require('os');
+const path = require('path');
+const processArgs = require('./processArgs');
+const isElectron = require('is-electron');
+const platform = process.env.OS_OVERRIDE ? process.env.OS_OVERRIDE : process.platform;
+const isWindows = platform === 'win32';
+const isMac = platform === 'darwin';
+const isLinux = platform === 'linux';
+const isDocker = fs.existsSync('/home/dbgate-docker/public');
+const isDevMode = process.env.DEVMODE == '1';
+const isBuiltWebMode = process.env.BUILTWEBMODE == '1';
+const isNpmDist = !!global['IS_NPM_DIST'];
+const isDbModel = !!global['IS_DB_MODEL'];
+const isForkedApi = processArgs.isForkedApi;
+const isAwsUbuntuLayout = fs.existsSync('/home/ubuntu/build/public');
+// function moduleAvailable(name) {
+//   try {
+//     require.resolve(name);
+//     return true;
+//   } catch (e) {
+//     return false;
+//   }
+// }
+const platformInfo = {
+  isWindows,
+  isMac,
+  isLinux,
+  isDocker,
+  isElectronBundle: isElectron() && !isDevMode,
+  isForkedApi,
+  isElectron: isElectron(),
+  isDevMode,
+  isNpmDist,
+  isSnap: process.env.ELECTRON_SNAP == 'true',
+  isPortable: isWindows && process.env.PORTABLE_EXECUTABLE_DIR,
+  isAppImage: process.env.DESKTOPINTEGRATION === 'AppImageLauncher',
+  sshAuthSock: process.env.SSH_AUTH_SOCK,
+  environment: process.env.NODE_ENV,
+  platform,
+  runningInWebpack: !!process.env.WEBPACK_DEV_SERVER_URL,
+  allowShellConnection:
+    (!processArgs.listenApiChild && !isNpmDist) ||
+    !!process.env.SHELL_CONNECTION ||
+    !!isElectron() ||
+    !!isDbModel ||
+    isDevMode,
+  allowShellScripting:
+    (!processArgs.listenApiChild && !isNpmDist) ||
+    !!process.env.SHELL_SCRIPTING ||
+    !!isElectron() ||
+    !!isDbModel ||
+    isDevMode,
+  allowConnectionFromEnvVariables: !!isDbModel,
+  defaultKeyfile: path.join(os.homedir(), '.ssh/id_rsa'),
+  isAwsUbuntuLayout,
+};
+module.exports = platformInfo;

package/src/utility/processArgs.js ADDED Viewed

@@ -0,0 +1,39 @@
+function getNamedArg(name) {
+  const argIndex = process.argv.indexOf(name);
+  if (argIndex > 0) {
+    return process.argv[argIndex + 1];
+  }
+  return null;
+}
+const checkParent = process.argv.includes('--checkParent');
+const startProcess = getNamedArg('--start-process');
+const isForkedApi = process.argv.includes('--is-forked-api');
+const pluginsDir = getNamedArg('--plugins-dir');
+const workspaceDir = getNamedArg('--workspace-dir');
+const processDisplayName = getNamedArg('--process-display-name');
+const listenApi = process.argv.includes('--listen-api');
+const listenApiChild = process.argv.includes('--listen-api-child') || listenApi;
+function getPassArgs() {
+  const res = [];
+  if (global['PLUGINS_DIR']) {
+    res.push('--plugins-dir', global['PLUGINS_DIR']);
+  }
+  if (listenApiChild) {
+    res.push('listen-api-child');
+  }
+  return res;
+}
+module.exports = {
+  checkParent,
+  startProcess,
+  isForkedApi,
+  getPassArgs,
+  pluginsDir,
+  workspaceDir,
+  listenApi,
+  listenApiChild,
+  processDisplayName,
+};

package/src/utility/processComm.js ADDED Viewed

@@ -0,0 +1,18 @@
+const { handleGetSshTunnelRequest, handleGetSshTunnelResponse } = require('./sshTunnelProxy');
+function handleProcessCommunication(message, subprocess) {
+  const { msgtype } = message;
+  if (msgtype == 'getsshtunnel-request') {
+    handleGetSshTunnelRequest(message, subprocess);
+    return true;
+  }
+  if (msgtype == 'getsshtunnel-response') {
+    handleGetSshTunnelResponse(message, subprocess);
+    return true;
+  }
+  return false;
+}
+module.exports = {
+  handleProcessCommunication,
+};

package/src/utility/requireEngineDriver.js ADDED Viewed

@@ -0,0 +1,26 @@
+const _ = require('lodash');
+const requirePlugin = require('../shell/requirePlugin');
+const { pickSafeConnectionInfo } = require('./crypting');
+/** @returns {import('dbgate-types').EngineDriver} */
+function requireEngineDriver(connection) {
+  let engine = null;
+  if (_.isString(connection)) {
+    engine = connection;
+  } else if (_.isPlainObject(connection)) {
+    engine = connection.engine;
+  }
+  if (!engine) {
+    throw new Error(`Could not get driver from connection ${JSON.stringify(pickSafeConnectionInfo(connection))}`);
+  }
+  if (engine.includes('@')) {
+    const [shortName, packageName] = engine.split('@');
+    const plugin = requirePlugin(packageName);
+    if (plugin.drivers) {
+      return plugin.drivers.find(x => x.engine == engine);
+    }
+  }
+  throw new Error(`Could not find engine driver ${engine}`);
+}
+module.exports = requireEngineDriver;

package/src/utility/requirePluginFunction.js ADDED Viewed

@@ -0,0 +1,16 @@
+const _ = require('lodash');
+const requirePlugin = require('../shell/requirePlugin');
+function requirePluginFunction(functionName) {
+  if (!functionName) return null;
+  if (functionName.includes('@')) {
+    const [shortName, packageName] = functionName.split('@');
+    const plugin = requirePlugin(packageName);
+    if (plugin.functions) {
+      return plugin.functions[shortName];
+    }
+  }
+  return null;
+}
+module.exports = requirePluginFunction;

package/src/utility/socket.js ADDED Viewed

@@ -0,0 +1,68 @@
+const _ = require('lodash');
+const stableStringify = require('json-stable-stringify');
+const sseResponses = {};
+let electronSender = null;
+let pingConfigured = false;
+module.exports = {
+  ensurePing() {
+    if (!pingConfigured) {
+      setInterval(() => this.emit('ping'), 29 * 1000);
+      pingConfigured = true;
+    }
+  },
+  addSseResponse(value, strmid) {
+    sseResponses[strmid] = {
+      ...sseResponses[strmid],
+      response: value,
+    };
+    this.ensurePing();
+  },
+  removeSseResponse(strmid) {
+    delete sseResponses[strmid];
+  },
+  setElectronSender(value) {
+    electronSender = value;
+    this.ensurePing();
+  },
+  emit(message, data) {
+    if (electronSender) {
+      electronSender.send(message, data == null ? null : data);
+    }
+    for (const strmid in sseResponses) {
+      if (data?.strmid && data?.strmid != strmid) {
+        continue;
+      }
+      let skipThisStream = false;
+      if (sseResponses[strmid].filter) {
+        for (const key in sseResponses[strmid].filter) {
+          if (data && data[key]) {
+            if (!sseResponses[strmid].filter[key].includes(data[key])) {
+              skipThisStream = true;
+              break;
+            }
+          }
+        }
+      }
+      if (skipThisStream) {
+        continue;
+      }
+      sseResponses[strmid].response?.write(
+        `event: ${message}\ndata: ${stableStringify(data == null ? null : _.omit(data, ['strmid']))}\n\n`
+      );
+    }
+  },
+  emitChanged(key, params = undefined) {
+    // console.log('EMIT CHANGED', key);
+    this.emit('changed-cache', { key, ...params });
+    // this.emit(key);
+  },
+  setStreamIdFilter(strmid, filter) {
+    sseResponses[strmid] = {
+      ...sseResponses[strmid],
+      filter,
+    };
+  },
+};