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

backend exporters/importers from plugins

Browse Source
This commit is contained in:
Jan Prochazka
2020-11-21 21:16:22 +01:00
parent 6ed3eaa896
commit 286cac066c
3 changed files with 47 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
packages/api/src/controllers/runners.js
View File

@@ -5,10 +5,19 @@ const uuidv1 = require('uuid/v1');
const byline = require('byline');
const socket = require('../utility/socket');
const { fork } = require('child_process');
const { rundir, uploadsdir } = require('../utility/directories');
const { rundir, uploadsdir, pluginsdir } = require('../utility/directories');
function extractPlugins(script) {
const requireRegex = /\s*\/\/\s*@require\s+([^\s]+)\s*\n/g;
const matches = [...script.matchAll(requireRegex)];
return matches.map((x) => x[1]);
}
const scriptTemplate = (script) => `
const dbgateApi = require(process.env.DBGATE_API || "dbgate-api");
const dbgateApi = require(process.env.DBGATE_API);
${extractPlugins(script)
.map((packageName) => `const ${_.camelCase(packageName)} = require(process.env.PLUGIN_${_.camelCase(packageName)});\n`)
.join('')}
require=null;
async function run() {
${script}
@@ -19,7 +28,7 @@ dbgateApi.runScript(run);
`;
const loaderScriptTemplate = (functionName, props, runid) => `
const dbgateApi = require(process.env.DBGATE_API || "dbgate-api");
const dbgateApi = require(process.env.DBGATE_API);
require=null;
async function run() {
const reader=await dbgateApi.${functionName}(${JSON.stringify(props)});
@@ -73,12 +82,14 @@ module.exports = {
const scriptFile = path.join(uploadsdir(), runid + '.js');
fs.writeFileSync(`${scriptFile}`, scriptText);
fs.mkdirSync(directory);
const pluginNames = fs.readdirSync(pluginsdir());
console.log(`RUNNING SCRIPT ${scriptFile}`);
const subprocess = fork(scriptFile, ['--checkParent'], {
cwd: directory,
stdio: ['ignore', 'pipe', 'pipe', 'ipc'],
env: {
DBGATE_API: process.argv[1],
..._.fromPairs(pluginNames.map((name) => [`PLUGIN_${_.camelCase(name)}`, path.join(pluginsdir(), name)])),
},
});
const pipeDispatcher = (severity) => (data) =>
@@ -153,4 +164,3 @@ module.exports = {
return promise;
},
};