firebase-tools
Version:
Command-Line Interface for Firebase
50 lines (49 loc) • 1.91 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.interactiveExecuteQuery = exports.confirmDangerousQuery = void 0;
const ora = require("ora");
const clc = require("colorette");
const logger_1 = require("../../logger");
const prompt_1 = require("../../prompt");
const Table = require("cli-table3");
const destructiveSqlKeywords = ["DROP", "DELETE"];
function checkIsDestructiveSql(query) {
const upperCaseQuery = query.toUpperCase();
return destructiveSqlKeywords.some((keyword) => upperCaseQuery.includes(keyword.toUpperCase()));
}
async function confirmDangerousQuery(query) {
if (checkIsDestructiveSql(query)) {
return await (0, prompt_1.confirm)({
message: clc.yellow("This query may be destructive. Are you sure you want to proceed?"),
default: false,
});
}
return true;
}
exports.confirmDangerousQuery = confirmDangerousQuery;
async function interactiveExecuteQuery(query, conn) {
const spinner = ora("Executing query...").start();
try {
const results = await conn.query(query);
spinner.succeed(clc.green("Query executed successfully"));
if (Array.isArray(results.rows) && results.rows.length > 0) {
const table = new Table({
head: Object.keys(results.rows[0]).map((key) => clc.cyan(key)),
style: { head: [], border: [] },
});
for (const row of results.rows) {
table.push(Object.values(row));
}
logger_1.logger.info(table.toString());
}
else {
if (query.toUpperCase().includes("SELECT")) {
logger_1.logger.info(clc.yellow("No results returned"));
}
}
}
catch (err) {
spinner.fail(clc.red(`Failed executing query: ${err}`));
}
}
exports.interactiveExecuteQuery = interactiveExecuteQuery;