dennii/dbgate
1
0
Fork 0
mirror of https://github.com/DeNNiiInc/dbgate.git synced 2026-04-21 19:56:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

runner openreader - support for plugins

Browse Source
This commit is contained in:
Jan Prochazka
2020-11-22 09:03:16 +01:00
parent 286cac066c
commit 7d1c0c5c18
5 changed files with 42 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
packages/web/src/impexp/ScriptWriter.js
View File

@@ -1,10 +1,11 @@
import _ from 'lodash';
import { extractShellApiFunctionName, extractShellApiPlugins } from 'dbgate-tools';
export default class ScriptWriter {
constructor() {
this.s = '';
this.packageNames = [];
this.engines = [];
// this.engines = [];
this.varCount = 0;
}
@@ -19,21 +20,8 @@ export default class ScriptWriter {
}
assign(variableName, functionName, props) {
const nsMatch = functionName.match(/^([^@]+)@([^@]+)/);
if (nsMatch) {
const packageName = nsMatch[2];
if (!this.packageNames.includes(packageName)) {
this.packageNames.push(packageName);
}
this.put(
`const ${variableName} = await ${_.camelCase(packageName)}.shellApi.${nsMatch[1]}(${JSON.stringify(props)});`
);
} else {
this.put(`const ${variableName} = await dbgateApi.${functionName}(${JSON.stringify(props)});`);
}
if (props && props.connection && props.connection.engine && !this.engines.includes(props.connection.engine)) {
this.engines.push(props.connection.engine);
}
this.put(`const ${variableName} = await ${extractShellApiFunctionName(functionName)}(${JSON.stringify(props)});`);
this.packageNames.push(...extractShellApiPlugins(functionName, props));
}
copyStream(sourceVar, targetVar) {
@@ -54,6 +42,6 @@ export default class ScriptWriter {
// this.comment(JSON.stringify(this.engines));
// }
const packageNames = this.packageNames;
return packageNames.map((packageName) => `// @require ${packageName}\n`).join('') + '\n' + this.s;
return _.uniq(packageNames).map((packageName) => `// @require ${packageName}\n`).join('') + '\n' + this.s;
}
}