@eagleoutice/flowr
Version:
Static Dataflow Analyzer and Program Slicer for the R Programming Language
177 lines • 8.06 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.optionHelp = exports.toolName = void 0;
const server_1 = require("./repl/server/server");
const net_1 = require("./repl/server/net");
const version_1 = require("../util/version");
const command_line_usage_1 = __importDefault(require("command-line-usage"));
const log_1 = require("../util/log");
const ansi_1 = require("../util/text/ansi");
const command_line_args_1 = __importDefault(require("command-line-args"));
const config_1 = require("../config");
const assert_1 = require("../util/assert");
const scripts_info_1 = require("./common/scripts-info");
const shell_1 = require("../r-bridge/shell");
const execute_1 = require("./repl/execute");
const repl_main_1 = require("./repl/commands/repl-main");
const core_1 = require("./repl/core");
const repl_version_1 = require("./repl/commands/repl-version");
const print_version_1 = require("./repl/print-version");
const flowr_main_options_1 = require("./flowr-main-options");
const tree_sitter_executor_1 = require("../r-bridge/lang-4.x/tree-sitter/tree-sitter-executor");
const fs_1 = __importDefault(require("fs"));
const path_1 = __importDefault(require("path"));
exports.toolName = 'flowr';
exports.optionHelp = [
{
header: `flowR (version ${(0, version_1.flowrVersion)().toString()})`,
content: 'A static dataflow analyzer and program slicer for R programs'
},
{
header: 'Synopsis',
content: [
`$ ${exports.toolName} {bold --help}`,
`$ ${exports.toolName} {bold --version}`,
`$ ${exports.toolName} {bold --server}`,
`$ ${exports.toolName} {bold --execute} {italic ":parse 2 - 4"}`,
`$ ${exports.toolName} {bold slicer} {bold --help}`,
]
},
{
header: 'Options',
optionList: flowr_main_options_1.flowrMainOptionDefinitions
}
];
const options = (0, command_line_args_1.default)(flowr_main_options_1.flowrMainOptionDefinitions);
log_1.log.updateSettings(l => l.settings.minLevel = options.verbose ? 1 /* LogLevel.Trace */ : 5 /* LogLevel.Error */);
log_1.log.info('running with options', options);
if (options['no-ansi']) {
log_1.log.info('disabling ansi colors');
(0, ansi_1.setFormatter)(ansi_1.voidFormatter);
}
function createConfig() {
let config;
if (options['config-json']) {
const passedConfig = (0, config_1.parseConfig)(options['config-json']);
if (passedConfig) {
log_1.log.info(`Using passed config ${JSON.stringify(passedConfig)}`);
config = passedConfig;
}
}
if (config == undefined) {
if (options['config-file']) {
// validate it exists
if (!fs_1.default.existsSync(path_1.default.resolve(options['config-file']))) {
log_1.log.error(`Config file '${options['config-file']}' does not exist`);
process.exit(1);
}
}
config = (0, config_1.getConfig)(options['config-file'] ?? flowr_main_options_1.defaultConfigFile);
}
// for all options that we manually supply that have a config equivalent, set them in the config
config = (0, config_1.amendConfig)(config, c => {
c.engines ??= [];
if (!options['engine.r-shell.disabled']) {
c.engines.push({ type: 'r-shell', rPath: options['r-path'] || options['engine.r-shell.r-path'] });
}
if (!options['engine.tree-sitter.disabled']) {
c.engines.push({
type: 'tree-sitter',
wasmPath: options['engine.tree-sitter.wasm-path'],
treeSitterWasmPath: options['engine.tree-sitter.tree-sitter-wasm-path'],
lax: options['engine.tree-sitter.lax']
});
}
if (options['default-engine']) {
c.defaultEngine = options['default-engine'];
}
return c;
});
return config;
}
async function retrieveEngineInstances(config) {
const engines = {};
if ((0, config_1.getEngineConfig)(config, 'r-shell')) {
// we keep an active shell session to allow other parse investigations :)
engines['r-shell'] = new shell_1.RShell((0, config_1.getEngineConfig)(config, 'r-shell'), {
revive: 2 /* RShellReviveOptions.Always */,
onRevive: (code, signal) => {
const signalText = signal == null ? '' : ` and signal ${signal}`;
console.log(ansi_1.formatter.format(`R process exited with code ${code}${signalText}. Restarting...`, { color: 5 /* Colors.Magenta */, effect: ansi_1.ColorEffect.Foreground }));
console.log((0, ansi_1.italic)(`If you want to exit, press either Ctrl+C twice, or enter ${(0, ansi_1.bold)(':quit')}`));
}
});
}
if ((0, config_1.getEngineConfig)(config, 'tree-sitter')) {
await tree_sitter_executor_1.TreeSitterExecutor.initTreeSitter((0, config_1.getEngineConfig)(config, 'tree-sitter'));
engines['tree-sitter'] = new tree_sitter_executor_1.TreeSitterExecutor();
}
let defaultEngine = config.defaultEngine;
if (!defaultEngine || !engines[defaultEngine]) {
// if a default engine isn't specified, we just take the first one we have
defaultEngine = Object.keys(engines)[0];
}
log_1.log.info(`Using engines ${Object.keys(engines).join(', ')} with default ${defaultEngine}`);
return { engines, default: defaultEngine };
}
function hookSignalHandlers(engines) {
const end = () => {
if (options.execute === undefined) {
console.log(`\n${(0, ansi_1.italic)('Exiting...')}`);
}
Object.values(engines.engines).forEach(e => e?.close());
process.exit(0);
};
process.on('SIGINT', end);
process.on('SIGTERM', end);
}
async function mainRepl() {
const config = createConfig();
if (options.script) {
const target = scripts_info_1.scripts[options.script].target;
(0, assert_1.guard)(target !== undefined, `Unknown script ${options.script}, pick one of ${(0, flowr_main_options_1.getScriptsText)()}.`);
console.log(`Running script '${ansi_1.formatter.format(options.script, { style: 1 /* FontStyles.Bold */ })}'`);
log_1.log.debug(`Script maps to "${target}"`);
await (0, execute_1.waitOnScript)(`${__dirname}/${target}`, process.argv.slice(3), undefined, true);
process.exit(0);
}
if (options.help) {
console.log((0, command_line_usage_1.default)(exports.optionHelp));
process.exit(0);
}
const engines = await retrieveEngineInstances(config);
const defaultEngine = engines.engines[engines.default];
if (options.version) {
for (const engine of Object.values(engines.engines)) {
await (0, repl_version_1.printVersionInformation)(repl_main_1.standardReplOutput, engine);
engine?.close();
}
process.exit(0);
}
hookSignalHandlers(engines);
const allowRSessionAccess = options['r-session-access'] ?? false;
if (options.execute) {
await (0, core_1.replProcessAnswer)(config, repl_main_1.standardReplOutput, options.execute, defaultEngine, allowRSessionAccess);
}
else {
await (0, print_version_1.printVersionRepl)(defaultEngine);
await (0, core_1.repl)(config, { parser: defaultEngine, allowRSessionAccess });
}
process.exit(0);
}
async function mainServer(backend = new net_1.NetServer()) {
const config = createConfig();
const engines = await retrieveEngineInstances(config);
hookSignalHandlers(engines);
await new server_1.FlowRServer(engines.engines, engines.default, options['r-session-access'], config, backend).start(options.port);
}
if (options.server) {
void mainServer(options.ws ? new net_1.WebSocketServerWrapper() : new net_1.NetServer());
}
else {
void mainRepl();
}
//# sourceMappingURL=flowr.js.map