@keepzen/orm-for-sqlite3/cli.js

ORM for Learn

#!/usr/bin/env node
const {
  createPromiseDB,
} = require('./index.js');
const {
  join: pathJoin
} = require('path');
const fs = require('fs');

const getTablesSQL = `SELECT name,type from sqlite_master
where (type="table" or type="view") and name not like 'sqlite_%';`
async function getTables(db) {
  return (
    await db.query(getTablesSQL)
  );
}
async function getMetaOf(db, table) {
  const { name, type } = table;
  const sql = `SELECT name, pk as isPK FROM Pragma_table_info('${name}')`;
  return (await db.query(sql)).
    reduce(
      (ret, { name, isPK }) => {
        const { fields, pks } = ret;
        if (isPK) {
          pks.push(name);
        }
        fields.push(name);
        return ret;
      },
      { fields: [], pks: [], name, type }
    )
}

function createClass({ fields, pks, name, type }, toDir = "./") {
  const otherFun = name => `${name}(){throw new Error("View can not be write")}`;
  let otherMethod = [];
  if (type == 'view') {
    otherMethod.push(otherFun('add'));
    otherMethod.push(otherFun('update'));
    otherMethod.push(otherFun('remove'));
  }
  const classContent = `const Table=require('@keepzen/orm-for-sqlite3/Table.js');
class ${name} extends Table {
  constructor(initValues = {}) {
    super(
      [${fields.map(JSON.stringify).join(',')}],
      [${pks.map(JSON.stringify).join(',')}],
      initValues
    );
  }
  ${otherMethod.join("\n  ")}
}
module.exports = ${name};
`;
  fs.writeFileSync(pathJoin(toDir, `${name}.${type}.js`), classContent);
}
async function main(...o) {
  const {
    dbPath,
    sqlScriptPath,
    toDir = "./"
  } = o.reduce((ret, i) => {
    switch (true) {
      case i.endsWith('.sql'):
        ret.sqlScriptPath = i;
        break;
      case i.endsWith(".db"):
        ret.dbPath = i;
        break;
      default:
        ret.toDir = i;
    }
    return ret;
  }, {})
  let db = await createPromiseDB(dbPath, sqlScriptPath);
  const tables = await getTables(db);
  for (let table of tables) {
    let meta = await getMetaOf(db, table);
    createClass(meta, toDir);
  }
  const files = tables.map(t => `${t.name}.${t.type}.js`);
  return `Have create fellow files:\n ${files.join("\n ")}.`;
}
if (module == require.main) {
  const [_1, _2, ...others] = process.argv;
  if (others.length < 1) {
    console.log(`Usage:
    Just export table from db:
      npx orm-sqlite3-cli path/name.db [export-class-dir="./"]
    Run sql script and export table scheme:
      npx orm-sqlite3-cli path/name.db path/script.sql [export-class-dir="./"]
    `);
    return;
  }
  main(...others).then(console.log, console.error);
}