trader-pgsql-ast-parser
Version:
Yet another simple Postgres SQL parser/modifier
108 lines (95 loc) • 3.38 kB
JavaScript
const rimraf = require('rimraf');
const fs = require('fs');
const path = require('path');
const outDir = path.join(__dirname, '.deno');
const excluded = [
path.join(__dirname, 'src', 'syntax', 'spec-utils.ts'),
]
function traverse(dir, relative, perform) {
const rdir = path.join(outDir, relative);
if (!(fs.existsSync(rdir))) {
fs.mkdirSync(rdir)
}
for (const iname of fs.readdirSync(dir)) {
const ipath = path.join(dir, iname);
if (excluded.includes(ipath)) {
continue;
}
const rpath = path.join(relative, iname);
const st = fs.statSync(ipath);
if (st.isDirectory()) {
traverse(ipath, rpath, perform);
continue;
}
if (/\.spec\.ts$/.test(iname)) {
// ignore tests
continue;
}
perform(iname, ipath, rpath);
}
}
if (process.argv.includes('--copy')) {
// ============== COPY
rimraf.sync(outDir);
if (!(fs.existsSync(outDir))) {
fs.mkdirSync(outDir)
}
const settingsJson = path.join(outDir, '.vscode', 'settings.json');
if (!fs.existsSync(path.dirname(settingsJson))) {
fs.mkdirSync(path.dirname(settingsJson));
}
if (!(fs.existsSync(settingsJson))) {
fs.writeFileSync(settingsJson, `{
"deno.enable": true
}`)
}
traverse(path.join(__dirname, 'src'), '', (iname, ipath, rpath) => {
const ext = path.extname(iname);
switch (ext) {
case '.ts':
fs.copyFileSync(ipath, path.join(outDir, rpath));
console.log('Copied ' + rpath);
break;
}
});
fs.writeFileSync(path.join(outDir, 'mod.ts'), `export * from './index';`);
fs.copyFileSync(path.join(__dirname, 'readme.md'), path.join(outDir, 'readme.md'));
} else if (process.argv.includes('--process')) {
// ============= TRANSPILE
const bindings = {
'moo': 'https://deno.land/x/moo@0.5.1-deno.2/mod.ts',
'nearley': 'https://deno.land/x/nearley@2.19.7-deno/mod.ts',
}
function handleTs(ipath, rpath) {
const content = fs.readFileSync(ipath, 'utf-8');
const newContent = content.replace(/^\s*(import|export)\s+([^\n]+)\s+from\s+['"]([^'"]+)['"]\;?$/mg, (_, op, what, where) => {
if (/^\./.test(where)) {
const asDir = path.join(path.dirname(ipath), where, 'index.ts');
if (fs.existsSync(asDir)) {
where = where + '/index';
}
return `${op} ${what} from '${where}.ts';`
}
let bound = bindings[where];
if (!bound) {
throw new Error('No Deno binding for dependency ' + where + ' in ' + rpath);
}
bound = typeof bound === 'string'
? { what: x => x, where: bound }
: bound
return `${op} ${bound.what(what)} from '${bound.where}';`;
});
fs.writeFileSync(ipath, newContent);
console.log('Transpiled ' + rpath);
}
traverse(outDir, '', (iname, ipath, rpath) => {
const ext = path.extname(iname);
switch (ext) {
case '.ts':
handleTs(ipath, rpath);
break;
}
});
} else {
throw new Error('Unkown transpile program arg');
}